Skip to content

Commit 32d6de1

Browse files
committed
Fix connection_deadline context manager
1 parent 1362c16 commit 32d6de1

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

neo4j/_io/deadline.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,8 +84,13 @@ def merge_deadlines_and_timeouts(*deadline):
8484
@contextmanager
8585
def connection_deadline(connection, deadline):
8686
original_deadline = connection.socket.get_deadline()
87-
if original_deadline is not None:
88-
deadline = min(original_deadline, original_deadline)
87+
if deadline is None and original_deadline is not None:
88+
# nothing to do here
89+
yield
90+
return
91+
deadline = merge_deadlines(
92+
*(d for d in (deadline, original_deadline) if d is not None)
93+
)
8994
connection.socket.set_deadline(deadline)
9095
try:
9196
yield

0 commit comments

Comments
 (0)