Add enter, exit function

tests/unit/test_dbapi.py

@@ -312,5 +312,5 @@ def test_hostname_parsing():
 
 def test_description_is_none_when_cursor_is_not_executed():
     connection = Connection("sample_trino_cluster:443")
-    cursor = connection.cursor()
-    assert hasattr(cursor, 'description')
+    with connection.cursor() as cursor:
+        assert hasattr(cursor, 'description')

trino/dbapi.py

@@ -382,6 +382,12 @@ def __init__(
     def __iter__(self):
         return self._iterator
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.close()
+
     @property
     def connection(self):
         return self._connection