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
