java - Junit 测试控制台中的输出

Question

我仍然找不到以下方法的测试方法失败的原因。

public static void print() {
    System.out.println("hello");
}

我的Junit测试方法如下

@Test
public void testPrint(){
   ByteArrayOutputStream baos = new ByteArrayOutputStream();
   PrintStream ps = new PrintStream(baos);

   PrintStream old = System.out;

   System.setOut(ps);

   //execute the above method
   obj.print();

   System.out.flush();
   System.setOut(old);

   //test method
   String expected = "hello" + "\n";
   assertEquals(expected, baos.toString());
}

AssertionFailed 预期错误:<hello but was <hello>

所以问题应该出在换行符上。有谁知道如何解决吗？

Answer 1

像这样编写测试是使用模拟的好地方:

// code with Mockito
@Test
public void testPrint(){
  PrintStream mockedPrintStream = Mockito.mock(PrintStream.class);
  PrintStream old = System.out;
  System.setOut(mockedPrintStream);

  obj.print();

  Mockito.verify(mockedPrintStream).println("hello");

  System.setOut(old); // add this line when edit
}

现在您检查是否使用参数调用了方法。

有关模拟的更多信息，请参阅 Mockito manual

<小时/>

编辑 - 回答评论中的问题

PrintStream old = System.out; 您需要此行，因为在 System.out 中是外部代码，测试后您应该清理您所使用的所有对象的状态测试过程中污染。

在大多数情况下，这种清理发生在@TearDown方法中，或者只是因为当@Test方法返回时GC删除了对象。

在这种情况下，您需要在测试方法中手动执行此操作，因为没有其他地方可以将 System.out 属性的状态恢复到测试之前的状态。