Harden system tests against 'ResourceExhausted' quota errors.

Closes #5303.
googleapis · Jun 14, 2018 · bbee15c · bbee15c
1 parent 71a0c9c
commit bbee15c
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/logging/tests/system/test_system.py b/logging/tests/system/test_system.py
@@ -16,11 +16,12 @@
 import logging
 import unittest
 
+from google.api_core.exceptions import Conflict
+from google.api_core.exceptions import NotFound
+from google.api_core.exceptions import TooManyRequests
+from google.api_core.exceptions import ResourceExhausted
+from google.api_core.exceptions import ServiceUnavailable
 from google.cloud._helpers import UTC
-from google.cloud.exceptions import Conflict
-from google.cloud.exceptions import NotFound
-from google.cloud.exceptions import TooManyRequests
-from google.cloud.exceptions import ServiceUnavailable
 import google.cloud.logging
 import google.cloud.logging.handlers.handlers
 from google.cloud.logging.handlers.handlers import CloudLoggingHandler
@@ -63,7 +64,7 @@ def _list_entries(logger):
     :returns: List of all entries consumed.
     """
     inner = RetryResult(_has_entries)(_consume_entries)
-    outer = RetryErrors(ServiceUnavailable)(inner)
+    outer = RetryErrors((ServiceUnavailable, ResourceExhausted))(inner)
     return outer(logger)