app home viewmodel test

AppHomeViewModelTest.kt

@Test
fun testIsTodoInHome() {
  whenever(firebaseRemoteConfig.getString(AbTestingUtils.FirebaseParamKeys.AB_TEST_HOME_SCREEN_VERSION_ANDROID))
    .thenReturn(AbTestingUtils.FirebaseParamValues.HOME_TODO)
  
  assertThat(viewModel.isTodoInHome()).isTrue()

  whenever(firebaseRemoteConfig.getString(AbTestingUtils.FirebaseParamKeys.AB_TEST_HOME_SCREEN_VERSION_ANDROID))
    .thenReturn(AbTestingUtils.FirebaseParamValues.HOME_DEFAULT)

  assertThat(viewModel.isTodoInHome())
    .isFalse()
}

@Test
fun testUpdateTodoCards() {
  viewModel.segmentsViewModels.add(getDummyAppSegmentViewModel1())
  viewModel.segmentsViewModels.add(getDummyTodoCardViewModel1())
  viewModel.segmentsViewModels.add(getDummyTodoCardViewModel1())
  viewModel.segmentsViewModels.add(getDummyTodoCardViewModel1())

  assertThat(viewModel.segmentsViewModels.size)
  .isEqualTo(4)

  val todoCards = ObservableArrayList<ItemViewModel>()
  todoCards.add(getDummyTodoCardViewModel1())

  viewModel.updateTodoCards(todoCards)

  assertThat(viewModel.segmentsViewModels.size)
  .isEqualTo(2)

}