Fix kt tests

decentralized-identity · Sep 24, 2024 · f9a69bd · f9a69bd
1 parent c08932a
commit f9a69bd
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 11 deletions.
diff --git a/bound/kt/src/test/kotlin/web5/sdk/dids/ResolutionResultTests.kt b/bound/kt/src/test/kotlin/web5/sdk/dids/ResolutionResultTests.kt
@@ -2,6 +2,7 @@ package web5.sdk.dids
 
 import okhttp3.mockwebserver.MockResponse
 import okhttp3.mockwebserver.MockWebServer
+import okhttp3.mockwebserver.SocketPolicy
 import org.junit.jupiter.api.*
 import org.junit.jupiter.api.Assertions.*
 import web5.sdk.dids.methods.jwk.DidJwk
@@ -43,6 +44,7 @@ class ResolutionResultTests {
                     .setResponseCode(200)
                     .addHeader("Content-Type", "application/json")
                     .setBody(bearerDid.document.toJsonString())
+                    .setSocketPolicy(SocketPolicy.DISCONNECT_AT_END)
             )
 
             val resolutionResult = ResolutionResult.resolve(bearerDid.did.uri)

diff --git a/bound/kt/src/test/kotlin/web5/sdk/dids/methods/dht/DidDhtTests.kt b/bound/kt/src/test/kotlin/web5/sdk/dids/methods/dht/DidDhtTests.kt
@@ -1,9 +1,6 @@
 package web5.sdk.dids.methods.dht
 
-import okhttp3.mockwebserver.Dispatcher
-import okhttp3.mockwebserver.MockResponse
-import okhttp3.mockwebserver.MockWebServer
-import okhttp3.mockwebserver.RecordedRequest
+import okhttp3.mockwebserver.*
 import org.junit.jupiter.api.*
 import org.junit.jupiter.api.Assertions.*
 import web5.sdk.Web5Exception
@@ -46,6 +43,7 @@ class DidDhtTests {
                 MockResponse()
                     .setResponseCode(200)
                     .addHeader("Content-Type", "application/octet-stream")
+                    .setSocketPolicy(SocketPolicy.DISCONNECT_AT_END)
             )
 
             val bearerDid = DidDht.create(
@@ -157,6 +155,7 @@ class DidDhtTests {
                 MockResponse()
                     .setResponseCode(200)
                     .addHeader("Content-Type", "application/octet-stream")
+                    .setSocketPolicy(SocketPolicy.DISCONNECT_AT_END)
             )
 
             val bearerDid = DidDht.create(
@@ -190,6 +189,7 @@ class DidDhtTests {
                 MockResponse()
                     .setResponseCode(500)
                     .addHeader("Content-Type", "application/octet-stream")
+                    .setSocketPolicy(SocketPolicy.DISCONNECT_AT_END)
             )
 
             val bearerDid = DidDht.create(
@@ -249,6 +249,7 @@ class DidDhtTests {
                 MockResponse()
                     .setResponseCode(404)
                     .addHeader("Content-Type", "application/octet-stream")
+                    .setSocketPolicy(SocketPolicy.DISCONNECT_AT_END)
             )
 
             val bearerDid = DidDht.create(
@@ -282,6 +283,7 @@ class DidDhtTests {
                 MockResponse()
                     .setResponseCode(500)
                     .addHeader("Content-Type", "application/octet-stream")
+                    .setSocketPolicy(SocketPolicy.DISCONNECT_AT_END)
             )
 
             val bearerDid = DidDht.create(
@@ -321,12 +323,14 @@ class DidDhtTests {
                             MockResponse()
                                 .setResponseCode(200)
                                 .addHeader("Content-Type", "application/octet-stream")
+                                .setSocketPolicy(SocketPolicy.DISCONNECT_AT_END)
                         }
 
                         request.method == "GET" -> {
                             MockResponse()
                                 .setResponseCode(200)
                                 .addHeader("Content-Type", "application/octet-stream")
+                                .setSocketPolicy(SocketPolicy.DISCONNECT_AT_END)
                                 .setBody(okio.Buffer().write(publishedBody.first()))
                         }
 

diff --git a/bound/kt/src/test/kotlin/web5/sdk/dids/methods/web/DidWebTests.kt b/bound/kt/src/test/kotlin/web5/sdk/dids/methods/web/DidWebTests.kt
@@ -3,6 +3,7 @@ package web5.sdk.dids.methods.web
 import web5.sdk.Web5Exception
 import okhttp3.mockwebserver.MockResponse
 import okhttp3.mockwebserver.MockWebServer
+import okhttp3.mockwebserver.SocketPolicy
 import org.junit.jupiter.api.*
 import org.junit.jupiter.api.Assertions.*
 import org.junit.jupiter.api.fail
@@ -224,6 +225,7 @@ class DidWebTests {
                     .setResponseCode(200)
                     .addHeader("Content-Type", "application/json")
                     .setBody(bearerDid.document.toJsonString())
+                    .setSocketPolicy(SocketPolicy.DISCONNECT_AT_END)
             )
 
             val resolveResult = DidWeb.resolve(bearerDid.did.uri)