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)