client/site_tests/desktopui_UrlFetchWithChromeDriver/desktopui_UrlFetchWithChromeDriver.py

# Copyright (c) 2013 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

import common
from autotest_lib.client.bin import test
from autotest_lib.client.common_lib import error
from autotest_lib.client.common_lib.cros import chromedriver
from autotest_lib.client.cros import httpd


class desktopui_UrlFetchWithChromeDriver(test.test):
    """Test fetching url and cookie using Chrome Driver."""
    version = 1

    def initialize(self, live=True):
        """Initialize the test.

        @param live: Set to True to access external websites. Otherwise, test
                     with localhost http server. Default value is set to True.
        """
        self._live = live
        super(desktopui_UrlFetchWithChromeDriver, self).initialize()

        if self._live:
            self._test_url = 'http://www.msn.com/'
            self._expected_title = 'MSN.com'
            self._domain = '.msn.com'
        else:
            self._test_url = 'http://localhost:8000/hello.html'
            self._expected_title = 'Hello World'
            self._domain = 'localhost'
            self._testServer = httpd.HTTPListener(8000, docroot=self.bindir)
            self._testServer.run()


    def cleanup(self):
        """Clean up the test environment, e.g., stop local http server."""
        if not self._live and hasattr(self, '_testServer'):
            self._testServer.stop()
        super(desktopui_UrlFetchWithChromeDriver, self).cleanup()


    def run_once(self):
        """Run the test code."""
        with chromedriver.chromedriver() as chromedriver_instance:
            driver = chromedriver_instance.driver
            driver.delete_all_cookies()
            driver.get(self._test_url)

            logging.info('Expected tab title: %s. Got: %s',
                         self._expected_title, driver.title)
            if driver.title != self._expected_title:
                raise error.TestError('Getting title failed, got title: %s'
                                      % driver.title)

            cookie_found = any([cookie for cookie in
                                driver.get_cookies()
                                if cookie['domain'] == self._domain])
            if not cookie_found:
                raise error.TestError('Expected cookie for %s' % self._test_url)