JUnit测试中如何将MockWebServer端口设置为WebClient？

Question

我将 spring-boot 与 WebClient 一起使用，它作为 bean 自动装配。

问题：在编写junit集成测试时，我必须使用okhttpMockWebServer。这个模拟总是在一个随机端口上启动，例如 localhost:14321.

现在我的 WebClient 当然有一个固定的 url 可以将请求发送到。 url 可能由 application.properties 参数给出，例如 webclient.url=https://my.domain.com/，因此我可以在 junit 测试中覆盖该字段。但只是静态的。

问题：如何重置 @SpringBootTest 中的 WebClient bean，以便它始终将请求发送到我的模拟服务器？

@Service
public class WebClientService {
     public WebClientService(WebClient.Builder builder, @Value("${webclient.url}" String url) {
          this.webClient = builder.baseUrl(url)...build();
     }

     public Response send() {
          return webClient.body().exchange().bodyToMono();
     }
}

@Service
public void CallingService {
      @Autowired
      private WebClientService service;

      public void call() {
           service.send();
      }
}


@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class MyWebTest {
        @Autowired
        private CallingService calling;

        @Test
        public void test() {
             MockWebServer mockWebServer = new MockWebServer();
             System.out.println("Current mock server url: " + mockWebServer.url("/").toString()); //this is random    

             mockWebServer.enqueue(new MockResponse()....);

             //TODO how to make the mocked server url public to the WebClient?
             calling.call();
        }
}

如您所见，我正在编写完整的真实世界 junit 集成测试。唯一的问题是：如何将 MockWebServer url 和端口传递给 WebClient 以便它自动将请求发送到我的模拟？

旁注：我绝对需要 MockWebServer 中的随机端口，以免干扰其他运行测试或应用程序。因此必须坚持使用随机端口，并找到一种方法将其传递给 webclient（或动态覆盖应用程序属性）。

更新：我想出了以下方法，它有效。但也许有人知道如何使 mockserver 字段 non-static?

@ContextConfiguration(initializers = RandomPortInitializer.class)
public abstract class AbstractITest {
    @ClassRule
    public static final MockWebServer mockWebServer = new MockWebServer();

    public static class RandomPortInitializer implements ApplicationContextInitializer<ConfigurableApplicationContext> {
        @Override
        public void initialize(ConfigurableApplicationContext applicationContext) {
            TestPropertySourceUtils.addInlinedPropertiesToEnvironment(applicationContext,
                    "webclient.url=" + mockWebServer.url("/").toString());
        }
    }
}

Answer 1

自从 Spring Framework 5.2.5 (Spring Boot 2.x) 你可以使用 DynamicPropertySource 非常方便的注解。

这是一个完整的示例，您可以如何将其与 MockWebServer 一起使用以绑定正确的端口：

@ExtendWith(SpringExtension.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS)
public abstract class AbstractIT {

    static MockWebServer mockWebServer;

    @DynamicPropertySource
    static void properties(DynamicPropertyRegistry r) throws IOException {
        r.add("some-service.url", () -> "http://localhost:" + mockWebServer.getPort());
    }

    @BeforeAll
    static void beforeAll() throws IOException {
        mockWebServer = new MockWebServer();
        mockWebServer.start();
    }

    @AfterAll
    static void afterAll() throws IOException {
        mockWebServer.shutdown();
    }
}

JUnit测试中如何将MockWebServer端口设置为WebClient？

How to set MockWebServer port to WebClient in JUnit test?

java

spring

spring-boot

mockwebserver

okhttp3