Use OPT_EAGER_TX_BEGIN for assert behaviour in Disconnects and Authorization tests

Author: bigmontz

tests/stub/authorization/test_authorization.py

@@ -193,8 +193,7 @@ def _fail_on_begin_using_tx_run(self, error, error_assertion):
 
         session = driver.session("r", database=self.get_db())
 
-        # TODO: remove block when all drivers behave the same way
-        if get_driver_name() in ["javascript", "go"]:
+        if not self.driver_supports_features(types.Feature.OPT_EAGER_TX_BEGIN):
             tx = session.begin_transaction()
             with self.assertRaises(types.DriverError) as exc:
                 tx.run("cypher").next()

tests/stub/disconnects/test_disconnects.py

@@ -177,10 +177,11 @@ def test_disconnect_on_tx_begin(self):
         self._server.done()
 
         expected_step = "after begin"
-        if self._driver_name in ["go"]:
-            expected_step = "after run"
-        elif self._driver_name in ["javascript"]:
-            expected_step = "after first next"
+        if not self.driver_supports_features(types.Feature.OPT_EAGER_TX_BEGIN):
+            if self._driver_name in ["go"]:
+                expected_step = "after run"
+            elif self._driver_name in ["javascript"]:
+                expected_step = "after first next"
         self.assertEqual(step, expected_step)
 
     def test_disconnect_on_tx_run(self):