Add test for sleeping pill error handling

javaBin · Feb 11, 2025 · 2112bc3 · 2112bc3
1 parent ce8d070
commit 2112bc3
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 4 deletions.
diff --git a/backend/src/test/kotlin/no/java/cupcake/TestExtensions.kt b/backend/src/test/kotlin/no/java/cupcake/TestExtensions.kt
@@ -67,6 +67,18 @@ fun buildMockEngine(
         )
     }
 
+fun buildErrorMockEngine(
+    httpStatusCode: HttpStatusCode,
+    message: String,
+): MockEngine =
+    MockEngine { request ->
+        respond(
+            content = ByteReadChannel(message),
+            status = httpStatusCode,
+            headers = headersOf(HttpHeaders.ContentType, "text/plain"),
+        )
+    }
+
 fun buildSlackService(
     fixture: String,
     channel: String,
@@ -81,20 +93,23 @@ fun buildSlackService(
 
 fun buildSleepingPillService(
     fixture: String,
+    client: HttpClient? = null,
+    bringService: BringService? = null,
     block: (suspend (request: HttpRequestData) -> Unit)? = null,
 ): SleepingPillService =
     SleepingPillService(
-        client = buildClient(buildMockEngine(fixture, block)),
-        bringService = buildBringService(fixture = "/postal_codes.json", postalCodeUrl = "/test"),
+        client = client ?: buildClient(buildMockEngine(fixture, block)),
+        bringService = bringService ?: buildBringService(fixture = "/postal_codes.json", postalCodeUrl = "/test"),
     )
 
 fun buildBringService(
     fixture: String,
     postalCodeUrl: String,
+    client: HttpClient? = null,
     block: (suspend (request: HttpRequestData) -> Unit)? = null,
 ): BringService =
     BringService(
-        client = buildClient(buildMockEngine(fixture, block)),
+        client = client ?: buildClient(buildMockEngine(fixture, block)),
         postalCodeUrl = postalCodeUrl,
         scheduler = false,
     )
diff --git a/backend/src/test/kotlin/no/java/cupcake/plugins/RoutingTest.kt b/backend/src/test/kotlin/no/java/cupcake/plugins/RoutingTest.kt
@@ -3,14 +3,18 @@ package no.java.cupcake.plugins
 import io.kotest.core.spec.style.FunSpec
 import io.kotest.matchers.shouldBe
 import io.kotest.matchers.string.shouldContain
+import io.ktor.client.HttpClient
 import io.ktor.client.call.body
 import io.ktor.client.request.HttpRequestData
 import io.ktor.client.request.accept
 import io.ktor.client.request.get
+import io.ktor.client.statement.bodyAsText
 import io.ktor.http.ContentType
 import io.ktor.http.HttpStatusCode
 import io.ktor.server.testing.ApplicationTestBuilder
 import io.ktor.server.testing.testApplication
+import no.java.cupcake.buildClient
+import no.java.cupcake.buildErrorMockEngine
 import no.java.cupcake.buildSlackService
 import no.java.cupcake.buildSleepingPillService
 import no.java.cupcake.buildTestClient
@@ -108,12 +112,45 @@ class RoutingTest :
                     }
             }
         }
+
+        test("Fetch sessions - sleeping pill error") {
+            testApplication {
+                val id = randomString()
+
+                buildTestApplication(
+                    buildService(
+                        "/sessions.json",
+                        client =
+                            buildClient(
+                                buildErrorMockEngine(HttpStatusCode.InternalServerError, "SleepingPill is asleep"),
+                            ),
+                    ),
+                )
+
+                val client = buildTestClient()
+
+                client
+                    .get("/api/conferences/$id/sessions") {
+                        accept(ContentType.Application.Json)
+                    }.apply {
+                        status shouldBe HttpStatusCode.InternalServerError
+
+                        val response = bodyAsText()
+
+                        response shouldContain "SleepingPill is asleep"
+                        response shouldContain "500"
+                        response shouldContain "Internal Server Error"
+                        response shouldContain "call to SleepingPill failed"
+                    }
+            }
+        }
     })
 
 private fun buildService(
     fixture: String,
+    client: HttpClient? = null,
     block: (suspend (request: HttpRequestData) -> Unit)? = null,
-) = buildSleepingPillService(fixture, block)
+) = buildSleepingPillService(fixture = fixture, client = client, block = block)
 
 private fun ApplicationTestBuilder.buildTestApplication(service: SleepingPillService) {
     serializedTestApplication {