JUnitで標準入力をつっこんで流す

mainメソッドに直接書いちゃった時のテスト用

引用元

っていうかこちらのサイトにわかりやすく書いてあるので皆さんこっちを参照してください。以下は私のメモ書きです。

テストしたいクラス

Main.java

import java.io.BufferedReader;
import java.io.InputStreamReader;

    public class Main {
        public static void main(String[] args) throws Exception {
            BufferedReader br = 
                    new BufferedReader(new InputStreamReader(System.in));
            String line = br.readLine();
            System.out.println(line);
            line = br.readLine();
            System.out.println(line);
        }
    }

JUnitクラス

uniTest.java

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class uniTest{

    private StandardInputSnatcher in = new StandardInputSnatcher();

    @Before
    public void before() {
        System.setIn(in);
    }

    @After
    public void after() {
        System.setIn(null);
    }

    @Test
    public void test() {
        in.inputln("Hello World");
        in.inputln("Hello Orange Sunshine");

        try {
            Main.main(null);
        } catch (Exception e) {
            // TODO: handle exception
        }
    }

}

InputStreamをオーバーライドしちゃうクラス

uniTest.java

import java.io.IOException;
import java.io.InputStream;

public class StandardInputSnatcher extends InputStream {

    private StringBuilder buffer = new StringBuilder();
    private static String crlf = System.getProperty("line.separator");

    public void inputln(String str) {
        buffer.append(str).append(crlf);
    }

    @Override
    public int read() throws IOException {
        if (buffer.length() == 0) {
            return -1;
        }
        int result = buffer.charAt(0);
        buffer.deleteCharAt(0);
        return result;
    }
}