diff --git a/translation/samples/snippets/translate_v3_batch_translate_text_test.py b/translation/samples/snippets/translate_v3_batch_translate_text_test.py index 2474f60d56a9..9dda7c04789a 100644 --- a/translation/samples/snippets/translate_v3_batch_translate_text_test.py +++ b/translation/samples/snippets/translate_v3_batch_translate_text_test.py @@ -33,6 +33,7 @@ def bucket(): bucket.delete(force=True) +@pytest.mark.flaky(max_runs=3, min_passes=1) def test_batch_translate_text(capsys, bucket): translate_v3_batch_translate_text.batch_translate_text( "gs://cloud-samples-data/translation/text.txt", diff --git a/translation/samples/snippets/translate_v3_batch_translate_text_with_glossary_test.py b/translation/samples/snippets/translate_v3_batch_translate_text_with_glossary_test.py index 0e79bf9b8693..e50105211892 100644 --- a/translation/samples/snippets/translate_v3_batch_translate_text_with_glossary_test.py +++ b/translation/samples/snippets/translate_v3_batch_translate_text_with_glossary_test.py @@ -52,6 +52,7 @@ def bucket(): bucket.delete(force=True) +@pytest.mark.flaky(max_runs=3, min_passes=1) def test_batch_translate_text_with_glossary(capsys, bucket, glossary): translate_v3_batch_translate_text_with_glossary.batch_translate_text_with_glossary( "gs://cloud-samples-data/translation/text_with_glossary.txt", diff --git a/translation/samples/snippets/translate_v3_create_glossary_test.py b/translation/samples/snippets/translate_v3_create_glossary_test.py index 77a2cf1435cc..b24488c80894 100644 --- a/translation/samples/snippets/translate_v3_create_glossary_test.py +++ b/translation/samples/snippets/translate_v3_create_glossary_test.py @@ -13,26 +13,32 @@ # limitations under the License. import os +import uuid + +import pytest + import translate_v3_create_glossary import translate_v3_delete_glossary -import uuid PROJECT_ID = os.environ["GCLOUD_PROJECT"] GLOSSARY_INPUT_URI = "gs://cloud-samples-data/translation/glossary_ja.csv" +@pytest.mark.flaky(max_runs=3, min_passes=1) def test_create_glossary(capsys): - glossary_id = "test-{}".format(uuid.uuid4()) - translate_v3_create_glossary.create_glossary( - PROJECT_ID, GLOSSARY_INPUT_URI, glossary_id - ) - out, _ = capsys.readouterr() - # assert - assert "Created:" in out - assert "gs://cloud-samples-data/translation/glossary_ja.csv" in out - - # clean up after use try: - translate_v3_delete_glossary.delete_glossary(PROJECT_ID, glossary_id) - except Exception: - pass + glossary_id = "test-{}".format(uuid.uuid4()) + translate_v3_create_glossary.create_glossary( + PROJECT_ID, GLOSSARY_INPUT_URI, glossary_id + ) + out, _ = capsys.readouterr() + # assert + assert "Created:" in out + assert "gs://cloud-samples-data/translation/glossary_ja.csv" in out + finally: + # clean up after use + try: + translate_v3_delete_glossary.delete_glossary( + PROJECT_ID, glossary_id) + except Exception: + pass diff --git a/translation/samples/snippets/translate_v3_delete_glossary_test.py b/translation/samples/snippets/translate_v3_delete_glossary_test.py index 8d7fb2d92451..4309ca01b959 100644 --- a/translation/samples/snippets/translate_v3_delete_glossary_test.py +++ b/translation/samples/snippets/translate_v3_delete_glossary_test.py @@ -13,14 +13,18 @@ # limitations under the License. import os +import uuid + +import pytest + import translate_v3_create_glossary import translate_v3_delete_glossary -import uuid PROJECT_ID = os.environ["GCLOUD_PROJECT"] GLOSSARY_INPUT_URI = "gs://cloud-samples-data/translation/glossary_ja.csv" +@pytest.mark.flaky(max_runs=3, min_passes=1) def test_delete_glossary(capsys): # setup glossary_id = "test-{}".format(uuid.uuid4())