Add a preference to control Windows desktop search redirection.

chrome/browser/ui/startup/startup_browser_creator_win_unittest.cc

Index: chrome/browser/ui/startup/startup_browser_creator_win_unittest.cc
diff --git a/chrome/browser/ui/startup/startup_browser_creator_win_unittest.cc b/chrome/browser/ui/startup/startup_browser_creator_win_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..9100ad81b06b20ad0387082264e3af78dc01abd0
--- /dev/null
+++ b/chrome/browser/ui/startup/startup_browser_creator_win_unittest.cc
@@ -0,0 +1,89 @@
+// Copyright 2015 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/startup/startup_browser_creator.h"
+
+#include <vector>
+
+#include "base/command_line.h"
+#include "base/feature_list.h"
+#include "base/macros.h"
+#include "base/prefs/pref_service.h"
+#include "chrome/browser/search_engines/template_url_service_factory.h"
+#include "chrome/browser/search_engines/template_url_service_factory_test_util.h"
+#include "chrome/test/base/testing_profile.h"
+#include "components/search_engines/desktop_search_win.h"
+#include "components/search_engines/util.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "url/gurl.h"
+
+class StartupBrowserCreatorWinUnitTest : public testing::Test {

[Peter Kasting, 2015/12/03 05:41:39]

Nit: I would probably remove "Unit" from this name; I think it's less common to
use that within the names of test classes.

[fdoray, 2015/12/03 17:16:17]

Done.

+ public:
+  StartupBrowserCreatorWinUnitTest() {}
+
+ protected:
+  void SetWindowsDesktopSearchFeatureEnabled(bool enabled) {
+    base::FeatureList::ClearInstanceForTesting();
+    scoped_ptr<base::FeatureList> feature_list(new base::FeatureList);
+    if (enabled) {
+      feature_list->InitializeFromCommandLine(
+          kWindowsDesktopSearchRedirectionFeature.name, std::string());
+    }
+    base::FeatureList::SetInstance(std::move(feature_list));
+  }
+
+ private:
+  content::TestBrowserThreadBundle thread_bundle_;
+
+  DISALLOW_COPY_AND_ASSIGN(StartupBrowserCreatorWinUnitTest);
+};
+
+TEST_F(StartupBrowserCreatorWinUnitTest,
+       GetURLsFromCommandLineWithDesktopSearchURL) {
+  const char kWindowsDesktopSearchRedirectionPref[] =
+      "windows_desktop_search_redirection";
+  const char kDesktopSearchURL[] =
+      "https://www.bing.com/search?q=keyword&form=WNSGPH";
+
+  TestingProfile profile;
+  TemplateURLServiceFactoryTestUtil template_url_service_factory_test_util(
+      &profile);
+
+  base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
+  command_line.AppendArg(kDesktopSearchURL);
+
+  // Expected vectors of URLs.
+  const std::vector<GURL> desktop_search_url_vector({GURL(kDesktopSearchURL)});
+  const std::vector<GURL> default_search_url_vector(
+      {GetDefaultSearchURLForSearchTerms(
+          TemplateURLServiceFactory::GetForProfile(&profile), L"keyword")});
+
+  // Preference unset, feature enabled.
+  SetWindowsDesktopSearchFeatureEnabled(true);
+  EXPECT_EQ(desktop_search_url_vector,
+            StartupBrowserCreator::GetURLsFromCommandLine(
+                command_line, base::FilePath(), &profile));
+
+  // Preference set to disabled, feature enabled.
+  profile.GetPrefs()->SetBoolean(kWindowsDesktopSearchRedirectionPref, false);
+  SetWindowsDesktopSearchFeatureEnabled(true);
+  EXPECT_EQ(desktop_search_url_vector,
+            StartupBrowserCreator::GetURLsFromCommandLine(
+                command_line, base::FilePath(), &profile));
+
+  // Preference set to enabled, feature enabled.
+  profile.GetPrefs()->SetBoolean(kWindowsDesktopSearchRedirectionPref, true);
+  SetWindowsDesktopSearchFeatureEnabled(true);
+  EXPECT_EQ(default_search_url_vector,
+            StartupBrowserCreator::GetURLsFromCommandLine(
+                command_line, base::FilePath(), &profile));
+
+  // Preference set to enabled, feature disabled.
+  profile.GetPrefs()->SetBoolean(kWindowsDesktopSearchRedirectionPref, true);
+  SetWindowsDesktopSearchFeatureEnabled(false);
+  EXPECT_EQ(desktop_search_url_vector,
+            StartupBrowserCreator::GetURLsFromCommandLine(
+                command_line, base::FilePath(), &profile));
+}