使用 MockRestServiceServer 验证多个 REST 请求

Question

假设我有以下端点： https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%20{a_number}

如您所见， a_number是一个变量，应该更改它的值，例如在以下循环中

public class Caller {

    public void callSomeThing() {
        int a_number = 0
        String URL = "https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%20";
        String updatedURL;
        while (a_number < 6) {
            updatedURL = URL + a_number;
            restTemplate.getForEntity(URI.create(updatedURL), String.class);
            a_number++;
        }
    }
}}

通过以上逻辑，我们对端点进行了 6 次 REST 调用

https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%200
https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%201
https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%202
https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%203
https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%204
https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%205

为了测试，我采用了MockRestServiceServer 。 测试的想法很简单：验证 REST 调用应该被请求 6 次。 为了实现这个想法，我想象如下。

@RestClientTest(Caller.class)
public class CallerTest {
    private static final String URL = "https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%20";

    @Autowired
    private Caller caller;

    @Autowired
    private MockRestServiceServer mockRestServiceServer;

    @BeforeEach
    void setUp() {
        mockRestServiceServer.reset();
    }

    @Test
    void shouldCall6Times() {
        int a_number_0 = 0;
        int a_number_1 = 1;
        int a_number_2 = 2;
        int a_number_3 = 3;
        int a_number_4 = 4;
        int a_number_5 = 5;

        mockRestServiceServer
                .expect(requestTo(URI.create(getUrl(a_number_0))))
                .andRespond(withSuccess());
                .expect(requestTo(URI.create(getUrl(a_number_1))))
                .andRespond(withSuccess());
                .expect(requestTo(URI.create(getUrl(a_number_2))))
                .andRespond(withSuccess());
                .expect(requestTo(URI.create(getUrl(a_number_3))))
                .andRespond(withSuccess());
                .expect(requestTo(URI.create(getUrl(a_number_4))))
                .andRespond(withSuccess());
                .expect(requestTo(URI.create(getUrl(a_number_5))))
                .andRespond(withSuccess());

        caller.callSomeThing();
    }

    private String getUrl(int a_number) {        
        return URL + a_number
    }
}

不幸的是，这个实现似乎不起作用。 你能帮帮我吗？

Answer 1

一种可能的测试方法即将出现。

@RestClientTest(Caller.class)
public class CallerTest {
    private static final String URL = "https://dummy-url/api/query/elements?MY_CONDITION_1%20LESSOREQUAL%20";

    @Autowired
    private Caller caller;

    @Autowired
    private MockRestServiceServer mockRestServiceServer;

    @BeforeEach
    void setUp() {
        mockRestServiceServer.reset();
    }

    @Test
    void shouldCall6Times() {
        IntStream.range(0, 6).forEach(i -> {
            mockRestServiceServer
                    .expect(requestTo(URI.create(getUrl(i))))
                    .andRespond(withSuccess());
        });


        caller.callSomeThing();
    }

    private String getUrl(int a_number) {        
        return URL + a_number
    }
}