Skip to content

Commit e181745

Browse files
Fixed test and naming
1 parent c2ebf99 commit e181745

File tree

2 files changed

+27
-20
lines changed

2 files changed

+27
-20
lines changed

compute/cloud-client/src/main/java/compute/reservation/CreateReservationFromVm.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -66,12 +66,12 @@ public static void createComputeReservationFromVm(
6666
// Extract properties from the existing VM
6767
List<AcceleratorConfig> guestAccelerators = new ArrayList<>();
6868
if (!existingVm.getGuestAcceleratorsList().isEmpty()) {
69-
for (AcceleratorConfig a : existingVm.getGuestAcceleratorsList()) {
69+
for (AcceleratorConfig accelatorConfig : existingVm.getGuestAcceleratorsList()) {
7070
guestAccelerators.add(
7171
AcceleratorConfig.newBuilder()
72-
.setAcceleratorCount(a.getAcceleratorCount())
73-
.setAcceleratorType(a.getAcceleratorType()
74-
.substring(a.getAcceleratorType().lastIndexOf('/') + 1))
72+
.setAcceleratorCount(accelatorConfig.getAcceleratorCount())
73+
.setAcceleratorType(accelatorConfig.getAcceleratorType()
74+
.substring(accelatorConfig.getAcceleratorType().lastIndexOf('/') + 1))
7575
.build());
7676
}
7777
}

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

Lines changed: 23 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -30,20 +30,26 @@
3030
import java.io.IOException;
3131
import java.util.UUID;
3232
import java.util.concurrent.ExecutionException;
33+
import java.util.concurrent.TimeUnit;
3334
import java.util.concurrent.TimeoutException;
3435
import org.junit.jupiter.api.AfterAll;
3536
import org.junit.jupiter.api.Assertions;
3637
import org.junit.jupiter.api.BeforeAll;
3738
import 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)
3945
public 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

Comments
 (0)