Writing data to System.in

2019-02-06 12:16发布

In our application, we expect user input within a Thread as follows :

BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

I want to pass that part in my unit test so that I can resume the thread to execute the rest of the code. How can I write something into System.in from junit?

4条回答
淡お忘
2楼-- · 2019-02-06 12:51

What you want to do is use the method setIn() from System. This will let you pass data into System.in from junit.

查看更多
我欲成王,谁敢阻挡
3楼-- · 2019-02-06 12:52

Replace it for the duration of your test:

String data = "the text you want to send";
InputStream testInput = new ByteArrayInputStream( data.getBytes("UTF-8") );
InputStream old = System.in;
try {
    System.setIn( testInput );

    ...
} finally {
    System.setIn( old );
}
查看更多
疯言疯语
4楼-- · 2019-02-06 12:57

My solution currently (in 2018) is:

 final byte[] passCode = "12343434".getBytes();
 final ByteArrayInputStream inStream = new ByteArrayInputStream(passCode);
        System.setIn(inStream);
查看更多
神经病院院长
5楼-- · 2019-02-06 13:03

Instead of the suggestions above (edit: I noticed that Bart left this idea in a comment as well), I would suggest making your class more unit testable by making the class accept the input source as a constructor parameter or similar (inject the dependency). A class shouldn't be so coupled to System.in anyway.

If your class is constructed from a Reader, you can just do this:

class SomeUnit {
   private final BufferedReader br;
   public SomeUnit(Reader r) {
       br = new BufferedReader(r);
   }
   //...
}

//in your real code:
SomeUnit unit = new SomeUnit(new InputStreamReader(System.in));

//in your JUnit test (e.g.):
SomeUnit unit = new SomeUnit(new StringReader("here's the input\nline 2"));
查看更多
登录 后发表回答