7

我需要对一个使用WebClient. 有什么好的方法来处理 WebClient 吗？有了RestTemplate我可以轻松使用 Mockito。模拟 WebClient 有点乏味，因为深度存根不适用于 webclient...

我想测试我的代码是否提供了正确的标题...缩短的示例代码：

public class MyOperations {
    private final WebClient webClient;

    public MyOperations(WebClient webClient) {
        this.webClient = webClient;
    }

    public Mono<ResponseEntity<String>> get( URI uri) {
        return webClient.get()
                        .uri(uri)
                        .headers(computeHeaders())
                        .accept(MediaType.APPLICATION_JSON)
                        .retrieve().toEntity(String.class);
    }

    private HttpHeaders computeHeaders() {
        ...
    }

}
4

2 回答 2

5

这是针对单元测试，而不是集成测试...

在Kotlin中实现，这有点初级，但很有效。这个想法可以从下面的这段代码中提取出来

一、一个WebClient的kotlin扩展

import org.mockito.ArgumentMatchers.anyString
import org.mockito.Mockito.*
import org.springframework.web.reactive.function.client.WebClient
import org.springframework.web.reactive.function.client.WebClientResponseException
import reactor.core.publisher.toMono

fun WebClient.mockAndReturn(data: Any) {
    val uriSpec = mock(WebClient.RequestBodyUriSpec::class.java)
    doReturn(uriSpec).`when`(this).get()
    doReturn(uriSpec).`when`(this).post()
    ...

    val headerSpec = mock(WebClient.RequestBodyUriSpec::class.java)
    doReturn(headerSpec).`when`(uriSpec).uri(anyString())
    doReturn(headerSpec).`when`(uriSpec).uri(anyString(), anyString())
    doReturn(headerSpec).`when`(uriSpec).uri(anyString(), any())
    doReturn(headerSpec).`when`(headerSpec).accept(any())
    doReturn(headerSpec).`when`(headerSpec).header(any(), any())
    doReturn(headerSpec).`when`(headerSpec).contentType(any())
    doReturn(headerSpec).`when`(headerSpec).body(any())

    val clientResponse = mock(WebClient.ResponseSpec::class.java)
    doReturn(clientResponse).`when`(headerSpec).retrieve()
    doReturn(data.toMono()).`when`(clientResponse).bodyToMono(data.javaClass)
}

fun WebClient.mockAndThrow() {
    doThrow(WebClientResponseException::class.java).`when`(this).get()
    doThrow(WebClientResponseException::class.java).`when`(this).post()
    ...
}

然后，单元测试

class MyRepositoryTest {

    lateinit var client: WebClient

    lateinit var repository: MyRepository

    @BeforeEach
    fun setUp() {
        client = mock(WebClient::class.java)
        repository = MyRepository(client)
    }

    @Test
    fun getError() {
        assertThrows(WebClientResponseException::class.java, {
            client.mockAndThrow()
            repository.get("x")
        })
    }

    @Test
    fun get() {
        val myType = MyType()
        client.mockAndReturn(myType)
        assertEquals(myType, repository.get("x").block())
    }
}

注意：在 JUnit 5 上的测试

于 2018-04-04T08:54:42.463 回答
1

这将在未来的 Spring Framework 版本中得到支持MockRestServiceServer；见SPR-15286

于 2017-08-30T08:52:54.807 回答