fix(health-sdk): Fix copying of extractions in the tests

IPC-295
gini · Jul 17, 2024 · dcb2041 · dcb2041
1 parent 33f6f99
commit dcb2041
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/health-sdk/sdk/src/test/java/net/gini/android/health/sdk/GiniHealthTest.kt b/health-sdk/sdk/src/test/java/net/gini/android/health/sdk/GiniHealthTest.kt
@@ -41,9 +41,29 @@ val extractions = ExtractionsContainer(
 )
 
 fun copyExtractions(extractions: ExtractionsContainer) = ExtractionsContainer(
-    extractions.specificExtractions.toMap(),
+    extractions.specificExtractions.toMap().mapValues { specificExtraction ->
+        SpecificExtraction(
+            specificExtraction.value.name,
+            specificExtraction.value.value,
+            specificExtraction.value.entity,
+            specificExtraction.value.box,
+            specificExtraction.value.candidate
+        )
+    },
     extractions.compoundExtractions.map { (name, compoundExtraction) ->
-        name to CompoundExtraction(name, compoundExtraction.specificExtractionMaps.map { it.toMap() })
+        name to CompoundExtraction(
+            name,
+            compoundExtraction.specificExtractionMaps.map { specificExtractionMap ->
+                specificExtractionMap.toMap().mapValues { specificExtraction ->
+                    SpecificExtraction(
+                        specificExtraction.value.name,
+                        specificExtraction.value.value,
+                        specificExtraction.value.entity,
+                        specificExtraction.value.box,
+                        specificExtraction.value.candidate
+                    )
+                }
+            })
     }.toMap()
 )