# Copyright 2016 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 logging

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


class policy_SpellCheckServiceEnabled(
        enterprise_policy_base.EnterprisePolicyTest):
    """
    Test effect of SpellCheckServiceEnabled policy on Chrome OS behavior.

    This test verifies the behavior of Chrome OS for all valid values of the
    SpellCheckServiceEnabled user policy: True, False, and Not set. 'Not set'
    indicates no value, and will induce the default behavior that is seen by
    an unmanaged user: unchecked and user-editable.

    When True or Not set, spelling suggestiongs are given. When False, spelling
    suggestions are not given. When set either True or False, the setting is
    disabled, so users cannot change or override the setting. When not set
    users can change the setting.

    """
    version = 1

    POLICY_NAME = 'SpellCheckServiceEnabled'
    STARTUP_URLS = ['chrome://policy']
    SUPPORTING_POLICIES = {
        'BookmarkBarEnabled': True,
        'RestoreOnStartupURLs': STARTUP_URLS,
        'RestoreOnStartup': 4
    }
    TEST_CASES = {
        'True_Enable': True,
        'False_Disable': False,
        'NotSet_Editable': None
    }


    def _test_spelling_suggestions_enabled(self, policy_value):
        """
        Verify CrOS enforces the SpellCheckServiceEnabled policy.

        @param policy_value: policy value expected.

        """
        setting_pref = 'spellcheck.use_spelling_service'
        properties = self._get_settings_checkbox_properties(setting_pref)
        setting_label = properties[self.SETTING_LABEL]
        setting_is_checked = properties[self.SETTING_CHECKED]
        setting_is_disabled = properties[self.SETTING_DISABLED]
        logging.info('Check box "%s" status: checked=%s, disabled=%s',
                     setting_label, setting_is_checked, setting_is_disabled)

        # Setting checked if policy is True, unchecked if False.
        if policy_value == True and not setting_is_checked:
            raise error.TestFail('Spelling Suggestions should be checked.')
        if policy_value == False and setting_is_checked:
            raise error.TestFail('Spelling Suggestions should be unchecked.')

        # Setting is enabled if policy is Not set, disabled if True or False.
        if policy_value == None:
            if setting_is_disabled:
                raise error.TestFail('Spelling Suggestions should '
                                     'be editable.')
        else:
            if not setting_is_disabled:
                raise error.TestFail('Spelling Suggestions should not '
                                     'be editable.')


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

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

        """
        case_value = self.TEST_CASES[case]
        self.setup_case(self.POLICY_NAME, case_value, self.SUPPORTING_POLICIES)
        self._test_spelling_suggestions_enabled(case_value)