Moved test to ReservationIT

Author: TetyanaYahodska

compute/cloud-client/src/test/java/compute/reservation/ConsumeReservationIT.java

@@ -16,14 +16,18 @@
 
 package compute.reservation;
 
+import static com.google.cloud.compute.v1.ReservationAffinity.ConsumeReservationType.SPECIFIC_RESERVATION;
 import static com.google.common.truth.Truth.assertThat;
 import static com.google.common.truth.Truth.assertWithMessage;
 
 import com.google.api.gax.rpc.NotFoundException;
+import com.google.cloud.compute.v1.Instance;
+import com.google.cloud.compute.v1.InstancesClient;
 import com.google.cloud.compute.v1.Reservation;
 import com.google.cloud.compute.v1.ReservationsClient;
 import compute.CreateInstanceTemplate;
 import compute.CreateRegionalInstanceTemplate;
+import compute.DeleteInstance;
 import compute.DeleteInstanceTemplate;
 import compute.DeleteRegionalInstanceTemplate;
 import compute.Util;
@@ -50,18 +54,20 @@ public class ReservationIT {
   private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
   private static final String ZONE = "us-west1-a";
   private static final String REGION = ZONE.substring(0, ZONE.lastIndexOf('-'));
+  static String javaVersion = System.getProperty("java.version").substring(0, 2);
   private static ReservationsClient reservationsClient;
+  private static InstancesClient instancesClient;
   private static String RESERVATION_NAME_GLOBAL;
   private static String RESERVATION_NAME_REGIONAL;
   private static String GLOBAL_INSTANCE_TEMPLATE_URI;
   private static String REGIONAL_INSTANCE_TEMPLATE_URI;
-  static String javaVersion = System.getProperty("java.version").substring(0, 2);
-  private static final String GLOBAL_INSTANCE_TEMPLATE_NAME =
-      "test-global-inst-temp-" + javaVersion + "-" + UUID.randomUUID().toString().substring(0, 8);
-  private static final String REGIONAL_INSTANCE_TEMPLATE_NAME =
-      "test-regional-inst-temp-" + javaVersion  + "-"
-          + UUID.randomUUID().toString().substring(0, 8);
+  private static String RESERVATION_SHARED_NAME;
+  private static String SPECIFIC_SHARED_INSTANCE_NAME;
+  private static  String GLOBAL_INSTANCE_TEMPLATE_NAME;
+  private static String REGIONAL_INSTANCE_TEMPLATE_NAME;
   private static final int NUMBER_OF_VMS = 3;
+  private static final String MACHINE_TYPE = "n2-standard-32";
+  private static final String MIN_CPU_PLATFORM = "Intel Cascade Lake";
 
   // Check if the required environment variables are set.
   public static void requireEnvVar(String envVarName) {
@@ -79,25 +85,38 @@ public static void setUp()
     System.setOut(new PrintStream(stdOut));
 
     // Cleanup existing stale resources.
+    Util.cleanUpExistingInstances("test-shared-instance-" + javaVersion, PROJECT_ID, ZONE);
     Util.cleanUpExistingInstanceTemplates("test-global-inst-temp-" + javaVersion, PROJECT_ID);
     Util.cleanUpExistingRegionalInstanceTemplates(
         "test-regional-inst-temp-" + javaVersion, PROJECT_ID, ZONE);
     Util.cleanUpExistingReservations(
         "test-reservation-global-" + javaVersion, PROJECT_ID, ZONE);
     Util.cleanUpExistingReservations("test-reservation-regional-" + javaVersion, PROJECT_ID, ZONE);
+    Util.cleanUpExistingReservations("test-shared-instance-" + javaVersion, PROJECT_ID, ZONE);
 
     // Initialize the client once for all tests
     reservationsClient = ReservationsClient.create();
+    instancesClient = InstancesClient.create();
 
     RESERVATION_NAME_GLOBAL = "test-reservation-global-" + javaVersion  + "-"
         + UUID.randomUUID().toString().substring(0, 8);
     RESERVATION_NAME_REGIONAL = "test-reservation-regional-" + javaVersion  + "-"
         + UUID.randomUUID().toString().substring(0, 8);
+    GLOBAL_INSTANCE_TEMPLATE_NAME =
+        "test-global-inst-temp-" + javaVersion + "-" + UUID.randomUUID().toString().substring(0, 8);
+    REGIONAL_INSTANCE_TEMPLATE_NAME =
+        "test-regional-inst-temp-" + javaVersion  + "-"
+            + UUID.randomUUID().toString().substring(0, 8);
     GLOBAL_INSTANCE_TEMPLATE_URI = String.format("projects/%s/global/instanceTemplates/%s",
         PROJECT_ID, GLOBAL_INSTANCE_TEMPLATE_NAME);
     REGIONAL_INSTANCE_TEMPLATE_URI =
         String.format("projects/%s/regions/%s/instanceTemplates/%s",
             PROJECT_ID, REGION, REGIONAL_INSTANCE_TEMPLATE_NAME);
+    RESERVATION_SHARED_NAME = "test-reservation-shared-" + javaVersion
+        + "-" + UUID.randomUUID().toString().substring(0, 8);
+    SPECIFIC_SHARED_INSTANCE_NAME =
+        "test-shared-instance-" + javaVersion  + "-"
+            + UUID.randomUUID().toString().substring(0, 8);
 
     // Create instance template with GLOBAL location.
     CreateInstanceTemplate.createInstanceTemplate(PROJECT_ID, GLOBAL_INSTANCE_TEMPLATE_NAME);
@@ -133,8 +152,10 @@ public static void cleanup()
             + REGIONAL_INSTANCE_TEMPLATE_NAME);
 
     // Delete all reservations created for testing.
+    DeleteInstance.deleteInstance(PROJECT_ID, ZONE, SPECIFIC_SHARED_INSTANCE_NAME);
     DeleteReservation.deleteReservation(PROJECT_ID, ZONE, RESERVATION_NAME_GLOBAL);
     DeleteReservation.deleteReservation(PROJECT_ID, ZONE, RESERVATION_NAME_REGIONAL);
+    DeleteReservation.deleteReservation(PROJECT_ID, ZONE, RESERVATION_SHARED_NAME);
 
     // Test that reservations are deleted
     Assertions.assertThrows(
@@ -143,9 +164,13 @@ public static void cleanup()
     Assertions.assertThrows(
         NotFoundException.class,
         () -> GetReservation.getReservation(PROJECT_ID, RESERVATION_NAME_REGIONAL, ZONE));
+    Assertions.assertThrows(
+        NotFoundException.class,
+        () -> GetReservation.getReservation(PROJECT_ID, RESERVATION_SHARED_NAME, ZONE));
 
     // Close the client after all tests
     reservationsClient.close();
+    instancesClient.close();
 
     stdOut.close();
     System.setOut(out);
@@ -177,4 +202,27 @@ public void testCreateReservationWithRegionInstanceTemplate()
     Assert.assertTrue(reservation.getZone().contains(ZONE));
     Assert.assertEquals(RESERVATION_NAME_REGIONAL, reservation.getName());
   }
+
+  @Test
+  public void testConsumeSpecificSharedReservation()
+      throws IOException, ExecutionException, InterruptedException, TimeoutException {
+    ConsumeSpecificSharedReservation.createReservation(PROJECT_ID,
+        RESERVATION_SHARED_NAME, NUMBER_OF_VMS, ZONE,
+        MACHINE_TYPE, MIN_CPU_PLATFORM, true);
+
+    Assertions.assertEquals(RESERVATION_SHARED_NAME,
+        reservationsClient.get(PROJECT_ID, ZONE, RESERVATION_SHARED_NAME).getName());
+
+    ConsumeSpecificSharedReservation.createInstance(
+        PROJECT_ID, ZONE, SPECIFIC_SHARED_INSTANCE_NAME, MACHINE_TYPE,
+        MIN_CPU_PLATFORM, RESERVATION_SHARED_NAME);
+
+    // Verify that the instance was created with the correct reservation and consumeReservationType
+    Instance instance = instancesClient.get(PROJECT_ID, ZONE, SPECIFIC_SHARED_INSTANCE_NAME);
+
+    Assertions.assertTrue(instance.getReservationAffinity()
+        .getValuesList().get(0).contains(RESERVATION_SHARED_NAME));
+    Assertions.assertEquals(SPECIFIC_RESERVATION.toString(),
+        instance.getReservationAffinity().getConsumeReservationType());
+  }
 }