fixed bug in UserLoginActivityTest

CMPUT301F17T11 · Nov 13, 2017 · a58d3e4 · a58d3e4
1 parent fe4dd6f
commit a58d3e4
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/app/src/androidTest/java/com/cmput301f17t11/cupofjava/UserLoginActivityTest.java b/app/src/androidTest/java/com/cmput301f17t11/cupofjava/UserLoginActivityTest.java
@@ -73,11 +73,12 @@ public void testLogin(){
 
 
         final ListView habitList = activity.getListView();
-        Habit habit = (Habit) habitList.getItemAtPosition(0);
+
+        Habit habit = (Habit) habitList.getItemAtPosition(habitList.getCount()-1);
         assertEquals("Disciplining Kids", habit.getHabitTitle());
         assertEquals("Good Nanny",habit.getHabitReason());
 
-        solo.clickInList(0);
+        solo.clickInList(habitList.getCount()-1);
         solo.assertCurrentActivity("wrong activity", HabitDetailViewActivity.class);
         assertTrue(solo.waitForText("Disciplining Kids", 1, 1000));
         assertTrue(solo.waitForText("Good Nanny", 1, 1000));
@@ -96,11 +97,12 @@ public void testLogin(){
 
 
         final ListView habitList2 = activity2.getListView();
-        Habit myhabit = (Habit) habitList.getItemAtPosition(0);
+
+        Habit myhabit = (Habit) habitList.getItemAtPosition(habitList.getCount()-1);
         assertEquals("Disciplining Kids", myhabit.getHabitTitle());
         assertEquals("Good Nanny",myhabit.getHabitReason());
 
-        solo.clickInList(0);
+        solo.clickInList(habitList.getCount()-1);
         solo.assertCurrentActivity("wrong activity", HabitDetailViewActivity.class);
         assertTrue(solo.waitForText("Disciplining Kids", 1, 1000));
         assertTrue(solo.waitForText("Good Nanny", 1, 1000));