Skip to content

Commit 733edb1

Browse files
committed
Fixes for 4.x branch
1 parent e11c48c commit 733edb1

File tree

1 file changed

+22
-23
lines changed

1 file changed

+22
-23
lines changed

okhttp/src/test/java/okhttp3/internal/http/SocketFailureTest.kt

Lines changed: 22 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -15,34 +15,32 @@
1515
*/
1616
package okhttp3.internal.http
1717

18-
import assertk.assertThat
19-
import assertk.assertions.isEqualTo
20-
import java.net.Socket
21-
import kotlin.test.assertFailsWith
22-
import mockwebserver3.MockResponse
23-
import mockwebserver3.MockWebServer
18+
import junit.framework.TestCase.fail
2419
import okhttp3.Call
2520
import okhttp3.Connection
2621
import okhttp3.EventListener
2722
import okhttp3.Headers
2823
import okhttp3.OkHttpClientTestRule
2924
import okhttp3.Request
25+
import okhttp3.mockwebserver.MockResponse
26+
import okhttp3.mockwebserver.MockWebServer
3027
import okhttp3.testing.PlatformRule
3128
import okio.IOException
32-
import org.junit.jupiter.api.BeforeEach
33-
import org.junit.jupiter.api.Tag
34-
import org.junit.jupiter.api.Test
35-
import org.junit.jupiter.api.extension.RegisterExtension
29+
import org.junit.Before
30+
import org.junit.Rule
31+
import org.junit.Test
32+
import java.net.Socket
33+
import org.assertj.core.api.Assertions.assertThat
3634

37-
@Tag("Slowish")
3835
class SocketFailureTest {
39-
@RegisterExtension
36+
@get:Rule
4037
val platform = PlatformRule()
4138

39+
@get:Rule
40+
val clientTestRule = OkHttpClientTestRule()
41+
4242
val listener = SocketClosingEventListener()
4343

44-
@RegisterExtension
45-
val clientTestRule = OkHttpClientTestRule()
4644
private lateinit var server: MockWebServer
4745
private var client =
4846
clientTestRule
@@ -68,9 +66,9 @@ class SocketFailureTest {
6866
}
6967
}
7068

71-
@BeforeEach
72-
fun setUp(server: MockWebServer) {
73-
this.server = server
69+
@Before
70+
fun setUp() {
71+
server = MockWebServer()
7472
}
7573

7674
@Test
@@ -86,7 +84,7 @@ class SocketFailureTest {
8684
.url(server.url("/"))
8785
.build(),
8886
)
89-
call1.execute().use { response -> response.body.string() }
87+
call1.execute().use { response -> response.body?.string() }
9088

9189
listener.shouldClose = true
9290
// Large headers are a likely reason the servers would cut off the connection before it completes sending
@@ -109,10 +107,11 @@ class SocketFailureTest {
109107
.build(),
110108
)
111109

112-
val exception =
113-
assertFailsWith<IOException> {
114-
call2.execute()
115-
}
116-
assertThat(exception.message).isEqualTo("Socket closed")
110+
try {
111+
call2.execute()
112+
fail()
113+
} catch (ioe: IOException) {
114+
assertThat(ioe.message).isEqualTo("Socket closed")
115+
}
117116
}
118117
}

0 commit comments

Comments
 (0)