Fix unit tests

rubensousa · May 24, 2023 · 05edfc9 · 05edfc9
1 parent 2c516ba
commit 05edfc9
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/...w/src/test/java/com/rubensousa/dpadrecyclerview/test/layoutmanager/mock/LayoutInfoMock.kt b/...w/src/test/java/com/rubensousa/dpadrecyclerview/test/layoutmanager/mock/LayoutInfoMock.kt
@@ -33,11 +33,19 @@ internal class LayoutInfoMock(
     var isInfinite = false
     var reverseLayout = false
     var isScrolling = false
+    var isLoopingAllowed = false
+    var isLoopingStart = false
     var isScrollingToTarget = false
     var hasCreatedFirstItem = false
     var hasCreatedLastItem = false
 
     init {
+        every { mock.isLoopingStart }.answers { isLoopingStart }
+        every { mock.isLoopingAllowed }.answers { isLoopingAllowed }
+        every { mock.updateLoopingState(any(), any()) }.answers {
+            isLoopingStart = it.invocation.args[0] as Boolean
+            isLoopingAllowed = it.invocation.args[1] as Boolean
+        }
         every { mock.isVertical() }.answers { realInstance.isVertical() }
         every { mock.isHorizontal() }.answers { realInstance.isHorizontal() }
         every { mock.isInfinite() }.answers { isInfinite }