Add roundtrip test for custom Hocon codecs

AVSystem · Sep 28, 2024 · 0ab4592 · 0ab4592
1 parent 65c5f75
commit 0ab4592
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/hocon/src/main/scala/com/avsystem/commons/hocon/HoconOutput.scala b/hocon/src/main/scala/com/avsystem/commons/hocon/HoconOutput.scala
@@ -2,7 +2,9 @@ package com.avsystem.commons
 package hocon
 
 import com.avsystem.commons.annotation.explicitGenerics
-import com.avsystem.commons.serialization._
+import com.avsystem.commons.serialization.*
+import com.avsystem.commons.serialization.cbor.RawCbor
+import com.avsystem.commons.serialization.json.RawJson
 import com.typesafe.config.{ConfigValue, ConfigValueFactory}
 
 object HoconOutput {
@@ -31,7 +33,14 @@ class HoconOutput(consumer: ConfigValue => Unit) extends OutputAndSimpleOutput {
   def writeList(): HoconListOutput = new HoconListOutput(consumer)
   def writeObject(): HoconObjectOutput = new HoconObjectOutput(consumer)
 
-  //TODO: writeCustom
+  // TODO: handle other markers in writeCustom? Unfortunately typesafe config does not provide a methods to write
+  // duration to nice string representation etc.
+  override def writeCustom[T](typeMarker: TypeMarker[T], value: T): Boolean =
+    typeMarker match {
+      case ConfigValueMarker => consumer(value); true
+      case PeriodMarker => anyRef(value.toString.toLowerCase.stripPrefix("p")); true
+      case _ => false
+    }
 }
 
 class HoconListOutput(consumer: ConfigValue => Unit) extends ListOutput {

diff --git a/hocon/src/test/scala/com/avsystem/commons/hocon/HoconGenCodecRoundtripTest.scala b/hocon/src/test/scala/com/avsystem/commons/hocon/HoconGenCodecRoundtripTest.scala
@@ -1,8 +1,12 @@
 package com.avsystem.commons
 package hocon
 
+import com.avsystem.commons.hocon.HoconInputTest.CustomCodecsClass
 import com.avsystem.commons.serialization.{GenCodecRoundtripTest, Input, Output}
-import com.typesafe.config.ConfigValue
+import com.typesafe.config.{ConfigFactory, ConfigValue}
+
+import java.time.{Duration, Period}
+import scala.concurrent.duration.*
 
 class HoconGenCodecRoundtripTest extends GenCodecRoundtripTest {
   type Raw = ConfigValue
@@ -15,4 +19,17 @@ class HoconGenCodecRoundtripTest extends GenCodecRoundtripTest {
 
   def createInput(raw: ConfigValue): Input =
     new HoconInput(raw)
+
+  test("custom codes class") {
+    val value = CustomCodecsClass(
+      duration = 1.minute,
+      jDuration = Duration.ofMinutes(5),
+      fileSize = SizeInBytes.`1KiB`,
+      embeddedConfig = ConfigFactory.parseMap(JMap("something" -> "abc")),
+      period = Period.ofWeeks(2),
+      clazz = classOf[HoconGenCodecRoundtripTest],
+      clazzMap = Map(classOf[HoconGenCodecRoundtripTest] -> "abc"),
+    )
+    testRoundtrip(value)
+  }
 }