From b1195cd20c0dc1f88c69fb4d3528a4aa0def63b2 Mon Sep 17 00:00:00 2001
From: Quentin Hibon <qh.public@yahoo.com>
Date: Thu, 19 Nov 2020 11:29:49 +0100
Subject: [PATCH] add sort toggle test

---
 .../list/ListHabitsMenuBehaviorTest.java      | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/android/uhabits-core/src/test/java/org/isoron/uhabits/core/ui/screens/habits/list/ListHabitsMenuBehaviorTest.java b/android/uhabits-core/src/test/java/org/isoron/uhabits/core/ui/screens/habits/list/ListHabitsMenuBehaviorTest.java
index 8b694f991..8751d2d1d 100644
--- a/android/uhabits-core/src/test/java/org/isoron/uhabits/core/ui/screens/habits/list/ListHabitsMenuBehaviorTest.java
+++ b/android/uhabits-core/src/test/java/org/isoron/uhabits/core/ui/screens/habits/list/ListHabitsMenuBehaviorTest.java
@@ -54,6 +54,9 @@ public class ListHabitsMenuBehaviorTest extends BaseUnitTest
     @Captor
     private ArgumentCaptor<HabitList.Order> orderCaptor;
 
+    @Captor
+    private ArgumentCaptor<HabitList.Order> secondaryOrderCaptor;
+
     @Override
     public void setUp() throws Exception
     {
@@ -134,9 +137,25 @@ public class ListHabitsMenuBehaviorTest extends BaseUnitTest
     @Test
     public void testOnSortStatus()
     {
+        when(adapter.getPrimaryOrder()).thenReturn(BY_NAME_ASC);
+
         behavior.onSortByStatus();
         verify(adapter).setPrimaryOrder(orderCaptor.capture());
+        verify(adapter).setSecondaryOrder(secondaryOrderCaptor.capture());
         assertThat(orderCaptor.getValue(), equalTo(BY_STATUS_ASC));
+        assertThat(secondaryOrderCaptor.getValue(), equalTo(BY_NAME_ASC));
+    }
+
+    @Test
+    public void testOnSortStatusToggle()
+    {
+        when(adapter.getPrimaryOrder()).thenReturn(BY_STATUS_ASC);
+
+        behavior.onSortByStatus();
+
+        verify(adapter).setPrimaryOrder(orderCaptor.capture());
+        verify(adapter, never()).setSecondaryOrder(any());
+        assertThat(orderCaptor.getValue(), equalTo(BY_STATUS_DESC));
     }
 
     @Test