SpamExperts
diff --git a/‎MySQLdb/connections.py
Copy file name to clipboardExpand all lines: MySQLdb/connections.py
+6Lines changed: 6 additions & 0 deletions b/‎MySQLdb/connections.py
Copy file name to clipboardExpand all lines: MySQLdb/connections.py
+6Lines changed: 6 additions & 0 deletions
diff --git a/‎tests/test_MySQLdb_nonstandard.py
Copy file name to clipboardExpand all lines: tests/test_MySQLdb_nonstandard.py
+5Lines changed: 5 additions & 0 deletions b/‎tests/test_MySQLdb_nonstandard.py
Copy file name to clipboardExpand all lines: tests/test_MySQLdb_nonstandard.py
+5Lines changed: 5 additions & 0 deletions
@@ -207,6 +207,12 @@ def unicode_literal(u, dummy=None):
                 self.autocommit(autocommit)
         self.messages = []
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.close()
+
     def autocommit(self, on):
         on = bool(on)
         if self.get_autocommit() != on:
 
@@ -99,3 +99,8 @@ def test_client_flag(self):
 
     def test_fileno(self):
         self.assertGreaterEqual(self.conn.fileno(), 0)
+
+    def test_context_manager(self):
+        with connection_factory() as conn:
+            self.assertFalse(conn.closed)
+        self.assertTrue(conn.closed)