Add enterprise E2E test for DefaultSearchProvider policies

Change-Id: Ie0e5103e4c952ef65110855de248e80a5a54468a
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/1725254Reviewed-by: Julian Pastarmov <pastarmovj@chromium.org>
Commit-Queue: Xiang Ji <jxiang@google.com>
Cr-Commit-Position: refs/heads/master@{#682736}

chrome/browser/policy/e2e_test/tests/__init__.py

@@ -3,6 +3,7 @@
 # found in the LICENSE file.
 
 from apps_shortcut.apps_shortcut import *
+from default_search_provider.default_search_provider import *
 from force_google_safe_search.force_google_safe_search import *
 from homepage.homepage import *
 from restore_on_startup.restore_on_startup import *

chrome/browser/policy/e2e_test/tests/default_search_provider/default_search_provider.py

+# 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.
+
+import os
+from chrome_ent_test.ent_tests import ChromeEnterpriseTestCase
+from chrome_ent_test.infra.core import environment, before_all, test
+
+
+@environment(file="../policy_test.asset.textpb")
+class DefaultSearchProviderTest(ChromeEnterpriseTestCase):
+  """Test the DefaultSearchProviderEnabled,
+              DefaultSearchProviderName,
+              DefaultSearchProviderSearchURL
+
+    https://cloud.google.com/docs/chrome-enterprise/policies/?policy=DefaultSearchProviderEnabled
+    https://cloud.google.com/docs/chrome-enterprise/policies/?policy=DefaultSearchProviderName
+    https://cloud.google.com/docs/chrome-enterprise/policies/?policy=DefaultSearchProviderSearchURL
+
+    """
+
+  @before_all
+  def setup(self):
+    self.InstallChrome('client2012')
+    self.EnableUITest('client2012')
+
+  def _get_search_url(self, instance_name):
+    local_dir = os.path.dirname(os.path.abspath(__file__))
+    output = self.RunUITest(
+        instance_name,
+        os.path.join(local_dir, 'default_search_provider_webdriver.py'))
+    return output
+
+  @test
+  def test_default_search_provider_bing(self):
+    self.SetPolicy('win2012-dc', 'DefaultSearchProviderEnabled', 1, 'DWORD')
+    self.SetPolicy('win2012-dc', 'DefaultSearchProviderName', 'Bing', 'String')
+    self.SetPolicy('win2012-dc', 'DefaultSearchProviderSearchURL',
+                   '"https://www.bing.com/search?q={searchTerms}"', 'String')
+    self.RunCommand('client2012', 'gpupdate /force')
+
+    output = self._get_search_url('client2012')
+    self.assertIn('www.bing.com', output)
+
+  @test
+  def test_default_search_provider_yahoo(self):
+    self.SetPolicy('win2012-dc', 'DefaultSearchProviderEnabled', 1, 'DWORD')
+    self.SetPolicy('win2012-dc', 'DefaultSearchProviderName', 'Yahoo', 'String')
+    self.SetPolicy('win2012-dc', 'DefaultSearchProviderSearchURL',
+                   '"https://search.yahoo.com/search?p={searchTerms}"',
+                   'String')
+    self.RunCommand('client2012', 'gpupdate /force')
+
+    output = self._get_search_url('client2012')
+    self.assertIn('search.yahoo.com', output)
+
+  @test
+  def test_default_search_provider_disabled(self):
+    self.SetPolicy('win2012-dc', 'DefaultSearchProviderEnabled', 0, 'DWORD')
+    self.RunCommand('client2012', 'gpupdate /force')
+
+    output = self._get_search_url('client2012')
+    self.assertIn('http://anything', output)

chrome/browser/policy/e2e_test/tests/default_search_provider/default_search_provider_webdriver.py

+# 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.
+
+from selenium import webdriver
+from pywinauto.application import Application
+import os
+
+# Set up ChromeDriver
+options = webdriver.ChromeOptions()
+options.add_argument("--force-renderer-accessibility")
+
+os.environ["CHROME_LOG_FILE"] = r"c:\temp\chrome_log.txt"
+
+driver = webdriver.Chrome(
+    "C:/ProgramData/chocolatey/lib/chromedriver/tools/chromedriver.exe",
+    chrome_options=options,
+    service_args=["--verbose", r"--log-path=c:\temp\chromedriver.log"])
+
+try:
+  app = Application(backend="uia")
+  app.connect(title_re='.*Chrome|.*Chromium')
+  omnibox = app.top_window() \
+            .child_window(title="Address and search bar", control_type="Edit")
+  omnibox.set_edit_text('anything').type_keys('{ENTER}')
+  print driver.current_url
+except Exception as error:
+  print error
+finally:
+  driver.quit()