[FIX/#241] 코드 수정

teamterning · Sep 16, 2024 · ad86cc9 · ad86cc9
1 parent c948c10
commit ad86cc9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/feature/src/main/java/com/terning/feature/filtering/filteringone/FilteringOneRoute.kt b/feature/src/main/java/com/terning/feature/filtering/filteringone/FilteringOneRoute.kt
@@ -58,10 +58,12 @@ fun FilteringOneRoute(
     FilteringOneScreen(
         name = name,
         onButtonClick = { grade ->
-            if (grade.isNotEmpty()) {
-                viewModel.updateGrade(grade)
-                viewModel.updateButton(true)
-            } else viewModel.updateButton(false)
+            with(viewModel) {
+                if (grade.isNotEmpty()) {
+                    updateGrade(grade)
+                    updateButton(true)
+                } else updateButton(false)
+            }
         },
         onNextClick = viewModel::navigateToFilteringTwo,
         navigateUp = viewModel::navigateUp,

diff --git a/feature/src/main/java/com/terning/feature/filtering/filteringtwo/FilteringTwoRoute.kt b/feature/src/main/java/com/terning/feature/filtering/filteringtwo/FilteringTwoRoute.kt
@@ -63,10 +63,12 @@ fun FilteringTwoRoute(
         onNextClick = viewModel::navigateToFilteringThree,
         navigateUp = viewModel::navigateUp,
         onButtonClick = { workingPeriod ->
-            if (workingPeriod.isNotEmpty()) {
-                viewModel.updateWorkingPeriod(workingPeriod)
-                viewModel.updateButton(true)
-            } else viewModel.updateButton(false)
+            with(viewModel) {
+                if (workingPeriod.isNotEmpty()) {
+                    updateWorkingPeriod(workingPeriod)
+                    updateButton(true)
+                } else updateButton(false)
+            }
         },
         buttonState = state.isButtonValid,
         workingPeriod = state.workingPeriod,