i

tlocke · Mar 31, 2024 · 97805c4 · 97805c4
1 parent 6e75f65
commit 97805c4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/native/auth/test_scram-sha-256.py b/test/native/auth/test_scram-sha-256.py
@@ -24,14 +24,16 @@ def setup(con):
 def test_scram_sha_256(setup, db_kwargs):
     db_kwargs["database"] = DB
 
-    Connection(**db_kwargs).close()
+    with Connection(**db_kwargs):
+        pass
 
 
 def test_scram_sha_256_ssl_False(setup, db_kwargs):
     db_kwargs["database"] = DB
     db_kwargs["ssl_context"] = False
 
-    Connection(**db_kwargs).close()
+    with Connection(**db_kwargs):
+        pass
 
 
 def test_scram_sha_256_ssl_True(setup, db_kwargs):