@@ -54,7 +54,7 @@ import org.wordpress.android.ui.reader.discover.ReaderPostCardActionType.LIKE
5454import org.wordpress.android.ui.reader.discover.ReaderPostCardActionType.REBLOG
5555import org.wordpress.android.ui.reader.discover.interests.TagUiState
5656import org.wordpress.android.ui.reader.reblog.ReblogUseCase
57- import org.wordpress.android.ui.reader.repository.ReaderAnnouncementRepository
57+ import org.wordpress.android.ui.reader.utils.ReaderAnnouncementHelper
5858import org.wordpress.android.ui.reader.repository.ReaderDiscoverCommunication
5959import org.wordpress.android.ui.reader.repository.ReaderDiscoverCommunication.Error.NetworkUnavailable
6060import org.wordpress.android.ui.reader.repository.ReaderDiscoverCommunication.Started
@@ -139,7 +139,7 @@ class ReaderDiscoverViewModelTest : BaseUnitTest() {
139139 private lateinit var readerImprovementsFeatureConfig: ReaderImprovementsFeatureConfig
140140
141141 @Mock
142- private lateinit var readerAnnouncementRepository : ReaderAnnouncementRepository
142+ private lateinit var mReaderAnnouncementHelper : ReaderAnnouncementHelper
143143
144144 private val fakeDiscoverFeed = ReactiveMutableLiveData <ReaderDiscoverCards >()
145145 private val fakeCommunicationChannel = MutableLiveData <Event <ReaderDiscoverCommunication >>()
@@ -164,7 +164,7 @@ class ReaderDiscoverViewModelTest : BaseUnitTest() {
164164 displayUtilsWrapper,
165165 getFollowedTagsUseCase,
166166 readerImprovementsFeatureConfig,
167- readerAnnouncementRepository ,
167+ mReaderAnnouncementHelper ,
168168 testDispatcher(),
169169 testDispatcher()
170170 )
@@ -408,7 +408,7 @@ class ReaderDiscoverViewModelTest : BaseUnitTest() {
408408 @Test
409409 fun `if Announcement does not exist then ReaderAnnouncementCardUiState will not be present` () = test {
410410 // Arrange
411- whenever(readerAnnouncementRepository .hasReaderAnnouncement()).thenReturn(false )
411+ whenever(mReaderAnnouncementHelper .hasReaderAnnouncement()).thenReturn(false )
412412 val uiStates = init (autoUpdateFeed = false ).uiStates
413413 // Act
414414 fakeDiscoverFeed.value = createDummyReaderCardsList() // mock finished loading
@@ -421,8 +421,8 @@ class ReaderDiscoverViewModelTest : BaseUnitTest() {
421421 @Test
422422 fun `if Announcement exists then ReaderAnnouncementCardUiState will be present` () = test {
423423 // Arrange
424- whenever(readerAnnouncementRepository .hasReaderAnnouncement()).thenReturn(true )
425- whenever(readerAnnouncementRepository .getReaderAnnouncementItems()).thenReturn(mock())
424+ whenever(mReaderAnnouncementHelper .hasReaderAnnouncement()).thenReturn(true )
425+ whenever(mReaderAnnouncementHelper .getReaderAnnouncementItems()).thenReturn(mock())
426426 val uiStates = init (autoUpdateFeed = false ).uiStates
427427 // Act
428428 fakeDiscoverFeed.value = createDummyReaderCardsList() // mock finished loading
@@ -435,8 +435,8 @@ class ReaderDiscoverViewModelTest : BaseUnitTest() {
435435 @Test
436436 fun `clicking done on ReaderAnnouncementCardUiState dismisses and updates the ContentUiState` () = test {
437437 // Arrange
438- whenever(readerAnnouncementRepository .hasReaderAnnouncement()).thenReturn(true )
439- whenever(readerAnnouncementRepository .getReaderAnnouncementItems()).thenReturn(mock())
438+ whenever(mReaderAnnouncementHelper .hasReaderAnnouncement()).thenReturn(true )
439+ whenever(mReaderAnnouncementHelper .getReaderAnnouncementItems()).thenReturn(mock())
440440 val uiStates = init (autoUpdateFeed = false ).uiStates
441441
442442 fakeDiscoverFeed.value = createDummyReaderCardsList() // mock finished loading
@@ -447,7 +447,7 @@ class ReaderDiscoverViewModelTest : BaseUnitTest() {
447447 announcementCard.onDoneClick()
448448
449449 // Assert
450- verify(readerAnnouncementRepository ).dismissReaderAnnouncement()
450+ verify(mReaderAnnouncementHelper ).dismissReaderAnnouncement()
451451
452452 val newContentUiState = uiStates.last() as ContentUiState
453453 assertThat(newContentUiState.cards.first())
0 commit comments