From b12d3dfeb89502a331e8318de094eac95c7e2748 Mon Sep 17 00:00:00 2001 From: Kevin Gordillo Date: Mon, 24 Aug 2026 21:51:58 +0000 Subject: [PATCH 1/3] fix(genai): allow configurable bucket and add safe cleanup in VideoGenerationIT --- .../genai/videogeneration/VideoGenerationIT.java | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/genai/snippets/src/test/java/genai/videogeneration/VideoGenerationIT.java b/genai/snippets/src/test/java/genai/videogeneration/VideoGenerationIT.java index 25e56317084..9febe9ad20c 100644 --- a/genai/snippets/src/test/java/genai/videogeneration/VideoGenerationIT.java +++ b/genai/snippets/src/test/java/genai/videogeneration/VideoGenerationIT.java @@ -38,7 +38,8 @@ public class VideoGenerationIT { private static final String VIDEO_GEN_MODEL = "veo-3.1-generate-001"; - private static final String BUCKET_NAME = "java-docs-samples-testing"; + private static final String BUCKET_NAME = + System.getenv().getOrDefault("BUCKET_NAME", "java-docs-samples-testing"); private static final String PREFIX = "genai-video-generation-" + UUID.randomUUID(); private static final String OUTPUT_GCS_URI = String.format("gs://%s/%s", BUCKET_NAME, PREFIX); private ByteArrayOutputStream bout; @@ -58,11 +59,15 @@ public static void checkRequirements() { @AfterClass public static void cleanup() { - Storage storage = StorageOptions.getDefaultInstance().getService(); - Page blobs = storage.list(BUCKET_NAME, Storage.BlobListOption.prefix(PREFIX)); + try { + Storage storage = StorageOptions.getDefaultInstance().getService(); + Page blobs = storage.list(BUCKET_NAME, Storage.BlobListOption.prefix(PREFIX)); - for (Blob blob : blobs.iterateAll()) { - storage.delete(blob.getBlobId()); + for (Blob blob : blobs.iterateAll()) { + storage.delete(blob.getBlobId()); + } + } catch (Exception e) { + System.err.println("Failed to cleanup bucket " + BUCKET_NAME + ": " + e.getMessage()); } } From 3921d9c70f3a62f2ba359abd58bebdb20b4452de Mon Sep 17 00:00:00 2001 From: Kevin Gordillo Date: Mon, 24 Aug 2026 22:28:46 +0000 Subject: [PATCH 2/3] fix(genai): allow configurable bucket, safe cleanup, and ignore deprecated Imagen tests in ImageGenerationIT --- .../imagegeneration/ImageGenerationIT.java | 24 ++++++++++++++----- 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/genai/snippets/src/test/java/genai/imagegeneration/ImageGenerationIT.java b/genai/snippets/src/test/java/genai/imagegeneration/ImageGenerationIT.java index a5e6d43ebc2..d8ef442ed87 100644 --- a/genai/snippets/src/test/java/genai/imagegeneration/ImageGenerationIT.java +++ b/genai/snippets/src/test/java/genai/imagegeneration/ImageGenerationIT.java @@ -33,6 +33,7 @@ import org.junit.AfterClass; import org.junit.Before; import org.junit.BeforeClass; +import org.junit.Ignore; import org.junit.Test; import org.junit.runner.RunWith; import org.junit.runners.JUnit4; @@ -41,7 +42,8 @@ public class ImageGenerationIT { private static final String IMAGEN_3_MODEL = "imagen-3.0-capability-001"; - private static final String BUCKET_NAME = "java-docs-samples-testing"; + private static final String BUCKET_NAME = + System.getenv().getOrDefault("BUCKET_NAME", "java-docs-samples-testing"); private static final String PREFIX = "genai-img-generation-" + UUID.randomUUID(); private static final String OUTPUT_GCS_URI = String.format("gs://%s/%s", BUCKET_NAME, PREFIX); private static final String IMAGEN_4_MODEL = "imagen-4.0-generate-001"; @@ -64,11 +66,15 @@ public static void checkRequirements() { @AfterClass public static void cleanup() { - Storage storage = StorageOptions.getDefaultInstance().getService(); - Page blobs = storage.list(BUCKET_NAME, Storage.BlobListOption.prefix(PREFIX)); - - for (Blob blob : blobs.iterateAll()) { - storage.delete(blob.getBlobId()); + try { + Storage storage = StorageOptions.getDefaultInstance().getService(); + Page blobs = storage.list(BUCKET_NAME, Storage.BlobListOption.prefix(PREFIX)); + + for (Blob blob : blobs.iterateAll()) { + storage.delete(blob.getBlobId()); + } + } catch (Exception e) { + System.err.println("Failed to cleanup bucket " + BUCKET_NAME + ": " + e.getMessage()); } } @@ -85,6 +91,7 @@ public void tearDown() { } @Test + @Ignore("Imagen models are deprecated and unavailable; migrated to Gemini in follow-up PR") public void testImageGenCannyCtrlTypeWithTextAndImage() { Optional response = ImageGenCannyCtrlTypeWithTextAndImage.cannyEdgeCustomization( @@ -94,6 +101,7 @@ public void testImageGenCannyCtrlTypeWithTextAndImage() { } @Test + @Ignore("Imagen models are deprecated and unavailable; migrated to Gemini in follow-up PR") public void testImageGenRawReferenceWithTextAndImage() { Optional response = ImageGenRawReferenceWithTextAndImage.styleTransferCustomization( @@ -103,6 +111,7 @@ public void testImageGenRawReferenceWithTextAndImage() { } @Test + @Ignore("Imagen models are deprecated and unavailable; migrated to Gemini in follow-up PR") public void testImageGenScribbleCtrlTypeWithTextAndImage() { Optional response = ImageGenScribbleCtrlTypeWithTextAndImage.scribbleCustomization( @@ -112,6 +121,7 @@ public void testImageGenScribbleCtrlTypeWithTextAndImage() { } @Test + @Ignore("Imagen models are deprecated and unavailable; migrated to Gemini in follow-up PR") public void testImageGenStyleReferenceWithTextAndImage() { Optional response = ImageGenStyleReferenceWithTextAndImage.styleCustomization( @@ -121,6 +131,7 @@ public void testImageGenStyleReferenceWithTextAndImage() { } @Test + @Ignore("Imagen models are deprecated and unavailable; migrated to Gemini in follow-up PR") public void testImageGenSubjectReferenceWithTextAndImage() { Optional response = ImageGenSubjectReferenceWithTextAndImage.subjectCustomization( @@ -141,6 +152,7 @@ public void testImageGenVirtualTryOnWithTextAndImage() throws IOException { } @Test + @Ignore("Imagen models are deprecated and unavailable; migrated to Gemini in follow-up PR") public void testImageGenWithText() throws IOException { Image image = ImageGenWithText.generateImage(IMAGEN_4_MODEL, "resources/output/dog_newspaper.png"); From 76f9325e4efc013897aa712ba43b878aabac8075 Mon Sep 17 00:00:00 2001 From: Kevin Gordillo Date: Mon, 24 Aug 2026 22:30:16 +0000 Subject: [PATCH 3/3] fix(genai): strip gs:// prefix from BUCKET_NAME in test classes --- .../src/test/java/genai/imagegeneration/ImageGenerationIT.java | 3 ++- .../src/test/java/genai/videogeneration/VideoGenerationIT.java | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/genai/snippets/src/test/java/genai/imagegeneration/ImageGenerationIT.java b/genai/snippets/src/test/java/genai/imagegeneration/ImageGenerationIT.java index d8ef442ed87..9946a4ff06b 100644 --- a/genai/snippets/src/test/java/genai/imagegeneration/ImageGenerationIT.java +++ b/genai/snippets/src/test/java/genai/imagegeneration/ImageGenerationIT.java @@ -43,7 +43,8 @@ public class ImageGenerationIT { private static final String IMAGEN_3_MODEL = "imagen-3.0-capability-001"; private static final String BUCKET_NAME = - System.getenv().getOrDefault("BUCKET_NAME", "java-docs-samples-testing"); + System.getenv().getOrDefault("BUCKET_NAME", "java-docs-samples-testing") + .replaceFirst("^gs://", ""); private static final String PREFIX = "genai-img-generation-" + UUID.randomUUID(); private static final String OUTPUT_GCS_URI = String.format("gs://%s/%s", BUCKET_NAME, PREFIX); private static final String IMAGEN_4_MODEL = "imagen-4.0-generate-001"; diff --git a/genai/snippets/src/test/java/genai/videogeneration/VideoGenerationIT.java b/genai/snippets/src/test/java/genai/videogeneration/VideoGenerationIT.java index 9febe9ad20c..159e529c6e5 100644 --- a/genai/snippets/src/test/java/genai/videogeneration/VideoGenerationIT.java +++ b/genai/snippets/src/test/java/genai/videogeneration/VideoGenerationIT.java @@ -39,7 +39,8 @@ public class VideoGenerationIT { private static final String VIDEO_GEN_MODEL = "veo-3.1-generate-001"; private static final String BUCKET_NAME = - System.getenv().getOrDefault("BUCKET_NAME", "java-docs-samples-testing"); + System.getenv().getOrDefault("BUCKET_NAME", "java-docs-samples-testing") + .replaceFirst("^gs://", ""); private static final String PREFIX = "genai-video-generation-" + UUID.randomUUID(); private static final String OUTPUT_GCS_URI = String.format("gs://%s/%s", BUCKET_NAME, PREFIX); private ByteArrayOutputStream bout;