diff options
Diffstat (limited to 'tests/utils.py')
-rw-r--r-- | tests/utils.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/utils.py b/tests/utils.py index 9b7376f1..158fa1e3 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,5 +1,5 @@ import pytest -import psycopg2 +import psycopg2cffi from pgcli.main import format_output # TODO: should this be somehow be divined from environment? @@ -7,7 +7,7 @@ POSTGRES_USER, POSTGRES_HOST = 'postgres', 'localhost' def db_connection(dbname=None): - conn = psycopg2.connect(user=POSTGRES_USER, host=POSTGRES_HOST, dbname=dbname) + conn = psycopg2cffi.connect(user=POSTGRES_USER, host=POSTGRES_HOST, dbname=dbname) conn.autocommit = True return conn |