from __future__ import annotations

import unittest

from arr_web.repository import DAILY_DOWNLOAD_SQL, JOBS_SQL, MONTHLY_RUNS_SQL


class PortalRepositorySchemaTests(unittest.TestCase):
    def test_queries_use_arr_mvp_v1_relations(self) -> None:
        sql = " ".join((JOBS_SQL, DAILY_DOWNLOAD_SQL, MONTHLY_RUNS_SQL))
        for required in (
            "ingestion.processing_runs",
            "ingestion.artifacts",
            "finance.daily_versions",
            "finance.v_active_daily_facts",
            "finance.current_daily_versions",
        ):
            with self.subTest(required=required):
                self.assertIn(required, sql)

        for removed in (
            "finance.processing_jobs",
            "booking.file_objects",
            "finance.report_versions",
            "finance.report_channel_manifest",
        ):
            with self.subTest(removed=removed):
                self.assertNotIn(removed, sql)


if __name__ == "__main__":
    unittest.main(verbosity=2)