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

Question

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

问题：在编写junit集成测试时，我必须使用okhttp MockWebServer。这个模拟总是在一个随机端口上启动，例如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 中设置一个随机端口，以免干扰其他正在运行的测试或应用程序。因此必须坚持随机端口，并找到一种方法将其传递给网络客户端（或动态覆盖应用程序属性）。

---

更新： 我想出了以下内容，这是可行的。但也许有人知道如何使模拟服务器字段非静态？

@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();
    }
}