Implement "Delete Budget" test

Ivy-Apps · Nov 24, 2021 · bd29c38 · bd29c38
1 parent 62edbee
commit bd29c38
Show file tree

Hide file tree

Showing 4 changed files with 57 additions and 6 deletions.
diff --git a/app/src/androidTest/java/com/ivy/wallet/compose/helpers/BudgetModal.kt b/app/src/androidTest/java/com/ivy/wallet/compose/helpers/BudgetModal.kt
@@ -39,4 +39,9 @@ class BudgetModal<A : ComponentActivity>(
         composeTestRule.onNodeWithText("Save")
             .performClick()
     }
+
+    fun clickDelete() {
+        composeTestRule.onNodeWithTag("modal_delete")
+            .performClick()
+    }
 }
diff --git a/app/src/androidTest/java/com/ivy/wallet/compose/helpers/BudgetsScreen.kt b/app/src/androidTest/java/com/ivy/wallet/compose/helpers/BudgetsScreen.kt
@@ -30,6 +30,9 @@ class BudgetsScreen<A : ComponentActivity>(
             appBudget != null && categoryBudget != null -> {
                 budgetInfoNode.assertTextEquals("Budget info: $categoryBudget $currency for categories / $appBudget $currency app budget")
             }
+            appBudget == null && categoryBudget == null -> {
+                budgetInfoNode.assertDoesNotExist()
+            }
             else -> error("Unexpected case")
         }
     }
@@ -42,4 +45,11 @@ class BudgetsScreen<A : ComponentActivity>(
             .performClick()
     }
 
+    fun assertBudgetDoesNotExist(
+        budgetName: String
+    ) {
+        composeTestRule.onNodeWithText(budgetName)
+            .assertDoesNotExist()
+    }
+
 }
diff --git a/app/src/androidTest/java/com/ivy/wallet/compose/scenario/BudgetsTests.kt b/app/src/androidTest/java/com/ivy/wallet/compose/scenario/BudgetsTests.kt
@@ -1,10 +1,7 @@
 package com.ivy.wallet.compose.scenario
 
 import com.ivy.wallet.compose.IvyComposeTest
-import com.ivy.wallet.compose.helpers.BudgetModal
-import com.ivy.wallet.compose.helpers.BudgetsScreen
-import com.ivy.wallet.compose.helpers.HomeMoreMenu
-import com.ivy.wallet.compose.helpers.OnboardingFlow
+import com.ivy.wallet.compose.helpers.*
 import dagger.hilt.android.testing.HiltAndroidTest
 import org.junit.Test
 
@@ -15,6 +12,7 @@ class BudgetsTests : IvyComposeTest() {
     private val homeMoreMenu = HomeMoreMenu(composeTestRule)
     private val budgetsScreen = BudgetsScreen(composeTestRule)
     private val budgetModal = BudgetModal(composeTestRule)
+    private val deleteConfirmationModal = DeleteConfirmationModal(composeTestRule)
 
     @Test
     fun CreateGlobalBudget() {
@@ -136,7 +134,42 @@ class BudgetsTests : IvyComposeTest() {
 
     @Test
     fun DeleteBudget() {
-        TODO("Implement")
+        onboardingFlow.quickOnboarding()
+
+        homeMoreMenu.clickOpenCloseArrow()
+        homeMoreMenu.clickBudgets()
+
+        budgetsScreen.clickAddBudget()
+
+        budgetModal.apply {
+            enterAmount("2,500")
+            clickCategory("Food & Drinks")
+            clickCategory("Bills & Fees")
+            enterName("Living (must)")
+            clickAdd()
+        }
+
+        budgetsScreen.assertBudgetsInfo(
+            appBudget = null,
+            categoryBudget = "2,500.00"
+        )
+
+        budgetsScreen.clickBudget(
+            budgetName = "Living (must)"
+        )
+
+        //Delete budget
+        budgetModal.clickDelete()
+        deleteConfirmationModal.confirmDelete()
+
+        budgetsScreen.assertBudgetsInfo(
+            appBudget = null,
+            categoryBudget = null
+        )
+
+        budgetsScreen.assertBudgetDoesNotExist(
+            budgetName = "Living (must)"
+        )
     }
 
     @Test

diff --git a/app/src/main/java/com/ivy/wallet/ui/theme/modal/IvyModalComponents.kt b/app/src/main/java/com/ivy/wallet/ui/theme/modal/IvyModalComponents.kt
@@ -6,6 +6,7 @@ import androidx.compose.foundation.layout.size
 import androidx.compose.material.Text
 import androidx.compose.runtime.Composable
 import androidx.compose.ui.Modifier
+import androidx.compose.ui.platform.testTag
 import androidx.compose.ui.text.font.FontWeight
 import androidx.compose.ui.unit.dp
 import com.ivy.wallet.R
@@ -193,7 +194,9 @@ fun ModalDelete(
     onClick: () -> Unit
 ) {
     IvyCircleButton(
-        modifier = Modifier.size(40.dp),
+        modifier = Modifier
+            .size(40.dp)
+            .testTag("modal_delete"),
         icon = R.drawable.ic_delete,
         backgroundGradient = GradientRed,
         enabled = enabled,