]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
fix: flush PG environment variables in test_connect{,_args}
authorMartin Weinelt <hexa@darmstadt.ccc.de>
Tue, 21 Nov 2023 02:00:36 +0000 (03:00 +0100)
committerMartin Weinelt <hexa@darmstadt.ccc.de>
Tue, 21 Nov 2023 02:00:36 +0000 (03:00 +0100)
These tests were previously affected by environment variables, that can
be used to configure a postgresql instance.

Closes: #681
tests/test_connection.py
tests/test_module.py
tests/test_psycopg_dbapi20.py

index 52808af038af75afca012af502e77b39a877c3bf..10f0a6ca7a05893f43f57ee6e3fb99c748f9b1bf 100644 (file)
@@ -409,6 +409,7 @@ def test_connect_args(conn_cls, monkeypatch, setpgenv, pgconn, args, kwargs, wan
         return pgconn
         yield
 
+    setpgenv({})
     monkeypatch.setattr(psycopg.connection, "connect", fake_connect)
     conn = conn_cls.connect(*args, **kwargs)
     got_params = drop_default_args_from_conninfo(got_conninfo)
index 030b75808aaf2e2613e3467e136d865f5ad9489f..948c00b938d8d91869220333227684d202c002cb 100644 (file)
@@ -15,7 +15,7 @@ from .test_connection import drop_default_args_from_conninfo
         ((), {"user": "foo", "dbname": None}, "user=foo"),
     ],
 )
-def test_connect(monkeypatch, dsn, args, kwargs, want):
+def test_connect(monkeypatch, dsn, args, kwargs, want, setpgenv):
     # Check the main args passing from psycopg.connect to the conn generator
     # Details of the params manipulation are in test_conninfo.
     import psycopg.connection
@@ -29,6 +29,7 @@ def test_connect(monkeypatch, dsn, args, kwargs, want):
         got_conninfo = conninfo
         return orig_connect(dsn)
 
+    setpgenv({})
     monkeypatch.setattr(psycopg.connection, "connect", mock_connect)
 
     conn = psycopg.connect(*args, **kwargs)
index bc8b1cc95bb244b319c50817557172f0512bcce8..69c2fa756708c680b60b22e033cbe4de3ba69add 100644 (file)
@@ -133,7 +133,7 @@ def test_time_from_ticks(ticks, want):
         (("host=foo",), {"user": None}, "host=foo"),
     ],
 )
-def test_connect_args(monkeypatch, pgconn, args, kwargs, want):
+def test_connect_args(monkeypatch, pgconn, args, kwargs, want, setpgenv):
     got_conninfo: str
 
     def fake_connect(conninfo):
@@ -142,6 +142,7 @@ def test_connect_args(monkeypatch, pgconn, args, kwargs, want):
         return pgconn
         yield
 
+    setpgenv({})
     monkeypatch.setattr(psycopg.connection, "connect", fake_connect)
     conn = psycopg.connect(*args, **kwargs)
     assert drop_default_args_from_conninfo(got_conninfo) == conninfo_to_dict(want)