#!/usr/bin/python
#
# 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.

'''A simple sanity test for Chrome.

This script logs in, ensures that the cryptohome is mounted,
and checks that the browser is functional.
'''

from __future__ import print_function

import argparse
import datetime
import logging
import sys

# This sets up import paths for autotest.
import common
from autotest_lib.client.bin import utils
from autotest_lib.client.common_lib.cros import arc, arc_common, chrome
from autotest_lib.client.common_lib.error import TestFail
from autotest_lib.client.cros import cryptohome


def Sanity(count=1):
    start = datetime.datetime.now()
    logging.info('Starting chrome and logging in.')
    is_arc_available = utils.is_arc_available()
    arc_mode = arc_common.ARC_MODE_ENABLED if is_arc_available else None
    for i in range(count):
        if count > 1:
            logging.info('Starting iteration %d.', i)
        with chrome.Chrome(arc_mode=arc_mode, num_tries=1) as cr:
            # Check that the cryptohome is mounted.
            # is_vault_mounted throws an exception if it fails.
            logging.info('Checking mounted cryptohome.')
            cryptohome.is_vault_mounted(user=cr.username, allow_fail=False)
            # Navigate to about:blank.
            tab = cr.browser.tabs[0]
            tab.Navigate('about:blank')

            # Evaluate some javascript.
            logging.info('Evaluating JavaScript.')
            if tab.EvaluateJavaScript('2+2') != 4:
                raise TestFail('EvaluateJavaScript failed')

            # ARC test.
            if is_arc_available:
                arc.wait_for_android_process('org.chromium.arc.intent_helper')
                arc.wait_for_adb_ready()
                logging.info('Android booted successfully.')
                if not arc.is_package_installed('android'):
                    raise TestFail('"android" system package was not listed by '
                                   'Package Manager.')

        if is_arc_available:
            utils.poll_for_condition(lambda: not arc.is_adb_connected(),
                                     timeout=15,
                                     desc='Android container still running '
                                          'after Chrome shutdown.')

        # Test incognito mode.
        with chrome.Chrome(logged_in=False):
            if not cryptohome.is_guest_vault_mounted():
                raise TestFail('Expected to find a guest vault mounted.')
        if cryptohome.is_guest_vault_mounted(allow_fail=True):
            raise TestFail('Expected to NOT find a guest vault mounted.')

        # Run Tast tests.
        utils.system('local_test_runner \'(!informational && !disabled && ' +
                     '("dep:chrome" || "dep:chrome_login") && ' +
                     '!"dep:android")\'')

        # GPU info collection via devtools SystemInfo.getInfo does not work
        # under mash due to differences in how the GPU process is configured
        # with mus hosting viz. http://crbug.com/669965
        mash_browser_args = ['--enable-features=Mash',
                             '--gpu-no-complete-info-collection']

        logging.info('Testing Chrome with Mash login.')
        with chrome.Chrome(extra_browser_args=mash_browser_args):
            logging.info('Chrome login with Mash succeeded.')

    elapsed = datetime.datetime.now() - start
    logging.info('Test succeeded in %s seconds.', elapsed.seconds)


def main(argv):
    '''The main function.'''

    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('--count', default=1,
                        help='Number of iterations of the test to run.')
    opts = parser.parse_args(argv)
    count = int(opts.count)
    Sanity(count)


if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))