Merge pull request #173 from lutovich/1.1-rollback-on-run-failure

Rollback transaction after run failure

Author: nigelsmall

src/v1/transaction.js

@@ -99,7 +99,13 @@ class Transaction {
   }
 
   _onError() {
-    this._onClose();
+    // rollback explicitly if tx.run failed, rollback
+    if (this._state == _states.ACTIVE) {
+        this.rollback();
+    } else {
+        // else just do the cleanup
+        this._onClose();
+    }
     this._state = _states.FAILED;
   }
 }

test/v1/transaction.test.js

@@ -236,4 +236,68 @@ describe('transaction', function() {
         done();
       });
   });
+
+  it('should rollback when very first run fails', done => {
+      const tx1 = session.beginTransaction();
+      tx1.run('RETURN foo')
+          .catch(error => {
+              expectSyntaxError(error);
+
+              const tx2 = session.beginTransaction();
+              tx2.run('RETURN 1')
+                  .then(result => {
+                      expect(result.records[0].get(0).toNumber()).toEqual(1);
+                      tx2.commit()
+                          .then(() => done());
+                  });
+          });
+  });
+
+  it('should rollback when some run fails', done => {
+      const tx1 = session.beginTransaction();
+      tx1.run('CREATE (:Person)')
+          .then(() => {
+              tx1.run('RETURN foo')
+                  .catch(error => {
+                      expectSyntaxError(error);
+
+                      const tx2 = session.beginTransaction();
+                      tx2.run('MATCH (n:Person) RETURN count(n)')
+                          .then(result => {
+                              expect(result.records[0].get(0).toNumber()).toEqual(0);
+                              tx2.commit()
+                                  .then(() => done());
+                          });
+                  });
+          });
+  });
+
+  it('should fail to commit transaction that had run failures', done => {
+      const tx1 = session.beginTransaction();
+      tx1.run('CREATE (:Person)')
+          .then(() => {
+              tx1.run('RETURN foo')
+                  .catch(error => {
+                      expectSyntaxError(error);
+                      tx1.commit()
+                          .catch(error => {
+                              const errorMessage = error.error;
+                              const index = errorMessage.indexOf('Cannot commit statements in this transaction');
+                              expect(index).not.toBeLessThan(0);
+
+                              const tx2 = session.beginTransaction();
+                              tx2.run('MATCH (n:Person) RETURN count(n)')
+                                  .then(result => {
+                                      expect(result.records[0].get(0).toNumber()).toEqual(0);
+                                      done();
+                                  });
+                          });
+                  });
+          });
+  });
+
+  function expectSyntaxError(error) {
+    const code = error.fields[0].code;
+    expect(code).toBe('Neo.ClientError.Statement.SyntaxError');
+  }
 });