def test_start_controller(*args):
"""Test activate method."""
(mock_pre_install_napps, mock_load_napps, mock_raw_event_handler,
mock_msg_in_event_handler, mock_msg_out_event_handler,
mock_app_event_handler, _) = args
napp = MagicMock()
loop = MagicMock()
options = KytosConfig().options['daemon']
options.napps_pre_installed = [napp]
controller = Controller(options, loop=loop)
controller.log = Mock()
controller.start_controller()
controller.server.serve_forever.assert_called()
calls = [
call(mock_raw_event_handler.return_value),
call(mock_msg_in_event_handler.return_value),
call(mock_msg_out_event_handler.return_value),
call(mock_app_event_handler.return_value)
]
loop.create_task.assert_has_calls(calls)
mock_pre_install_napps.assert_called_with([napp])
mock_load_napps.assert_called()
def _get_controller_mock(self):
"""Return a controller mock."""
loop = asyncio.new_event_loop()
asyncio.set_event_loop(None)
options = KytosConfig().options['daemon']
options.jwt_secret = 'jwt_secret'
controller = Controller(options, loop=loop)
controller.log = Mock()
# Patch event callback trigger.
controller.buffers.app.put = self._patch_event_trigger
return controller
def test_start_controller(*args):
"""Test activate method."""
(mock_pre_install_napps, mock_load_napps, _, _, _, _, _) = args
napp = MagicMock()
loop = MagicMock()
options = KytosConfig().options['daemon']
options.napps_pre_installed = [napp]
controller = Controller(options, loop=loop)
controller.log = Mock()
controller.start_controller()
controller.server.serve_forever.assert_called()
assert loop.create_task.call_count == 5
assert len(controller._tasks) == 5
mock_pre_install_napps.assert_called_with([napp])
mock_load_napps.assert_called()
def get_controller_mock():
"""Return a controller mock."""
options = KytosConfig().options['daemon']
controller = Controller(options)
controller.log = Mock()
return controller
