"""Shared pytest fixtures.""" import json import pytest TEST_EVENT = 'tests/sample_event.json' TEST_TRANSFORMED_EVENT = 'tests/sample_event_transformed.json' @pytest.fixture def test_event(): """Test event with valid data.""" with open(TEST_EVENT, 'r') as f: event = json.load(f) return event @pytest.fixture def test_transformed_event(): """Test transformed event with valid data.""" with open(TEST_TRANSFORMED_EVENT, 'r') as f: event = json.load(f) return event @pytest.fixture def test_transformed_event_no_extra_fields(): """Test transformed event with valid data.""" with open(TEST_TRANSFORMED_EVENT, 'r') as f: event = json.load(f) del event['CorrelationId__c'] del event['RecordTypeId'] del event['KafkaMessageHeaders__c'] return event @pytest.fixture def test_event_no_required_field(): """Test event with not valid data.""" with open(TEST_EVENT, 'r') as f: event = json.load(f) body = json.loads(event['body']) del body['email'] event['body'] = json.dumps(body) return event