Make rollback optional when dealing with unknown time zones

Co-authored-by: Rouven Bauer <[email protected]>

Author: fbiville

tests/stub/types/scripts/v4x4/echo_unknown_zoned_date_time.script

@@ -29,6 +29,11 @@ S: <RAW> 8C 45 75 72 6f 70 65 2f 4e 65 6f 34 6a
 S: <RAW> 00 00
 
 S: SUCCESS {"type": "r"}
-A: ROLLBACK
+# TODO: fix this once Go rolls back managed transactions on error (instead of relying on connection resets when they're put back to the pool)
+{{
+    A: ROLLBACK
+----
+    A: RESET
+}}
 *: RESET
 ?: GOODBYE

tests/stub/types/scripts/v4x4/echo_unknown_zoned_date_time_patched.script

@@ -30,6 +30,11 @@ S: <RAW> 8C 45 75 72 6f 70 65 2f 4e 65 6f 34 6a
 S: <RAW> 00 00
 
 S: SUCCESS {"type": "r"}
-A: ROLLBACK
+# TODO: fix this once Go rolls back managed transactions on error (instead of relying on connection resets when they're put back to the pool)
+{{
+    A: ROLLBACK
+----
+    A: RESET
+}}
 *: RESET
 ?: GOODBYE

tests/stub/types/scripts/v5x0/echo_unknown_zoned_date_time.script

@@ -29,6 +29,11 @@ S: <RAW> 8C 45 75 72 6f 70 65 2f 4e 65 6f 34 6a
 S: <RAW> 00 00
 
 S: SUCCESS {"type": "r"}
-A: ROLLBACK
+# TODO: fix this once Go rolls back managed transactions on error (instead of relying on connection resets when they're put back to the pool)
+{{
+    A: ROLLBACK
+----
+    A: RESET
+}}
 *: RESET
 ?: GOODBYE

tests/stub/types/test_temporal_types.py

@@ -2,6 +2,7 @@
 from nutkit.frontend import Driver
 from tests.shared import (
     driver_feature,
+    get_driver_name,
     TestkitTestCase,
 )
 from tests.stub.shared import StubServer
@@ -87,6 +88,10 @@ def work(tx):
         with self.assertRaises(types.DriverError):
             self._session.read_transaction(work)
         self._server.done()
+        # TODO: Remove once all drivers roll back failed managed transactions
+        if get_driver_name() not in ["go"]:
+            rollback_count = self._server.count_requests("ROLLBACK")
+            self.assertEqual(rollback_count, 1)
         self.assertEqual(tx_count, 1)
 
     def _test_unknown_then_known_zoned_date_time(self, patched):