解决在IDEA中无法使用Scanner输入的问题

今天写了一段代码用来测试输入，发现不管怎么样搜无法输入数据，代码如下：

@Test
public void func01(){
    Scanner scanner = new Scanner(System.in);
    while (scanner.hasNextLine()){
        String msg = scanner.nextLine();
        System.out.println(msg);
    }
}

感觉写得没有错，以前也能运行，但是就是不能输入，后来发现这段代码是在单元测试里面运行的，只要把这段代码放到main函数里面就能运行了