MD Bookmarks: Support policies for disabling bookmark editing

chrome/browser/ui/webui/md_bookmarks/md_bookmarks_browsertest.cc

Index: chrome/browser/ui/webui/md_bookmarks/md_bookmarks_browsertest.cc
diff --git a/chrome/browser/ui/webui/md_bookmarks/md_bookmarks_browsertest.cc b/chrome/browser/ui/webui/md_bookmarks/md_bookmarks_browsertest.cc
index fa9ce17a176a60958f79f7f7f3bee2b4e85a33c7..7dea1b0f14581fd4aa14f3bb2d346f9402b29553 100644
--- a/chrome/browser/ui/webui/md_bookmarks/md_bookmarks_browsertest.cc
+++ b/chrome/browser/ui/webui/md_bookmarks/md_bookmarks_browsertest.cc
@@ -8,6 +8,7 @@
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/common/pref_names.h"
+#include "components/bookmarks/common/bookmark_pref_names.h"
 #include "components/prefs/pref_service.h"
 
 MdBookmarksBrowserTest::MdBookmarksBrowserTest() {}
@@ -19,6 +20,10 @@ void MdBookmarksBrowserTest::RegisterMessages() {
       "testSetIncognito",
       base::Bind(&MdBookmarksBrowserTest::HandleSetIncognitoAvailability,
                  base::Unretained(this)));
+  web_ui()->RegisterMessageCallback(
+      "testSetCanEdit",
+      base::Bind(&MdBookmarksBrowserTest::HandleSetCanEditBookmarks,
+                 base::Unretained(this)));
 }
 
 void MdBookmarksBrowserTest::SetIncognitoAvailability(int availability) {
@@ -28,6 +33,11 @@ void MdBookmarksBrowserTest::SetIncognitoAvailability(int availability) {
       prefs::kIncognitoModeAvailability, availability);
 }
 
+void MdBookmarksBrowserTest::SetCanEditBookmarks(bool canEdit) {
+  browser()->profile()->GetPrefs()->SetBoolean(
+      bookmarks::prefs::kEditBookmarksEnabled, canEdit);
+}
+
 void MdBookmarksBrowserTest::HandleSetIncognitoAvailability(
     const base::ListValue* args) {
   AllowJavascript();
@@ -43,6 +53,21 @@ void MdBookmarksBrowserTest::HandleSetIncognitoAvailability(
   ResolveJavascriptCallback(*callback_id, base::Value());
 }
 
+void MdBookmarksBrowserTest::HandleSetCanEditBookmarks(
+    const base::ListValue* args) {
+  AllowJavascript();
+
+  ASSERT_EQ(2U, args->GetSize());
+  const base::Value* callback_id;
+  ASSERT_TRUE(args->Get(0, &callback_id));
+  bool pref_value;
+  ASSERT_TRUE(args->GetBoolean(1, &pref_value));
+
+  SetCanEditBookmarks(pref_value);
+
+  ResolveJavascriptCallback(*callback_id, base::Value());
+}
+
 content::WebUIMessageHandler* MdBookmarksBrowserTest::GetMockMessageHandler() {
   return this;
 }