Fix updateExpense test

larmie56 · Sep 4, 2021 · 408286d · 408286d
1 parent d53e13b
commit 408286d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/cache/src/main/java/com/example/expenselogger/cache/entity/ExpenseEntity.kt b/cache/src/main/java/com/example/expenselogger/cache/entity/ExpenseEntity.kt
@@ -13,7 +13,7 @@ public data class ExpenseEntity(
     @ColumnInfo(name = "date")
     val date: Long,
     @ColumnInfo(name = "info")
-    var info: String
+    val info: String
 ) {
     @PrimaryKey(autoGenerate = true)
     @ColumnInfo(name = "id")

diff --git a/cache/src/test/java/com/example/expenselogger/cache/repository/ExpenseRepositoryTest.kt b/cache/src/test/java/com/example/expenselogger/cache/repository/ExpenseRepositoryTest.kt
@@ -66,8 +66,11 @@ internal class ExpenseRepositoryTest {
         val id = expenseRepository.insertExpense(expenseEntity)
         val newExpenseEntity = expenseRepository.getExpense(id)
         val newInfo = "Valentine outing with now ex bae"
-        newExpenseEntity?.info = newInfo
-        expenseRepository.updateExpense(newExpenseEntity!!)
+        val newExpenseEntityCopy = newExpenseEntity?.copy(info = newInfo)
+        newExpenseEntityCopy?.let {
+            it.id = id
+            expenseRepository.updateExpense(it)
+        }
         val actual = expenseRepository.getExpense(id)
         assertThat(actual?.info).isEqualTo(newInfo)
     }