System.in / out 테스트

Java의 표준 입출력

• System.in / out

• System.err / setErr()

• System.setIn() / Out()

System.In

---

예제

public String inputCarNames() {
    String input = Console.readLine();
    checkEmptyAndBlank(input);
    return input;
}

private void checkEmptyAndBlank(String input) {
    inputValidator.validateInputEmpty(input);
    inputValidator.validateInputBlank(input);
}

다음과 같은 메소드를 테스트 하려고한다. Console.readLine()은 Scanner의 readLine()과 동일하다.

❓ 테스트는 input(문자열을 입력받는 상황)을 어떻게 구현할까?

ByteArrayInputStream과 ByteArrayOutputStream은 메모리, 즉 바이트 배열에 데이터를 입출력하는데 사용되는 스트림이다. 주로 다른곳에 입출력하기 전에 데이터를 임시로 바이트 배열에 담아서 변환등의 작업을 하는데 사용된다. 인스턴스를 생성하기 위해서는 byte[] 배열을 인자로 넘겨줘야 한다.

@Test
void testInput() {
    // String empty = "";
    String empty = "\n";
    String blank = " ";
    
    // InputStream empty = new ByteArrayInputStream(empty.getBytes());
    ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(empty.getBytes());
    System.setIn(byteArrayInputStream);
    
    inputView.inputCarNames();
}

• getBytes()는 byte arrary로 변환해서 반환해준다.

• ByteArrayInputStream은 InputStream을 상속받기 때문에 타입은 InputStream을 사용해도 된다.

System.setIn(InputStream in) :

• 표준 입력을 키보드, 파일, 다른 소스에서 입력 받고록 설정한다.

• 표준 입력 스트림(System.in)을 재지정하는데 사용된다. 프로그램 실행 중 입력 스트림을 원하는 입력 데이터로 변경할 수 있다.

• System.setIn()을 사용해서 System.in을 empty, blank의 byte arrary로 설정했다. 이것은 readLine()과 동일한 기능을 한다.

하지만 ""를 입력으로 주어지게 되면 java.util.NoSuchElementException: No line found Exception을 발생하게 된다. 입력 받은것이 아무것도 없다고 인식하는 것이다. 그래서 빈 문자열을 테스트 하기 위해서는 empty = "\n"을 설정해준다.

프리코스의 Console.readLine()

우테코 프리코스에서는 Console이라는 모듈을 제공해서 이것으로 입력을 받는다.

콘솔의 내부 구현은 다음과 같다.

public class Console {
    private static Scanner scanner;

    private Console() {
    }

    public static String readLine() {
        return getInstance().nextLine();
    }

    public static void close() {
        if (scanner != null) {
            scanner.close();
            scanner = null;
        }
    }

    private static Scanner getInstance() {
        if (scanner == null) {
            scanner = new Scanner(System.in);
        }
        return scanner;
    }
}

@Test
void test1() {
    System.setIn(new ByteArrayInputStream(input.getBytes());
    testMethod();    // System.in에 값을 할당한다.
}

@Test
void test2() {
    System.setIn(new ByteArrayInputStream(input.getBytes());
    testMethod();    // System.in에 값을 할당과정
}

private void testMethod() {
    Console.readLine();    // input값 스캔
}

입력받는 테스트를 한번에 여러개 실행하면 NoSuchElementException 을 뱉어버리는데, 그 이유는 getInstance().nextLine()에 있다.

콘솔의 Scanner는 stacic으로 선언되어 있고, 첫 번째 테스트에서 System.in에 "blank" 값을 넣는다.

• 테스트에서는 Console.readLine()으로 입력을 넣는 코드가 존재한다.

그러면 Scanner가 생성되면서 "blank"를 스캔한다. 그 다음 두 번쨰 테스트에서 또 Console.readLine()을 호출한다. 여기서 문제가 발생해버린다. getInstance()가 호출될 때 Scanner는 이미 만들어진 상태이기 때문에 두 번째에서 Console.read()을 호출하면 이미 생성된 Scanner를 반환 해버린다는 것이다.

결국, 첫 번째 테스트에서 Scanner가 생성되고 값을 할당한다. 두 번째 테스트에서 Console.reaLine()을 타게 되면 이미 System.in에 첫 번째 input값이 할당 되어있기 때문에, 두 번쨰 테스트에서는 예외를 발생시킨다.

어떻게 해결하나?

@AfterEach
void clean() {
    Console.close();
}

@AfterEach 어노테이션을 적용시켜서 메소드가 실행되고 난 후, static 영역에 존재하는 Scanner를 null로 바꿔야한다. 그러면 다음 테스트도 잘 수행된다.

System.out

---

System.in은 InputStream을 사용해서 System.in을 스캔하며 테스트 했다.

System.out은 출력될 문구를 PrintStream에 담는다.

private final ByteArrayOutputStream output = new ByteArrayOutputStream();
private OutputView outputView;

@BeforeEach
void setup() {
    System.setOut(new PrintStream(output));
    outputView = new OutputView();
}

@AfterEach
void clean() {
    System.setOut(System.out);
    output.reset();
}

outputView.method()에는 System.out.print()메소드가 담겨있다. 메소드가 호출되면 OutputStream output에 byte로 저장된다. 원하는 출력 문구와 output.toString()으로 출력 문구를 비교한다.

두 개 이상의 메소드를 연속으로 테스트 하기 위해서는 @AfterEach를 설정하자.

@ParameterizedTest
@CsvSource(value = {
    "4,3,0/pobi : -ruby : dobbi :",
    "4,4,0/pobi : -ruby : -dobbi :",
    "9,8,7/pobi : -ruby : -dobbi : -"
},delimiter  = '/')
void testPrintProcessMessage(String numbers, String expectedMessage) {
    outputView.printProcess(cars);
    String outputString = output.toString().replace("\n", "").strip();
    assertThat(outputString).isEqualTo(expectedMessage.strip());
}

• replaceAll("\n", "")을 작성하면 output.toString() 결과에서 개행이 사라진다.

• strip()은 문자열의 시작과 끝에서 공백을 제거하는 메서드이다.

• @CsvSourece에 옵션을 추가하면, 좌우 공백을 자동으로 삭제하지 않는다. ignoreLeadingAndTrailingWhitespace = false

output.toString()으로는 원하는 결과가 비교되지 않아서 두 메소드를 활용해서 비교한다.