From 881c2b373453f3ed438fcfdab7ea64ba58c707c5 Mon Sep 17 00:00:00 2001
From: Phillip Thelen <phillip@habitica.com>
Date: Mon, 16 May 2022 15:56:13 +0200
Subject: [PATCH] fix UI tests

---
 .../ui/fragments/tasks/TaskRecyclerViewFragmentTest.kt    | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/Habitica/src/androidTest/java/com/habitrpg/android/habitica/ui/fragments/tasks/TaskRecyclerViewFragmentTest.kt b/Habitica/src/androidTest/java/com/habitrpg/android/habitica/ui/fragments/tasks/TaskRecyclerViewFragmentTest.kt
index 58d055200..16cad30c1 100644
--- a/Habitica/src/androidTest/java/com/habitrpg/android/habitica/ui/fragments/tasks/TaskRecyclerViewFragmentTest.kt
+++ b/Habitica/src/androidTest/java/com/habitrpg/android/habitica/ui/fragments/tasks/TaskRecyclerViewFragmentTest.kt
@@ -5,7 +5,6 @@ import android.view.View
 import androidx.fragment.app.testing.launchFragmentInContainer
 import com.habitrpg.android.habitica.R
 import com.habitrpg.android.habitica.databinding.FragmentRefreshRecyclerviewBinding
-import com.habitrpg.android.habitica.helpers.TaskFilterHelper
 import com.habitrpg.android.habitica.models.tasks.Task
 import com.habitrpg.android.habitica.models.tasks.TaskList
 import com.habitrpg.android.habitica.models.tasks.TaskType
@@ -43,7 +42,6 @@ internal class TaskRecyclerViewFragmentTest : FragmentTestCase<TaskRecyclerViewF
         tasks = loadJsonFile<TaskList>("tasks", TaskList::class.java).tasks.values
         fragment = spyk()
         fragment.shouldInitializeComponent = false
-        fragment.taskFilterHelper = TaskFilterHelper()
     }
 
     override fun launchFragment(args: Bundle?) {
@@ -142,7 +140,7 @@ internal class TaskRecyclerViewFragmentTest : FragmentTestCase<TaskRecyclerViewF
     @Test
     fun displaysEmptyFilteredHabitScreen() {
         fragment.taskType = TaskType.HABIT
-        fragment.taskFilterHelper.setActiveFilter(TaskType.HABIT, Task.FILTER_WEAK)
+        fragment.viewModel?.setActiveFilter(TaskType.HABIT, Task.FILTER_WEAK)
         launchFragment()
         screen {
             recycler {
@@ -155,7 +153,7 @@ internal class TaskRecyclerViewFragmentTest : FragmentTestCase<TaskRecyclerViewF
     @Test
     fun displaysEmptyFilteredDailyScreen() {
         fragment.taskType = TaskType.DAILY
-        fragment.taskFilterHelper.setActiveFilter(TaskType.DAILY, Task.FILTER_GRAY)
+        fragment.viewModel?.setActiveFilter(TaskType.DAILY, Task.FILTER_GRAY)
         launchFragment()
         screen {
             recycler {
@@ -168,7 +166,7 @@ internal class TaskRecyclerViewFragmentTest : FragmentTestCase<TaskRecyclerViewF
     @Test
     fun displaysEmptyFilteredTodoScreen() {
         fragment.taskType = TaskType.TODO
-        fragment.taskFilterHelper.tags = mutableListOf("test")
+        fragment.viewModel?.tags = mutableListOf("test")
         launchFragment()
         screen {
             recycler {