MSSQLConnection: Add with (context manager) support

Author: msabramo

_mssql.pyx

@@ -569,6 +569,12 @@ cdef class MSSQLConnection:
         log("_mssql.MSSQLConnection.__dealloc__()")
         self.close()
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.close()
+
     def __iter__(self):
         assert_connected(self)
         clr_err(self)

tests/test_context_managers.py

@@ -8,7 +8,7 @@
     import unittest
 
 from pymssql import InterfaceError
-from .helpers import pymssqlconn
+from .helpers import pymssqlconn, mssqlconn
 
 
 class TestContextManagers(unittest.TestCase):
@@ -35,3 +35,10 @@ def test_pymssql_Cursor_with(self):
             cursor.execute("SELECT @@version AS version")
 
         self.assertEqual(str(context.exception), "Cursor is closed.")
+
+    def test_mssql_Connection_with(self):
+        with mssqlconn() as conn:
+            conn.execute_query("SELECT @@version AS version")
+            self.assertTrue(conn.connected)
+
+        self.assertFalse(conn.connected)