feat: support objectmapper customization

manosbatsis · Dec 28, 2023 · 83ecc79 · 83ecc79
1 parent ca598bf
commit 83ecc79
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/...in/kotlin/com/github/manosbatsis/corda5/testutils/integration/junit5/Corda5NodesConfig.kt b/...in/kotlin/com/github/manosbatsis/corda5/testutils/integration/junit5/Corda5NodesConfig.kt
@@ -1,5 +1,6 @@
 package com.github.manosbatsis.corda5.testutils.integration.junit5
 
+import com.fasterxml.jackson.databind.ObjectMapper
 import com.github.manosbatsis.corda5.testutils.rest.client.loggerFor
 import java.io.File
 
@@ -39,8 +40,9 @@ data class Corda5NodesConfig(
     val baseUrl: String = "https://localhost:8888/api/v1/",
     val httpMaxWaitSeconds: Int = 60,
     val debug: Boolean = false,
-    val projectDir: File = Corda5NodesConfig.gradleRootDir,
-    val combinedWorkerMode: CombinedWorkerMode = CombinedWorkerMode.SHARED
+    val projectDir: File = gradleRootDir,
+    val combinedWorkerMode: CombinedWorkerMode = CombinedWorkerMode.SHARED,
+    val objectMapperConfigurer: ((ObjectMapper) -> Unit)? = null
 ) {
     companion object {
         private val logger = loggerFor(Corda5NodesConfig::class.java)

diff --git a/...m/github/manosbatsis/corda5/testutils/integration/junit5/nodehandles/NodeHandlesHelper.kt b/...m/github/manosbatsis/corda5/testutils/integration/junit5/nodehandles/NodeHandlesHelper.kt
@@ -22,11 +22,17 @@ class NodeHandlesHelper(
 
     companion object {
         private val logger = loggerFor(NodeHandlesHelper::class.java)
-        private val objectMapper = ObjectMapper().registerModules(JavaTimeModule(), kotlinModule(), Corda5Module())
 
         var nodeHandlesCache: NodeHandles? = null
     }
 
+    private val objectMapper by lazy {
+        ObjectMapper().registerModules(JavaTimeModule(), kotlinModule(), Corda5Module())
+            .also {
+                if (config.objectMapperConfigurer != null) config.objectMapperConfigurer.invoke(it)
+            }
+    }
+
     val nodeHandles: NodeHandles
         get() {
             when (config.combinedWorkerMode) {