# -*- coding: utf-8 -*-
"""init_db：幂等创建请假考勤业务表与元数据。"""
from __future__ import annotations

import unittest

from _support import IsolatedDataRoot


class TestDbSmoke(unittest.TestCase):
    def test_init_db_idempotent_with_business_tables(self) -> None:
        with IsolatedDataRoot():
            from db.connection import get_conn, init_db
            from db.display_metadata import METADATA_TABLES, iter_user_tables
            from db.schema import BUSINESS_TABLES

            init_db()
            init_db()

            conn = get_conn()
            try:
                cur = conn.cursor()
                cur.execute(
                    "SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%'"
                )
                tables = {str(row[0]) for row in cur.fetchall()}
                self.assertIn("_jiangchang_tables", tables)
                self.assertIn("_jiangchang_columns", tables)
                for name in BUSINESS_TABLES:
                    self.assertIn(name, tables)
                self.assertNotIn("task_logs", tables)
                user_tables = list(iter_user_tables(cur))
                self.assertEqual(len(user_tables), len(BUSINESS_TABLES))
                leftover = tables - METADATA_TABLES - set(user_tables)
                self.assertEqual(leftover, set())
            finally:
                conn.close()


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