Update SnippetsIT.java

GoogleCloudPlatform · Dec 20, 2022 · 34fd127 · 34fd127
1 parent a20dba4
commit 34fd127
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/webrisk/src/test/java/webrisk/SnippetsIT.java b/webrisk/src/test/java/webrisk/SnippetsIT.java
@@ -45,8 +45,7 @@
 public class SnippetsIT {
 
   private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
-  private static ByteArrayOutputStream originalOut = new ByteArrayOutputStream();
-  private static PrintStream printStream = new PrintStream(originalOut);
+  private final PrintStream originalOut = System.out;
 
   private ByteArrayOutputStream stdOut;
 
@@ -59,7 +58,6 @@ public static void requireEnvVar(String envVarName) {
 
   @BeforeClass
   public static void setUp() throws IOException {
-    printStream.println(System.out);
     requireEnvVar("GOOGLE_APPLICATION_CREDENTIALS");
     requireEnvVar("GOOGLE_CLOUD_PROJECT");
   }
@@ -72,8 +70,8 @@ public void beforeEach() {
 
   @After
   public void afterEach() {
-    stdOut = originalOut;
-    System.setOut(new PrintStream(originalOut));
+    stdOut.reset();
+    System.setOut(originalOut);
   }
 
   @Test