Merge pull request #150 from technige/1.2-fix-broken-disconnect

Fixes #144

Author: srbaker

neo4j/v1/api.py

@@ -242,8 +242,9 @@ def _disconnect(self, sync):
         if self._connection:
             if sync:
                 self._connection.sync()
-            self._connection.in_use = False
-            self._connection = None
+            if self._connection:
+                self._connection.in_use = False
+                self._connection = None
             self._connection_access_mode = None
 
     def close(self):

test/integration/test_session.py

@@ -551,6 +551,11 @@ def test_peek_at_different_stages(self):
                 upcoming = result.peek()
                 assert upcoming is None
 
+    def test_can_safely_exit_session_without_consuming_result(self):
+        with self.driver.session() as session:
+            session.run("RETURN 1")
+        assert True
+
 
 class SessionCompletionTestCase(DirectIntegrationTestCase):