From d5de113387ffef7ef39282862bb84cac7e4d1f5d Mon Sep 17 00:00:00 2001
From: kean <grebenyuk.alexander@gmail.com>
Date: Mon, 6 Nov 2023 11:38:43 -0500
Subject: [PATCH] Fix tests

---
 .../ViewRelated/Pages/PageMenuViewModelTests.swift     | 10 +++++-----
 .../Post/Views/PostCardStatusViewModelTests.swift      |  7 +++++--
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/WordPress/Classes/ViewRelated/Pages/PageMenuViewModelTests.swift b/WordPress/Classes/ViewRelated/Pages/PageMenuViewModelTests.swift
index 4ea70295de37..9f666889c8b5 100644
--- 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
index 9c81160653da..109d4c3b09d2 100644
--- 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))