def test_faraday_dictionary_dispatcher_result(self):
mock_raw_hosts = MagicMock()
mock_raw_hosts.return_value = {'rows': [{'a': 'host', 'value': {'stuff': 'other_stuff'}}], 'total_rows': 4}
with patch('persistence.server.server._get_raw_hosts', mock_raw_hosts):
list_of_dicts = server._get_faraday_ready_dictionaries('some_workspace', 'hosts', 'rows', full_table=False)
with patch('persistence.server.server._get_raw_hosts', mock_raw_hosts):
full_list_of_dicts = server._get_faraday_ready_dictionaries('some_workspace', 'hosts',
'rows', full_table=True)
self.assertTrue(len(list_of_dicts) == 1 == len(full_list_of_dicts))
self.assertEqual(list_of_dicts, [mock_raw_hosts.return_value['rows'][0]['value']])
self.assertEqual(full_list_of_dicts, mock_raw_hosts.return_value['rows'])
def test_faraday_dictionary_dispatcher_result(self):
mock_raw_hosts = MagicMock()
mock_raw_hosts.return_value = {'rows': [{'a': 'host', 'value': {'stuff': 'other_stuff'}}], 'total_rows': 4}
with patch('persistence.server.server._get_raw_hosts', mock_raw_hosts):
list_of_dicts = server._get_faraday_ready_dictionaries('some_workspace', 'hosts', 'rows', full_table=False)
with patch('persistence.server.server._get_raw_hosts', mock_raw_hosts):
full_list_of_dicts = server._get_faraday_ready_dictionaries('some_workspace', 'hosts',
'rows', full_table=True)
self.assertTrue(len(list_of_dicts) == 1 == len(full_list_of_dicts))
self.assertEqual(list_of_dicts, [mock_raw_hosts.return_value['rows'][0]['value']])
self.assertEqual(full_list_of_dicts, mock_raw_hosts.return_value['rows'])
def test_faraday_dictionary_dispatcher_result(self):
mock_raw_hosts = MagicMock()
mock_raw_hosts.return_value = {"rows": [{"a": "host", "value": {"stuff": "other_stuff"}}], "total_rows": 4}
with patch("persistence.server.server._get_raw_hosts", mock_raw_hosts):
list_of_dicts = server._get_faraday_ready_dictionaries("some_workspace", "hosts", "rows", full_table=False)
with patch("persistence.server.server._get_raw_hosts", mock_raw_hosts):
full_list_of_dicts = server._get_faraday_ready_dictionaries(
"some_workspace", "hosts", "rows", full_table=True
)
self.assertTrue(len(list_of_dicts) == 1 == len(full_list_of_dicts))
self.assertEqual(list_of_dicts, [mock_raw_hosts.return_value["rows"][0]["value"]])
self.assertEqual(full_list_of_dicts, mock_raw_hosts.return_value["rows"])
def test_faraday_dictionary_dispatcher_calls(self, mock_hosts, mock_vulns,
mock_services, mock_notes,
mock_credentials, mock_commands):
# NOTE: if you finds any bugs here, i have the suspipcion that mock_host is actually mock_commands
# i mean that the parameters names are wrong. I'd check for that. Good luck.
server._get_faraday_ready_dictionaries('a', 'hosts', 'whatever')
server._get_faraday_ready_dictionaries('a', 'vulns', 'whatever')
server._get_faraday_ready_dictionaries('a', 'services', 'whatever')
server._get_faraday_ready_dictionaries('a', 'notes', 'whatever')
server._get_faraday_ready_dictionaries('a', 'credentials', 'whatever')
server._get_faraday_ready_dictionaries('a', 'commands', 'whatever')
mock_hosts.assert_called_once_with('a')
mock_vulns.assert_called_once_with('a')
mock_services.assert_called_once_with('a')
mock_notes.assert_called_once_with('a')
mock_credentials.assert_called_once_with('a')
mock_commands.assert_called_once_with('a')
def test_faraday_dictionary_dispatcher_calls(
self, mock_hosts, mock_vulns, mock_interfaces, mock_services, mock_notes, mock_credentials, mock_commands
):
# NOTE: if you finds any bugs here, i have the suspipcion that mock_host is actually mock_commands
# i mean, the parameters names are wrong. I'd check for that. Good luck.
server._get_faraday_ready_dictionaries("a", "hosts", "whatever")
server._get_faraday_ready_dictionaries("a", "interfaces", "whatever")
server._get_faraday_ready_dictionaries("a", "vulns", "whatever")
server._get_faraday_ready_dictionaries("a", "services", "whatever")
server._get_faraday_ready_dictionaries("a", "notes", "whatever")
server._get_faraday_ready_dictionaries("a", "credentials", "whatever")
server._get_faraday_ready_dictionaries("a", "commands", "whatever")
mock_hosts.assert_called_once_with("a")
mock_vulns.assert_called_once_with("a")
mock_interfaces.assert_called_once_with("a")
mock_services.assert_called_once_with("a")
mock_notes.assert_called_once_with("a")
mock_credentials.assert_called_once_with("a")
mock_commands.assert_called_once_with("a")
def test_faraday_dictionary_dispatcher_calls(self, mock_hosts, mock_vulns,
mock_services, mock_notes,
mock_credentials, mock_commands):
# NOTE: if you finds any bugs here, i have the suspipcion that mock_host is actually mock_commands
# i mean that the parameters names are wrong. I'd check for that. Good luck.
server._get_faraday_ready_dictionaries('a', 'hosts', 'whatever')
server._get_faraday_ready_dictionaries('a', 'vulns', 'whatever')
server._get_faraday_ready_dictionaries('a', 'services', 'whatever')
server._get_faraday_ready_dictionaries('a', 'notes', 'whatever')
server._get_faraday_ready_dictionaries('a', 'credentials', 'whatever')
server._get_faraday_ready_dictionaries('a', 'commands', 'whatever')
mock_hosts.assert_called_once_with('a')
mock_vulns.assert_called_once_with('a')
mock_services.assert_called_once_with('a')
mock_notes.assert_called_once_with('a')
mock_credentials.assert_called_once_with('a')
mock_commands.assert_called_once_with('a')
