testkit: Support for routing stub tests

Author: Peter Wilhelmsson

testkitbackend/backend.py

@@ -26,6 +26,7 @@ def __init__(self, rd, wr):
         self.drivers = {}
         self.sessions = {}
         self.results = {}
+        self.transactions = {}
         self.key = 0
         # Collect all request handlers
         self._requestHandlers = dict(

testkitbackend/requests.py

@@ -68,7 +68,6 @@ def NewSession(backend, data):
 
 def SessionRun(backend, data):
     session = backend.sessions[data["sessionId"]].session
-    print(data)
     cypher, params = fromtestkit.toCypherAndParams(data)
     # TODO: txMeta, timeout
     result = session.run(cypher, parameters=params)
@@ -85,10 +84,47 @@ def SessionClose(backend, data):
     backend.send_response("Session", {"id": key})
 
 
+def SessionBeginTransaction(backend, data):
+    key = data["sessionId"]
+    session = backend.sessions[key].session
+    metadata = data.get('txMeta', None)
+    timeout = data.get('timeout', None)
+    if timeout:
+        timeout = int(timeout)
+    tx = session.begin_transaction(metadata=metadata, timeout=timeout)
+    key = backend.next_key()
+    backend.transactions[key] = tx
+    backend.send_response("Transaction", {"id": key})
+
+
 def ResultNext(backend, data):
     result = backend.results[data["resultId"]]
     try:
         record = next(iter(result))
     except StopIteration:
         backend.send_response("NullRecord", {})
     backend.send_response("Record", totestkit.record(record))
+
+
+def TransactionRun(backend, data):
+    key = data["txId"]
+    tx = backend.transactions[key]
+    cypher, params = fromtestkit.toCypherAndParams(data)
+    result = tx.run(cypher, parameters=params)
+    key = backend.next_key()
+    backend.results[key] = result
+    backend.send_response("Result", {"id": key})
+
+
+def TransactionCommit(backend, data):
+    key = data["txId"]
+    tx = backend.transactions[key]
+    tx.commit()
+    backend.send_response("Transaction", {"id": key})
+
+
+def TransactionRollback(backend, data):
+    key = data["txId"]
+    tx = backend.transactions[key]
+    tx.rollback()
+    backend.send_response("Transaction", {"id": key})