black

Author: gbtami

server/db_wrapper.py

@@ -32,15 +32,15 @@
 # Retryable error codes for OperationFailure in single-node setup
 RETRYABLE_ERROR_CODES = {
     # Connection and network related (most relevant for single node)
-    6,    # HostUnreachable
-    7,    # HostNotFound
-    89,   # NetworkInterfaceExceededTimeLimit / NetworkTimeout
-    91,   # ShutdownInProgress
-    9001, # SocketException
+    6,  # HostUnreachable
+    7,  # HostNotFound
+    89,  # NetworkInterfaceExceededTimeLimit / NetworkTimeout
+    91,  # ShutdownInProgress
+    9001,  # SocketException
     # Time limits
     262,  # ExceededTimeLimit
     # Write concern related (certain cases)
-    64,   # WriteConcernFailed (certain cases)
+    64,  # WriteConcernFailed (certain cases)
     # Some replica set codes that might occur (for broader compatibility)
     189,  # PrimarySteppedDown (may occur during failover-like scenarios)
 }

tests/test_db_wrapper_comprehensive.py

@@ -67,7 +67,9 @@ async def test_retry_logic_with_operation_failure(self, mock_sleep):
         wrapped_collection = wrapped_db["test_collection"]
 
         # Simulate an OperationFailure with retryable code on the first call, then succeed
-        retryable_exc = OperationFailure("Network timeout", code=89)  # NetworkInterfaceExceededTimeLimit
+        retryable_exc = OperationFailure(
+            "Network timeout", code=89
+        )  # NetworkInterfaceExceededTimeLimit
         mock_collection.find_one.side_effect = [
             retryable_exc,
             {"_id": "123", "name": "test"},