前言：junit測試預設是不支援控制檯輸入的，如果有需要，可以通過變相的方式實現控制檯輸入，詳情看下文

今天我寫了一個鍵盤類的輸入功能，這個類長這樣，它繼承自一個鍵盤介面。

public interface KeyBoard {
    String input();
}

public class KeyBoardImpl implements KeyBoard {
    @Override
    public String input() {
        Scanner sc = new Scanner(System.in);
        String input = sc.nextLine();
        return input;
    }
}
 

為了測試這個類的輸入功能，我寫了這樣一個測試用例，用的是junit

public class KeyBoardImplTest {
    KeyBoard keyBoard = new KeyBoardImpl();

    @Test
    public void test() {
        String input = keyBoard.input();
        System.out.println(input);
    }
}

但是我在執行該測試用例後，發現無法在控制檯輸入任何內容，程式停在那裡不動了。

我嘗試了一下在main函式裡面寫個用例，發現能夠輸入。

public class KeyBoardImpl implements KeyBoard {
    @Override
    public String input() {
        Scanner sc = new Scanner(System.in);
        String input = sc.nextLine();
        return input;
    }

    public static void main(String[] args) {
        KeyBoard keyBoard = new KeyBoardImpl();
        String input = keyBoard.input();
        System.out.printf(input);
    }
}
 

image.png

這是為啥呢，經過網上搜索，我找到了解決方案，並且大致瞭解了原因: junit是用例自動化單元測試的，那麼控制檯輸入這個人工操作就應該不會支援，解決方案如下

public class KeyBoardImplTest {
    KeyBoard keyBoard = new KeyBoardImpl();

    @Test
    public void testInput() throws Exception {
        String data = "Hello, World!\r\n";
        String input;
        InputStream stdin = System.in;
        try {
            System.setIn(new ByteArrayInputStream(data.getBytes()));
            input = keyBoard.input();
        } finally {
            System.setIn(stdin);
        }
        System.out.println("input is----" + input);
    }
}

執行該測試用例，得到了希望的結果：