Skip to content

Commit c12beba

Browse files
Fixed test
1 parent 40403be commit c12beba

File tree

1 file changed

+3
-17
lines changed

1 file changed

+3
-17
lines changed

tpu/src/test/java/tpu/CreateQueuedResourceWithNetworkIT.java

Lines changed: 3 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,7 @@
2020

2121
import com.google.api.gax.rpc.NotFoundException;
2222
import com.google.cloud.tpu.v2alpha1.QueuedResource;
23-
import java.io.ByteArrayOutputStream;
2423
import java.io.IOException;
25-
import java.io.PrintStream;
2624
import java.util.UUID;
2725
import java.util.concurrent.TimeUnit;
2826
import org.junit.Test;
@@ -66,28 +64,16 @@ public static void cleanup() {
6664
}
6765

6866
@Test
69-
public void shouldCreateQueuedResourceWithSpecifiedNetwork() throws Exception {
70-
final PrintStream out = System.out;
71-
ByteArrayOutputStream stdOut = new ByteArrayOutputStream();
72-
System.setOut(new PrintStream(stdOut));
67+
public void testCreateQueuedResourceWithSpecifiedNetwork() throws Exception {
7368

7469
QueuedResource queuedResource = CreateQueuedResourceWithNetwork.createQueuedResourceWithNetwork(
75-
PROJECT_ID,
76-
ZONE,
77-
QUEUED_RESOURCE_NAME,
78-
NODE_NAME,
79-
TPU_TYPE,
80-
TPU_SOFTWARE_VERSION,
81-
NETWORK_NAME);
70+
PROJECT_ID, ZONE, QUEUED_RESOURCE_NAME, NODE_NAME,
71+
TPU_TYPE, TPU_SOFTWARE_VERSION, NETWORK_NAME);
8272

83-
assertThat(stdOut.toString()).contains("Queued Resource created: " + QUEUED_RESOURCE_NAME);
8473
assertThat(queuedResource.getTpu().getNodeSpec(0).getNode().getName()).isEqualTo(NODE_NAME);
8574
assertThat(queuedResource.getTpu().getNodeSpec(0).getNode().getNetworkConfig().getNetwork()
8675
.contains(NETWORK_NAME));
8776
assertThat(queuedResource.getTpu().getNodeSpec(0).getNode().getNetworkConfig().getSubnetwork()
8877
.contains(NETWORK_NAME));
89-
90-
stdOut.close();
91-
System.setOut(out);
9278
}
9379
}

0 commit comments

Comments
 (0)