mirror of
https://github.com/neondatabase/neon.git
synced 2026-05-28 02:20:42 +00:00
47bd307cb8
For better ergonomics. I always found it weird that we used UUID to actually mean a tenant or timeline ID. It worked because it happened to have the same length, 16 bytes, but it was hacky.
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
import os
|
|
from pathlib import Path
|
|
|
|
from fixtures.neon_fixtures import (
|
|
NeonEnvBuilder,
|
|
PgBin,
|
|
PortDistributor,
|
|
VanillaPostgres,
|
|
base_dir,
|
|
pg_distrib_dir,
|
|
)
|
|
from fixtures.types import ZTenantId
|
|
|
|
|
|
def test_wal_restore(
|
|
neon_env_builder: NeonEnvBuilder,
|
|
pg_bin: PgBin,
|
|
test_output_dir: Path,
|
|
port_distributor: PortDistributor,
|
|
):
|
|
env = neon_env_builder.init_start()
|
|
env.neon_cli.create_branch("test_wal_restore")
|
|
pg = env.postgres.create_start("test_wal_restore")
|
|
pg.safe_psql("create table t as select generate_series(1,300000)")
|
|
tenant_id = ZTenantId(pg.safe_psql("show neon.tenant_id")[0][0])
|
|
env.neon_cli.pageserver_stop()
|
|
port = port_distributor.get_port()
|
|
data_dir = test_output_dir / "pgsql.restored"
|
|
with VanillaPostgres(data_dir, PgBin(test_output_dir), port) as restored:
|
|
pg_bin.run_capture(
|
|
[
|
|
os.path.join(base_dir, "libs/utils/scripts/restore_from_wal.sh"),
|
|
os.path.join(pg_distrib_dir, "bin"),
|
|
str(test_output_dir / "repo" / "safekeepers" / "sk1" / str(tenant_id) / "*"),
|
|
str(data_dir),
|
|
str(port),
|
|
]
|
|
)
|
|
restored.start()
|
|
assert restored.safe_psql("select count(*) from t", user="cloud_admin") == [(300000,)]
|