samples(tests): added retry logic to flaky batch tests (#587)

GoogleCloudPlatform · Oct 9, 2020 · 39adafb · 39adafb
1 parent d3f17c4
commit 39adafb
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/translate/test/v3/translate_batch_translate_text.test.js b/translate/test/v3/translate_batch_translate_text.test.js
@@ -25,7 +25,8 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const REGION_TAG = 'translate_batch_translate_text';
 
-describe(REGION_TAG, () => {
+describe(REGION_TAG, function () {
+  this.retries(3);
   const translationClient = new TranslationServiceClient();
   const location = 'us-central1';
   const bucketUuid = uuid.v4();

diff --git a/translate/test/v3/translate_batch_translate_text_with_glossary.test.js b/translate/test/v3/translate_batch_translate_text_with_glossary.test.js
@@ -25,7 +25,8 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const REGION_TAG = 'translate_batch_translate_text_with_glossary';
 
-describe(REGION_TAG, () => {
+describe(REGION_TAG, function () {
+  this.retries(3);
   const translationClient = new TranslationServiceClient();
   const location = 'us-central1';
   const glossaryId = `my_test_glossary_${uuid.v4()}`;

diff --git a/translate/test/v3/translate_batch_translate_text_with_glossary_and_model.test.js b/translate/test/v3/translate_batch_translate_text_with_glossary_and_model.test.js
@@ -25,7 +25,8 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const REGION_TAG = 'translate_batch_translate_text_with_glossary_and_model';
 
-describe(REGION_TAG, () => {
+describe(REGION_TAG, function () {
+  this.retries(3);
   const translationClient = new TranslationServiceClient();
   const location = 'us-central1';
   const glossaryId = `my_test_glossary_${uuid.v4()}`;

diff --git a/translate/test/v3/translate_batch_translate_text_with_model.test.js b/translate/test/v3/translate_batch_translate_text_with_model.test.js
@@ -25,7 +25,8 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const REGION_TAG = 'translate_batch_translate_text_with_model';
 
-describe(REGION_TAG, () => {
+describe(REGION_TAG, function () {
+  this.retries(3);
   const translationClient = new TranslationServiceClient();
   const location = 'us-central1';
   const modelId = 'TRL1218052175389786112';