首页 > 解决方案 > 来自 System.in 的 Java UnitTest 多命令行输入

问题描述

我正在尝试测试一个永远循环直到QUIT输入的函数,但我无法弄清楚如何模拟来自系统的多行用户输入。

这是我目前拥有的

String validStartFlow = "START-FLOW\r\nQUIT\r\n";
ByteArrayInputStream in = new ByteArrayInputStream(validStartFlow.getBytes());
System.setIn(in);
CommandLineListener cmdLineListener = new CommandLineListener(eventBus,logger);
cmdLineListener.startCommandLineListener(in);

永远循环的方法是

while (!userCmd.equals("QUIT")) {
    userCmd = "";
    Scanner scanner = new Scanner(in);
    // BufferedReader reader = new BufferedReader(in);

    while (userCmd == null || userCmd.equals("")) {
        userCmd = scanner.nextLine();
    }

    ...
}

START-FLOW 完美读取,但之后当它到达scanner.nextLine()时崩溃并出现以下错误

找不到行 java.util.NoSuchElementException: 找不到行

如何让它QUITvalidStartFlow字符串中读取?

标签: javaunit-testingjunit

解决方案


Scanner您在每次迭代中创建一个新的并且每次nextLine()只访问一次。

尝试从循环中提取它:

Scanner scanner = new Scanner(in);
scanner.useDelimiter("\r\n");
String userCmd = "";

while (!userCmd.equals("QUIT")) {
        userCmd = scanner.nextLine();
...

推荐阅读