Fixed test

Author: TetyanaYahodska

compute/cloud-client/src/main/java/compute/snapshot/DeleteSnapshotSchedule.java

@@ -17,6 +17,7 @@
 package compute.snapshot;
 
 // [START compute_snapshot_schedule_delete]
+import com.google.cloud.compute.v1.Operation;
 import com.google.cloud.compute.v1.ResourcePoliciesClient;
 import java.io.IOException;
 import java.util.concurrent.ExecutionException;
@@ -38,13 +39,20 @@ public static void main(String[] args)
   }
 
   // Deletes a snapshot schedule policy.
-  public static void deleteSnapshotSchedule(String projectId, String region, String scheduleName)
+  public static Operation.Status deleteSnapshotSchedule(
+          String projectId, String region, String scheduleName)
           throws IOException, ExecutionException, InterruptedException, TimeoutException {
-
+    // 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 (ResourcePoliciesClient resourcePoliciesClient = ResourcePoliciesClient.create()) {
-      resourcePoliciesClient.deleteAsync(projectId, region, scheduleName).get(3, TimeUnit.MINUTES);
+      Operation response =  resourcePoliciesClient.deleteAsync(projectId, region, scheduleName)
+              .get(3, TimeUnit.MINUTES);
 
-      System.out.println("Snapshot schedule deleted successfully: " + scheduleName);
+      if (response.hasError()) {
+        System.out.printf("Snapshot schedule deletion failed: %s%n", response.getError());
+        return null;
+      }
+      return response.getStatus();
     }
   }
 }

compute/cloud-client/src/test/java/compute/snapshot/SnapshotIT.java

@@ -20,9 +20,7 @@
 import static com.google.common.truth.Truth.assertWithMessage;
 
 import com.google.cloud.compute.v1.Operation;
-import java.io.ByteArrayOutputStream;
 import java.io.IOException;
-import java.io.PrintStream;
 import java.util.UUID;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
@@ -63,16 +61,11 @@ public static void setUp()
   @AfterAll
   public static void cleanup()
           throws IOException, ExecutionException, InterruptedException, TimeoutException {
-    ByteArrayOutputStream stdOut = new ByteArrayOutputStream();
-    System.setOut(new PrintStream(stdOut));
-
     // Delete snapshot schedule created for testing.
-    DeleteSnapshotSchedule.deleteSnapshotSchedule(PROJECT_ID, REGION, SCHEDULE_NAME);
-
-    assertThat(stdOut.toString())
-            .contains("Snapshot schedule deleted successfully: " + SCHEDULE_NAME);
+    Operation.Status status = DeleteSnapshotSchedule
+            .deleteSnapshotSchedule(PROJECT_ID, REGION, SCHEDULE_NAME);
 
-    stdOut.close();
+    assertThat(status).isEqualTo(Operation.Status.DONE);
   }
 
   @Test
@@ -81,6 +74,7 @@ public void testCreateSnapshotScheduleHourly()
     Operation.Status status = CreateSnapshotSchedule.createSnapshotSchedule(
             PROJECT_ID, REGION, SCHEDULE_NAME, SCHEDULE_DESCRIPTION,
             MAX_RETENTION_DAYS, STORAGE_LOCATION, ON_SOURCE_DISK_DELETE);
+
     assertThat(status).isEqualTo(Operation.Status.DONE);
   }
 }