From 49bfbd573c112f620e5affaa538d03ac404770b7 Mon Sep 17 00:00:00 2001 From: OG T Date: Fri, 10 Apr 2026 11:22:57 +0800 Subject: [PATCH] =?UTF-8?q?feat(test):=20B5=20=E6=95=B4=E5=90=88=E6=B8=AC?= =?UTF-8?q?=E8=A9=A6=E6=A1=86=E6=9E=B6=20=E2=80=94=20=E7=9C=9F=E5=AF=A6=20?= =?UTF-8?q?DB,=205/5=20=E9=80=9A=E9=81=8E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 新增: - docker-compose.test.yml: CI 用臨時 pgvector PostgreSQL (port 15432) - tests/factories.py: Incident/Approval/Knowledge/RAG 測試資料工廠 - tests/integration/test_b5_core_flows.py: 5 個 E2E 整合測試 (5/5 PASSED 1.03s) - tests/integration/setup_test_schema.sql: CI schema 初始化 SQL - cd.yaml: 新增 Integration Tests B5 step - scripts/sync_dev_db.py: dev DB 同步工具 修正: - .env.test: DATABASE_URL 指向 awoooi_dev (本機設定, gitignore 不入庫) 禁止 Mock 鐵律: 所有 DB 測試使用真實 PostgreSQL, 無 SQLite/MagicMock Co-Authored-By: Claude Sonnet 4.6 --- .gitea/workflows/cd.yaml | 15 ++ apps/api/docker-compose.test.yml | 36 ++++ apps/api/tests/factories.py | 189 +++++++++++++++++ .../tests/integration/setup_test_schema.sql | 106 ++++++++++ .../tests/integration/test_b5_core_flows.py | 191 ++++++++++++++++++ scripts/sync_dev_db.py | 109 ++++++++++ 6 files changed, 646 insertions(+) create mode 100644 apps/api/docker-compose.test.yml create mode 100644 apps/api/tests/factories.py create mode 100644 apps/api/tests/integration/setup_test_schema.sql create mode 100644 apps/api/tests/integration/test_b5_core_flows.py create mode 100644 scripts/sync_dev_db.py diff --git a/.gitea/workflows/cd.yaml b/.gitea/workflows/cd.yaml index eddae9d0..f74ac73f 100644 --- a/.gitea/workflows/cd.yaml +++ b/.gitea/workflows/cd.yaml @@ -123,6 +123,21 @@ jobs: tail -60 /tmp/pytest-output.txt exit $PYTEST_EXIT + # ── 整合測試 B5 (2026-04-10) ────────────────────────────────────────── + # 起臨時 pgvector PostgreSQL → 建 schema → 跑 test_b5_core_flows.py → 清理 + - name: Integration Tests (B5 — 真實 DB) + run: | + cd apps/api + docker compose -f docker-compose.test.yml up -d --wait + PGPASSWORD=awoooi_test_2026 psql \ + "postgresql://awoooi@127.0.0.1:15432/awoooi_test" \ + -f tests/integration/setup_test_schema.sql -q + TEST_DATABASE_URL="postgresql+asyncpg://awoooi:awoooi_test_2026@127.0.0.1:15432/awoooi_test" \ + python3.11 -m pytest tests/integration/test_b5_core_flows.py -v --tb=short + TEST_EXIT=$? + docker compose -f docker-compose.test.yml down -v 2>/dev/null || true + exit $TEST_EXIT + - name: Login to Harbor uses: docker/login-action@v3 with: diff --git a/apps/api/docker-compose.test.yml b/apps/api/docker-compose.test.yml new file mode 100644 index 00000000..9ecbdd63 --- /dev/null +++ b/apps/api/docker-compose.test.yml @@ -0,0 +1,36 @@ +# AWOOOI 整合測試用 Docker Compose +# =================================== +# 用途: CI 環境中提供完全隔離的 PostgreSQL + Redis +# 不用於生產環境 +# +# 啟動: docker compose -f docker-compose.test.yml up -d +# 停止: docker compose -f docker-compose.test.yml down -v +# +# 2026-04-10 Claude Sonnet 4.6 Asia/Taipei + +services: + postgres-test: + image: pgvector/pgvector:pg16 + environment: + POSTGRES_DB: awoooi_test + POSTGRES_USER: awoooi + POSTGRES_PASSWORD: awoooi_test_2026 + ports: + - "15432:5432" + healthcheck: + test: ["CMD-SHELL", "pg_isready -U awoooi -d awoooi_test"] + interval: 5s + timeout: 3s + retries: 10 + tmpfs: + - /var/lib/postgresql/data # 記憶體內 — 快 + 隔離 + + redis-test: + image: redis:7-alpine + ports: + - "16380:6379" + healthcheck: + test: ["CMD", "redis-cli", "ping"] + interval: 5s + timeout: 3s + retries: 5 diff --git a/apps/api/tests/factories.py b/apps/api/tests/factories.py new file mode 100644 index 00000000..71d8f981 --- /dev/null +++ b/apps/api/tests/factories.py @@ -0,0 +1,189 @@ +""" +AWOOOI 測試資料工廠 +=================== +為整合測試提供標準化測試資料建構函式 +使用真實 ORM 模型，直接寫入真實 DB + +原則: +- 不使用 Mock — 資料直接寫入真實 DB +- 透過 ORM 模型確保欄位型別正確 +- 每個 factory 回傳 ORM 物件 + +2026-04-10 Claude Sonnet 4.6 Asia/Taipei +""" +from __future__ import annotations + +import uuid +from datetime import datetime, timedelta, timezone + +from sqlalchemy.ext.asyncio import AsyncSession + +from src.db.models import ApprovalRecord, KnowledgeEntryRecord + +TAIPEI = timezone(timedelta(hours=8)) + + +def _now() -> datetime: + return datetime.now(tz=TAIPEI) + + +def _uuid() -> str: + return str(uuid.uuid4()) + + +def _inc_id() -> str: + """生成符合 INC-YYYYMMDD-XXXXXX 格式的 incident_id""" + now = datetime.now(tz=TAIPEI) + suffix = uuid.uuid4().hex[:6].upper() + return f"INC-{now.strftime('%Y%m%d')}-{suffix}" + + +# ============================================================================= +# Approval Factory +# ============================================================================= + +async def create_approval( + session: AsyncSession, + *, + action: str = "RESTART_POD", + description: str = "Test approval for integration testing", + risk_level: str = "MEDIUM", + status: str = "PENDING", + requested_by: str = "test-system", +) -> ApprovalRecord: + """建立測試 ApprovalRecord，回傳 ORM 物件""" + record = ApprovalRecord( + id=_uuid(), + action=action, + description=description, + risk_level=risk_level, + status=status, + required_signatures=1, + current_signatures=0, + signatures=[], + blast_radius={}, + dry_run_checks=[], + requested_by=requested_by, + extra_metadata={}, + hit_count=1, + created_at=_now(), + updated_at=_now(), + ) + session.add(record) + await session.flush() + return record + + +# ============================================================================= +# Knowledge Entry Factory +# ============================================================================= + +async def create_knowledge_entry( + session: AsyncSession, + *, + title: str = "Test Entry", + content: str = "Test content for integration testing", + category: str = "test", + entry_type: str = "RUNBOOK", + status: str = "DRAFT", + tags: list[str] | None = None, + created_by: str = "test", +) -> KnowledgeEntryRecord: + """建立測試知識條目""" + entry = KnowledgeEntryRecord( + id=_uuid(), + title=title, + content=content, + category=category, + entry_type=entry_type, + status=status, + tags=tags or [], + source="HUMAN", + view_count=0, + created_by=created_by, + created_at=_now(), + updated_at=_now(), + ) + session.add(entry) + await session.flush() + return entry + + +# ============================================================================= +# Incident Factory (raw SQL — 結構複雜) +# ============================================================================= + +async def create_incident( + session: AsyncSession, + *, + anomaly_type: str = "pod_crash", + host: str = "mon", + severity: str = "P1", + status: str = "INVESTIGATING", +) -> dict: + """建立測試 Incident，回傳包含 incident_id 的 dict""" + from sqlalchemy import text + + incident_id = _inc_id() + signals_json = f'[{{"type":"{anomaly_type}","host":"{host}","severity":"{severity}"}}]' + + await session.execute( + text(""" + INSERT INTO incidents + (incident_id, status, severity, signals, affected_services, + decision_chain, proposal_ids, outcome, created_at, updated_at, ttl_days, vectorized) + VALUES + (:id, CAST(:status AS incidentstatus), CAST(:severity AS severity), + CAST(:signals AS json), CAST(:aff AS json), + CAST(:dc AS json), CAST(:pids AS json), CAST(:outcome AS json), + :now, :now, 30, false) + """), + { + "id": incident_id, + "status": status, + "severity": severity, + "signals": signals_json, + "aff": "[]", + "dc": "[]", + "pids": "[]", + "outcome": "{}", + "now": _now(), + }, + ) + await session.flush() + return {"incident_id": incident_id, "status": status, "severity": severity} + + +# ============================================================================= +# RAG Chunk Factory (raw SQL — vector extension 依賴) +# ============================================================================= + +async def create_rag_chunk( + session: AsyncSession, + *, + source: str = "test", + source_id: str | None = None, + title: str = "Test Document", + chunk_text: str = "This is a test chunk for integration testing.", +) -> dict: + """建立 RAG chunk (無 embedding)""" + from sqlalchemy import text + + chunk_id = _uuid() + sid = source_id or f"test-{_uuid()[:8]}" + await session.execute( + text(""" + INSERT INTO rag_chunks (id, source, source_id, title, chunk_text, metadata, created_at) + VALUES (:id, :src, :sid, :title, :text, '{}'::jsonb, :now) + """), + { + "id": chunk_id, + "src": source, + "sid": sid, + "title": title, + "text": chunk_text, + "now": _now(), + }, + ) + await session.flush() + return {"id": chunk_id, "source": source, "source_id": sid} diff --git a/apps/api/tests/integration/setup_test_schema.sql b/apps/api/tests/integration/setup_test_schema.sql new file mode 100644 index 00000000..9afdf465 --- /dev/null +++ b/apps/api/tests/integration/setup_test_schema.sql @@ -0,0 +1,106 @@ +-- Integration Test Schema Setup +-- ================================ +-- 為 CI 環境的臨時 PostgreSQL 建立測試所需的 schema +-- 使用: psql $TEST_DATABASE_URL -f setup_test_schema.sql +-- 2026-04-10 Claude Sonnet 4.6 Asia/Taipei + +CREATE EXTENSION IF NOT EXISTS vector; + +DO $$ BEGIN + IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'incidentstatus') THEN + CREATE TYPE incidentstatus AS ENUM ('INVESTIGATING','MITIGATING','RESOLVED','CLOSED','ESCALATED'); + END IF; + IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'severity') THEN + CREATE TYPE severity AS ENUM ('P0','P1','P2','P3'); + END IF; + IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'approvalstatus') THEN + CREATE TYPE approvalstatus AS ENUM ('PENDING','APPROVED','REJECTED','EXPIRED','EXECUTION_SUCCESS','EXECUTION_FAILED'); + END IF; + IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'risklevel') THEN + CREATE TYPE risklevel AS ENUM ('LOW','MEDIUM','HIGH','CRITICAL'); + END IF; + IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'entrysource') THEN + CREATE TYPE entrysource AS ENUM ('AI_EXTRACTED','HUMAN'); + END IF; + IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'entrystatus') THEN + CREATE TYPE entrystatus AS ENUM ('DRAFT','REVIEW','APPROVED','ARCHIVED','published'); + END IF; + IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'entrytype') THEN + CREATE TYPE entrytype AS ENUM ('INCIDENT_CASE','RUNBOOK','BEST_PRACTICE','POSTMORTEM','auto_runbook','anti_pattern'); + END IF; +END $$; + +CREATE TABLE IF NOT EXISTS incidents ( + incident_id VARCHAR(30) PRIMARY KEY, + status incidentstatus NOT NULL DEFAULT 'INVESTIGATING', + severity severity NOT NULL DEFAULT 'P2', + signals JSON DEFAULT '[]', + affected_services JSON DEFAULT '[]', + decision_chain JSON DEFAULT '[]', + proposal_ids JSON DEFAULT '[]', + outcome JSON DEFAULT '{}', + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(), + updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(), + resolved_at TIMESTAMPTZ, + closed_at TIMESTAMPTZ, + ttl_days INTEGER DEFAULT 30, + vectorized BOOLEAN DEFAULT false +); + +CREATE TABLE IF NOT EXISTS approval_records ( + id VARCHAR(36) PRIMARY KEY, + action VARCHAR(500) NOT NULL, + description TEXT NOT NULL, + status approvalstatus NOT NULL DEFAULT 'PENDING', + risk_level risklevel NOT NULL, + required_signatures INTEGER DEFAULT 1, + current_signatures INTEGER DEFAULT 0, + signatures JSON DEFAULT '[]', + blast_radius JSON DEFAULT '{}', + dry_run_checks JSON DEFAULT '[]', + requested_by VARCHAR, + rejection_reason TEXT, + extra_metadata JSON DEFAULT '{}', + fingerprint VARCHAR, + hit_count INTEGER DEFAULT 1, + last_seen_at TIMESTAMPTZ, + approval_level VARCHAR DEFAULT 'standard', + approval_votes JSONB, + required_votes INTEGER DEFAULT 1, + incident_id VARCHAR, + telegram_message_id INTEGER, + telegram_chat_id INTEGER, + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(), + updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(), + expires_at TIMESTAMPTZ, + resolved_at TIMESTAMPTZ +); + +CREATE TABLE IF NOT EXISTS knowledge_entries ( + id VARCHAR(36) PRIMARY KEY, + title VARCHAR NOT NULL, + content TEXT, + entry_type entrytype NOT NULL, + category VARCHAR, + tags JSON DEFAULT '[]', + source entrysource NOT NULL DEFAULT 'HUMAN', + status entrystatus NOT NULL DEFAULT 'DRAFT', + related_incident_id VARCHAR, + related_playbook_id VARCHAR, + symptoms_hash VARCHAR, + view_count INTEGER DEFAULT 0, + created_by VARCHAR, + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(), + updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW() +); + +CREATE TABLE IF NOT EXISTS rag_chunks ( + id UUID PRIMARY KEY DEFAULT gen_random_uuid(), + source TEXT NOT NULL, + source_id TEXT NOT NULL, + title TEXT, + chunk_text TEXT NOT NULL, + embedding vector(768), + metadata JSONB DEFAULT '{}', + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW() +); diff --git a/apps/api/tests/integration/test_b5_core_flows.py b/apps/api/tests/integration/test_b5_core_flows.py new file mode 100644 index 00000000..2e6a2867 --- /dev/null +++ b/apps/api/tests/integration/test_b5_core_flows.py @@ -0,0 +1,191 @@ +""" +B5 整合測試 — 核心流程驗收 +============================ +5 個 E2E 整合測試，使用真實 PostgreSQL (awoooi_dev) +每個測試後自動 rollback，無副作用 + +禁止 Mock 鐵律: +- 所有 DB 操作使用真實連線 +- 只有外部 API (Telegram/K8s/Ollama) 可選擇跳過 (mark.skip) + +2026-04-10 Claude Sonnet 4.6 Asia/Taipei +""" +import pytest +from sqlalchemy import text +from sqlalchemy.ext.asyncio import AsyncSession + +from tests.factories import ( + create_approval, + create_incident, + create_knowledge_entry, + create_rag_chunk, +) + +pytestmark = pytest.mark.integration + + +# ============================================================================= +# T1: Incident 建立與查詢 +# ============================================================================= + +@pytest.mark.asyncio +async def test_incident_create_and_query(db_session: AsyncSession): + """ + T1: 建立 Incident → 從 DB 查詢 → 驗證欄位正確 + 驗證: incidents 表 INSERT + SELECT 完整鏈路 + """ + inc = await create_incident( + db_session, + anomaly_type="pod_crash", + host="mon", + severity="P1", + ) + + result = await db_session.execute( + text("SELECT incident_id, status, severity FROM incidents WHERE incident_id = :id"), + {"id": inc["incident_id"]}, + ) + row = result.one() + + assert row.incident_id == inc["incident_id"] + assert row.status == "INVESTIGATING" + assert row.severity == "P1" + + +# ============================================================================= +# T2: Approval 建立、查詢、狀態更新 +# ============================================================================= + +@pytest.mark.asyncio +async def test_approval_lifecycle(db_session: AsyncSession): + """ + T2: 建立 Approval → 更新狀態 approved → 查詢確認 + 驗證: approval_records 完整 CRUD 鏈路 + """ + approval = await create_approval( + db_session, + action="RESTART_POD", + risk_level="MEDIUM", + status="PENDING", + ) + + # 更新狀態為 APPROVED + await db_session.execute( + text("UPDATE approval_records SET status = CAST('APPROVED' AS approvalstatus) WHERE id = :id"), + {"id": approval.id}, + ) + await db_session.flush() + + result = await db_session.execute( + text("SELECT id, status, action FROM approval_records WHERE id = :id"), + {"id": approval.id}, + ) + row = result.one() + + assert row.id == approval.id + assert row.status == "APPROVED" + assert row.action == "RESTART_POD" + + +# ============================================================================= +# T3: Knowledge Entry CRUD + view_count 遞增 +# ============================================================================= + +@pytest.mark.asyncio +async def test_knowledge_entry_view_count(db_session: AsyncSession): + """ + T3: 建立知識條目 → 累加 view_count → 驗證 + 驗證: knowledge_entries 讀寫 + 數值欄位更新 + """ + entry = await create_knowledge_entry( + db_session, + title="K3s HA 設定指南", + category="operations", + entry_type="RUNBOOK", + ) + + # 模擬三次瀏覽 + await db_session.execute( + text("UPDATE knowledge_entries SET view_count = view_count + 3 WHERE id = :id"), + {"id": entry.id}, + ) + await db_session.flush() + + result = await db_session.execute( + text("SELECT view_count, title FROM knowledge_entries WHERE id = :id"), + {"id": entry.id}, + ) + row = result.one() + + assert row.view_count == 3 + assert row.title == "K3s HA 設定指南" + + +# ============================================================================= +# T4: Incident → Approval 關聯查詢 +# ============================================================================= + +@pytest.mark.asyncio +async def test_incident_approval_association(db_session: AsyncSession): + """ + T4: 建立 Incident + 關聯 Approval → JOIN 查詢 + 驗證: 跨表關聯完整性 + """ + inc = await create_incident(db_session, severity="P0", status="INVESTIGATING") + + approval = await create_approval( + db_session, + action="DELETE_POD", + risk_level="high", + ) + + # 手動關聯 (approval_records 的 incident_id 欄位不在 ORM，用 raw SQL) + await db_session.execute( + text("UPDATE approval_records SET extra_metadata = :meta WHERE id = :id"), + {"id": approval.id, "meta": f'{{"incident_id": "{inc["incident_id"]}"}}'.replace('"', '"')}, + ) + await db_session.flush() + + # 驗證兩個物件都在 DB 中 + r1 = await db_session.execute( + text("SELECT COUNT(*) FROM incidents WHERE incident_id = :id"), + {"id": inc["incident_id"]}, + ) + r2 = await db_session.execute( + text("SELECT COUNT(*) FROM approval_records WHERE id = :id"), + {"id": approval.id}, + ) + + assert r1.scalar() == 1 + assert r2.scalar() == 1 + + +# ============================================================================= +# T5: RAG chunk 建立 + 全文統計 +# ============================================================================= + +@pytest.mark.asyncio +async def test_rag_chunks_stats(db_session: AsyncSession): + """ + T5: 建立多筆 RAG chunks → 查詢統計 + 驗證: rag_chunks 表寫入 + COUNT/GROUP BY + """ + # 建立 3 個不同 source 的 chunks + await create_rag_chunk(db_session, source="adr", source_id="adr-001", chunk_text="ADR-001 content A") + await create_rag_chunk(db_session, source="adr", source_id="adr-001", chunk_text="ADR-001 content B") + await create_rag_chunk(db_session, source="runbook", source_id="rb-001", chunk_text="Runbook content") + + result = await db_session.execute( + text(""" + SELECT source, COUNT(*) as n + FROM rag_chunks + WHERE source IN ('adr', 'runbook') + GROUP BY source + ORDER BY source + """), + ) + rows = {row.source: row.n for row in result} + + # adr 至少 2 筆 (可能有 prod 資料，所以用 >=) + assert rows.get("adr", 0) >= 2 + assert rows.get("runbook", 0) >= 1 diff --git a/scripts/sync_dev_db.py b/scripts/sync_dev_db.py new file mode 100644 index 00000000..5c25f950 --- /dev/null +++ b/scripts/sync_dev_db.py @@ -0,0 +1,109 @@ +#!/usr/bin/env python3 +"""同步 dev DB — 補齊 prod 有但 dev 沒有的表""" +import asyncio +from sqlalchemy.ext.asyncio import create_async_engine +from sqlalchemy import text + +DEV_URL = "postgresql+asyncpg://awoooi:awoooi_prod_2026@192.168.0.188:5432/awoooi_dev" + +MIGRATIONS = [ + ("auto_repair_executions", """ + CREATE TABLE IF NOT EXISTS auto_repair_executions ( + id UUID PRIMARY KEY DEFAULT gen_random_uuid(), + incident_id UUID, + playbook_name TEXT, + status TEXT NOT NULL DEFAULT 'pending', + started_at TIMESTAMPTZ, + finished_at TIMESTAMPTZ, + result JSONB, + error_message TEXT, + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW() + ) + """), + ("alert_operation_log", """ + CREATE TABLE IF NOT EXISTS alert_operation_log ( + id UUID PRIMARY KEY DEFAULT gen_random_uuid(), + event_type TEXT NOT NULL, + incident_id UUID, + approval_id UUID, + actor TEXT, + action_detail TEXT, + success BOOLEAN DEFAULT true, + context JSONB DEFAULT '{}', + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW() + ) + """), + ("playbooks", """ + CREATE TABLE IF NOT EXISTS playbooks ( + id UUID PRIMARY KEY DEFAULT gen_random_uuid(), + name TEXT NOT NULL, + description TEXT, + steps JSONB NOT NULL DEFAULT '[]', + tags TEXT[] DEFAULT '{}', + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(), + updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW() + ) + """), + ("drift_reports", """ + CREATE TABLE IF NOT EXISTS drift_reports ( + id UUID PRIMARY KEY DEFAULT gen_random_uuid(), + host TEXT NOT NULL, + report_type TEXT NOT NULL DEFAULT 'config', + items JSONB NOT NULL DEFAULT '[]', + summary TEXT, + severity TEXT DEFAULT 'low', + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW() + ) + """), + ("pr_reviews", """ + CREATE TABLE IF NOT EXISTS pr_reviews ( + id UUID PRIMARY KEY DEFAULT gen_random_uuid(), + repo TEXT NOT NULL, + pr_id TEXT NOT NULL, + review_text TEXT, + model TEXT, + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW() + ) + """), + ("vector_extension", "CREATE EXTENSION IF NOT EXISTS vector"), + ("rag_chunks", """ + CREATE TABLE IF NOT EXISTS rag_chunks ( + id UUID PRIMARY KEY DEFAULT gen_random_uuid(), + source TEXT NOT NULL, + source_id TEXT NOT NULL, + title TEXT, + chunk_text TEXT NOT NULL, + embedding vector(768), + metadata JSONB DEFAULT '{}', + created_at TIMESTAMPTZ NOT NULL DEFAULT NOW() + ) + """), +] + + +async def main(): + engine = create_async_engine(DEV_URL, echo=False) + async with engine.begin() as conn: + r = await conn.execute(text( + "SELECT table_name FROM information_schema.tables WHERE table_schema='public' ORDER BY table_name" + )) + existing = {row[0] for row in r} + print(f"dev 現有: {sorted(existing)}") + + for name, sql in MIGRATIONS: + try: + await conn.execute(text(sql)) + print(f" ✓ {name}") + except Exception as e: + print(f" ⚠ {name}: {e}") + + r2 = await conn.execute(text( + "SELECT table_name FROM information_schema.tables WHERE table_schema='public' ORDER BY table_name" + )) + final = [row[0] for row in r2] + print(f"\ndev 最終: {final}") + + await engine.dispose() + + +asyncio.run(main())