Function main

compat/src/test/python/runner_psycopg3.py:211–238

(yaml_file)

def main(yaml_file):
    data = load_yaml(yaml_file)
    global_variables = data.get('variables', {})
    tests = data.get('tests', [])

    port = int(os.getenv('PGPORT', 8812))
    for binary in [True, False]:
        binary_mode = "binary" if binary else "text"
        print(f"\n=== Running tests with {binary_mode} mode ===\n")
        for test in tests:
            iterations = test.get('iterations', 50)
            exclusions = test.get('exclude', [])
            if 'psycopg3' in exclusions:
                print(f"Skipping test '{test['name']}' because it is excluded for psycopg3.")
                continue
            for i in range(iterations):
                print(f"Running test '{test['name']}' [{binary_mode}] (iteration {i + 1})")
                connection = psycopg.connect(
                    host='localhost',
                    port=port,
                    user='admin',
                    password='quest',
                    dbname='qdb',
                    autocommit=True
                )
                register_varchar_array_type(connection, binary)
                run_test(test, global_variables, connection, binary)
                connection.close()


if __name__ == '__main__':