From 0318ebf127adece22b38b83dc4e6439e97806fa5 Mon Sep 17 00:00:00 2001 From: malinajirka Date: Thu, 1 Aug 2019 13:01:25 +0200 Subject: [PATCH] Fix PostListItemUiStateHelper unit tests --- .../posts/PostListItemUiStateHelperTest.kt | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/WordPress/src/test/java/org/wordpress/android/viewmodel/posts/PostListItemUiStateHelperTest.kt b/WordPress/src/test/java/org/wordpress/android/viewmodel/posts/PostListItemUiStateHelperTest.kt index bf92ff18c471..ed3a25885825 100644 --- a/WordPress/src/test/java/org/wordpress/android/viewmodel/posts/PostListItemUiStateHelperTest.kt +++ b/WordPress/src/test/java/org/wordpress/android/viewmodel/posts/PostListItemUiStateHelperTest.kt @@ -238,16 +238,9 @@ class PostListItemUiStateHelperTest { post = createPostModel(status = POST_STATE_TRASHED) ) - assertThat(state.actions[0].buttonType).isEqualTo(PostListButtonType.BUTTON_EDIT) - assertThat(state.actions[1].buttonType).isEqualTo(PostListButtonType.BUTTON_VIEW) - assertThat(state.actions[2].buttonType).isEqualTo(PostListButtonType.BUTTON_MORE) - assertThat(state.actions).hasSize(3) - - assertThat((state.actions[2] as MoreItem).actions[0].buttonType) - .isEqualTo(PostListButtonType.BUTTON_DELETE) - assertThat((state.actions[2] as MoreItem).actions[1].buttonType) - .isEqualTo(PostListButtonType.BUTTON_MOVE_TO_DRAFT) - assertThat((state.actions[2] as MoreItem).actions).hasSize(2) + assertThat(state.actions[0].buttonType).isEqualTo(PostListButtonType.BUTTON_MOVE_TO_DRAFT) + assertThat(state.actions[1].buttonType).isEqualTo(PostListButtonType.BUTTON_DELETE_PERMANENTLY) + assertThat(state.actions).hasSize(2) } @Test @@ -537,7 +530,7 @@ class PostListItemUiStateHelperTest { uploadStatus = createUploadStatus(uploadError = UploadError(MediaError(MediaErrorType.GENERIC_ERROR))) ) assertThat(state.actions[0].buttonType).isEqualTo(PostListButtonType.BUTTON_EDIT) - assertThat(state.actions[1].buttonType).isEqualTo(PostListButtonType.BUTTON_SUBMIT) + assertThat(state.actions[1].buttonType).isEqualTo(PostListButtonType.BUTTON_RETRY) assertThat(state.actions[2].buttonType).isEqualTo(PostListButtonType.BUTTON_DELETE) assertThat(state.actions).hasSize(3) }