Update test-db-connection.py
#!/usr/bin/env python3
import os
-import psycopg2
+import psycopg
-con = psycopg2.connect(
+conn = psycopg.connect(
+ dbname=os.environ.get("POSTGRES_DB", "ldbcsnb"),
host=os.environ.get("POSTGRES_HOST", "localhost"),
user=os.environ.get("POSTGRES_USER", "postgres"),
password=os.environ.get("POSTGRES_PASSWORD", "mysecretpassword"),
- port=int(os.environ.get("POSTGRES_PORT", 5432)),
+ port=int(os.environ.get("POSTGRES_PORT", 5432))
)
-con.close()
+conn.close()
\ No newline at end of file
