mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-06 13:02:55 +00:00
66ec135676
Instead of having a lot of separate fixtures for setting up the page server, the compute nodes, the safekeepers etc., have one big ZenithEnv object that encapsulates the whole environment. Every test either uses a shared "zenith_simple_env" fixture, which contains the default setup of a pageserver with no authentication, and no safekeepers. Tests that want to use safekeepers or authentication set up a custom test-specific ZenithEnv fixture. Gathering information about the whole environment into one object makes some things simpler. For example, when a new compute node is created, you no longer need to pass the 'wal_acceptors' connection string as argument to the 'postgres.create_start' function. The 'create_start' function fetches that information directly from the ZenithEnv object.
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
from contextlib import closing
|
|
|
|
import pytest
|
|
|
|
from fixtures.zenith_fixtures import ZenithEnvBuilder
|
|
|
|
|
|
@pytest.mark.parametrize('with_wal_acceptors', [False, True])
|
|
def test_tenants_normal_work(zenith_env_builder: ZenithEnvBuilder, with_wal_acceptors: bool):
|
|
if with_wal_acceptors:
|
|
zenith_env_builder.num_safekeepers = 3
|
|
|
|
env = zenith_env_builder.init()
|
|
"""Tests tenants with and without wal acceptors"""
|
|
tenant_1 = env.create_tenant()
|
|
tenant_2 = env.create_tenant()
|
|
|
|
env.zenith_cli([
|
|
"branch",
|
|
f"test_tenants_normal_work_with_wal_acceptors{with_wal_acceptors}",
|
|
"main",
|
|
f"--tenantid={tenant_1}"
|
|
])
|
|
env.zenith_cli([
|
|
"branch",
|
|
f"test_tenants_normal_work_with_wal_acceptors{with_wal_acceptors}",
|
|
"main",
|
|
f"--tenantid={tenant_2}"
|
|
])
|
|
|
|
pg_tenant1 = env.postgres.create_start(
|
|
f"test_tenants_normal_work_with_wal_acceptors{with_wal_acceptors}",
|
|
None, # branch name, None means same as node name
|
|
tenant_1,
|
|
)
|
|
pg_tenant2 = env.postgres.create_start(
|
|
f"test_tenants_normal_work_with_wal_acceptors{with_wal_acceptors}",
|
|
None, # branch name, None means same as node name
|
|
tenant_2,
|
|
)
|
|
|
|
for pg in [pg_tenant1, pg_tenant2]:
|
|
with closing(pg.connect()) as conn:
|
|
with conn.cursor() as cur:
|
|
# we rely upon autocommit after each statement
|
|
# as waiting for acceptors happens there
|
|
cur.execute("CREATE TABLE t(key int primary key, value text)")
|
|
cur.execute("INSERT INTO t SELECT generate_series(1,100000), 'payload'")
|
|
cur.execute("SELECT sum(key) FROM t")
|
|
assert cur.fetchone() == (5000050000, )
|