add wait log

spotify · Jan 16, 2021 · 8f36797 · 8f36797
1 parent 715ed32
commit 8f36797
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/luigi/rpc.py b/luigi/rpc.py
@@ -149,6 +149,7 @@ def retry_logging(retry_state):
             if self._rpc_log_retries:
                 logger.warning("Failed connecting to remote scheduler %r", self._url, exc_info=True)
                 logger.info("Retrying attempt %r of %r (max)" % (retry_state.attempt_number, self._rpc_retry_attempts))
+                logger.info("Wait for %d seconds" % self._rpc_retry_wait)
 
         return Retrying(wait=wait_fixed(self._rpc_retry_wait),
                         stop=stop_after_attempt(self._rpc_retry_attempts),

diff --git a/test/rpc_test.py b/test/rpc_test.py
@@ -75,8 +75,10 @@ def test_log_rpc_retries_enabled(self, mock_logger):
         self.assertEqual([
             mock.call.warning('Failed connecting to remote scheduler %r', 'http://zorg.com', exc_info=True),
             mock.call.info('Retrying attempt 1 of 3 (max)'),
+            mock.call.info('Wait for 1 seconds'),
             mock.call.warning('Failed connecting to remote scheduler %r', 'http://zorg.com', exc_info=True),
             mock.call.info('Retrying attempt 2 of 3 (max)'),
+            mock.call.info('Wait for 1 seconds'),
         ], mock_logger.mock_calls)
 
     @with_config({'core': {'rpc-log-retries': 'false'}})