From 3045a8581ad645f62c267a44d8a0be1e79248ff9 Mon Sep 17 00:00:00 2001
From: Maxime NATUREL <46314705+mnaturel@users.noreply.github.com>
Date: Thu, 26 Jan 2023 13:58:30 +0100
Subject: [PATCH] Adding unit tests for DefaultGetLoadedPollsStatusTaskTest

---
 .../DefaultGetLoadedPollsStatusTaskTest.kt    | 121 ++++++++++++++++++
 1 file changed, 121 insertions(+)
 create mode 100644 matrix-sdk-android/src/test/java/org/matrix/android/sdk/internal/session/room/poll/DefaultGetLoadedPollsStatusTaskTest.kt

diff --git a/matrix-sdk-android/src/test/java/org/matrix/android/sdk/internal/session/room/poll/DefaultGetLoadedPollsStatusTaskTest.kt b/matrix-sdk-android/src/test/java/org/matrix/android/sdk/internal/session/room/poll/DefaultGetLoadedPollsStatusTaskTest.kt
new file mode 100644
index 0000000000..2f58973eca
--- /dev/null
+++ b/matrix-sdk-android/src/test/java/org/matrix/android/sdk/internal/session/room/poll/DefaultGetLoadedPollsStatusTaskTest.kt
@@ -0,0 +1,121 @@
+/*
+ * Copyright (c) 2023 The Matrix.org Foundation C.I.C.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.matrix.android.sdk.internal.session.room.poll
+
+import io.mockk.unmockkAll
+import kotlinx.coroutines.ExperimentalCoroutinesApi
+import kotlinx.coroutines.test.runTest
+import org.amshove.kluent.shouldBeEqualTo
+import org.junit.After
+import org.junit.Test
+import org.matrix.android.sdk.api.session.room.poll.LoadedPollsStatus
+import org.matrix.android.sdk.internal.database.model.PollHistoryStatusEntity
+import org.matrix.android.sdk.internal.database.model.PollHistoryStatusEntityFields
+import org.matrix.android.sdk.test.fakes.FakeMonarchy
+import org.matrix.android.sdk.test.fakes.givenEqualTo
+import org.matrix.android.sdk.test.fakes.givenFindFirst
+
+private const val A_ROOM_ID = "room-id"
+/**
+ * 2023/01/26
+ */
+private const val A_TIMESTAMP = 1674737619290L
+
+@OptIn(ExperimentalCoroutinesApi::class)
+internal class DefaultGetLoadedPollsStatusTaskTest {
+
+    private val fakeMonarchy = FakeMonarchy()
+
+    private val defaultGetLoadedPollsStatusTask = DefaultGetLoadedPollsStatusTask(
+            monarchy = fakeMonarchy.instance,
+    )
+
+    @After
+    fun tearDown() {
+        unmockkAll()
+    }
+
+    @Test
+    fun `given poll history status exists in db with an oldestTimestamp reached when execute then the computed status is returned`() = runTest {
+        // Given
+        val params = givenTaskParams()
+        // 2023/01/20
+        val oldestTimestampReached = 1674169200000
+        val pollHistoryStatus = aPollHistoryStatusEntity(
+                isEndOfPollsBackward = false,
+                oldestTimestampReached = oldestTimestampReached,
+        )
+        fakeMonarchy.fakeRealm
+                .givenWhere<PollHistoryStatusEntity>()
+                .givenEqualTo(PollHistoryStatusEntityFields.ROOM_ID, A_ROOM_ID)
+                .givenFindFirst(pollHistoryStatus)
+        val expectedStatus = LoadedPollsStatus(
+                canLoadMore = true,
+                daysSynced = 6,
+                hasCompletedASyncBackward = true,
+        )
+
+        // When
+        val result = defaultGetLoadedPollsStatusTask.execute(params)
+
+        // Then
+        result shouldBeEqualTo expectedStatus
+    }
+
+    @Test
+    fun `given poll history status exists in db and no oldestTimestamp reached when execute then the computed status is returned`() = runTest {
+        // Given
+        val params = givenTaskParams()
+        val pollHistoryStatus = aPollHistoryStatusEntity(
+                isEndOfPollsBackward = false,
+                oldestTimestampReached = null,
+        )
+        fakeMonarchy.fakeRealm
+                .givenWhere<PollHistoryStatusEntity>()
+                .givenEqualTo(PollHistoryStatusEntityFields.ROOM_ID, A_ROOM_ID)
+                .givenFindFirst(pollHistoryStatus)
+        val expectedStatus = LoadedPollsStatus(
+                canLoadMore = true,
+                daysSynced = 0,
+                hasCompletedASyncBackward = false,
+        )
+
+        // When
+        val result = defaultGetLoadedPollsStatusTask.execute(params)
+
+        // Then
+        result shouldBeEqualTo expectedStatus
+    }
+
+    private fun givenTaskParams(): GetLoadedPollsStatusTask.Params {
+        return GetLoadedPollsStatusTask.Params(
+                roomId = A_ROOM_ID,
+                currentTimestampMs = A_TIMESTAMP,
+        )
+    }
+
+    private fun aPollHistoryStatusEntity(
+            isEndOfPollsBackward: Boolean,
+            oldestTimestampReached: Long?,
+    ): PollHistoryStatusEntity {
+        return PollHistoryStatusEntity(
+                roomId = A_ROOM_ID,
+                isEndOfPollsBackward = isEndOfPollsBackward,
+                oldestTimestampTargetReachedMs = oldestTimestampReached,
+        )
+    }
+}