Adjust unit tests

Author: robsdedude

tests/unit/async_/test_driver.py

@@ -24,6 +24,7 @@
     AsyncBoltDriver,
     AsyncGraphDatabase,
     AsyncNeo4jDriver,
+    ExperimentalWarning,
     TRUST_ALL_CERTIFICATES,
     TRUST_SYSTEM_CA_SIGNED_CERTIFICATES,
     TrustAll,
@@ -224,18 +225,19 @@ async def test_verify_connectivity(uri, mocker):
     "neo4j://127.0.0.1:9000",
 ))
 @pytest.mark.parametrize("kwargs", (
-    {"access_mode": WRITE_ACCESS},
-    {"access_mode": READ_ACCESS},
+    {"default_access_mode": WRITE_ACCESS},
+    {"default_access_mode": READ_ACCESS},
     {"fetch_size": 69},
 ))
 @mark_async_test
-async def test_verify_connectivity_parameters_are_deprecated(uri, kwargs,
-                                                             mocker):
+async def test_verify_connectivity_parameters_are_experimental(
+    uri, kwargs, mocker
+):
     driver = AsyncGraphDatabase.driver(uri)
     mocker.patch.object(driver, "_pool", autospec=True)
 
     try:
-        with pytest.warns(DeprecationWarning, match="configuration"):
+        with pytest.warns(ExperimentalWarning, match="configuration"):
             await driver.verify_connectivity(**kwargs)
     finally:
         await driver.close()

tests/unit/sync/test_driver.py

@@ -22,6 +22,7 @@
 
 from neo4j import (
     BoltDriver,
+    ExperimentalWarning,
     GraphDatabase,
     Neo4jDriver,
     TRUST_ALL_CERTIFICATES,
@@ -224,18 +225,19 @@ def test_verify_connectivity(uri, mocker):
     "neo4j://127.0.0.1:9000",
 ))
 @pytest.mark.parametrize("kwargs", (
-    {"access_mode": WRITE_ACCESS},
-    {"access_mode": READ_ACCESS},
+    {"default_access_mode": WRITE_ACCESS},
+    {"default_access_mode": READ_ACCESS},
     {"fetch_size": 69},
 ))
 @mark_sync_test
-def test_verify_connectivity_parameters_are_deprecated(uri, kwargs,
-                                                             mocker):
+def test_verify_connectivity_parameters_are_experimental(
+    uri, kwargs, mocker
+):
     driver = GraphDatabase.driver(uri)
     mocker.patch.object(driver, "_pool", autospec=True)
 
     try:
-        with pytest.warns(DeprecationWarning, match="configuration"):
+        with pytest.warns(ExperimentalWarning, match="configuration"):
             driver.verify_connectivity(**kwargs)
     finally:
         driver.close()