@@ -48,9 +48,8 @@ def test_config_with_uri_in_flask_conf_var(self):
48
48
mongo = flask_pymongo .PyMongo (self .app , connect = True )
49
49
50
50
_wait_until_connected (mongo )
51
- assert mongo .cx is not None
52
- self .addCleanup (mongo .cx .close )
53
51
assert mongo .db is not None
52
+ assert mongo .cx is not None
54
53
assert mongo .db .name == self .dbname
55
54
assert ("localhost" , self .port ) == mongo .cx .address or (
56
55
"127.0.0.1" ,
@@ -63,9 +62,8 @@ def test_config_with_uri_passed_directly(self):
63
62
mongo = flask_pymongo .PyMongo (self .app , uri , connect = True )
64
63
65
64
_wait_until_connected (mongo )
66
- assert mongo .cx is not None
67
- self .addCleanup (mongo .cx .close )
68
65
assert mongo .db is not None
66
+ assert mongo .cx is not None
69
67
assert mongo .db .name == self .dbname
70
68
assert ("localhost" , self .port ) == mongo .cx .address or (
71
69
"127.0.0.1" ,
@@ -93,8 +91,6 @@ class CustomDict(dict[str, Any]):
93
91
94
92
uri = f"mongodb://localhost:{ self .port } /{ self .dbname } "
95
93
mongo = flask_pymongo .PyMongo (self .app , uri , document_class = CustomDict )
96
- assert mongo .cx is not None
97
- self .addCleanup (mongo .cx .close )
98
94
assert mongo .db is not None
99
95
assert mongo .db .things .find_one () is None , "precondition failed"
100
96
0 commit comments