Java 命令行程序和嵌套 readLine() 的 TDD

Question

基本上，我正在编写一个 Java 命令行应用程序，它通过 readLine 接受来自用户的参数。例如，如果用户按“1”，则会询问用户要借阅哪本书，用户可以按“1”或“2”按书的编号...因此，应用程序接受用户参数两次。我也尝试使用 TDD 来测试应用程序。

问题就在这里。 如果我做这样的事情来模拟用户输入

System.setIn(new ByteArrayInputStream(PRESS_TWO.getBytes())); // set the first option

Program.main(new String[]{}); // run the program

System.setIn(new ByteArrayInputStream(PRESS_ONE.getBytes())); // set the second option

等待用户选择选项的第一步是可以的，但是它将直接执行第二步而不等待第二个输入。我如何在JUnit中模拟这个？

谢谢

这就是我读取输入的方式

 try {

            i1 = Integer.parseInt(reader.readLine());
        }
        catch (Exception e) {
            System.out.println("Enter a valid integer!!");
        }

        if (i1 == 1) {
            System.out.println(" 1. Book1 ");
            System.out.println(" 2. eBook2 ");
            System.out.println(" 3. Book3 ");
        }
        else if (i1 == 2) {
            System.out.println(" Which one do you want?: ");
            int i2 = 0;
            try {
                i2 = Integer.parseInt(reader.readLine());
            }
            catch (Exception e) {
                // Do you know what numbers are!!!
                System.out.println("Enter a valid integer!!");

            }
       }

Answer 1

这取决于您的代码如何读取输入，但我假设一旦到达第一个 ByteArrayInputStream 的末尾，您的代码就会看到文件结尾并终止。尝试将两个流合并为一个（用换行符分隔）。