def test_types_of_inner_apis(_):
client = ThreatResponse('CLIENT_ID', 'CLIENT_PASSWORD')
assert isinstance(client.inspect, InspectAPI)
assert isinstance(client.enrich, EnrichAPI)
assert isinstance(client.response, ResponseAPI)
assert isinstance(client.private_intel, IntelAPI)
assert isinstance(client.global_intel, IntelAPI)
def test_that_client_with_valid_credentials_succeeds(inner_session_request):
inner_session_request.return_value = auth_response(200)
logger = MagicMock()
ThreatResponse(
client_id='CLIENT_ID',
client_password='******',
logger=logger,
)
# Verify that only a single request has been made to an external API.
# Don't check the actual arguments since we're not interested in any
# auth-specific details here.
inner_session_request.assert_called_once()
logger.info.assert_called_once_with(
'POST https://visibility.amp.cisco.com/iroh/oauth2/token 200 OK')
def TR(region):
return ThreatResponse('CLIENT_ID', 'CLIENT_PASSWORD', region=region)
def test_types_of_inner_apis(_):
client = ThreatResponse('CLIENT_ID', 'CLIENT_PASSWORD')
assert isinstance(client.inspect, InspectAPI)
assert isinstance(client.enrich, EnrichAPI)