Chromium Code Reviews| 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 |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..7ce9c1e8c7ac8a8b31c64bb57d527e9c5a2a195e |
| --- /dev/null |
| +++ b/chrome/browser/ui/webui/md_bookmarks/md_bookmarks_browsertest.cc |
| @@ -0,0 +1,45 @@ |
| +// 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::A
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; |
| +} |