Merge pull request #70 from TurboGears/mim_conn_context_mgr

MIM: Connection supports `with` like pymongo MongoClient does

ming/mim.py

@@ -127,6 +127,11 @@ def _is_writable(self):
     def is_mongos(self):
         return False
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb) -> None:
+        self.close()
 
 class Database(database.Database):
     def __init__(self, client, name, **__):

ming/tests/test_mim.py

@@ -190,6 +190,16 @@ def test_search(self):
         assert coll.count_documents({'$text': {'$search': 'searched'}}) == 1
 
 
+class TestConnection(TestCase):
+
+    def setUp(self):
+        self.conn = create_datastore('mim:///testdb').conn
+
+    def test_context_manager(self):
+        with self.conn:
+            pass
+
+
 class TestDottedOperators(TestCase):
 
     def setUp(self):