Fixed test

Author: TetyanaYahodska

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

@@ -76,7 +76,6 @@ public static QueuedResource createQueuedResource(
                 .setTotalTimeout(Duration.ofHours(24L))
                 .build());
     String parent = String.format("projects/%s/locations/%s", projectId, zone);
-    // Read the startup script content from the file
     String startupScriptContent = "#!/bin/bash\necho \"Hello from the startup script!\"";
     // Add startup script to metadata
     Map<String, String> metadata = new HashMap<>();

tpu/src/test/java/tpu/QueuedResourceIT.java

@@ -87,8 +87,10 @@ public void testCreateQueuedResourceWithStartupScript() throws Exception {
         PROJECT_ID, ZONE, QUEUED_RESOURCE_FOR_SCRIPT, NODE_FOR_SCRIPT,
         TPU_TYPE, TPU_SOFTWARE_VERSION);
 
-    assertThat(queuedResource.getTpu().getNodeSpec(0).getNode().containsLabels("startup-script"));
-    assertThat(queuedResource.getTpu().getNodeSpec(0).getNode().getLabelsMap()
-        .containsValue("Hello from the startup script!"));
+    System.out.println(queuedResource.getTpu().getNodeSpec(0).getNode().getMetadataMap());
+    assertThat(queuedResource.getTpu().getNodeSpec(0).getNode().containsMetadata("startup-script"));
+    assertThat(queuedResource.getTpu().getNodeSpec(0).getNode().getMetadataMap()
+        .containsValue("#!/bin/bash\n"
+        + "echo \"Hello from the startup script!\""));
   }
 }