Benoit Marty
2 weeks ago
committed by
Benoit Marty
1 changed files with 55 additions and 0 deletions
@ -0,0 +1,55 @@
@@ -0,0 +1,55 @@
|
||||
/* |
||||
* Copyright 2024 New Vector Ltd. |
||||
* |
||||
* SPDX-License-Identifier: AGPL-3.0-only |
||||
* Please see LICENSE in the repository root for full details. |
||||
*/ |
||||
|
||||
package io.element.android.features.messages.impl.timeline.protection |
||||
|
||||
import com.google.common.truth.Truth.assertThat |
||||
import io.element.android.libraries.matrix.test.AN_EVENT_ID |
||||
import io.element.android.libraries.preferences.api.store.AppPreferencesStore |
||||
import io.element.android.libraries.preferences.test.InMemoryAppPreferencesStore |
||||
import io.element.android.tests.testutils.WarmUpRule |
||||
import io.element.android.tests.testutils.test |
||||
import kotlinx.collections.immutable.persistentSetOf |
||||
import kotlinx.coroutines.test.runTest |
||||
import org.junit.Rule |
||||
import org.junit.Test |
||||
|
||||
class TimelineProtectionPresenterTest { |
||||
@get:Rule |
||||
val warmUpRule = WarmUpRule() |
||||
|
||||
@Test |
||||
fun `present - initial state`() = runTest { |
||||
val presenter = createPresenter() |
||||
presenter.test { |
||||
val initialState = awaitItem() |
||||
assertThat(initialState.protectionState).isEqualTo(ProtectionState.RenderAll) |
||||
} |
||||
} |
||||
|
||||
@Test |
||||
fun `present - protected`() = runTest { |
||||
val appPreferencesStore = InMemoryAppPreferencesStore(hideImagesAndVideos = true) |
||||
val presenter = createPresenter(appPreferencesStore) |
||||
presenter.test { |
||||
skipItems(1) |
||||
val initialState = awaitItem() |
||||
assertThat(initialState.protectionState).isEqualTo(ProtectionState.RenderOnly(persistentSetOf())) |
||||
// ShowContent with null should have no effect. |
||||
initialState.eventSink(TimelineProtectionEvent.ShowContent(eventId = null)) |
||||
initialState.eventSink(TimelineProtectionEvent.ShowContent(eventId = AN_EVENT_ID)) |
||||
val finalState = awaitItem() |
||||
assertThat(finalState.protectionState).isEqualTo(ProtectionState.RenderOnly(persistentSetOf(AN_EVENT_ID))) |
||||
} |
||||
} |
||||
|
||||
private fun createPresenter( |
||||
appPreferencesStore: AppPreferencesStore = InMemoryAppPreferencesStore(), |
||||
) = TimelineProtectionPresenter( |
||||
appPreferencesStore = appPreferencesStore, |
||||
) |
||||
} |
Loading…
Reference in new issue