Spring Cloud gateway 单元测试，模拟http请求,并mock

矫情吗；* 2022-12-19 06:12 220阅读 0赞

Spring Cloud gateway使用的是reactive,使用WebTestClient模拟发送请求。

待测试的方法如下，方法中会访问其它的服务，在单元测试时，我们要把这个语法mock掉。

public class EmployeeService { 
        private WebClient.Builder webClient;
        public Mono<Employee> getEmployeeById(Integer employeeId) { 
            return webClient
                    .get()
                    .uri("http://auth_user/v2/role/resources")
                    .retrieve()
                    .bodyToMono(Employee.class);
        }
    }

@SpringBootTest(webEnvironment = RANDOM_PORT)
    @ExtendWith(MockitoExtension.class)
    public class GatewayTests { 
    
        protected static WebTestClient webClient;
        @MockBean
        private WebClient webClientMock;
        @MockBean
        private WebClient.RequestHeadersSpec requestHeadersMock;
        @MockBean
        private WebClient.RequestHeadersUriSpec requestHeadersUriMock;
        @MockBean
        private WebClient.ResponseSpec responseMock;
    	@LocalServerPort
        protected int port = 0;
        
        @BeforeEach
        public void setup()
        { 
            String baseUri = "http://localhost:"+ port;
            webClient = WebTestClient.bindToServer()
                    .baseUrl(baseUri)
                    .responseTimeout(Duration.ofSeconds(12))
                    .build();
        }
    
        /** * 测试 */
        @Test
        public void test()  { 
            GotRoleResourceDTO dto = new GotRoleResourceDTO();
            dto.setMethod("post");
            dto.setRoleId("ROLE-test");
            dto.setUrl("/auth_user/v2/test503");
            List<GotRoleResourceDTO> list = new ArrayList<>();
            list.add(dto);
            when(webClientBuilderMock.build()).thenReturn(webClientMock);
    
            when(webClientMock.get())
                    .thenReturn(requestHeadersUriMock);
            when(requestHeadersUriMock.uri("http://auth_user/v2/role/resources")).thenReturn(requestHeadersMock);
            when(requestHeadersMock.retrieve()).thenReturn(responseMock);
            when(responseMock.bodyToFlux(GotRoleResourceDTO.class)).thenReturn(Flux.just(dto));
    
    
            webTestClient.get().uri("/auth_user/v2/test503")
                    .header("Authorization","Bearer "+TOKEN)
                    .exchange()
                    .expectStatus().isEqualTo(503)
                    .expectBody(String.class).consumeWith(result ->{ 
                String body = result.getResponseBody();
                log.debug("body:{}",body);
            });
    
        }
    
    }

[官方例子][Link 1]

[Link 1]: https://github.com/spring-cloud/spring-cloud-gateway/blob/master/spring-cloud-gateway-sample/src/test/java/org/springframework/cloud/gateway/sample/GatewaySampleApplicationTests.java