Fix test again again

neondatabase · Jan 10, 2024 · ac39a37 · ac39a37
1 parent 76837dd
commit ac39a37
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/test_runner/regress/test_migrations.py b/test_runner/regress/test_migrations.py
@@ -6,16 +6,24 @@
 def test_migrations(neon_simple_env: NeonEnv):
     env = neon_simple_env
     env.neon_cli.create_branch("test_migrations", "empty")
+
     endpoint = env.endpoints.create("test_migrations")
+    log_path = endpoint.endpoint_path() / "compute.log"
+
     endpoint.respec(skip_pg_catalog_updates=False)
     endpoint.start()
+
     time.sleep(1)  # Sleep to let migrations run
 
     with endpoint.cursor() as cur:
         cur.execute("SELECT id FROM neon_migration.migration_id")
         migration_id = cur.fetchall()
         assert migration_id[0][0] == 2
 
+    with open(log_path, "r") as log_file:
+        logs = log_file.read()
+        assert "INFO start_compute:apply_config:handle_migrations: Ran 2 migrations" in logs
+
     endpoint.stop()
     endpoint.start()
     time.sleep(1)  # Sleep to let migrations run
@@ -24,8 +32,6 @@ def test_migrations(neon_simple_env: NeonEnv):
         migration_id = cur.fetchall()
         assert migration_id[0][0] == 2
 
-    log_path = endpoint.endpoint_path() / "compute.log"
     with open(log_path, "r") as log_file:
         logs = log_file.read()
-        assert "INFO start_compute:apply_config:handle_migrations: Ran 2 migrations" in logs
         assert "INFO start_compute:apply_config:handle_migrations: Ran 0 migrations" in logs