Skip to content

Commit 7c21019

Browse files
nnegreyShabirmean
authored andcommitted
samples: translate: use unique gcs folders for batch translate tests (#2053)
* translate: use unique gcs folders for batch translate tests * update based on feedback * fix typo
1 parent 61702a1 commit 7c21019

File tree

4 files changed

+22
-8
lines changed

4 files changed

+22
-8
lines changed

translate/snippets/src/test/java/com/example/translate/BatchTranslateTextTests.java

+6-2
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import java.io.ByteArrayOutputStream;
2828
import java.io.IOException;
2929
import java.io.PrintStream;
30+
import java.util.UUID;
3031
import java.util.concurrent.ExecutionException;
3132
import java.util.concurrent.TimeoutException;
3233

@@ -43,6 +44,9 @@
4344
public class BatchTranslateTextTests {
4445
private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
4546
private static final String INPUT_URI = "gs://cloud-samples-data/translation/text.txt";
47+
private static final String PREFIX = "BATCH_TRANSLATION_OUTPUT/";
48+
private static final String OUTPUT_URI =
49+
String.format("gs://%s/%s%s/", PROJECT_ID, PREFIX, UUID.randomUUID());
4650

4751
private ByteArrayOutputStream bout;
4852
private PrintStream out;
@@ -53,7 +57,7 @@ private static void cleanUpBucket() {
5357
storage.list(
5458
PROJECT_ID,
5559
Storage.BlobListOption.currentDirectory(),
56-
Storage.BlobListOption.prefix("BATCH_TRANSLATION_OUTPUT/"));
60+
Storage.BlobListOption.prefix(PREFIX));
5761

5862
deleteDirectory(storage, blobs);
5963
}
@@ -102,7 +106,7 @@ public void tearDown() {
102106
public void testBatchTranslateText()
103107
throws InterruptedException, ExecutionException, IOException, TimeoutException {
104108
BatchTranslateText.batchTranslateText(
105-
PROJECT_ID, "en", "es", INPUT_URI, "gs://" + PROJECT_ID + "/BATCH_TRANSLATION_OUTPUT/");
109+
PROJECT_ID, "en", "es", INPUT_URI, OUTPUT_URI);
106110
String got = bout.toString();
107111
assertThat(got).contains("Total Characters: 13");
108112
}

translate/snippets/src/test/java/com/example/translate/BatchTranslateTextWithGlossaryAndModelTests.java

+5-2
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,9 @@ public class BatchTranslateTextWithGlossaryAndModelTests {
6464
private static final String MODEL_ID = "TRL2188848820815848149";
6565
private static final String GLOSSARY_INPUT_URI =
6666
"gs://cloud-samples-data/translation/glossary_ja.csv";
67+
private static final String PREFIX = "BATCH_TRANSLATION_OUTPUT/";
68+
private static final String OUTPUT_URI =
69+
String.format("gs://%s/%s%s/", PROJECT_ID, PREFIX, UUID.randomUUID());
6770

6871
private ByteArrayOutputStream bout;
6972
private PrintStream out;
@@ -74,7 +77,7 @@ private static final void cleanUpBucket() {
7477
storage.list(
7578
PROJECT_ID,
7679
Storage.BlobListOption.currentDirectory(),
77-
Storage.BlobListOption.prefix("BATCH_TRANSLATION_OUTPUT/"));
80+
Storage.BlobListOption.prefix(PREFIX));
7881

7982
deleteDirectory(storage, blobs);
8083
}
@@ -137,7 +140,7 @@ public void testBatchTranslateTextWithGlossaryAndModel()
137140
"en",
138141
"ja",
139142
INPUT_URI,
140-
"gs://" + PROJECT_ID + "/BATCH_TRANSLATION_OUTPUT/",
143+
OUTPUT_URI,
141144
GLOSSARY_ID,
142145
MODEL_ID);
143146
String got = bout.toString();

translate/snippets/src/test/java/com/example/translate/BatchTranslateTextWithGlossaryTests.java

+5-2
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,9 @@ public class BatchTranslateTextWithGlossaryTests {
6363
"gs://cloud-samples-data/translation/glossary_ja.csv";
6464
private static final String GLOSSARY_ID =
6565
String.format("test_%s", UUID.randomUUID().toString().replace("-", "_").substring(0, 26));
66+
private static final String PREFIX = "BATCH_TRANSLATION_OUTPUT/";
67+
private static final String OUTPUT_URI =
68+
String.format("gs://%s/%s%s/", PROJECT_ID, PREFIX, UUID.randomUUID());
6669

6770
private ByteArrayOutputStream bout;
6871
private PrintStream out;
@@ -73,7 +76,7 @@ private static final void cleanUpBucket() {
7376
storage.list(
7477
PROJECT_ID,
7578
Storage.BlobListOption.currentDirectory(),
76-
Storage.BlobListOption.prefix("BATCH_TRANSLATION_OUTPUT/"));
79+
Storage.BlobListOption.prefix(PREFIX));
7780

7881
deleteDirectory(storage, blobs);
7982
}
@@ -137,7 +140,7 @@ public void testBatchTranslateTextWithGlossary()
137140
"en",
138141
"ja",
139142
INPUT_URI,
140-
"gs://" + PROJECT_ID + "/BATCH_TRANSLATION_OUTPUT/",
143+
OUTPUT_URI,
141144
GLOSSARY_ID);
142145
String got = bout.toString();
143146
assertThat(got).contains("Total Characters: 9");

translate/snippets/src/test/java/com/example/translate/BatchTranslateTextWithModelTests.java

+6-2
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import java.io.ByteArrayOutputStream;
2828
import java.io.IOException;
2929
import java.io.PrintStream;
30+
import java.util.UUID;
3031
import java.util.concurrent.ExecutionException;
3132
import java.util.concurrent.TimeoutException;
3233

@@ -45,6 +46,9 @@ public class BatchTranslateTextWithModelTests {
4546
private static final String INPUT_URI =
4647
"gs://cloud-samples-data/translation/custom_model_text.txt";
4748
private static final String MODEL_ID = "TRL2188848820815848149";
49+
private static final String PREFIX = "BATCH_TRANSLATION_OUTPUT/";
50+
private static final String OUTPUT_URI =
51+
String.format("gs://%s/%s%s/", PROJECT_ID, PREFIX, UUID.randomUUID());
4852

4953
private ByteArrayOutputStream bout;
5054
private PrintStream out;
@@ -55,7 +59,7 @@ private static final void cleanUpBucket() {
5559
storage.list(
5660
PROJECT_ID,
5761
Storage.BlobListOption.currentDirectory(),
58-
Storage.BlobListOption.prefix("BATCH_TRANSLATION_OUTPUT/"));
62+
Storage.BlobListOption.prefix(PREFIX));
5963

6064
deleteDirectory(storage, blobs);
6165
}
@@ -108,7 +112,7 @@ public void testBatchTranslateTextWithModel()
108112
"en",
109113
"ja",
110114
INPUT_URI,
111-
"gs://" + PROJECT_ID + "/BATCH_TRANSLATION_OUTPUT/",
115+
OUTPUT_URI,
112116
MODEL_ID);
113117
String got = bout.toString();
114118
assertThat(got).contains("Total Characters: 15");

0 commit comments

Comments
 (0)