fix actions for a homepage

wordpress-mobile · Nov 3, 2023 · 72b44d1 · 72b44d1
1 parent bd77fc9
commit 72b44d1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/WordPress/Classes/ViewRelated/Pages/PageMenuViewModelTests.swift b/WordPress/Classes/ViewRelated/Pages/PageMenuViewModelTests.swift
@@ -106,10 +106,9 @@ final class PageMenuViewModelTests: CoreDataTestCase {
             .map { $0.buttons }
         let expectedButtons: [[AbstractPostButton]] = [
             [.view],
-            [.moveToDraft, .duplicate],
+            [.duplicate],
             [.blaze],
-            [.setParent, .setPostsPage],
-            [.trash]
+            [.setParent, .setPostsPage]
         ]
         expect(buttons).to(equal(expectedButtons))
     }

diff --git a/WordPress/Classes/ViewRelated/Post/PageMenuViewModel.swift b/WordPress/Classes/ViewRelated/Post/PageMenuViewModel.swift
@@ -49,7 +49,7 @@ final class PageMenuViewModel: AbstractPostMenuViewModel {
     private func createSecondarySection() -> AbstractPostButtonSection {
         var buttons = [AbstractPostButton]()
 
-        if page.status != .draft {
+        if page.status != .draft && isSiteHomepage {
             buttons.append(.moveToDraft)
         }
 
@@ -106,6 +106,10 @@ final class PageMenuViewModel: AbstractPostMenuViewModel {
     }
 
     private func createTrashSection() -> AbstractPostButtonSection {
+        guard !isSiteHomepage else {
+            return AbstractPostButtonSection(buttons: [])
+        }
+
         return AbstractPostButtonSection(buttons: [.trash])
     }
 }