def test_getResults_fullResults(self, s3):
client, bucket, bucket_name = s3
target = "infosec.mozilla.org"
# Ensure we have all matching objects in the test S3 bucket
client.put_object(Bucket=bucket_name,
Body=b'ABCD',
Key='{}_direnum.json'.format(target))
client.put_object(Bucket=bucket_name,
Body=b'ABCD',
Key='{}_websearch.json'.format(target))
client.put_object(Bucket=bucket_name,
Body=b'ABCD',
Key='{}_portscan.json'.format(target))
client.put_object(Bucket=bucket_name,
Body=b'ABCD',
Key='{}_httpobservatory.json'.format(target))
client.put_object(Bucket=bucket_name,
Body=b'ABCD',
Key='{}_tlsobservatory.json'.format(target))
client.put_object(Bucket=bucket_name,
Body=b'ABCD',
Key='{}_sshobservatory.json'.format(target))
test_event = {"body": '{"target": "' + target + '"}'}
test_context = None
results_handler = ResultsHandler(
client, bucket_name, results_path=TEST_SCAN_RESULTS_BASE_PATH)
response = results_handler.getResults(test_event, test_context)
assert type(response) is dict
assert response['statusCode'] == 200
assert response['isBase64Encoded'] is True
assert 'Content-Disposition' in response['headers']
assert response['headers']['Content-Type'] == "application/gzip"
def test_getResults_unableToDownload(self, s3):
client, bucket, bucket_name = s3
target = "infosec.mozilla.org"
path = "/abc/def" # A path that does not exist
# Ensure we have matching objects in the test S3 bucket to rule out other failure scenario
client.put_object(Bucket=bucket_name,
Body=b'ABCD',
Key='{}_direnum.json'.format(target))
client.put_object(Bucket=bucket_name,
Body=b'ABCD',
Key='{}_websearch.json'.format(target))
test_event = {"body": '{"target": "' + target + '"}'}
test_context = None
results_handler = ResultsHandler(client,
bucket_name,
results_path=path)
response = results_handler.getResults(test_event, test_context)
assert type(response) is dict
assert response['statusCode'] == 500
assert 'isBase64Encoded' not in response
assert 'Content-Disposition' not in response['headers']
assert response['headers']['Content-Type'] == "application/json"
assert 'unable' in response['body'].lower()
def test_getResults_signedURL(self, s3):
client, bucket, bucket_name = s3
target = "infosec.mozilla.org"
# Ensure we have all matching objects in the test S3 bucket
client.put_object(ACL='authenticated-read',
Bucket=bucket_name,
Body=b'ABCD',
Key='{}_direnum.json'.format(target))
client.put_object(ACL='authenticated-read',
Bucket=bucket_name,
Body=b'ABCD',
Key='{}_websearch.json'.format(target))
client.put_object(ACL='authenticated-read',
Bucket=bucket_name,
Body=b'ABCD',
Key='{}_portscan.json'.format(target))
client.put_object(ACL='authenticated-read',
Bucket=bucket_name,
Body=b'ABCD',
Key='{}_httpobservatory.json'.format(target))
client.put_object(ACL='authenticated-read',
Bucket=bucket_name,
Body=b'ABCD',
Key='{}_tlsobservatory.json'.format(target))
client.put_object(ACL='authenticated-read',
Bucket=bucket_name,
Body=b'ABCD',
Key='{}_sshobservatory.json'.format(target))
test_event = {"target": target}
test_context = None
results_handler = ResultsHandler(
s3_client=client,
bucket=bucket_name,
results_path=TEST_SCAN_RESULTS_BASE_PATH)
response = results_handler.getResults(test_event, test_context)
print(response)
assert type(response) is dict
assert response['statusCode'] == 200
url = urlparse(json.loads(response['body'])['url'])
query = parse_qs(url.query)
assert type(query) is dict
assert url.scheme == "https"
assert url.netloc == "{}.s3.amazonaws.com".format(bucket_name)
assert url.path == "/results/{}.tgz".format(target)
assert "AWSAccessKeyId" in query
assert "Expires" in query
assert "Signature" in query
def test_getResults_noResults(self, s3):
client, bucket, bucket_name = s3
target = "infosec.mozilla.org"
# No objects in the S3 bucket, should return no results
test_event = {"body": '{"target": "' + target + '"}'}
test_context = None
results_handler = ResultsHandler(
client, bucket_name, results_path=TEST_SCAN_RESULTS_BASE_PATH)
response = results_handler.getResults(test_event, test_context)
assert type(response) is dict
assert response['statusCode'] == 404
assert 'isBase64Encoded' not in response
assert 'Content-Disposition' not in response['headers']
assert response['headers']['Content-Type'] == "application/json"
assert 'no' in response['body'].lower()
def download_results(event, context):
results_handler = ResultsHandler(s3_client=S3_CLIENT, logger=logger)
response = results_handler.getResults(event, context)
return response