SFR-2481/save_test_data

processes/__init__.py

@@ -8,7 +8,6 @@
 from .cluster import ClusterProcess
 from .local_development.local_development_setup import LocalDevelopmentSetupProcess
 from .local_development.seed_local_data import SeedLocalDataProcess
-from .local_development.seed_test_data import SeedTestDataProcess
 from .file.s3_files import S3Process
 from .api import APIProcess
 from .ingest.muse import MUSEProcess

tests/conftest.py

@@ -1,8 +1,17 @@
 import os
 import pytest
-
+from datetime import datetime, timezone
+import json
+from uuid import uuid4
+from processes import ClusterProcess
+from model import Record, Item
+from logger import create_log
+from managers import DBManager
 from load_env import load_env_file
 
+logger = create_log(__name__)
+
+TEST_SOURCE = 'test_source'
 
 def pytest_addoption(parser):
     parser.addoption('--env', action='store', default='local', help='Environment to use for tests')
@@ -16,3 +25,69 @@ def setup_env(pytestconfig, request):
 
     if not running_unit_tests and environment in ['local', 'qa']:
         load_env_file(environment, file_string=f'config/{environment}.yaml')
+
+@pytest.fixture(scope='module')
+def db_manager():
+    db_manager = DBManager()
+    db_manager.createSession()
+    yield db_manager
+    db_manager.close_connection()
+
+@pytest.fixture(scope='module')
+def seed_test_data(db_manager):
+    flags = { 'catalog': False, 'download': False, 'reader': False, 'embed': True }
+    test_data = {
+        'title': 'test data 1',
+        'uuid': uuid4(),
+        'frbr_status': 'complete',
+        'cluster_status': False,
+        "source": TEST_SOURCE,
+        'authors': ['Ayan||true'],
+        'languages': ['Serbian'],
+        'dates': ['1907-|publication_date'],
+        'publisher': ['Project Gutenberg Literary Archive Foundation||'],
+        'identifiers': [],
+        'source_id': '4064148285|test',
+        'contributors': ['Metropolitan Museum of Art (New York, N.Y.)|||contributor','Metropolitan Museum of Art (New York, N.Y.)|||repository','Thomas J. Watson Library|||provider'],
+        'extent': ('11, 164 p. ;'),
+        'is_part_of': ['Tauchnitz edition|Vol. 4560|volume'],
+        'abstract': ['test abstract 1', 'test abstract 2'],
+        'subjects': ['test subjects 1||'],
+        'rights': ('hathitrust|public_domain|expiration of copyright term for non-US work with corporate author|Public Domain|2021-10-02 05:25:13'),
+        'has_part': [f'1|example.com/1.pdf|{TEST_SOURCE}|text/html|{json.dumps(flags)}']
+    }
+
+    existing_record = db_manager.session.query(Record).filter_by(source_id=test_data['source_id']).first()
+
+    if existing_record:
+        for key, value in test_data.items():
+            if key != 'uuid' and hasattr(existing_record, key):
+                setattr(existing_record, key, value)
+        existing_record.date_modified = datetime.now(timezone.utc).replace(tzinfo=None)
+        test_data['uuid'] = existing_record.uuid
+        test_record = existing_record
+    else:
+        test_record = Record(**test_data)
+        db_manager.session.add(test_record)
+
+    db_manager.session.commit()
+
+    item = db_manager.session.query(Item).filter_by(record_id=test_record.id).first()
+    edition_id = str(item.edition_id) if item else None
+
+    return {
+        'edition_id': edition_id,
+        'uuid': str(test_data['uuid'])
+    }
+
+@pytest.fixture(scope='module')
+def seeded_edition_id(request, seed_test_data):
+    if 'functional' in request.keywords or 'integration' in request.keywords:
+        return seed_test_data['edition_id']
+    return None
+
+@pytest.fixture(scope='module')
+def seeded_uuid(request, seed_test_data):
+    if 'functional' in request.keywords or 'integration' in request.keywords:
+        return seed_test_data['uuid']
+    return None

tests/integration/api/test_get_edition.py

@@ -1,17 +1,19 @@
 import pytest
 import requests
+import json
 from .constants import API_URL
 from .utils import assert_response_status
 
+@pytest.mark.integration
 @pytest.mark.parametrize("endpoint, expected_status", [
-    ("/editions/1982731", 200),
+    ("/editions/{seeded_edition_id}", 200),
     ("/editions/00000000-0000-0000-0000-000000000000", 400),
     ("/editions/invalid_id_format", 400),
     ("/editions/", 404),
     ("/editions/%$@!*", 400)
 ])
-def test_get_edition(endpoint, expected_status):
-    url = API_URL + endpoint
+def test_get_edition(endpoint, expected_status, seeded_edition_id):
+    url = API_URL + endpoint.format(seeded_edition_id=seeded_edition_id)
     response = requests.get(url)
 
     assert response.status_code is not None