testing_postgresql_temporary_instance_quickstart_with_psycopg2.py

1import psycopg2
2import testing.postgresql
3
4# Launch new PostgreSQL server
5with testing.postgresql.Postgresql() as postgresql:
6    # Connect to PostgreSQL
7    conn = psycopg2.connect(**postgresql.dsn())
8
9    # Create a table and insert data
10    cursor = conn.cursor()
11    cursor.execute("CREATE TABLE test (id int, name varchar(256))")
12    cursor.execute("INSERT INTO test VALUES (1, 'hello'), (2, 'world')")
13    conn.commit()
14
15    # Query the database
16    cursor.execute("SELECT * FROM test")
17    print(cursor.fetchall())
18
19    cursor.close()
20    conn.close()
21
22# The PostgreSQL server is automatically terminated and temporary files are deleted here