Java+JUnit+main関数での標準入出力

目的

main関数で標準入出力処理を行う場合の、JUnitテストケースの書き方。

必要ライブラリ

• JUnit
• commons.io
  • ストリームから文字列に変換するのに使用。独自実装すれば不要。

補足

• それぞれ以下の値を設定します。
  • 「in.txt」に標準出力からの入力値。
  • 「expected.txt」標準出力に出力される予測値
• 「in.txt」「expected.txt」は、それぞれテストクラスと同じフォルダにおいてください。

コード

package template;

import static org.junit.Assert.assertEquals;

import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.io.PrintStream;

import org.apache.commons.io.IOUtils;
import org.junit.Test;

/**
 * @author umed0025
 *
 */
public class MainTest {

    /**
     * {@link template.Main#main(java.lang.String[])} のためのテスト・メソッド。
     */
    @Test
    public void testMain() throws Exception {
        InputStream defaultSysin = System.in;
        PrintStream defaultSysout = System.out;

        // put "in.txt" in the same path as this class file.
        try (InputStream sysin = this.getClass().getResourceAsStream("in.txt")) {
            try (ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream()) {
                try (PrintStream sysout = new PrintStream(byteArrayOutputStream)) {
                    System.setIn(sysin);
                    System.setOut(sysout);
                    // call target method.
                    String[] param = new String[] {};
                    Main.main(param);
                    String actual = new String(byteArrayOutputStream.toByteArray());
                    // put "expected.txt" in the same path as this class file.
                    try (InputStream expected = this.getClass().getResourceAsStream("expected.txt")) {
                        // assert stdout.
                        assertEquals(IOUtils.toString(expected), actual);
                    }

                }
            }
        }

        System.setIn(defaultSysin);
        System.setOut(defaultSysout);
    }
}