fixing tests

Author: Guiners

genai/test/imggen-canny-ctrl-type-with-txt-img.test.js

@@ -20,6 +20,7 @@ const {describe, it} = require('mocha');
 const projectId = process.env.CAIP_PROJECT_ID;
 
 const sample = require('../image-generation/imggen-canny-ctrl-type-with-txt-img');
+const location = 'us-central1';
 const {delay} = require('./util');
 const {createOutputGcsUri} = require('./imggen-util');
 
@@ -32,7 +33,8 @@ describe('imggen-canny-ctrl-type-with-txt-img', () => {
     await delay(this.test);
     const generatedFileNames = await sample.generateImage(
       output.uri,
-      projectId
+      projectId,
+      location
     );
     assert(generatedFileNames.length > 0);
   });

genai/test/imggen-raw-reference-with-txt-img.test.js

@@ -22,11 +22,12 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const sample = require('../image-generation/imggen-raw-reference-with-txt-img');
 const {delay} = require('./util');
 const {createOutputGcsUri} = require('./imggen-util');
+const location = 'us-central1';
 
 describe('imggen-raw-reference-with-txt-img', () => {
   it('should return an array of generated image URIs', async function () {
     this.timeout(600000);
-    // this.retries(3);
+    this.retries(3);
 
     const output = await createOutputGcsUri();
     console.log('Output GCS URI:', output.uri);
@@ -35,7 +36,8 @@ describe('imggen-raw-reference-with-txt-img', () => {
       await delay(this.test);
       const generatedFileNames = await sample.generateImage(
         output.uri,
-        projectId
+        projectId,
+        location
       );
       console.log('Generated files:', generatedFileNames);
 

genai/test/imggen-scribble-ctrl-type-with-txt-img.test.js

@@ -21,6 +21,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const sample = require('../image-generation/imggen-scribble-ctrl-type-with-txt-img');
 const {delay} = require('./util');
 const {createOutputGcsUri} = require('./imggen-util');
+const location = 'us-central1';
 
 describe('imggen-scribble-ctrl-type-with-txt-img', async () => {
   it('should generate images from a text prompt with control reference image', async function () {
@@ -34,7 +35,8 @@ describe('imggen-scribble-ctrl-type-with-txt-img', async () => {
       await delay(this.test);
       const generatedFileNames = await sample.generateImage(
         output.uri,
-        projectId
+        projectId,
+        location
       );
       console.log('Generated files:', generatedFileNames);
 

genai/test/imggen-style-reference-with-txt-img.test.js

@@ -21,6 +21,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const sample = require('../image-generation/imggen-style-reference-with-txt-img');
 const {delay} = require('./util');
 const {createOutputGcsUri} = require('./imggen-util');
+const location = 'us-central1';
 
 describe('imggen-style-reference-with-txt-img', async () => {
   it('should generate images from a text prompt with style reference', async function () {
@@ -31,7 +32,8 @@ describe('imggen-style-reference-with-txt-img', async () => {
     await delay(this.test);
     const generatedFileNames = await sample.generateImage(
       output.uri,
-      projectId
+      projectId,
+      location
     );
     assert(generatedFileNames.length > 0);
   });

genai/test/imggen-subj-refer-ctrl-refer-with-txt-imgs.test.js

@@ -21,6 +21,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const sample = require('../image-generation/imggen-subj-refer-ctrl-refer-with-txt-imgs');
 const {delay} = require('./util');
 const {createOutputGcsUri} = require('./imggen-util');
+const location = 'us-central1';
 
 describe('imggen-subj-refer-ctrl-refer-with-txt-imgs', async () => {
   it('should generate images from a text prompt with subject reference image and control reference image', async function () {
@@ -31,7 +32,8 @@ describe('imggen-subj-refer-ctrl-refer-with-txt-imgs', async () => {
     await delay(this.test);
     const generatedFileNames = await sample.generateImage(
       output.uri,
-      projectId
+      projectId,
+      location
     );
     assert(generatedFileNames.length > 0);
   });

genai/test/imggen-util.js

@@ -15,9 +15,8 @@
 const {Storage} = require('@google-cloud/storage');
 const {format} = require('date-fns');
 
-// const gcsOutputBucket = 'nodejs-docs-samples-tests';
-const gcsOutputBucket =
-    'example-cloud-bucket-5439474/nodejs-docs-samples-tests';
+const gcsOutputBucket = 'nodejs-docs-samples-tests';
+
 module.exports.createOutputGcsUri = async function () {
   const prefix = `text_output/${format(new Date(), 'yyyy-MM-dd-HH-mm-ss')}`;
   const gcsUri = `gs://${gcsOutputBucket}/${prefix}`;