"""Conftest.""" import json import os from bulk_metadata_ingester_common.models.bulk_release import BulkRelease import pytest KEY = os.environ.get('KEY', 'parsed_csv/ALL_DDEX_TO_CSV_2023-02-10_2023-03-01.json') # noqa BUCKET = os.environ.get('BUCKET', 'prod-orcd-fluve-drop') ITEM_INDEX = '7892920191004_____1362655' DATA_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data',) def get_json(json_file='single_release.json'): """Get JSON helper method.""" with open(os.path.join(DATA_DIR, json_file)) as fd: return fd.read() @pytest.fixture def test_release_json(json_file='single_release.json'): """Parse given XML file using xmltodict.parse().""" return get_json(json_file) @pytest.fixture def test_event(): """JSON representing a single release event.""" return { 'key': KEY, 'bucket': BUCKET, 'item_index': ITEM_INDEX, 'execution_name': 'exec-1234', 'state_machine_name': 'sfn-bulk-metadata-ingester', 'correlation_id': 'unittest-1234' } @pytest.fixture def test_model(json_file='single_release.json'): """Named Tuple model representing a release.""" test_json = json.loads(get_json())[ITEM_INDEX] return BulkRelease(json_release_rows=test_json)