Added return value for CreateQueuedResource

Author: TetyanaYahodska

tpu/src/main/java/tpu/CreateQueuedResource.java

@@ -55,7 +55,7 @@ public static void main(String[] args)
   }
 
   // Creates a Queued Resource
-  public static void createQueuedResource(String projectId, String zone,
+  public static QueuedResource createQueuedResource(String projectId, String zone,
       String queuedResourceId, String nodeName, String tpuType, String tpuSoftwareVersion)
       throws IOException, ExecutionException, InterruptedException {
     // With these settings the client library handles the Operation's polling mechanism
@@ -116,6 +116,7 @@ public static void createQueuedResource(String projectId, String zone,
       // You can wait until TPU Node is READY,
       // and check its status using getTpuVm() from "tpu_vm_get" sample.
       System.out.printf("Queued Resource created: %s\n", response.getName());
+      return response;
     }
   }
 }

tpu/src/test/java/tpu/QueuedResourcesIT.java

@@ -93,15 +93,11 @@ public static void cleanup() throws IOException {
   @Order(1)
   public void testCreateQueuedResource()
       throws IOException, ExecutionException, InterruptedException {
-    final PrintStream out = System.out;
-    ByteArrayOutputStream stdOut = new ByteArrayOutputStream();
-    System.setOut(new PrintStream(stdOut));
-    CreateQueuedResource.createQueuedResource(PROJECT_ID, ZONE,
+    QueuedResource queuedResource = CreateQueuedResource.createQueuedResource(PROJECT_ID, ZONE,
         QUEUED_RESOURCE_NAME, NODE_NAME, TPU_TYPE, TPU_SOFTWARE_VERSION);
 
-    assertThat(stdOut.toString()).contains("Queued Resource created: " + QUEUED_RESOURCE_PATH_NAME);
-    stdOut.close();
-    System.setOut(out);
+    assertThat(queuedResource.getName()).isEqualTo(QUEUED_RESOURCE_PATH_NAME);
+    assertThat(queuedResource.getTpu().getNodeSpec(0).getNode().getName()).isEqualTo(NODE_NAME);
   }
 
   @Test