Fix flaky test

kizitonwose · Aug 3, 2024 · b1d6c4f · b1d6c4f
1 parent fbc3fb2
commit b1d6c4f
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/sample/src/androidTest/java/com/kizitonwose/calendar/sample/CalendarViewTest.kt b/sample/src/androidTest/java/com/kizitonwose/calendar/sample/CalendarViewTest.kt
@@ -236,11 +236,12 @@ class CalendarViewTest {
 
     @Test
     fun findVisibleDaysAndMonthsWorksOnVerticalOrientation() {
-        openExampleAt(1)
+        openExampleAt(7)
 
-        val calendarView = getView<CalendarView>(R.id.exTwoCalendar)
+        val calendarView = getView<CalendarView>(R.id.exEightCalendar)
 
         runOnMain {
+            calendarView.orientation = RecyclerView.VERTICAL
             // Scroll to a random date
             calendarView.scrollToDate(LocalDate.now().plusDays(120))
         }
@@ -266,11 +267,12 @@ class CalendarViewTest {
 
     @Test
     fun findVisibleDaysAndMonthsWorksOnHorizontalOrientation() {
-        openExampleAt(0)
+        openExampleAt(7)
 
-        val calendarView = getView<CalendarView>(R.id.exOneCalendar)
+        val calendarView = getView<CalendarView>(R.id.exEightCalendar)
 
         runOnMain {
+            calendarView.orientation = RecyclerView.HORIZONTAL
             // Scroll to a random date
             calendarView.scrollToDate(LocalDate.now().plusDays(120))
         }