Complete cache contract test

larmie56 · Sep 4, 2021 · d53e13b · d53e13b
2 parents d73fde1 + b6bad85
commit d53e13b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
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")
-    val info: String
+    var 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
@@ -59,4 +59,25 @@ internal class ExpenseRepositoryTest {
         val actual = expenseRepository.getExpense(id)
         assertThat(actual).isEqualTo(expenseEntity)
     }
+
+    @Test
+    fun `verify that updateExpense updates an expense`(): Unit = runBlocking {
+        val expenseEntity = DummyData.expenseEntity
+        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 actual = expenseRepository.getExpense(id)
+        assertThat(actual?.info).isEqualTo(newInfo)
+    }
+
+    @Test
+    fun `verify that deleteExpense deletes an expense`(): Unit = runBlocking {
+        val expenseEntity = DummyData.expenseEntity
+        val id = expenseRepository.insertExpense(expenseEntity)
+        expenseRepository.deleteExpense(id)
+        val actual = expenseRepository.getExpense(id)
+        assertThat(actual).isNull()
+    }
 }