client/site_tests/desktopui_UrlFetch/desktopui_UrlFetch.py

# Copyright (c) 2011 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.bin import utils
from autotest_lib.client.common_lib import error
from autotest_lib.client.cros import cros_ui_test, httpd


class desktopui_UrlFetch(cros_ui_test.UITest):
    version = 1


    def initialize(self, live=True):
        self._live = live
        super(desktopui_UrlFetch, self).initialize(creds='$default')
        if self._live:
            self._test_url = 'http://www.msn.com/'
            self._expected_title = 'MSN.com'
        else:
            self._test_url = 'http://localhost:8000/hello.html'
            self._expected_title = 'Hello World'
            self._testServer = httpd.HTTPListener(8000, docroot=self.bindir)
            self._testServer.run()


    def cleanup(self):
        if not self._live and hasattr(self, '_testServer'):
            self._testServer.stop()
        super(desktopui_UrlFetch, self).cleanup()


    def start_authserver(self):
        # Do nothing in live mode so that we do real login.
        # In non-live mode, setup local authserver which handles login.
        if not self._live:
            super(desktopui_UrlFetch, self).start_authserver()


    def run_once(self):
        import pyauto

        assert not self.pyauto.GetCookie(pyauto.GURL(self._test_url))

        def _OpenUrl():
            self.pyauto.NavigateToURL(self._test_url)
            tab_title = self.pyauto.GetActiveTabTitle()
            logging.info('Expected tab title: %s. Got: %s' % (
                self._expected_title, tab_title))
            return tab_title == self._expected_title

        # TODO(nirnimesh): This polling is a stop-gap fix and should not be
        # necessary.  Looks like the http server takes a while to startup.
        # Remove after http can guarantee its startup.
        # crosbug.com/20034
        utils.poll_for_condition(
            _OpenUrl,
            error.TestError('Timeout waiting to open %s' % self._test_url),
            timeout=60,
            sleep_interval=1)

        cookie = self.pyauto.GetCookie(pyauto.GURL(self._test_url))
        if not cookie:
            raise error.TestError('Expected cookie for %s' % self._test_url)