Skip to content

Commit b34eae1

Browse files
Fixed test, deleted cleanup method
1 parent 8fb7f26 commit b34eae1

File tree

3 files changed

+6
-102
lines changed

3 files changed

+6
-102
lines changed

tpu/src/main/java/tpu/CreateQueuedResourceWithStartupScript.java

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -80,17 +80,16 @@ public static QueuedResource createQueuedResource(
8080
.setMaxRetryDelay(Duration.ofMillis(45000L))
8181
.setTotalTimeout(Duration.ofHours(24L))
8282
.build());
83-
// Initialize client that will be used to send requests. This client only needs to be created
84-
// once, and can be reused for multiple requests.
85-
try (TpuClient tpuClient = TpuClient.create(clientSettings.build())) {
8683
String parent = String.format("projects/%s/locations/%s", projectId, zone);
8784
// Read the startup script content from the file
8885
String startupScriptContent = new String(Files.readAllBytes(Paths.get(startupScriptPath)));
89-
9086
// Add startup script to metadata
9187
Map<String, String> metadata = new HashMap<>();
9288
metadata.put("startup-script", startupScriptContent);
9389

90+
// Initialize client that will be used to send requests. This client only needs to be created
91+
// once, and can be reused for multiple requests.
92+
try (TpuClient tpuClient = TpuClient.create(clientSettings.build())) {
9493
Node node =
9594
Node.newBuilder()
9695
.setName(nodeName)

tpu/src/test/java/tpu/CreateQueuedResourceWithStartupScriptIT.java

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121

2222
import com.google.api.gax.rpc.NotFoundException;
2323
import com.google.cloud.tpu.v2alpha1.QueuedResource;
24-
import java.io.IOException;
2524
import java.util.UUID;
2625
import java.util.concurrent.TimeUnit;
2726
import org.junit.jupiter.api.AfterAll;
@@ -38,13 +37,10 @@ public class CreateQueuedResourceWithStartupScriptIT {
3837

3938
private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
4039
private static final String ZONE = "europe-west4-a";
41-
static String javaVersion = System.getProperty("java.version").substring(0, 2);
42-
private static final String NODE_NAME = "test-tpu-queued-resource-script-" + javaVersion + "-"
43-
+ UUID.randomUUID().toString().substring(0, 8);
40+
private static final String NODE_NAME = "test-tpu-queued-resource-script-" + UUID.randomUUID();
4441
private static final String TPU_TYPE = "v2-8";
4542
private static final String TPU_SOFTWARE_VERSION = "tpu-vm-tf-2.14.1";
46-
private static final String QUEUED_RESOURCE_NAME = "queued-resource-script-" + javaVersion + "-"
47-
+ UUID.randomUUID().toString().substring(0, 8);
43+
private static final String QUEUED_RESOURCE_NAME = "queued-resource-script-" + UUID.randomUUID();
4844
private static final String STARTUP_SCRIPT_PATH = "src/test/java/tpu/startup-script.sh";
4945

5046
public static void requireEnvVar(String envVarName) {
@@ -53,19 +49,15 @@ public static void requireEnvVar(String envVarName) {
5349
}
5450

5551
@BeforeAll
56-
public static void setUp() throws IOException {
52+
public static void setUp() {
5753
requireEnvVar("GOOGLE_APPLICATION_CREDENTIALS");
5854
requireEnvVar("GOOGLE_CLOUD_PROJECT");
59-
60-
// Cleanup existing stale resources.
61-
Util.cleanUpExistingQueuedResources("queued-resource-script-", PROJECT_ID, ZONE);
6255
}
6356

6457
@AfterAll
6558
public static void cleanup() {
6659
DeleteForceQueuedResource.deleteForceQueuedResource(PROJECT_ID, ZONE, QUEUED_RESOURCE_NAME);
6760

68-
// Test that resource is deleted
6961
Assertions.assertThrows(
7062
NotFoundException.class,
7163
() -> GetQueuedResource.getQueuedResource(PROJECT_ID, ZONE, QUEUED_RESOURCE_NAME));

tpu/src/test/java/tpu/Util.java

Lines changed: 0 additions & 87 deletions
This file was deleted.

0 commit comments

Comments
 (0)