总结在Junit中测试控制台输出的方法

在Junit中无法调用一个断言方法直接判断程序的控制台输出与预期是否相等，因此需要以下方法劫持控制台输出，并用assertEquals方法测试：

    private PrintStream console = null;
    private ByteArrayOutputStream bytes = null;
    FriendshipGraph graph = new FriendshipGraph();

    @Before/*获取控制台输出*/
    public void init() {
        bytes = new ByteArrayOutputStream();
        console = System.out;
        System.setOut(new PrintStream(bytes));
    }

    @After/*恢复控制台输出*/
    public void tearDown() {
        System.setOut(console);
    }

进行测试的例子如下：

assertEquals("The person is already added.",bytes.toString().trim());

如果要清空已经获取的输出，则使用：

bytes.reset();