1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
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
|
