chore(tests): add sleep mocks (#1943)

daniel-sanche · gemini-code-assist[bot] · web-flow · commit a43a18c27c5a · 2026-01-16T09:28:16.000-08:00
Some google3 tests were failing due to timeouts. It seems like there
were some tests that preformed up to 10 retries, with exponential
backoff between each one

This PR mocks out sleep, so these tests will complete ~ instantly,
rather than &gt; 10 mins

---------

Co-authored-by: gemini-code-assist[bot] &lt;176961590+gemini-code-assist[bot]@users.noreply.github.com&gt;
diff --git a/packages/google-auth/tests/compute_engine/test__metadata.py b/packages/google-auth/tests/compute_engine/test__metadata.py
@@ -201,8 +201,9 @@ def test_ping_success_custom_root(mock_metrics_header_value):
     )
 
 
+@mock.patch("time.sleep", return_value=None)
 @mock.patch("google.auth.metrics.mds_ping", return_value=MDS_PING_METRICS_HEADER_VALUE)
-def test_ping_failure_custom_retry(mock_metrics_header_value):
+def test_ping_failure_custom_retry(mock_metrics_header_value, _mock_sleep):
     request = make_request("")
     request.side_effect = exceptions.TransportError()
 
@@ -450,7 +451,8 @@ def test_get_failure_connection_failed(mock_sleep):
     assert request.call_count == 5
 
 
-def test_get_too_many_requests_retryable_error_failure():
+@mock.patch("time.sleep", return_value=None)
+def test_get_too_many_requests_retryable_error_failure(_mock_sleep):
     request = make_request("too many requests", status=http_client.TOO_MANY_REQUESTS)
 
     with pytest.raises(exceptions.TransportError) as excinfo:
@@ -546,7 +548,8 @@ def test_get_universe_domain_not_found():
     assert universe_domain == "googleapis.com"
 
 
-def test_get_universe_domain_retryable_error_failure():
+@mock.patch("time.sleep", return_value=None)
+def test_get_universe_domain_retryable_error_failure(_mock_sleep):
     # Test that if the universe domain endpoint returns a retryable error
     # we should retry.
     #
