def test_context_command(mocker, args, test_scenario, api_response, status_code, expected_output): """ Test various inputs for context command """ client = GreyNoise.Client( api_key="true_api_key", api_server="dummy_server", timeout=10, proxy="proxy", use_cache=False, integration_name="dummy_integration", ) dummy_response = DummyResponse({"Content-Type": "application/json"}, json.dumps(expected_output), status_code) mocker.patch("requests.Session.get", return_value=dummy_response) if test_scenario == "positive": response = GreyNoise.context_command(client, args) assert response.outputs == expected_output else: mocker.patch("requests.Session.get", return_value=dummy_response) with pytest.raises(Exception) as err: _ = GreyNoise.ip_reputation_command(client, args) print("this is err: " + str(err)) assert str(err.value) == expected_output
def test_ip_reputation_command(args, test_scenario, api_response, status_code, expected_output, mocker): """ Tests various combinations of vald and invalid responses for IPReputation command. """ client = GreyNoise.Client("true_api_key", "dummy_server", 10, "proxy", False, "dummy_integration") dummy_response = DummyResponse({"Content-Type": "application/json"}, json.dumps(api_response), status_code) if test_scenario == "positive": mocker.patch("requests.Session.get", return_value=dummy_response) response = GreyNoise.ip_reputation_command(client, args) assert response[0].outputs == expected_output else: mocker.patch("requests.Session.get", return_value=dummy_response) with pytest.raises(Exception) as err: _ = GreyNoise.ip_reputation_command(client, args) assert str(err.value) == expected_output