From e745b959caea80dd0a1d5743ef5d0282bb9a798f Mon Sep 17 00:00:00 2001
From: "Alinson S. Xavier" <git@axavier.org>
Date: Thu, 18 Mar 2021 21:29:51 -0500
Subject: [PATCH] Fix multiple selection

---
 .../java/org/isoron/uhabits/acceptance/HabitsTest.kt  | 11 +++++++++++
 .../habits/list/views/HabitCardListController.kt      |  2 +-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/uhabits-android/src/androidTest/java/org/isoron/uhabits/acceptance/HabitsTest.kt b/uhabits-android/src/androidTest/java/org/isoron/uhabits/acceptance/HabitsTest.kt
index ed04edcea..171a9ed41 100644
--- a/uhabits-android/src/androidTest/java/org/isoron/uhabits/acceptance/HabitsTest.kt
+++ b/uhabits-android/src/androidTest/java/org/isoron/uhabits/acceptance/HabitsTest.kt
@@ -199,4 +199,15 @@ class HabitsTest : BaseUserInterfaceTest() {
         verifyDisplaysText("Track time")
         verifyDisplaysText("Wake up early")
     }
+
+    @Test
+    @Throws(Exception::class)
+    fun shouldAllowMultipleSelection() {
+        launchApp()
+        verifyShowsScreen(LIST_HABITS)
+        verifyDisplaysText("Track time")
+        longClickText("Wake up early")
+        longClickText("Track time")
+        verifyDisplaysText("2")
+    }
 }
diff --git a/uhabits-android/src/main/java/org/isoron/uhabits/activities/habits/list/views/HabitCardListController.kt b/uhabits-android/src/main/java/org/isoron/uhabits/activities/habits/list/views/HabitCardListController.kt
index ad2ebb270..1cb3462f9 100644
--- a/uhabits-android/src/main/java/org/isoron/uhabits/activities/habits/list/views/HabitCardListController.kt
+++ b/uhabits-android/src/main/java/org/isoron/uhabits/activities/habits/list/views/HabitCardListController.kt
@@ -156,7 +156,7 @@ class HabitCardListController @Inject constructor(
         }
 
         private fun notifyListener() {
-            if (activeMode === SelectionMode())
+            if (activeMode is SelectionMode)
                 selectionMenu.get().onSelectionChange()
             else
                 selectionMenu.get().onSelectionFinish()