mocking out cursor

src/py/crankshaft/test/mock_plpy.py

@@ -1,5 +1,16 @@
 import re
 
+class MockCursor:
+    def __init__(self, data):
+        self.cursor_pos =0
+        self.data = data
+
+    def fetch(self, batch_size):
+        batch = self.data[self.cursor_pos : self.cursor_pos + batch_size]
+        self.cursor_pos += batch_size
+        return batch
+
+
 class MockPlPy:
     def __init__(self):
         self._reset()
@@ -27,6 +38,10 @@ class MockPlPy:
     def info(self, msg):
         self.infos.append(msg)
 
+    def cursor(self,query):
+        data = self.execute(query)
+        return MockCursor(data)
+
     def execute(self, query): # TODO: additional arguments
        for result in self.results:
           if result[0].match(query):