3030import java .io .IOException ;
3131import java .util .UUID ;
3232import java .util .concurrent .ExecutionException ;
33+ import java .util .concurrent .TimeUnit ;
3334import java .util .concurrent .TimeoutException ;
3435import org .junit .jupiter .api .AfterAll ;
3536import org .junit .jupiter .api .Assertions ;
3637import org .junit .jupiter .api .BeforeAll ;
3738import org .junit .jupiter .api .Test ;
39+ import org .junit .jupiter .api .Timeout ;
40+ import org .junit .runner .RunWith ;
41+ import org .junit .runners .JUnit4 ;
3842
43+ @ RunWith (JUnit4 .class )
44+ @ Timeout (value = 3 , unit = TimeUnit .MINUTES )
3945public class CreateReservationFromVmIT {
4046
4147 private static final String PROJECT_ID = System .getenv ("GOOGLE_CLOUD_PROJECT" );
4248 private static final String ZONE = "us-east4-c" ;
4349 private static ReservationsClient reservationsClient ;
4450 private static InstancesClient instancesClient ;
45- private static String RESERVATION_NAME ;
46- private static String INSTANCE_FOR_RESERVATION ;
51+ private static String reservationName ;
52+ private static String instanceForReservation ;
4753 static String javaVersion = System .getProperty ("java.version" ).substring (0 , 2 );
4854
4955 // Check if the required environment variables are set.
@@ -60,29 +66,23 @@ public static void setUp()
6066 reservationsClient = ReservationsClient .create ();
6167 instancesClient = InstancesClient .create ();
6268
63- RESERVATION_NAME = "test-reservation-from-vm-" + javaVersion + "-"
69+ reservationName = "test-reservation-from-vm-" + javaVersion + "-"
6470 + UUID .randomUUID ().toString ().substring (0 , 8 );
65- INSTANCE_FOR_RESERVATION = "test-instance-for-reserv-" + javaVersion + "-"
71+ instanceForReservation = "test-instance-for-reserv-" + javaVersion + "-"
6672 + UUID .randomUUID ().toString ().substring (0 , 8 );
6773
6874 // Cleanup existing stale resources.
6975 Util .cleanUpExistingInstances ("test-instance-for-reserv-" + javaVersion , PROJECT_ID , ZONE );
7076 Util .cleanUpExistingReservations ("test-reservation-from-vm-" + javaVersion , PROJECT_ID , ZONE );
7177
72- CreateInstance .createInstance (PROJECT_ID , ZONE , INSTANCE_FOR_RESERVATION );
78+ CreateInstance .createInstance (PROJECT_ID , ZONE , instanceForReservation );
7379 }
7480
7581 @ AfterAll
7682 public static void cleanup ()
7783 throws IOException , ExecutionException , InterruptedException , TimeoutException {
7884 // Delete resources created for testing.
79- DeleteInstance .deleteInstance (PROJECT_ID , ZONE , INSTANCE_FOR_RESERVATION );
80- DeleteReservation .deleteReservation (PROJECT_ID , ZONE , RESERVATION_NAME );
81-
82- // Test that reservation is deleted
83- Assertions .assertThrows (
84- NotFoundException .class ,
85- () -> GetReservation .getReservation (PROJECT_ID , RESERVATION_NAME , ZONE ));
85+ DeleteInstance .deleteInstance (PROJECT_ID , ZONE , instanceForReservation );
8686
8787 reservationsClient .close ();
8888 instancesClient .close ();
@@ -92,17 +92,24 @@ public static void cleanup()
9292 public void testCreateComputeReservationFromVm ()
9393 throws IOException , ExecutionException , InterruptedException , TimeoutException {
9494 CreateReservationFromVm .createComputeReservationFromVm (
95- PROJECT_ID , ZONE , RESERVATION_NAME , INSTANCE_FOR_RESERVATION );
95+ PROJECT_ID , ZONE , reservationName , instanceForReservation );
9696
97- Instance instance = instancesClient .get (PROJECT_ID , ZONE , INSTANCE_FOR_RESERVATION );
97+ Instance instance = instancesClient .get (PROJECT_ID , ZONE , instanceForReservation );
9898 Reservation reservation =
99- reservationsClient .get (PROJECT_ID , ZONE , RESERVATION_NAME );
99+ reservationsClient .get (PROJECT_ID , ZONE , reservationName );
100100
101101 Assertions .assertNotNull (reservation );
102- assertThat (reservation .getName ()).isEqualTo (RESERVATION_NAME );
102+ assertThat (reservation .getName ()).isEqualTo (reservationName );
103103 Assertions .assertEquals (instance .getMinCpuPlatform (),
104104 reservation .getSpecificReservation ().getInstanceProperties ().getMinCpuPlatform ());
105105 Assertions .assertEquals (instance .getGuestAcceleratorsList (),
106106 reservation .getSpecificReservation ().getInstanceProperties ().getGuestAcceleratorsList ());
107+
108+ DeleteReservation .deleteReservation (PROJECT_ID , ZONE , reservationName );
109+
110+ // Test that reservation is deleted
111+ Assertions .assertThrows (
112+ NotFoundException .class ,
113+ () -> GetReservation .getReservation (PROJECT_ID , reservationName , ZONE ));
107114 }
108115}
0 commit comments