Skip to content

Commit

Permalink
testing(automl): retry upon InternalServerError [(#4350)](GoogleCloud…
Browse files Browse the repository at this point in the history
  • Loading branch information
Takashi Matsuo authored and busunkim96 committed Aug 11, 2020
1 parent 28d13a6 commit 4b8e11b
Showing 1 changed file with 8 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,8 @@

import os

import backoff
from google.api_core.exceptions import InternalServerError
import pytest

import list_operation_status
Expand All @@ -23,6 +25,11 @@

@pytest.mark.slow
def test_list_operation_status(capsys):
list_operation_status.list_operation_status(PROJECT_ID)
# We saw 500 InternalServerError. Now we just retry few times.
@backoff.on_exception(backoff.expo, InternalServerError, max_time=120)
def run_sample():
list_operation_status.list_operation_status(PROJECT_ID)

run_sample()
out, _ = capsys.readouterr()
assert "Operation details" in out

0 comments on commit 4b8e11b

Please sign in to comment.