When deserializing a MutableState, also deserialize its value

cashapp · Sep 28, 2023 · b1e3dcc · b1e3dcc
1 parent a48e231
commit b1e3dcc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/redwood-treehouse/src/commonMain/kotlin/app/cash/redwood/treehouse/StateSnapshot.kt b/redwood-treehouse/src/commonMain/kotlin/app/cash/redwood/treehouse/StateSnapshot.kt
@@ -34,7 +34,7 @@ public class StateSnapshot(
     return content.mapValues { entry ->
       entry.value.map {
         if (it.isMutableState) {
-          mutableStateOf(it.value)
+          mutableStateOf(it.value.fromJsonElement())
         } else {
           it.value.fromJsonElement()
         }

diff --git a/redwood-treehouse/src/commonTest/kotlin/app/cash/redwood/treehouse/StateSnapshotTest.kt b/redwood-treehouse/src/commonTest/kotlin/app/cash/redwood/treehouse/StateSnapshotTest.kt
@@ -32,12 +32,12 @@ class StateSnapshotTest {
     val valuesMap = stateSnapshot.toValuesMap()
     assertThat(valuesMap.entries.size).isEqualTo(4)
     assertTrue(valuesMap["key1"]!![0] is MutableState<*>)
-    assertThat((valuesMap["key1"]!![0] as MutableState<*>).value).isEqualTo(JsonPrimitive(1))
+    assertThat((valuesMap["key1"]!![0] as MutableState<*>).value).isEqualTo(1.0)
 
     assertThat(valuesMap["key2"]).isEqualTo(listOf(1.0))
 
     assertThat(valuesMap["key3"]!![0] is MutableState<*>)
-    assertThat((valuesMap["key3"]!![0] as MutableState<*>).value).isEqualTo(JsonPrimitive("str"))
+    assertThat((valuesMap["key3"]!![0] as MutableState<*>).value).isEqualTo("str")
 
     assertThat(valuesMap["key4"]).isEqualTo(listOf("str"))
   }