store transaction status after query

Author: leafo

pgmoon/init.lua

@@ -735,6 +735,7 @@ do
           end
           row_desc, data_rows, command_complete = nil
         elseif MSG_TYPE_B.ready_for_query == _exp_0 then
+          self.transaction_status = msg
           break
         elseif MSG_TYPE_B.notification == _exp_0 then
           if not (notifications) then
@@ -936,6 +937,7 @@ do
           return nil, self:parse_error(msg)
         end
         if MSG_TYPE_B.ready_for_query == t then
+          self.transaction_status = msg
           break
         end
       end

pgmoon/init.moon

@@ -745,6 +745,7 @@ class Postgres
 
           row_desc, data_rows, command_complete = nil
         when MSG_TYPE_B.ready_for_query
+          @transaction_status = msg
           break
         when MSG_TYPE_B.notification
           notifications = {} unless notifications
@@ -924,7 +925,9 @@ class Postgres
       if MSG_TYPE_B.error == t
         return nil, @parse_error(msg)
 
-      break if MSG_TYPE_B.ready_for_query == t
+      if MSG_TYPE_B.ready_for_query == t
+        @transaction_status = msg
+        break
 
     true
 

spec/pgmoon_spec.moon

@@ -220,6 +220,42 @@ describe "pgmoon with server", ->
         assert.falsy status, "connection should fail if it could not establish ssl"
         assert.same [[the server does not support SSL connections]], err
 
+      describe "transaction_status", ->
+        it "is idle after connect", ->
+          assert.same "I", pg.transaction_status
+
+        it "is in transaction after BEGIN", ->
+          assert pg\query "BEGIN"
+          assert.same "T", pg.transaction_status
+          assert pg\query "ROLLBACK"
+          assert.same "I", pg.transaction_status
+
+        it "is in error state after failed query in transaction", ->
+          assert pg\query "BEGIN"
+          assert.same "T", pg.transaction_status
+
+          -- query a non-existent table to trigger an error
+          res, err = pg\query "SELECT * FROM nonexistent_table_12345"
+          assert.falsy res
+          assert.same "E", pg.transaction_status
+
+          -- further queries should fail until rollback
+          res, err = pg\query "SELECT 1"
+          assert.falsy res
+          assert.same "E", pg.transaction_status
+
+          -- rollback should restore to idle
+          assert pg\query "ROLLBACK"
+          assert.same "I", pg.transaction_status
+
+        it "returns to idle after COMMIT", ->
+          assert pg\query "BEGIN"
+          assert.same "T", pg.transaction_status
+          assert pg\query "SELECT 1"
+          assert.same "T", pg.transaction_status
+          assert pg\query "COMMIT"
+          assert.same "I", pg.transaction_status
+
       describe "extended_query", ->
         it "query with no params", ->
           res = assert pg\extended_query "select 1 as one"