From 05edfc97c608221cc7139d3ee34e5bdf476ebbff Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=BAben=20Sousa?= <ruben@rubensousa.com>
Date: Wed, 24 May 2023 16:35:51 +0100
Subject: [PATCH] Fix unit tests

---
 .../test/layoutmanager/mock/LayoutInfoMock.kt             | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/dpadrecyclerview/src/test/java/com/rubensousa/dpadrecyclerview/test/layoutmanager/mock/LayoutInfoMock.kt b/dpadrecyclerview/src/test/java/com/rubensousa/dpadrecyclerview/test/layoutmanager/mock/LayoutInfoMock.kt
index 1f651f6e..ad84215b 100644
--- a/dpadrecyclerview/src/test/java/com/rubensousa/dpadrecyclerview/test/layoutmanager/mock/LayoutInfoMock.kt
+++ b/dpadrecyclerview/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 }