知識庫 / Spring / Spring Boot RSS 訂閱

<h3><strong>5.1. 注入單個 WireMock 服務器</strong></h3>
<p>首先，讓我們將一個 WireMock 服務器注入到我們的測試類中。當需要模擬單個外部服務時，此方法非常有用：</p>

@SpringBootTest(classes = SimpleWiremockTest.AppConfiguration.class,
  webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@EnableWireMock({
    @ConfigureWireMock(name = "user-service", port = 8081),
})
public class InjectedWiremockTest {
    @InjectWireMock("user-service")
    WireMockServer mockUserService;

    @Autowired
    private TestRestTemplate restTemplate;

    @Test
    void givenEmptyUserList_whenFetchingUsers_thenReturnsEmptyList() {
        mockUserService.stubFor(get("/users").willReturn(okJson("[]")));

        ResponseEntity<String> response = restTemplate.getForEntity(
          "http://localhost:8081/users",
          String.class);

        Assertions.assertEquals(HttpStatus.OK, response.getStatusCode());
        Assertions.assertEquals("[]", response.getBody());
    }
}

與之前的方法不同，通過使用 @EnableWireMock 並在測試類級別啓用 WireMock，無需顯式注入，這種方法允許更精細的控制，通過注入一個命名的 WireMock 服務器實例。 @ConfigureWireMock 註解明確定義了 WireMock 實例的名稱和端口，從而方便在不同的測試用例中管理多個外部服務。

@InjectWireMock(“user-service”) 允許我們直接訪問 WireMockServer 實例，並在測試方法中動態配置和管理其行為。

5.2. 使用多個 WireMock 服務器

在我們的應用程序與多個外部服務交互時，我們可能需要使用多個 WireMock 實例來模擬多個 API。WireMock 允許我們為每個實例配置和指定不同的名稱和端口：

@SpringBootTest(classes = SimpleWiremockTest.AppConfiguration.class,
  webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@EnableWireMock({
    @ConfigureWireMock(name = "user-service", port = 8081),
    @ConfigureWireMock(name = "product-service", port = 8082)
})
public class InjectedWiremockTest {
    @InjectWireMock("user-service")
    WireMockServer mockUserService;

    @InjectWireMock("product-service")
    WireMockServer mockProductService;

    @Autowired
    private TestRestTemplate restTemplate;

    @Test
    void givenUserAndProductLists_whenFetchingUsersAndProducts_thenReturnsMockedData() {
        mockUserService.stubFor(get("/users")
          .willReturn(okJson("[{\"id\": 1, \"name\": \"John\"}]")));
        mockProductService.stubFor(get("/products")
          .willReturn(okJson("[{\"id\": 101, \"name\": \"Laptop\"}]")));

        ResponseEntity<String> userResponse = restTemplate
          .getForEntity("http://localhost:8081/users", String.class);
        ResponseEntity<String> productResponse = restTemplate
          .getForEntity("http://localhost:8082/products", String.class);

        Assertions.assertEquals(HttpStatus.OK, userResponse.getStatusCode());
        Assertions.assertEquals("[{\"id\": 1, \"name\": \"John\"}]", userResponse.getBody());

        Assertions.assertEquals(HttpStatus.OK, productResponse.getStatusCode());
        Assertions.assertEquals("[{\"id\": 101, \"name\": \"Laptop\"}]", productResponse.getBody());
    }
}

我們隔離了服務，確保對一個模擬服務器的更改不會干擾其他服務器。通過注入多個 WireMock 實例，我們可以完全模擬複雜的服務交互，從而使我們的測試更準確、更可靠。這種方法在微服務架構中尤其有益，因為不同的組件會與各種外部服務進行通信。

6. 結論

WireMock 是一款強大的工具，用於在 Spring Boot 應用程序中測試外部依賴項。 在本文中，我們創建了可靠、可重複且獨立的測試，而無需依賴實際的第三方服務。 我們從一個簡單的測試開始，並將其演變為更高級的場景，包括注入多個 WireMock 服務器。

通過這些技術，我們可以確保我們的應用程序正確處理外部 API 響應，無論它們返回預期數據或錯誤。