support uuid as arg encoder (ghostdogpr#1266)

* support uuid as arg encoder * test the uuid encoder * reuse string value for uuid encoder
Fluxx · Jan 27, 2022 · 56e63f2 · 56e63f2
1 parent ad0db0b
commit 56e63f2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/client/src/main/scala/caliban/client/ArgEncoder.scala b/client/src/main/scala/caliban/client/ArgEncoder.scala
@@ -4,6 +4,7 @@ import caliban.client.__Value.{ __BooleanValue, __ListValue, __NullValue, __Numb
 import io.circe.Json
 
 import scala.annotation.implicitNotFound
+import java.util.UUID
 
 /**
  * Typeclass that defines how to encode an argument of type `A` into a valid [[caliban.client.__Value]].
@@ -42,6 +43,8 @@ object ArgEncoder {
 
   implicit val unit: ArgEncoder[Unit] = (_: Unit) => __ObjectValue(Nil)
 
+  implicit val uuid: ArgEncoder[UUID] = (value: UUID) => __StringValue(value.toString())
+
   implicit def option[A](implicit ev: ArgEncoder[A]): ArgEncoder[Option[A]] = (value: Option[A]) =>
     value.fold(__NullValue: __Value)(ev.encode)
 

diff --git a/client/src/test/scala/caliban/client/ArgEncoderSpec.scala b/client/src/test/scala/caliban/client/ArgEncoderSpec.scala
@@ -3,6 +3,7 @@ package caliban.client
 import zio.test.Assertion.equalTo
 import zio.test._
 import zio.test.environment.TestEnvironment
+import java.util.UUID
 
 object ArgEncoderSpec extends DefaultRunnableSpec {
   override def spec: ZSpec[TestEnvironment, Any] =
@@ -20,6 +21,13 @@ object ArgEncoderSpec extends DefaultRunnableSpec {
         test("string with null characters") {
           assert(ArgEncoder.string.encode("abcde who am i\u0000").toString)(equalTo("\"abcde who am i\\u0000\""))
         }
+      ),
+      suite("__UUIDValue")(
+        test("regular uuid") {
+          assert(ArgEncoder.uuid.encode(UUID.fromString("20a69d87-6d68-4779-a4da-601f4c04ebf3")).toString)(
+            equalTo(""""20a69d87-6d68-4779-a4da-601f4c04ebf3"""")
+          )
+        }
       )
     )
 }