Removed fetch_all

Author: technige

neo4j/connection.py

@@ -235,7 +235,9 @@ def on_failure(metadata):
 
         self.append(INIT, (user_agent,), response=response)
         self.send()
-        self.fetch_all(response)
+        fetch_next = self.fetch_next
+        while not response.complete:
+            fetch_next()
 
     def append(self, signature, fields=(), response=None):
         """ Add a message to the outgoing queue.
@@ -283,11 +285,6 @@ def fetch_next(self):
                 self.append(ACK_FAILURE, response=AckFailureResponse(self))
         raw.close()
 
-    def fetch_all(self, response):
-        fetch_next = self.fetch_next
-        while not response.complete:
-            fetch_next()
-
     def close(self):
         """ Shut down and close the connection.
         """

neo4j/session.py

@@ -172,8 +172,10 @@ def run(self, statement, parameters=None):
         self.connection.send()
         t.end_send = perf_counter()
 
-        self.connection.fetch_all(run_response)
-        self.connection.fetch_all(pull_all_response)
+        fetch_next = self.connection.fetch_next
+        while not pull_all_response.complete:
+            fetch_next()
+
         t.done = perf_counter()
         self.bench_tests.append(t)