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")