Fix testHeadAfterPut failures (#7565)

square · Dec 24, 2022 · 158c66b · 158c66b
1 parent f27a291
commit 158c66b
Showing 1 changed file with 9 additions and 17 deletions.
diff --git a/okhttp/src/jvmTest/java/okhttp3/CallKotlinTest.kt b/okhttp/src/jvmTest/java/okhttp3/CallKotlinTest.kt
@@ -147,15 +147,17 @@ class CallKotlinTest {
         .header("Content-Type", "application/xml")
         .put(ValidRequestBody())
         .build()
-    // 201
-    client.newCall(request).execute()
+    client.newCall(request).execute().use {
+      assertEquals(201, it.code)
+    }
 
     request = Request.Builder()
         .url(endpointUrl)
         .head()
         .build()
-    // 204
-    client.newCall(request).execute()
+    client.newCall(request).execute().use {
+      assertEquals(204, it.code)
+    }
 
     request = Request.Builder()
         .url(endpointUrl)
@@ -174,19 +176,9 @@ class CallKotlinTest {
         .head()
         .build()
 
-    client.newCall(request).execute()
-
-    var recordedRequest = server.takeRequest()
-    assertEquals("PUT", recordedRequest.method)
-
-    recordedRequest = server.takeRequest()
-    assertEquals("HEAD", recordedRequest.method)
-
-    recordedRequest = server.takeRequest()
-    assertThat(recordedRequest.failure).isNotNull()
-
-    recordedRequest = server.takeRequest()
-    assertEquals("HEAD", recordedRequest.method)
+    client.newCall(request).execute().use {
+      assertEquals(204, it.code)
+    }
   }
 
   @Test