如何使用 PrintWriter 测试方法？

Question

我有以下方法：@Componentpublic class WriteCsvToResponse {    private static final Logger LOGGER = LoggerFactory.getLogger(WriteCsvToResponse.class);    public void writeStatus(PrintWriter writer, Status status) {        try {            ColumnPositionMappingStrategy mapStrategy                = new ColumnPositionMappingStrategy();            mapStrategy.setType(Status.class);            String[] columns = new String[]{"id", "storeId", "status"};            mapStrategy.setColumnMapping(columns);            StatefulBeanToCsv btcsv = new StatefulBeanToCsvBuilder(writer)                .withQuotechar(CSVWriter.NO_QUOTE_CHARACTER)                .withMappingStrategy(mapStrategy)                .withSeparator(',')                .build();            btcsv.write(status);        } catch (CsvException ex) {            LOGGER.error("Error mapping Bean to CSV", ex);        }    }我不知道如何正确使用mockito.使用它来将对象状态包装成csv格式。我使用 StringWriter 将响应包装在其中。没有更多细节了，但似乎我必须创建一些单词才能通过验证:)

Answer 1

你不需要 mockito 来测试这个方法，只需要一个java.io.StringWriter.

以下是如何编写用于名义用途的测试：

@Test

void status_written_in_csv_format() {

    // Setup

    WriteCsvToResponse objectUnderTest = new WriteCsvToResponse ();

    StringWriter stringWriter = new StringWriter();

    PrintWriter printWriter = new PrintWriter(stringWriter);

    // Given

    Status status = ...

    // When

    objectUnderTest.writeStatus(printWriter, status);

    // Then

    String actualCsv = stringWriter.toString();

    assertThat(actualCsv.split("\n"))

       .as("Produced CSV")

       .containsExactly(

         "id,storeId,status",

         "42,142,OK");

}

这个例子假设了一些关于你的Status对象的事情，但你有一般的想法。对于断言，我使用AssertJ，但您可以使用 JUnit5 内置断言执行相同操作。

Answer 2

您可以编写这样的测试并更改 write 方法中的输入：

    @Test

    public void test() {

        WriteCsvToResponse writeCsvToResponse = mock(WriteCsvToResponse.class);

        doAnswer(new Answer() {

            public Object answer(InvocationOnMock invocation) {

                Object[] args = invocation.getArguments();

                write((Status)args[1]);

                return null;

            }

        }).when(writeCsvToResponse).writeStatus(any(PrintWriter.class),any(Status.class));

        writeCsvToResponse.writeStatus(writer, status);

    }

    public void write(Status status) {

//      do anythings you like with status

    }

Answer 3

稍微重构一下，其中 Builder 是注入到该组件中的 Spring Bean。

然后，您可以模拟该构建器以返回模拟的 StatefulBeanToCsv，特别是 write 方法，您可以在其中编写条件和断言。如果遇到错误，则抛出一些未经检查的异常，例如 IllegalStateException，如果一切正常，则不抛出任何内容