1 files changed, 51 insertions, 0 deletions
diff --git a/chromium/chrome/browser/ui/webui/ntp/cookie_controls_handler_unittest.cc b/chromium/chrome/browser/ui/webui/ntp/cookie_controls_handler_unittest.cc
new file mode 100644
index 00000000000..e269fef1122
--- /dev/null
+++ b/chromium/chrome/browser/ui/webui/ntp/cookie_controls_handler_unittest.cc
@@ -0,0 +1,51 @@
+// Copyright 2019 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/ntp/cookie_controls_handler.h"
+
+#include "base/values.h"
+#include "chrome/test/base/chrome_render_view_host_test_harness.h"
+#include "components/content_settings/core/browser/cookie_settings.h"
+#include "components/content_settings/core/common/pref_names.h"
+#include "components/prefs/pref_service.h"
+#include "content/public/test/test_web_ui.h"
+
+class CookieControlsHandlerTest : public ChromeRenderViewHostTestHarness {
+ public:
+  void SetUp() override {
+    ChromeRenderViewHostTestHarness::SetUp();
+    web_ui_.set_web_contents(web_contents());
+    handler_ = std::make_unique<CookieControlsHandler>();
+    handler_->set_web_ui(&web_ui_);
+  }
+
+  void TearDown() override {
+    handler_->set_web_ui(nullptr);
+    ChromeRenderViewHostTestHarness::TearDown();
+  }
+
+ protected:
+  content::TestWebUI web_ui_;
+  std::unique_ptr<CookieControlsHandler> handler_;
+};
+
+TEST_F(CookieControlsHandlerTest, HandleCookieControlsToggleChanged) {
+  EXPECT_EQ(
+      static_cast<int>(content_settings::CookieControlsMode::kIncognitoOnly),
+      Profile::FromWebUI(&web_ui_)->GetPrefs()->GetInteger(
+          prefs::kCookieControlsMode));
+  base::ListValue args_false;
+  args_false.AppendBoolean(false);
+  handler_->HandleCookieControlsToggleChanged(&args_false);
+  EXPECT_EQ(static_cast<int>(content_settings::CookieControlsMode::kOff),
+            Profile::FromWebUI(&web_ui_)->GetPrefs()->GetInteger(
+                prefs::kCookieControlsMode));
+  base::ListValue args_true;
+  args_true.AppendBoolean(true);
+  handler_->HandleCookieControlsToggleChanged(&args_true);
+  EXPECT_EQ(
+      static_cast<int>(content_settings::CookieControlsMode::kIncognitoOnly),
+      Profile::FromWebUI(&web_ui_)->GetPrefs()->GetInteger(
+          prefs::kCookieControlsMode));
+}