65 lines
2.4 KiB
Python
Executable File
65 lines
2.4 KiB
Python
Executable File
# 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)
|
|
|