Other changes for 4.2 support

Author: technige

neo4j/io/__init__.py

@@ -134,12 +134,13 @@ def protocol_handlers(cls, protocol_version=None):
 
         # Carry out Bolt subclass imports locally to avoid circular dependency issues.
         from neo4j.io._bolt3 import Bolt3
-        from neo4j.io._bolt4 import Bolt4x0, Bolt4x1
+        from neo4j.io._bolt4 import Bolt4x0, Bolt4x1, Bolt4x2
 
         handlers = {
             Bolt3.PROTOCOL_VERSION: Bolt3,
             Bolt4x0.PROTOCOL_VERSION: Bolt4x0,
             Bolt4x1.PROTOCOL_VERSION: Bolt4x1,
+            Bolt4x2.PROTOCOL_VERSION: Bolt4x2,
         }
 
         if protocol_version is None:
@@ -209,6 +210,10 @@ def open(cls, address, *, auth=None, timeout=None, routing_context=None, **pool_
             # Carry out Bolt subclass imports locally to avoid circular dependency issues.
             from neo4j.io._bolt4 import Bolt4x1
             connection = Bolt4x1(address, s, pool_config.max_connection_lifetime, auth=auth, user_agent=pool_config.user_agent, routing_context=routing_context)
+        elif pool_config.protocol_version == (4, 2):
+            # Carry out Bolt subclass imports locally to avoid circular dependency issues.
+            from neo4j.io._bolt4 import Bolt4x2
+            connection = Bolt4x2(address, s, pool_config.max_connection_lifetime, auth=auth, user_agent=pool_config.user_agent, routing_context=routing_context)
         else:
             log.debug("[#%04X]  S: <CLOSE>", s.getpeername()[1])
             s.shutdown(SHUT_RDWR)
@@ -672,7 +677,7 @@ def fetch_routing_info(self, *, address, timeout, database):
 
         # Carry out Bolt subclass imports locally to avoid circular dependency issues.
         from neo4j.io._bolt3 import Bolt3
-        from neo4j.io._bolt4 import Bolt4x0, Bolt4x1
+        from neo4j.io._bolt4 import Bolt4x0, Bolt4x1, Bolt4x2
 
         from neo4j.api import (
             SYSTEM_DATABASE,

tests/unit/io/test_class_bolt.py

@@ -28,7 +28,7 @@
 def test_class_method_protocol_handlers():
     # python -m pytest tests/unit/io/test_class_bolt.py -s -v -k test_class_method_protocol_handlers
     protocol_handlers = Bolt.protocol_handlers()
-    assert len(protocol_handlers) == 3
+    assert len(protocol_handlers) == 4
 
 
 @pytest.mark.parametrize(
@@ -53,7 +53,7 @@ def test_class_method_protocol_handlers_with_invalid_protocol_version():
 def test_class_method_get_handshake():
     # python -m pytest tests/unit/io/test_class_bolt.py -s -v -k test_class_method_get_handshake
     handshake = Bolt.get_handshake()
-    assert handshake == b"\x00\x00\x01\x04\x00\x00\x00\x04\x00\x00\x00\x03\x00\x00\x00\x00"
+    assert handshake == b"\x00\x00\x02\x04\x00\x00\x01\x04\x00\x00\x00\x04\x00\x00\x00\x03"
 
 
 def test_magic_preamble():