@MethodSource() 👊

@ParameterizedTest - @MethodSource

@ValueSource(), @CsvSource() 를 통해 기본 ints, strings 등 다양한 자바의 타입을 파라미터를 통해 넘겨주고, 여러개의 인자를 csv형식으로 넘겨줄 수 있었다!

그런데 좀 더 “복잡한” 녀석들을 넘겨주고 싶다면? 🤨

예를 들어 List 객체, 또는 사용자 정의 객체 같은 녀석들 말이다!

*@MethodSouce : 파라미터를 정의하고, 반환해주는 method를 정의하자. 그리고 Stream을 반환받자! (static)*

사용 예시와 함께 보면 더 빠르게 이해할 수 있다.

@ParameterizedTest
@MethodSource("generateTestLottoInput")
void 로또번호를_받는다(List<Integer> input) {
     System.out.println(input);
}
	
static Stream<List<Integer>> **generateTestLottoInput**() {
        return Stream.of(
                List.of(8, 21, 23, 41, 42, 43),
                List.of(3, 5, 11, 16, 32, 38),
                List.of(7, 11, 16, 35, 36, 44)
        );
    }

@MethodSource("generateTestLottoInput")는 generateTestLottoInput() 를 호출해 인자를 받아올 것임을 명시하고 있다.

generateTestLottoInput() 는 List<Integer>로 구성된 Stream을 반환한다!

그리고 Stream, List와 같은 객체는 of() (static)함수를 통해 생성 가능하다.

잠깐! 그럼 복잡한 객체를 여러개 받고싶다면요? CsvSource+Method…?!

당연히 그럴 필요 없이, 여러개의 인자를 받고 싶을때는 Arguments객체를 사용한다.

Arguments를 사용해 여러개의 인자를 한번에 묶고, 이를 Stream으로 구성해 반환하자!

@ParameterizedTest
@MethodSource("generateTestLottoInput")
void 로또와_구입금액_인자로받기(List<Integer> input, int money) {
    System.out.println(input);
    System.out.println(money);
}

static Stream<Arguments> **generateTestLottoInput**() {
    return Stream.of(
        Arguments.arguments(List.of(7, 11, 16, 35, 36, 44), 8000),
        Arguments.arguments(List.of(3, 5, 11, 16, 32, 38), 1000),
    );
}

NsTest 👊

assertRandomNumberInRangeTest

    @Test
    void 기능_테스트() {
        assertRandomUniqueNumbersInRangeTest(
                () -> {
                    run("8000", "1,2,3,4,5,6", "7");
                    assertThat(output()).contains(
                            "8개를 구매했습니다.",
                            "[8, 21, 23, 41, 42, 43]",
                            "[3, 5, 11, 16, 32, 38]",
                            "[7, 11, 16, 35, 36, 44]",
                            "[1, 8, 11, 31, 41, 42]",
                            "[13, 14, 16, 38, 42, 45]",
                            "[7, 11, 30, 40, 42, 43]",
                            "[2, 13, 22, 32, 38, 45]",
                            "[1, 3, 5, 14, 22, 45]",
                            "3개 일치 (5,000원) - 1개",
                            "4개 일치 (50,000원) - 0개",
                            "5개 일치 (1,500,000원) - 0개",
                            "5개 일치, 보너스 볼 일치 (30,000,000원) - 0개",
                            "6개 일치 (2,000,000,000원) - 0개",
                            "총 수익률은 62.5%입니다."
                    );
                },
                List.of(8, 21, 23, 41, 42, 43),
                List.of(3, 5, 11, 16, 32, 38),
                List.of(7, 11, 16, 35, 36, 44),
                List.of(1, 8, 11, 31, 41, 42),
                List.of(13, 14, 16, 38, 42, 45),
                List.of(7, 11, 30, 40, 42, 43),
                List.of(2, 13, 22, 32, 38, 45),
                List.of(1, 3, 5, 14, 22, 45)
        );
    }

위 코드는 로또 게임의 시작 부터 종료를 출력하는 코드이다.

1. "8000"을 입력한다.

2. 로또넘버, 보너스 넘버를 입력한다."1,2,3,4,5,6", "7"

3. 시스템에서는

           List.of(8, 21, 23, 41, 42, 43),
           List.of(3, 5, 11, 16, 32, 38),
           List.of(7, 11, 16, 35, 36, 44),
           List.of(1, 8, 11, 31, 41, 42),
           List.of(13, 14, 16, 38, 42, 45),
           List.of(7, 11, 30, 40, 42, 43),
           List.of(2, 13, 22, 32, 38, 45),
           List.of(1, 3, 5, 14, 22, 45)

8개의 로또를 발행한다.

1. 결과를 출력한다.

위에서 run, output 메서드가 어떤 로직인지 파헤쳐 보기 위해 NsTest를 살펴보자.

public abstract class NsTest {
    private PrintStream standardOut;
    private OutputStream captor;

    @BeforeEach
    protected final void init() {
        standardOut = System.out;
        captor = new ByteArrayOutputStream();// 기존 Sytem.out의 스트림 저장
        System.setOut(new PrintStream(captor));// System.out으로 출력하는 스트림을 captor로 변경
    }

    @AfterEach
    protected final void printOutput() {
        System.setOut(standardOut); // 기존 Sytem.out스트림으로 출력 스트림 변경
        System.out.println(output());
    }

    protected final String output() {
        return captor.toString().trim();
    }

    protected final void run(final String... args) {
        try {
            command(args);
            runMain();
        } finally {
            Console.close();
        }
    }

    protected final void runException(final String... args) {
        try {
            run(args);
        } catch (final NoSuchElementException ignore) {
        }
    }

    private void command(final String... args) {
        final byte[] buf = String.join("\n", args).getBytes();
        System.setIn(new ByteArrayInputStream(buf));
    }

    protected abstract void runMain();
}

NsTest는 테스트 어플리케이션에서 콘솔에서 입력받거나 출력하는 값들을 내부적으로 처리하기 위해 System.setOut과 System.setIn을 사용한다. 먼저 콘솔 출력값(System.out.println)을 내부 변수에 어떻게 담는지 알아보자.

NsTest는 BeforeEach와 AfterEach 어노테이션을 사용해 Test 중에 System.out.println 메서드로 콘솔에 출력하는 값을 captor에 담아둔다.

1. init 메서드에서 System.out의 스트림을 captor를 향하도록 바꾼다. 즉, 테스트 코드의 출력 값을 captor로 담아둔다.
2. printOutput에서 System.out의 스트림을 원상복구한다.

콘솔에서 입력해야 하는 값은 미리 command 메서드에 String 값(String... args)으로 전달한다. 전달한 String 값은 byte 배열인 buf 변수에 담는다. 그리고 buf 값을 ByteArrayInputStream 형태로 System.in 입력 스트림으로 사용한다. 즉, Scanner.nextLine 등을 호출해 콘솔에서 입력값을 받을 때 실제로 콘솔에서 입력하는 게 아니라 buf 배열 안에 값들을 차례로 전달하는 것이다.

ApplicationTest에서 사용하는 run메서드는 command메서드와 runMain메서드로 구성된다. command 메서드는 게임 유저가 입력해야 하는 사려는 값(8000), 로또 당첨 번호(1,2,3,4,5,6), 보너스 번호(7) 을 받는다. 위에서 설명한 것처럼 command는 전달받은 문자열을 System.in의 입력 스트림으로 대기시킨다. runMain메서드는 ApplicationTest에서 오버라이딩한 메서드로 테스트 클래스 Application을 실행한다. Application을 실행하면서 Scanner.nextLine()을 호출할 때마다 command에 전달한 문자열 값을 리턴받는 것이다.

그럼 이제 assertRandomNumberInRangeTest가 어떤 동작을 수행하는지 살펴보자.

public static void assertRandomUniqueNumbersInRangeTest(
        final Executable executable,
        final List<Integer> value,
        final List<Integer>... values
    ) {
        assertRandomTest(
            () -> Randoms.**pickUniqueNumbersInRange**(anyInt(), anyInt(), anyInt()),
            executable,
            value,
            values
        );

assertRandomUniqueNumbersInRangeTest 메서드는 Executable, value, values를 전달받는다. 이것을 그대로 "() -> Ra*ndoms.pickUniqueNumbersInRange(anyInt(), anyInt(), anyInt())"와 함께 assertRandomTest 메서드에 전달한다. assertRandomTest는 한마디로 verification이 호출된다면 value, values를* 묶은 어레이의 값들을 차례로 리턴해, Executable을 실행하는 것이다.

private static <T> void assertRandomTest(
        final Verification verification,
        final Executable executable,
        final T value,
        final T... values
    ) {
        assertTimeoutPreemptively(RANDOM_TEST_TIMEOUT, () -> {
            try (final MockedStatic<Randoms> mock = mockStatic(Randoms.class)) {
                mock.when(verification).thenReturn(value, Arrays.stream(values).toArray());
                executable.execute();
            }
        });
    }