diff --git a/.gitignore b/.gitignore index 940014d..666da99 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,4 @@ .idea venv/ .pytest_cache/ - +__pycache__/ diff --git a/requirements.txt b/requirements.txt index d6e342d..5ae5865 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,6 +7,8 @@ iniconfig==2.0.0 packaging==24.0 pluggy==1.5.0 psycopg==3.1.18 +psycopg-binary==3.1.18 +psycopg-pool==3.2.2 pytest==8.1.1 requests==2.31.0 testcontainers==4.4.0 diff --git a/tests/test_customers.py b/tests/test_customers.py index 11f159f..275855e 100644 --- a/tests/test_customers.py +++ b/tests/test_customers.py @@ -18,9 +18,9 @@ def remove_container(): os.environ["DB_CONN"] = postgres.get_connection_url() os.environ["DB_HOST"] = postgres.get_container_host_ip() os.environ["DB_PORT"] = postgres.get_exposed_port(5432) - os.environ["DB_USERNAME"] = postgres.POSTGRES_USER - os.environ["DB_PASSWORD"] = postgres.POSTGRES_PASSWORD - os.environ["DB_NAME"] = postgres.POSTGRES_DB + os.environ["DB_USERNAME"] = postgres.username + os.environ["DB_PASSWORD"] = postgres.password + os.environ["DB_NAME"] = postgres.dbname customers.create_table()