如何使用Mockito测试REST服务？

Question

我对Java单元测试非常陌生，听说Mockito框架对于测试非常有用。

我已经开发了一个REST Server（CRUD方法），现在我想对其进行测试，但是我不知道如何？

我什至不知道该测试程序应该如何开始。我的服务器应该在本地主机上运行，​​然后在该URL上进行调用（例如，本地主机：8888）？

这是我到目前为止尝试过的方法，但是我很确定这不是正确的方法。

    @Test
    public void testInitialize() {
        RESTfulGeneric rest = mock(RESTfulGeneric.class);

        ResponseBuilder builder = Response.status(Response.Status.OK);

        builder = Response.status(Response.Status.OK).entity(
                "Your schema was succesfully created!");

        when(rest.initialize(DatabaseSchema)).thenReturn(builder.build());

        String result = rest.initialize(DatabaseSchema).getEntity().toString();

        System.out.println("Here: " + result);

        assertEquals("Your schema was succesfully created!", result);

    }

这是initialize方法的代码。

    @POST
    @Produces(MediaType.APPLICATION_JSON)
    @Path("/initialize")
    public Response initialize(String DatabaseSchema) {

        /** Set the LogLevel to Info, severe, warning and info will be written */
        LOGGER.setLevel(Level.INFO);

        ResponseBuilder builder = Response.status(Response.Status.OK);

        LOGGER.info("POST/initialize - Initialize the " + user.getUserEmail()
                + " namespace with a database schema.");

        /** Get a handle on the datastore itself */
        DatastoreService datastore = DatastoreServiceFactory
                .getDatastoreService();


        datastore.put(dbSchema);

        builder = Response.status(Response.Status.OK).entity(
                "Your schema was succesfully created!");
        /** Send response */
        return builder.build();
    }

在此测试用例中，我想将Json字符串发送到服务器（POST）。如果一切顺利，则服务器应回答“您的模式已成功创建！”。

有人可以帮帮我吗？

Answer 1

好。因此，该方法的约定如下：将输入字符串解析为JSON，BAD_REQUEST如果无效，则将其发送回。如果有效，请在中创建datastore具有各种属性的实体（您知道它们），然后将其发送回去OK。

并且您需要验证该方法是否满足了该合同。

Mockito在这里有什么帮助？好吧，如果您在没有Mockito的情况下测试此方法，则需要一个real
DataStoreService，并且需要验证已在此real中正确创建了实体DataStoreService。这是您的测试不再是单元测试的地方，这也是测试太复杂，太长且难以运行的原因，因为它需要一个复杂的环境。

Mockito可以通过DataStoreService模拟对的依赖来提供帮助：您可以创建的模拟DataStoreService，并initialize()在测试中调用方法时验证是否确实使用适当的entity参数调用了该模拟。

为此，您需要能够将注入DataStoreService到被测对象中。可以通过以下方式重构对象一样简单：

public class MyRestService {
    private DataStoreService dataStoreService;

    // constructor used on the server
    public MyRestService() {
        this.dataStoreService = DatastoreServiceFactory.getDatastoreService();
    }

    // constructor used by the unit tests
    public MyRestService(DataStoreService dataStoreService) {
        this.dataStoreService = dataStoreService;
    }

    public Response initialize(String DatabaseSchema) {
         ...
         // use this.dataStoreService instead of datastore
    }
}

现在，在您的测试方法中，您可以执行以下操作：

@Test
public void testInitializeWithGoodInput() {
    DataStoreService mockDataStoreService = mock(DataStoreService.class);
    MyRestService service = new MyRestService(mockDataStoreService);
    String goodInput = "...";
    Response response = service.initialize(goodInput);
    assertEquals(Response.Status.OK, response.getStatus());

    ArgumentCaptor<Entity> argument = ArgumentCaptor.forClass(Entity.class);
    verify(mock).put(argument.capture());
    assertEquals("the correct kind", argument.getValue().getKind());
    // ... other assertions
}