使用扫描程序对用户输入进行junit测试

Question

我必须测试一个类中的一个方法，它接受一个使用扫描器类的输入。

package com.math.calculator;

import java.util.Scanner;

public class InputOutput {

    public String getInput() {
        Scanner sc = new Scanner(System.in);
        return sc.nextLine();
    }
}

我想使用JUnit测试它，但不确定如何进行。

我尝试使用下面的代码，但它不起作用。

package com.math.calculator;

import org.junit.Test;

import static org.junit.Assert.assertEquals;

public class InputOutputTest {

    @Test
    public void shouldTakeUserInput() {
        InputOutput inputOutput= new InputOutput();

        assertEquals("add 5", inputOutput.getInput());
    }
}

我也想用Mockito（使用mock...when...thenReturn)，但不确定如何做。

Answer 1

可以使用system.setin()方法更改system.in流。

试试这个,

@Test
public void shouldTakeUserInput() {
    InputOutput inputOutput= new InputOutput();

    String input = "add 5";
    InputStream in = new ByteArrayInputStream(input.getBytes());
    System.setIn(in);

    assertEquals("add 5", inputOutput.getInput());
}

您刚刚修改了system.in字段。system.in基本上是一个InputStream，它从控制台读取（因此您在控制台中输入）。但您只是修改了它，让系统读取提供的inputstream。因此它不再从控制台读取,而是从提供的inputstream读取。