client/site_tests/policy_DefaultSearchProvider/policy_DefaultSearchProvider.py

# Copyright 2019 The Chromium OS 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 time

from autotest_lib.client.common_lib import error
from autotest_lib.client.cros.enterprise import enterprise_policy_base
from autotest_lib.client.cros.input_playback import keyboard


class policy_DefaultSearchProvider(
        enterprise_policy_base.EnterprisePolicyTest):
    """
    Verify effects of the DefaultSearchProviderSearchURL and
    DefaultSearchProviderKeyword policy. When the
    DefaultSearchProviderSearchURL policy is set, the specified search url
    will be used when a value is entered in the omnibox. When the
    DefaultSearchProviderKeyword is set, the value will trigger the shortcut
    used in the omnibox to trigger the search for this provider.

    """
    version = 1

    def _search_check(self, case):
        """
        Open a new tab, use the omnibox as a search box, and check the URL.

        @param case: Value of the test being run.

        """
        self.ui.start_ui_root(self.cr)
        self.keyboard = keyboard.Keyboard()
        self.ui.doDefault_on_obj(name='Address and search bar')
        # The keys to be pressed for the test
        if case == 'Keyword':
            buttons = ['d', 'a', 'd', 'tab', 's', 's', 'enter']
            expected = '{}{}'.format(self.BASE_URL, 'ss')
        else:
            buttons = ['f', 's', 'w', 'enter']
            expected = '{}{}'.format(self.BASE_URL, 'fsw')

        # Enter the buttons
        for button in buttons:
            self.keyboard.press_key(button)

        tabFound = False
        startTime = time.time()
        while time.time() - startTime < 1:
            tabs = set([tab.GetUrl() for tab in self.cr.browser.tabs])
            if expected in tabs:
                tabFound = True
                break

        if not tabFound:
            raise error.TestFail(
                'Search not formated correctly. expected {} got {}'
                .format(expected, tabs))

    def run_once(self, case):
        """
        Setup and run the test configured for the specified test case.

        @param case: Name of the test case to run.

        """
        self.BASE_URL = 'https://fakeurl/search?q='
        POLICIES = {'DefaultSearchProviderEnabled': True,
                    'DefaultSearchProviderSearchURL':
                       '%s{searchTerms}' % (self.BASE_URL)}
        if case == 'Keyword':
            POLICIES['DefaultSearchProviderKeyword'] = 'dad'
        self.setup_case(user_policies=POLICIES)
        self._search_check(case)