Remove incorrect toString() for URL in RestClientGraphQLClient error …

…handling and added more tests
Netflix · Nov 22, 2024 · 5d3990a · 5d3990a
1 parent f112648
commit 5d3990a
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/graphql-dgs-client/src/main/kotlin/com/netflix/graphql/dgs/client/RestClientGraphQLClient.kt b/graphql-dgs-client/src/main/kotlin/com/netflix/graphql/dgs/client/RestClientGraphQLClient.kt
@@ -99,7 +99,7 @@ class RestClientGraphQLClient(
         if (!responseEntity.statusCode.is2xxSuccessful) {
             throw GraphQLClientException(
                 statusCode = responseEntity.statusCode.value(),
-                url = restClient.toString(),
+                url = "",
                 response = responseEntity.body ?: "",
                 request = serializedRequest,
             )

diff --git a/...-dgs-client/src/test/kotlin/com/netflix/graphql/dgs/client/RestClientGraphQLClientTest.kt b/...-dgs-client/src/test/kotlin/com/netflix/graphql/dgs/client/RestClientGraphQLClientTest.kt
@@ -34,6 +34,7 @@ import graphql.schema.idl.RuntimeWiring
 import graphql.schema.idl.SchemaParser
 import graphql.schema.idl.TypeDefinitionRegistry
 import org.assertj.core.api.Assertions.assertThat
+import org.assertj.core.api.Assertions.assertThatException
 import org.intellij.lang.annotations.Language
 import org.junit.jupiter.api.BeforeEach
 import org.junit.jupiter.api.Test
@@ -45,6 +46,7 @@ import org.springframework.boot.test.web.server.LocalServerPort
 import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder
 import org.springframework.web.bind.annotation.RequestHeader
 import org.springframework.web.bind.annotation.RequestParam
+import org.springframework.web.client.HttpClientErrorException
 import org.springframework.web.client.RestClient
 import java.time.LocalDateTime
 import java.time.format.DateTimeFormatter
@@ -82,6 +84,27 @@ class RestClientGraphQLClientTest {
         assertThat(result).isEqualTo("Hi!")
     }
 
+    @Test
+    fun `Unsuccessful request with default status handling`() {
+        val client = RestClientGraphQLClient(restClient.mutate().baseUrl("http://localhost:$port/wrongpath").build())
+        assertThatException().isThrownBy { client.executeQuery("{hello}") }.isInstanceOf(HttpClientErrorException::class.java)
+    }
+
+    @Test
+    fun `Unsuccessful request with non default status handling`() {
+        val client =
+            RestClientGraphQLClient(
+                restClient
+                    .mutate()
+                    .defaultStatusHandler(
+                        { true },
+                        { _, _ -> },
+                    ).baseUrl("http://localhost:$port/wrongpath")
+                    .build(),
+            )
+        assertThatException().isThrownBy { client.executeQuery("{hello}") }.isInstanceOf(GraphQLClientException::class.java)
+    }
+
     @Test
     fun `Extra header can be provided`() {
         val client =