mirror of
https://github.com/neondatabase/neon.git
synced 2026-06-01 12:30:38 +00:00
257ade0688
This patch aims to:
* Unify connection & querying logic of ZenithPagerserver and Postgres.
* Mitigate changes to transaction machinery introduced in `psycopg2 >= 2.9`.
Now it's possible to acquire db connection using the corresponding
method:
```python
pg = postgres.create_start('main')
conn = pg.connect()
...
conn.close()
```
This pattern can be further improved with the help of `closing`:
```python
from contextlib import closing
pg = postgres.create_start('main')
with closing(pg.connect()) as conn:
...
```
All connections produced by this method will have autocommit
enabled by default.
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from contextlib import closing
|
|
|
|
pytest_plugins = ("fixtures.zenith_fixtures")
|
|
|
|
|
|
#
|
|
# Test restarting and recreating a postgres instance
|
|
#
|
|
def test_restart_compute(zenith_cli, pageserver, postgres, pg_bin):
|
|
zenith_cli.run(["branch", "test_restart_compute", "empty"])
|
|
|
|
pg = postgres.create_start('test_restart_compute')
|
|
print("postgres is running on 'test_restart_compute' branch")
|
|
|
|
with closing(pg.connect()) as conn:
|
|
with conn.cursor() as cur:
|
|
# Create table, and insert a row
|
|
cur.execute('CREATE TABLE foo (t text)')
|
|
cur.execute("INSERT INTO foo VALUES ('bar')")
|
|
|
|
# Stop and restart the Postgres instance
|
|
pg.stop().start()
|
|
|
|
with closing(pg.connect()) as conn:
|
|
with conn.cursor() as cur:
|
|
# We can still see the row
|
|
cur.execute('SELECT count(*) FROM foo')
|
|
assert cur.fetchone() == (1, )
|
|
|
|
# Insert another row
|
|
cur.execute("INSERT INTO foo VALUES ('bar2')")
|
|
cur.execute('SELECT count(*) FROM foo')
|
|
assert cur.fetchone() == (2, )
|
|
|
|
# Stop, and destroy the Postgres instance. Then recreate and restart it.
|
|
pg.stop_and_destroy().create_start('test_restart_compute')
|
|
|
|
with closing(pg.connect()) as conn:
|
|
with conn.cursor() as cur:
|
|
# We can still see the rows
|
|
cur.execute('SELECT count(*) FROM foo')
|
|
assert cur.fetchone() == (2, )
|