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)
