def test_cleanup(self, Popen):
backend1 = Backend.objects.create(name="foo")
backend2 = Backend.objects.create(name="bar")
manager = ListenerManager()
manager.start(backend1)
manager.start(backend2)
manager.stop = MagicMock()
manager.cleanup()
manager.stop.assert_has_calls([call(backend1.id), call(backend2.id)], any_order=True)
def test_cleanup(self, Popen):
backend1 = Backend.objects.create(name="foo")
backend2 = Backend.objects.create(name="bar")
manager = ListenerManager()
manager.start(backend1)
manager.start(backend2)
manager.stop = MagicMock()
manager.cleanup()
manager.stop.assert_has_calls(
[call(backend1.id), call(backend2.id)], any_order=True)
def assert_regex_is_valid(self, test_case):
charset = test_case[
'charset'] if 'charset' in test_case else MH.Charset.ASCII_EXT
self._parser.parse(test_case['regex'], "name", charset)
calls = []
nodes = test_case['nodes']
for i in range(0, len(nodes)):
type = nodes[i]['type'] if 'type' in nodes[i] else vt.String
values = nodes[i]['values'] if 'values' in nodes[i] else None
alphabet = nodes[i]['alphabet'] if 'alphabet' in nodes[i] else None
qty = nodes[i]['qty'] if 'qty' in nodes[i] else (1, 1)
calls.append(
mock.call("name" + "_" + str(i + 1),
type,
values=values,
alphabet=alphabet,
qty=qty))
self._parser._create_terminal_node.assert_has_calls(calls)
self.assertEquals(self._parser._create_terminal_node.call_count,
len(test_case['nodes']))
def test_keep_listeners_running_added(self, Popen):
manager = ListenerManager()
backend1 = Backend.objects.create(name="foo")
manager.start = MagicMock()
manager.stop = MagicMock()
# start existing backends
manager.keep_listeners_running()
manager.start.assert_called_with(backend1)
# new backend, start it too
backend2 = Backend.objects.create(name="bar")
manager.keep_listeners_running()
manager.start.assert_has_calls(
[call(backend1), call(backend2)], any_order=True)
manager.stop.assert_not_called()
def assert_regex_is_valid(self, test_case):
charset = test_case['charset'] if 'charset' in test_case else MH.Charset.ASCII_EXT
self._parser.parse(test_case['regex'], "name", charset)
calls = []
nodes = test_case['nodes']
for i in range(0, len(nodes)):
type = nodes[i]['type'] if 'type' in nodes[i] else vt.String
values = nodes[i]['values'] if 'values' in nodes[i] else None
alphabet = nodes[i]['alphabet'] if 'alphabet' in nodes[i] else None
qty = nodes[i]['qty'] if 'qty' in nodes[i] else (1, 1)
calls.append(mock.call("name" + "_" + str(i + 1), type, values=values, alphabet=alphabet, qty=qty))
self._parser._create_terminal_node.assert_has_calls(calls)
self.assertEquals(self._parser._create_terminal_node.call_count, len(test_case['nodes']))