fix flaky test: add httperror to backoff check (GoogleCloudPlatform#9161)

leahecole · engelke · nicain · web-flow · commit f23eb8480ed4 · 2023-02-23T11:50:39.000-05:00
* add httperror to backoff check

* fix import order

---------

Co-authored-by: Charles Engelke &lt;engelke@google.com&gt;
Co-authored-by: nicain &lt;nicholascain@google.com&gt;
diff --git a/jobs/v3/api_client/commute_search_sample_test.py b/jobs/v3/api_client/commute_search_sample_test.py
@@ -15,6 +15,7 @@
 import re
 
 import backoff
+from googleapiclient.errors import HttpError
 import pytest
 
 import commute_search_sample
@@ -28,7 +29,7 @@ def company_name():
 
 
 def test_commute_search_sample(company_name, capsys):
-    @backoff.on_exception(backoff.expo, AssertionError, max_time=240)
+    @backoff.on_exception(backoff.expo, (AssertionError, HttpError), max_time=240)
     def eventually_consistent_test():
         commute_search_sample.run_sample(company_name)
         out, _ = capsys.readouterr()
