From 1808a5128723258248f20c7bb99dfefd972bc458 Mon Sep 17 00:00:00 2001 From: Mike <45373284+munkhuushmgl@users.noreply.github.com> Date: Thu, 7 Jan 2021 16:34:05 -0800 Subject: [PATCH] chore: fixed wrong method name (#100) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Due to wrong method usage one method, it led to glossary max [1000] from not deleting glossary in the teardown. Fixes #95, #96, #97, #98 🦕 --- ...slate_text_with_glossary_and_model_test.py | 21 ++++++++++++++----- ...late_v3_batch_translate_text_with_model.py | 2 +- .../snippets/translate_v3_delete_glossary.py | 4 +++- 3 files changed, 20 insertions(+), 7 deletions(-) diff --git a/packages/google-cloud-translate/samples/snippets/translate_v3_batch_translate_text_with_glossary_and_model_test.py b/packages/google-cloud-translate/samples/snippets/translate_v3_batch_translate_text_with_glossary_and_model_test.py index 6579831af275..dc12e2d1b69b 100644 --- a/packages/google-cloud-translate/samples/snippets/translate_v3_batch_translate_text_with_glossary_and_model_test.py +++ b/packages/google-cloud-translate/samples/snippets/translate_v3_batch_translate_text_with_glossary_and_model_test.py @@ -15,7 +15,10 @@ import os import uuid +import backoff +from google.api_core.exceptions import DeadlineExceeded, GoogleAPICallError from google.cloud import storage +from google.cloud.exceptions import NotFound import pytest import translate_v3_batch_translate_text_with_glossary_and_model @@ -37,16 +40,24 @@ def glossary(): yield glossary_id - try: - translate_v3_delete_glossary.sample_delete_glossary(PROJECT_ID, glossary_id) - except Exception: - pass + # clean up + @backoff.on_exception( + backoff.expo, (DeadlineExceeded, GoogleAPICallError), max_time=60 + ) + def delete_glossary(): + try: + translate_v3_delete_glossary.delete_glossary(PROJECT_ID, glossary_id) + except NotFound as e: + # Ignoring this case. + print("Got NotFound, detail: {}".format(str(e))) + + delete_glossary() @pytest.fixture(scope="function") def bucket(): """Create a temporary bucket to store annotation output.""" - bucket_name = "mike-test-delete-" + str(uuid.uuid1()) + bucket_name = "test-bucket-for-glossary-" + str(uuid.uuid1()) storage_client = storage.Client() bucket = storage_client.create_bucket(bucket_name) diff --git a/packages/google-cloud-translate/samples/snippets/translate_v3_batch_translate_text_with_model.py b/packages/google-cloud-translate/samples/snippets/translate_v3_batch_translate_text_with_model.py index 07d967d71774..a5b94866623f 100644 --- a/packages/google-cloud-translate/samples/snippets/translate_v3_batch_translate_text_with_model.py +++ b/packages/google-cloud-translate/samples/snippets/translate_v3_batch_translate_text_with_model.py @@ -24,7 +24,7 @@ def batch_translate_text_with_model( model_id="YOUR_MODEL_ID", ): """Batch translate text using Translation model. - Model can be AutoML or General[built-in] model. """ + Model can be AutoML or General[built-in] model.""" client = translate.TranslationServiceClient() diff --git a/packages/google-cloud-translate/samples/snippets/translate_v3_delete_glossary.py b/packages/google-cloud-translate/samples/snippets/translate_v3_delete_glossary.py index 336b7a06b087..b5c55b2ba09b 100644 --- a/packages/google-cloud-translate/samples/snippets/translate_v3_delete_glossary.py +++ b/packages/google-cloud-translate/samples/snippets/translate_v3_delete_glossary.py @@ -17,7 +17,9 @@ def delete_glossary( - project_id="YOUR_PROJECT_ID", glossary_id="YOUR_GLOSSARY_ID", timeout=180, + project_id="YOUR_PROJECT_ID", + glossary_id="YOUR_GLOSSARY_ID", + timeout=180, ): """Delete a specific glossary based on the glossary ID.""" client = translate.TranslationServiceClient()