Re-use run context.

jaraco · Apr 24, 2024 · 691819b · 691819b
1 parent f14608e
commit 691819b
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/jaraco/mongodb/tests/test_service.py b/jaraco/mongodb/tests/test_service.py
@@ -10,25 +10,19 @@
 @pytest.mark.xfail(reason="#31")
 def test_MongoDBReplicaSet_writable():
     rs = service.MongoDBReplicaSet()
-    rs.start()
-    try:
+    with rs.run():
         conn = pymongo.MongoClient(rs.get_connect_hosts())
         conn.database.collection.insert_one({'foo': 'bar'})
-    finally:
-        rs.stop()
 
 
 def test_MongoDBReplicaSet_starts_quickly():
     pytest.skip("Takes 20-30 seconds")
     sw = timing.Stopwatch()
     rs = service.MongoDBReplicaSet()
-    rs.start()
-    try:
+    with rs.run():
         elapsed = sw.split()
         limit = datetime.timedelta(seconds=5)
         assert elapsed < limit
-    finally:
-        rs.stop()
 
 
 def test_fixture(mongodb_instance):