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