MD Bookmarks: Disable 'Open in Incognito Window' when Incognito is disabled

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

+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/ui/webui/md_bookmarks/md_bookmarks_browsertest.h"
+
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/common/pref_names.h"
+#include "components/prefs/pref_service.h"
+
+MdBookmarksBrowserTest::MdBookmarksBrowserTest() {}
+
+MdBookmarksBrowserTest::~MdBookmarksBrowserTest() {}
+
+void MdBookmarksBrowserTest::RegisterMessages() {
+  web_ui()->RegisterMessageCallback(
+      "testSetIncognito",
+      base::Bind(&MdBookmarksBrowserTest::HandleSetIncognitoAvailability,
+                 base::Unretained(this)));
+}
+
+void MdBookmarksBrowserTest::SetIncognitoAvailability(int availability) {
+  browser()->profile()->GetPrefs()->SetInteger(
+      prefs::kIncognitoModeAvailability, availability);
+}
+
+void MdBookmarksBrowserTest::HandleSetIncognitoAvailability(
+    const base::ListValue* args) {
+  AllowJavascript();
+
+  ASSERT_EQ(2U, args->GetSize());
+  const base::Value* callback_id;
+  ASSERT_TRUE(args->Get(0, &callback_id));
+  int pref_value;
+  ASSERT_TRUE(args->GetInteger(1, &pref_value));
+

[calamity, 2017/05/29 06:52:28]

Assert that this is >=0 and <IncognitoModePrefs::AVAILBILITY_NUM_TYPES.

[tsergeant, 2017/05/30 00:18:21]

Done.

+  SetIncognitoAvailability(pref_value);
+
+  CallJavascriptFunction("cr.webUIResponse", *callback_id, base::Value(true));

[calamity, 2017/05/29 06:52:27]

Use ResolveJavascriptCallback.

[tsergeant, 2017/05/30 00:18:21]

Done.

+}
+
+content::WebUIMessageHandler* MdBookmarksBrowserTest::GetMockMessageHandler() {
+  return this;
+}