fix(main): Fix psycopg unit test

Uziel Silva · thameezb · commit ef46cf0526a4 · 2026-02-23T22:43:42.000+01:00
diff --git a/tests/unit/test_psycopg.py b/tests/unit/test_psycopg.py
@@ -33,7 +33,7 @@ async def test_psycopg(context: ssl.SSLContext, kwargs: Any) -> None:
         socket.create_connection((ip_addr, 3307)),
         server_hostname=ip_addr,
     )
-    with patch("psycopg.connect") as mock_connect:
+    with patch("psycopg.Connection.connect") as mock_connect:
         type(mock_connect.return_value).autocommit = PropertyMock(return_value=True)
         connection = connect(ip_addr, sock, **kwargs)
         assert connection.autocommit is True

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@ async def test_psycopg(context: ssl.SSLContext, kwargs: Any) -> None:`
`33`	`33`	`socket.create_connection((ip_addr, 3307)),`
`34`	`34`	`server_hostname=ip_addr,`
`35`	`35`	`)`
`36`		`- with patch("psycopg.connect") as mock_connect:`
	`36`	`+ with patch("psycopg.Connection.connect") as mock_connect:`
`37`	`37`	`type(mock_connect.return_value).autocommit = PropertyMock(return_value=True)`
`38`	`38`	`connection = connect(ip_addr, sock, **kwargs)`
`39`	`39`	`assert connection.autocommit is True`