Fix tests

wordpress-mobile · Nov 6, 2023 · d5de113 · d5de113
1 parent 6364b50
commit d5de113
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/WordPress/Classes/ViewRelated/Pages/PageMenuViewModelTests.swift b/WordPress/Classes/ViewRelated/Pages/PageMenuViewModelTests.swift
@@ -26,7 +26,7 @@ final class PageMenuViewModelTests: CoreDataTestCase {
             [.view],
             [.moveToDraft, .duplicate],
             [.blaze],
-            [.setParent, .setHomepage, .setPostsPage],
+            [.setParent, .setHomepage, .setPostsPage, .settings],
             [.trash]
         ]
         expect(buttons).to(equal(expectedButtons))
@@ -53,7 +53,7 @@ final class PageMenuViewModelTests: CoreDataTestCase {
         let expectedButtons: [[AbstractPostButton]] = [
             [.view],
             [.moveToDraft, .duplicate],
-            [.setParent, .setHomepage, .setPostsPage],
+            [.setParent, .setHomepage, .setPostsPage, .settings],
             [.trash]
         ]
         expect(buttons).to(equal(expectedButtons))
@@ -80,7 +80,7 @@ final class PageMenuViewModelTests: CoreDataTestCase {
         let expectedButtons: [[AbstractPostButton]] = [
             [.view],
             [.moveToDraft, .duplicate],
-            [.setParent, .setHomepage, .setPostsPage],
+            [.setParent, .setHomepage, .setPostsPage, .settings],
             [.trash]
         ]
         expect(buttons).to(equal(expectedButtons))
@@ -108,7 +108,7 @@ final class PageMenuViewModelTests: CoreDataTestCase {
             [.view],
             [.duplicate],
             [.blaze],
-            [.setParent, .setPostsPage]
+            [.setParent, .setPostsPage, .settings]
         ]
         expect(buttons).to(equal(expectedButtons))
     }
@@ -135,7 +135,7 @@ final class PageMenuViewModelTests: CoreDataTestCase {
             [.view],
             [.moveToDraft, .duplicate],
             [.blaze],
-            [.setParent, .setHomepage, .setRegularPage],
+            [.setParent, .setHomepage, .setRegularPage, .settings],
             [.trash]
         ]
         expect(buttons).to(equal(expectedButtons))

diff --git a/WordPress/WordPressTest/ViewRelated/Post/Views/PostCardStatusViewModelTests.swift b/WordPress/WordPressTest/ViewRelated/Post/Views/PostCardStatusViewModelTests.swift
@@ -20,7 +20,7 @@ class PostCardStatusViewModelTests: CoreDataTestCase {
             [.view],
             [.moveToDraft, .duplicate, .share],
             [.blaze],
-            [.stats, .comments],
+            [.stats, .comments, .settings],
             [.trash]
         ]
         expect(buttons).to(equal(expectedButtons))
@@ -41,7 +41,7 @@ class PostCardStatusViewModelTests: CoreDataTestCase {
         let expectedButtons: [[AbstractPostButton]] = [
             [.view],
             [.moveToDraft, .duplicate, .share],
-            [.stats, .comments],
+            [.stats, .comments, .settings],
             [.trash]
         ]
         expect(buttons).to(equal(expectedButtons))
@@ -62,6 +62,7 @@ class PostCardStatusViewModelTests: CoreDataTestCase {
         let expectedButtons: [[AbstractPostButton]] = [
             [.view],
             [.moveToDraft, .duplicate, .share],
+            [.settings],
             [.trash]
         ]
         expect(buttons).to(equal(expectedButtons))
@@ -81,6 +82,7 @@ class PostCardStatusViewModelTests: CoreDataTestCase {
         let expectedButtons: [[AbstractPostButton]] = [
             [.view],
             [.duplicate, .publish],
+            [.settings],
             [.trash]
         ]
         expect(buttons).to(equal(expectedButtons))
@@ -100,6 +102,7 @@ class PostCardStatusViewModelTests: CoreDataTestCase {
         let expectedButtons: [[AbstractPostButton]] = [
             [.view],
             [.moveToDraft],
+            [.settings],
             [.trash]
         ]
         expect(buttons).to(equal(expectedButtons))