Fixed imports

Author: TetyanaYahodska

tpu/src/main/java/tpu/ListQueuedResources.java

@@ -18,7 +18,9 @@
 
 //[START tpu_queued_resources_list]
 import com.google.cloud.tpu.v2alpha1.ListQueuedResourcesRequest;
+import com.google.cloud.tpu.v2alpha1.QueuedResource;
 import com.google.cloud.tpu.v2alpha1.TpuClient;
+import com.google.cloud.tpu.v2alpha1.TpuClient.ListQueuedResourcesPage;
 import java.io.IOException;
 
 public class ListQueuedResources {
@@ -33,16 +35,20 @@ public static void main(String[] args) throws IOException {
   }
 
   // List Queued Resources.
-  public static TpuClient.ListQueuedResourcesPage listQueuedResources(
+  public static ListQueuedResourcesPage listQueuedResources(
       String projectId, String zone) throws IOException {
     String parent = String.format("projects/%s/locations/%s", projectId, zone);
     // Initialize client that will be used to send requests. This client only needs to be created
     // once, and can be reused for multiple requests.
     try (TpuClient tpuClient = TpuClient.create()) {
       ListQueuedResourcesRequest request =
           ListQueuedResourcesRequest.newBuilder().setParent(parent).build();
+      ListQueuedResourcesPage response =  tpuClient.listQueuedResources(request).getPage();
 
-      return tpuClient.listQueuedResources(request).getPage();
+      for (QueuedResource queuedResource : response.iterateAll()) {
+        System.out.println(queuedResource.getName());
+      }
+      return response;
     }
   }
 }

tpu/src/test/java/tpu/QueuedResourceIT.java

@@ -33,6 +33,8 @@
 import com.google.cloud.tpu.v2alpha1.ListQueuedResourcesRequest;
 import com.google.cloud.tpu.v2alpha1.QueuedResource;
 import com.google.cloud.tpu.v2alpha1.TpuClient;
+import com.google.cloud.tpu.v2alpha1.TpuClient.ListQueuedResourcesPage;
+import com.google.cloud.tpu.v2alpha1.TpuClient.ListQueuedResourcesPagedResponse;
 import com.google.cloud.tpu.v2alpha1.TpuSettings;
 import java.io.IOException;
 import java.util.Arrays;
@@ -134,16 +136,16 @@ public void testListTpuVm() throws IOException {
 
       TpuClient mockClient = mock(TpuClient.class);
       mockedTpuClient.when(TpuClient::create).thenReturn(mockClient);
-      TpuClient.ListQueuedResourcesPagedResponse mockListQueuedResourcesResponse =
-          mock(TpuClient.ListQueuedResourcesPagedResponse.class);
+      ListQueuedResourcesPagedResponse mockListQueuedResourcesResponse =
+          mock(ListQueuedResourcesPagedResponse.class);
       when(mockClient.listQueuedResources(any(ListQueuedResourcesRequest.class)))
           .thenReturn(mockListQueuedResourcesResponse);
-      TpuClient.ListQueuedResourcesPage mockQueuedResourcesPage =
-          mock(TpuClient.ListQueuedResourcesPage.class);
+      ListQueuedResourcesPage mockQueuedResourcesPage =
+          mock(ListQueuedResourcesPage.class);
       when(mockListQueuedResourcesResponse.getPage()).thenReturn(mockQueuedResourcesPage);
       when(mockQueuedResourcesPage.getValues()).thenReturn(mockListQueuedResources);
 
-      TpuClient.ListQueuedResourcesPage returnedList =
+      ListQueuedResourcesPage returnedList =
           ListQueuedResources.listQueuedResources(PROJECT_ID, ZONE);
 
       assertThat(returnedList.getValues()).isEqualTo(mockListQueuedResources);