def test_insert_data(self, mock_nl_start, mock_dm_start): fw = Framework(config_path) fw.start() test_dict = {'a': 1, 'b': 2} with patch('database.datamanager.DataManager.insert_data') as mock_insert_data: fw.insert_data(test_dict) mock_insert_data.assert_called_with(test_dict)
def test_plugins_enabled(self, mock_nl_start, mock_dm_start): framework = Framework(config_path) framework.start() expected = { 8082: make_mock_config(8082, 'HTTPPlugin', 'HTTPPlugin') } self.assertEqual(expected, framework.global_config.get_plugin_dictionary()) self.assertEqual(1, mock_nl_start.call_count) self.assertTrue(mock_dm_start.called)
def test_spawn(self, mock_gpn, mock_gsn, mock_nl_start, mock_dm_start): framework = Framework(config_path) framework.start() with patch('plugins.http.HTTPPlugin.start') as mock_http_start: framework.spawn(socket.socket(), {'port': 8082}) self.assertTrue(mock_http_start.called)
def test_get_config(self, mock_nl_start, mock_dm_start): framework = Framework(config_path) framework.start() expected = make_mock_config(8082, 'HTTPPlugin', 'HTTPPlugin') self.assertEqual(expected, framework.get_config(8082))
def test_plugins_disabled(self, mock_nl_start, mock_dm_start): framework = Framework(config_path) framework.start() self.assertTrue(8083 not in framework.global_config.get_plugin_dictionary()) self.assertTrue(8082 in framework.global_config.get_plugin_dictionary()) self.assertEqual(1, mock_nl_start.call_count)