Adding tests to ensure transient error caused by user canceling execution is not retried

Author: Zhen

test/stub/scripts/user_canceled_tx.script.script

@@ -0,0 +1,20 @@
+!: AUTO INIT
+!: AUTO RESET
+
+C: RUN "BEGIN" {}
+   PULL_ALL
+S: SUCCESS {"fields": []}
+   SUCCESS {}
+
+C: RUN "RETURN 1" {}
+   PULL_ALL
+S: FAILURE {"code": "Neo.TransientError.Transaction.LockClientStopped", "message": "X"}
+   IGNORED {}
+
+C: ACK_FAILURE
+S: SUCCESS {}
+
+C: RUN "ROLLBACK" {}
+   PULL_ALL
+S: SUCCESS {}
+   SUCCESS {}

test/stub/test_accesslevel.py

@@ -19,7 +19,7 @@
 # limitations under the License.
 
 
-from neo4j.v1 import GraphDatabase, CypherError
+from neo4j.v1 import GraphDatabase, CypherError, TransientError
 
 from test.stub.tools import StubTestCase, StubCluster
 
@@ -159,3 +159,25 @@ def unit_of_work_2(tx):
                     assert value == 1
                     value = session.read_transaction(unit_of_work_2)
                     assert value == 2
+
+    def test_no_retry_read_on_user_canceled_tx(self):
+        with StubCluster({9001: "router.script", 9004: "user_canceled_tx.script.script"}):
+            uri = "bolt+routing://127.0.0.1:9001"
+            with GraphDatabase.driver(uri, auth=self.auth_token, encrypted=False) as driver:
+                with driver.session() as session:
+                    def unit_of_work(tx):
+                        tx.run("RETURN 1")
+
+                    with self.assertRaises(TransientError):
+                        _ = session.read_transaction(unit_of_work)
+
+    def test_no_retry_write_on_user_canceled_tx(self):
+        with StubCluster({9001: "router.script", 9006: "user_canceled_tx.script.script"}):
+            uri = "bolt+routing://127.0.0.1:9001"
+            with GraphDatabase.driver(uri, auth=self.auth_token, encrypted=False) as driver:
+                with driver.session() as session:
+                    def unit_of_work(tx):
+                        tx.run("RETURN 1")
+
+                    with self.assertRaises(TransientError):
+                        _ = session.write_transaction(unit_of_work)