class TestHandler(TestCase):
"""Test the Handler."""
def setUp(self): #pylint: disable=C0103
"""Initialize"""
self.test_file = MagicMock()
self.test_parser = MagicMock()
self.test_sink = MagicMock()
self.files = { '/test': self.test_file }
self.parsers = { '/test': [ self.test_parser ] }
self.sinks = { self.test_parser: self.test_sink }
self.handler = Handler(self.files, self.parsers, self.sinks)
def test_reload(self):
"""Test Handler reloading"""
self.handler.reload('/test')
self.test_file.reload.assert_called_with()
def test_handle(self):
"""Test Handler handle"""
lines = [ 'test' ]
results = [ {'test': 'test'} ]
self.test_file.read.return_value = lines
self.test_parser.parse.return_value = results
self.handler.handle('/test')
self.test_file.read.assert_called_with()
self.test_parser.parse.assert_called_with(lines)
self.test_sink.sink.assert_called_with(results)
def test_handle_all(self):
"""Test Handler handle all"""
self.handler.handle = MagicMock()
self.handler.handle_all()
self.handler.handle.assert_called_with('/test')
