Improving tests.

Team-Kujira · Sep 14, 2023 · 2145114 · 2145114
1 parent 50f710d
commit 2145114
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/...hummingbot/connector/gateway/clob_spot/data_sources/kujira/test_kujira_api_data_source.py b/...hummingbot/connector/gateway/clob_spot/data_sources/kujira/test_kujira_api_data_source.py
@@ -32,10 +32,21 @@
 )
 from hummingbot.core.network_iterator import NetworkStatus
 
-kujira_constants_module = importlib.import_module("hummingbot.connector.gateway.clob_spot.data_sources.kujira.kujira_constants")
-kujira_constants_module.NUMBER_OF_RETRIES = 0
-kujira_constants_module.DELAY_BETWEEN_RETRIES = 0
-kujira_constants_module.TIMEOUT = None
+
+def mock_automatic_retry_with_timeout(retries=0, delay=0, timeout=None):
+    def decorator(func):
+        return func
+
+    return decorator
+
+
+kujira_helpers_module = importlib.import_module("hummingbot.connector.gateway.clob_spot.data_sources.kujira.kujira_helpers")
+kujira_helpers_module.automatic_retry_with_timeout = mock_automatic_retry_with_timeout
+
+# kujira_constants_module = importlib.import_module("hummingbot.connector.gateway.clob_spot.data_sources.kujira.kujira_constants")
+# kujira_constants_module.NUMBER_OF_RETRIES = 0
+# kujira_constants_module.DELAY_BETWEEN_RETRIES = 0
+# kujira_constants_module.TIMEOUT = None
 
 from hummingbot.connector.gateway.clob_spot.data_sources.kujira.kujira_api_data_source import (  # noqa: E402
     KujiraAPIDataSource,