def test_anonymise_user(app, mock_user_ctx, request_headers, objects):
"""Test anonymize_user preprocessor."""
with app.test_request_context(headers=request_headers['user']):
event = file_download_event_builder({}, app, objects[0])
event = anonymize_user(event)
assert 'user_id' not in event
assert 'user_agent' not in event
assert 'ip_address' not in event
assert event['visitor_id'] == \
'78d8045d684abd2eece923758f3cd781489df3a48e1278982466017f'
def test_anonymise_user(app, mock_user_ctx, request_headers, objects):
"""Test anonymize_user preprocessor."""
with app.test_request_context(headers=request_headers['user']):
event = file_download_event_builder({}, app, objects[0])
event = anonymize_user(event)
assert 'user_id' not in event
assert 'user_agent' not in event
assert 'ip_address' not in event
assert event['visitor_id'] == \
'78d8045d684abd2eece923758f3cd781489df3a48e1278982466017f'
def test_file_download_event_builder(app, mock_user_ctx, sequential_ids,
objects):
"""Test the file-download event builder."""
file_obj = objects[0]
file_obj.bucket_id = sequential_ids[0]
with app.test_request_context(headers=headers):
event = {}
with patch('datetime.datetime', NewDate):
file_download_event_builder(event, app, file_obj)
assert event == dict(
# When:
timestamp=NewDate.utcnow().isoformat(),
# What:
bucket_id=str(file_obj.bucket_id),
file_id=str(file_obj.file_id),
file_key=file_obj.key,
# Who:
**get_user())
def mock_event_queue(app, mock_datetime, request_headers, objects,
event_entrypoints, mock_user_ctx):
"""Create a mock queue containing a few file download events."""
mock_queue = Mock()
mock_queue.routing_key = 'stats-file-download'
with patch('datetime.datetime', mock_datetime), \
app.test_request_context(headers=request_headers['user']):
events = [file_download_event_builder({}, app, objects[0]) for idx
in range(100)]
mock_queue.consume.return_value = iter(events)
# Save the queued events for later tests
mock_queue.queued_events = deepcopy(events)
return mock_queue
def mock_event_queue(app, mock_datetime, request_headers, objects,
event_entrypoints, mock_user_ctx):
"""Create a mock queue containing a few file download events."""
mock_queue = Mock()
mock_queue.routing_key = 'stats-file-download'
with patch('datetime.datetime', mock_datetime), \
app.test_request_context(headers=request_headers['user']):
events = [
file_download_event_builder({}, app, objects[0])
for idx in range(100)
]
mock_queue.consume.return_value = iter(events)
# Save the queued events for later tests
mock_queue.queued_events = deepcopy(events)
return mock_queue
def build_event(headers):
with app.test_request_context(headers=headers):
event = file_download_event_builder({}, app, objects[0])
return flag_robots(event)
def test_referrer(app, mock_user_ctx, request_headers, objects):
"""Test referrer header."""
request_headers['user']['REFERER'] = 'example.com'
with app.test_request_context(headers=request_headers['user']):
event = file_download_event_builder({}, app, objects[0])
assert event['referrer'] == 'example.com'
def build_event(headers):
with app.test_request_context(headers=headers):
event = file_download_event_builder({}, app, objects[0])
return flag_machines(event)
