Fix test mocking

elastic · Nov 28, 2023 · bd7d324 · bd7d324
1 parent cd29883
commit bd7d324
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...-common-spring5/src/test/java/co/elastic/apm/agent/webfluxcommon/SpringWeb5UtilsTest.java b/...-common-spring5/src/test/java/co/elastic/apm/agent/webfluxcommon/SpringWeb5UtilsTest.java
@@ -45,7 +45,7 @@ void testWrongResponseType() {
     @Test
     void testGetClientStatusCode() throws Exception {
         ClientResponse mockResponse = mock(ClientResponse.class);
-        doReturn(HttpStatus.IM_USED).when(mockResponse).statusCode();
+        doReturn(226).when(mockResponse).rawStatusCode();
         assertThat(SpringWebVersionUtils.getClientStatusCode(mockResponse)).isEqualTo(226);
     }
 }