i

tlocke · Mar 31, 2024 · 80a6502 · 80a6502
1 parent 8cc4925
commit 80a6502
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/test/native/auth/test_scram-sha-256_ssl.py b/test/native/auth/test_scram-sha-256_ssl.py
@@ -18,36 +18,36 @@ def setup(con):
         pass
 
 
-def test_scram_sha_256_plus(db_kwargs):
+def test_scram_sha_256_plus(setup, db_kwargs):
     db_kwargs["database"] = DB
 
-    con = Connection(**db_kwargs)
-    assert isinstance(con._usock, SSLSocket)
+    with Connection(**db_kwargs) as con:
+        assert isinstance(con._usock, SSLSocket)
 
 
-def test_scram_sha_256_plus_ssl_True(db_kwargs):
+def test_scram_sha_256_plus_ssl_True(setup, db_kwargs):
     db_kwargs["ssl_context"] = True
     db_kwargs["database"] = DB
 
-    con = Connection(**db_kwargs)
-    assert isinstance(con._usock, SSLSocket)
+    with Connection(**db_kwargs) as con:
+        assert isinstance(con._usock, SSLSocket)
 
 
-def test_scram_sha_256_plus_ssl_custom(db_kwargs):
+def test_scram_sha_256_plus_ssl_custom(setup, db_kwargs):
     context = create_default_context()
     context.check_hostname = False
     context.verify_mode = CERT_NONE
 
     db_kwargs["ssl_context"] = context
     db_kwargs["database"] = DB
 
-    con = Connection(**db_kwargs)
-    assert isinstance(con._usock, SSLSocket)
+    with Connection(**db_kwargs) as con:
+        assert isinstance(con._usock, SSLSocket)
 
 
-def test_scram_sha_256_plus_ssl_False(db_kwargs):
+def test_scram_sha_256_plus_ssl_False(setup, db_kwargs):
     db_kwargs["ssl_context"] = False
     db_kwargs["database"] = DB
 
-    con = Connection(**db_kwargs)
-    assert not isinstance(con._usock, SSLSocket)
+    with Connection(**db_kwargs) as con:
+        assert not isinstance(con._usock, SSLSocket)