Make connection stub use actual Deadline objects

Author: robsdedude

tests/unit/async_/work/_fake_connection.py

@@ -22,6 +22,7 @@
 
 from neo4j import ServerInfo
 from neo4j._async.io import AsyncBolt
+from neo4j._io.deadline import Deadline
 
 
 @pytest.fixture
@@ -54,6 +55,7 @@ def close_side_effect():
             )
 
             def set_deadline_side_effect(deadline):
+                deadline = Deadline.from_timeout_or_deadline(deadline)
                 self.socket.get_deadline.return_value = deadline
 
             self.socket.attach_mock(

tests/unit/sync/work/_fake_connection.py

@@ -21,6 +21,7 @@
 import pytest
 
 from neo4j import ServerInfo
+from neo4j._io.deadline import Deadline
 from neo4j._sync.io import Bolt
 
 
@@ -54,6 +55,7 @@ def close_side_effect():
             )
 
             def set_deadline_side_effect(deadline):
+                deadline = Deadline.from_timeout_or_deadline(deadline)
                 self.socket.get_deadline.return_value = deadline
 
             self.socket.attach_mock(