chore(cloud): verify integration change

2026-07-13 08:06:15 +08:00
parent 3c5f5509c2
commit 22d37ca95b
10 changed files with 60 additions and 33 deletions
@@ -74,7 +74,11 @@ def test_legacy_data_survives_upgrade_and_downgrade(tmp_path) -> None:
                    "('task-a', 'goal', null, :constraints, 'queued', null, null, "
                    "'2026-01-01T00:00:00+00:00')"
                ),
-                {"constraints": json.dumps({"driver_type": None, "capability_tags": []})},
+                {
+                    "constraints": json.dumps(
+                        {"driver_type": None, "capability_tags": []}
+                    )
+                },
            )
    finally:
        engine.dispose()
@@ -83,7 +87,9 @@ def test_legacy_data_survives_upgrade_and_downgrade(tmp_path) -> None:
    engine = create_engine(database_url)
    try:
        with engine.connect() as connection:
-            assert connection.scalar(text("select count(*) from host_registrations")) == 1
+            assert (
+                connection.scalar(text("select count(*) from host_registrations")) == 1
+            )
            assert connection.scalar(text("select count(*) from scheduled_tasks")) == 1
        task_columns = {
            column["name"] for column in inspect(engine).get_columns("scheduled_tasks")
@@ -96,7 +102,9 @@ def test_legacy_data_survives_upgrade_and_downgrade(tmp_path) -> None:
    engine = create_engine(database_url)
    try:
        with engine.connect() as connection:
-            assert connection.scalar(text("select count(*) from host_registrations")) == 1
+            assert (
+                connection.scalar(text("select count(*) from host_registrations")) == 1
+            )
            assert connection.scalar(text("select count(*) from scheduled_tasks")) == 1
        task_columns = {
            column["name"] for column in inspect(engine).get_columns("scheduled_tasks")