1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
from unittest import TestCase
from prospector import blender
from prospector.message import Location, Message
class TestBlendLine(TestCase):
BLEND = (
(("s1", "s1c01"), ("s2", "s2c12")),
(("s3", "s3c81"), ("s1", "s1c04"), ("s2", "s2c44")),
)
def _do_test(self, messages, expected):
def _msg(source, code):
loc = Location("path.py", "path", None, 1, 0)
return Message(source, code, loc, "Test Message")
messages = [_msg(*m) for m in messages]
expected = set(expected)
blended = blender.blend_line(messages, TestBlendLine.BLEND)
result = {(msg.source, msg.code) for msg in blended}
self.assertEqual(expected, result)
def test_blend_line(self):
messages = (("s2", "s2c12"), ("s2", "s2c11"), ("s1", "s1c01"))
expected = (
("s1", "s1c01"),
("s2", "s2c11"), # s2c12 should be blended with s1c01
)
self._do_test(messages, expected)
def test_single_blend(self):
# these three should be blended together
messages = (
("s1", "s1c04"),
("s2", "s2c44"),
("s3", "s3c81"),
)
# the s3 message is the highest priority
expected = (("s3", "s3c81"),)
self._do_test(messages, expected)
def test_nothing_to_blend(self):
"""
Verifies that messages pass through if there is nothing to blend
"""
messages = (("s4", "s4c99"), ("s4", "s4c01"), ("s5", "s5c51"), ("s6", "s6c66"))
self._do_test(messages, messages) # expected = messages
def test_no_messages(self):
"""
Ensures that the blending works fine when there are no messages to blend
"""
self._do_test((), ())
def test_multiple_lines():
def _msg(source, code, line_number):
loc = Location("path.py", "path", None, line_number, 0)
return Message(source, code, loc, "Test Message")
messages = [
_msg("s1", "s1c001", 4),
_msg("s2", "s2c001", 6),
_msg("s2", "s2c101", 4),
_msg("s1", "s1c001", 6),
]
result = blender.blend(messages, ((("s1", "s1c001"), ("s2", "s2c101")),))
result = [(msg.source, msg.code, msg.location.line) for msg in result]
result = set(result)
expected = {("s1", "s1c001", 4), ("s1", "s1c001", 6), ("s2", "s2c001", 6)}
assert expected == result
|
