Commit 0f1aa790 authored by Jan Reimes's avatar Jan Reimes
Browse files

refactor(tests): update database imports for consistency and clarity

* Added MeetingDatabase import to conftest.py for proper database usage.
* Moved Specification-related imports to the correct module in test_specs_database.py.
parent d3bd9acc
Loading
Loading
Loading
Loading
+3 −2
Original line number Diff line number Diff line
@@ -9,7 +9,7 @@ from pathlib import Path
import pytest

from tdoc_crawler.config import CacheManager, register_cache_manager
from tdoc_crawler.database import TDocDatabase
from tdoc_crawler.database import TDocDatabase, MeetingDatabase
from tdoc_crawler.meetings.models import MeetingMetadata
from tdoc_crawler.models import WorkingGroup
from tdoc_crawler.tdocs.models import TDocMetadata
@@ -176,7 +176,7 @@ def populated_db(test_db_path: Path, sample_meetings: list[dict], sample_tdocs:
    return TDocDatabase(test_db_path)


def insert_sample_meetings(database: TDocDatabase, meetings: list[dict]) -> None:
def insert_sample_meetings(database: MeetingDatabase, meetings: list[dict]) -> None:
    """Insert sample meetings into database using the proper database API.

    This function uses db.upsert_meeting() instead of raw SQL to ensure
@@ -251,6 +251,7 @@ def insert_sample_meetings(database: TDocDatabase, meetings: list[dict]) -> None
            location=meeting_dict.get("location"),
            files_url=meeting_dict.get("files_url"),
            portal_url=None,
            tdoc_count=0,
        )

        # Use the proper database API instead of raw SQL
+1 −2
Original line number Diff line number Diff line
@@ -2,9 +2,8 @@

from pathlib import Path

from tdoc_crawler.models.specs import Specification, SpecificationDownload, SpecificationSourceRecord, SpecificationVersion

from tdoc_crawler.database.specs import SpecDatabase
from tdoc_crawler.specs.models import Specification, SpecificationDownload, SpecificationSourceRecord, SpecificationVersion
from tdoc_crawler.specs.query import SpecQueryFilters