From ccb2eea7fded8d89d82c8d0f38f2eacfc365878d Mon Sep 17 00:00:00 2001
From: anastasia <lubennikovaav@gmail.com>
Date: Tue, 25 May 2021 12:09:10 +0300
Subject: [PATCH] Add test_isolation that runs pg_isolation_regress for zenith

---
 .../batch_pg_regress/test_isolation.py        | 57 +++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 test_runner/batch_pg_regress/test_isolation.py

diff --git a/test_runner/batch_pg_regress/test_isolation.py b/test_runner/batch_pg_regress/test_isolation.py
new file mode 100644
index 0000000000..1e66a2a654
--- /dev/null
+++ b/test_runner/batch_pg_regress/test_isolation.py
@@ -0,0 +1,57 @@
+import pytest
+from fixtures.utils import mkdir_if_needed
+import getpass
+import os
+import psycopg2
+
+pytest_plugins = ("fixtures.zenith_fixtures")
+
+
+def test_isolation(pageserver, postgres, pg_bin, zenith_cli, test_output_dir, pg_distrib_dir, base_dir, capsys):
+
+    # Create a branch for us
+    zenith_cli.run(["branch", "test_isolation", "empty"]);
+
+    # Connect to postgres and create a database called "regression".
+    # isolation tests use prepared transactions, so enable them
+    pg = postgres.create_start('test_isolation', ['max_prepared_transactions=100'])
+    pg_conn = psycopg2.connect(pg.connstr())
+    pg_conn.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
+    cur = pg_conn.cursor()
+    cur.execute('CREATE DATABASE isolation_regression')
+    pg_conn.close()
+
+    # Create some local directories for pg_isolation_regress to run in.
+    runpath = os.path.join(test_output_dir, 'regress')
+    mkdir_if_needed(runpath)
+    mkdir_if_needed(os.path.join(runpath, 'testtablespace'))
+
+    # Compute all the file locations that pg_isolation_regress will need.
+    build_path = os.path.join(
+        pg_distrib_dir, 'build/src/test/isolation')
+    src_path = os.path.join(
+        base_dir, 'vendor/postgres/src/test/isolation')
+    bindir = os.path.join(pg_distrib_dir, 'bin')
+    schedule = os.path.join(src_path, 'isolation_schedule')
+    pg_isolation_regress = os.path.join(build_path, 'pg_isolation_regress')
+
+    pg_isolation_regress_command = [
+        pg_isolation_regress,
+        '--use-existing',
+        '--bindir={}'.format(bindir),
+        '--dlpath={}'.format(build_path),
+        '--inputdir={}'.format(src_path),
+        '--schedule={}'.format(schedule)
+    ]
+
+    env = {
+        'PGPORT': str(pg.port),
+        'PGUSER': pg.username,
+        'PGHOST': pg.host,
+    }
+
+    # Run the command.
+    # We don't capture the output. It's not too chatty, and it always
+    # logs the exact same data to `regression.out` anyway.
+    with capsys.disabled():
+        pg_bin.run(pg_isolation_regress_command, env=env, cwd=runpath)