mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-16 01:42:55 +00:00
2fd8e24c8f
## Problem I didn't know about `wait_until` and was relying on `sleep` to wait for stuff. This caused some tests to be flaky. https://github.com/neondatabase/neon/issues/6561 ## Summary of changes Switch to `wait_until`, this should make it tests less flaky
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import time
|
|
|
|
from fixtures.neon_fixtures import NeonEnv
|
|
|
|
|
|
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, features=["migrations"])
|
|
endpoint.start()
|
|
|
|
endpoint.wait_for_migrations()
|
|
|
|
num_migrations = 3
|
|
|
|
with endpoint.cursor() as cur:
|
|
cur.execute("SELECT id FROM neon_migration.migration_id")
|
|
migration_id = cur.fetchall()
|
|
assert migration_id[0][0] == num_migrations
|
|
|
|
with open(log_path, "r") as log_file:
|
|
logs = log_file.read()
|
|
assert "INFO handle_migrations: Ran 3 migrations" in logs
|
|
|
|
endpoint.stop()
|
|
endpoint.start()
|
|
# We don't have a good way of knowing that the migrations code path finished executing
|
|
# in compute_ctl in the case that no migrations are being run
|
|
time.sleep(1)
|
|
with endpoint.cursor() as cur:
|
|
cur.execute("SELECT id FROM neon_migration.migration_id")
|
|
migration_id = cur.fetchall()
|
|
assert migration_id[0][0] == num_migrations
|
|
|
|
with open(log_path, "r") as log_file:
|
|
logs = log_file.read()
|
|
assert "INFO handle_migrations: Ran 0 migrations" in logs
|