Adding upstream version 3.1.7.upstream/3.1.7upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 57 insertions, 0 deletions

diff --git a/tests/test_module.py b/tests/test_module.py
new file mode 100644
index 0000000..794ef0f
--- /dev/null
+++ b/tests/test_module.py
@@ -0,0 +1,57 @@
+import pytest
+
+from psycopg._cmodule import _psycopg
+
+
+@pytest.mark.parametrize(
+    "args, kwargs, want_conninfo",
+    [
+        ((), {}, ""),
+        (("dbname=foo",), {"user": "bar"}, "dbname=foo user=bar"),
+        ((), {"port": 15432}, "port=15432"),
+        ((), {"user": "foo", "dbname": None}, "user=foo"),
+    ],
+)
+def test_connect(monkeypatch, dsn, args, kwargs, want_conninfo):
+    # Check the main args passing from psycopg.connect to the conn generator
+    # Details of the params manipulation are in test_conninfo.
+    import psycopg.connection
+
+    orig_connect = psycopg.connection.connect  # type: ignore
+
+    got_conninfo = None
+
+    def mock_connect(conninfo):
+        nonlocal got_conninfo
+        got_conninfo = conninfo
+        return orig_connect(dsn)
+
+    monkeypatch.setattr(psycopg.connection, "connect", mock_connect)
+
+    conn = psycopg.connect(*args, **kwargs)
+    assert got_conninfo == want_conninfo
+    conn.close()
+
+
+def test_version(mypy):
+    cp = mypy.run_on_source(
+        """\
+from psycopg import __version__
+assert __version__
+"""
+    )
+    assert not cp.stdout
+
+
+@pytest.mark.skipif(_psycopg is None, reason="C module test")
+def test_version_c(mypy):
+    # can be psycopg_c, psycopg_binary
+    cpackage = _psycopg.__name__.split(".")[0]
+
+    cp = mypy.run_on_source(
+        f"""\
+from {cpackage} import __version__
+assert __version__
+"""
+    )
+    assert not cp.stdout