]> 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 01:55:02 +0000 (02:55 +0100)
committerMartin Weinelt <hexa@darmstadt.ccc.de>
Tue, 21 Nov 2023 01:55:02 +0000 (02:55 +0100)
These tests were previously affected by environment variables, that can
be used to configure a postgresql instance.

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

index 7f667d08a50d2345e4717aee7af5519dd908399d..b039ee6ac292fa28876c0ad78e9721c402329c8f 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.generators, "connect", mock_connect)
 
     conn = psycopg.connect(*args, **kwargs)
index 8e51466d682580f805f7af6324731382934b9724..e87937dafcd9c2b16397ae2dcb589dd7d2cb1342 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.generators, "connect", fake_connect)
     conn = psycopg.connect(*args, **kwargs)
     assert drop_default_args_from_conninfo(got_conninfo) == conninfo_to_dict(want)