blob: 5331697b2aebc182867f37894fa98cc2d2e184b9 [file] [log] [blame]
# Copyright (c) 2014 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 collections
import logging
from autotest_lib.client.common_lib import error
from autotest_lib.client.common_lib import global_config
from autotest_lib.client.common_lib import utils
from autotest_lib.client.common_lib.cros import dev_server
from import ping_runner
from autotest_lib.server import hosts
from autotest_lib.server import site_linux_router
from autotest_lib.server import test
from autotest_lib.server.cros import autoupdater
from autotest_lib.server.cros import dnsname_mangler
from import wifi_test_context_manager
# Stable versions come from the builders.
# The builder version is used to build the URL of the corresponding image
# in Google Storage.
# The image version is a line from /etc/lsb-release in the corresponding image.
StableVersion = collections.namedtuple('StableVersion',
['builder_version', 'release_version'])
class network_WiFi_UpdateRouter(test.test):
"""Updates a router to the most recent stable version.
This is not a test per se, since it does not test client behavior. However
it is advantageous to write this as a test so that we can schedule it to
run periodically via the same infrastructure we use to run tests.
Note that this test is very much patterned on provision_AutoUpdate.
version = 1
'stumpy': StableVersion('trybot-stumpy-test-ap/R47-7424.0.0-b10',
'panther': StableVersion('trybot-panther-test-ap/R47-7424.0.0-b10',
'whirlwind': StableVersion('trybot-whirlwind-test-ap-tryjob/'
'gale': StableVersion('gale-test-ap-tryjob/R74-11895.11.0-b3621467',
def get_release_version(self, host):
result ='cat /etc/lsb-release')
for line in result.stdout.splitlines():
if line.startswith('CHROMEOS_RELEASE_VERSION='):
return line.split('=', 1)[1]
def get_update_url(self, ds_url, image):
CONFIG = global_config.global_config
IMAGE_URL_PATTERN = CONFIG.get_config_value(
'CROS', 'image_url_pattern', type=str)
return IMAGE_URL_PATTERN % (ds_url, image)
def warmup(self, raw_cmdline_args):
"""Possibly parse the router hostname from the commandline.
@param raw_cmdline_args raw input from autotest.
cmdline_args = utils.args_to_dict(raw_cmdline_args)'Running wifi test with commandline arguments: %r',
self._router_hostname_from_cmdline = cmdline_args.get(
wifi_test_context_manager.WiFiTestContextManager. \
def run_once(self, host, is_pcap=False):
"""Update router / packet capture associated with host.
@param host DUT connected to AP/Pcap that needs update
if is_pcap:
device_hostname = dnsname_mangler.get_pcap_addr(
device_hostname = site_linux_router.build_router_hostname(
ping_helper = ping_runner.PingRunner()
if not ping_helper.simple_ping(device_hostname):
# Pcap devices aren't always present. Just claim Not Applicable if
# we couldn't find it.
e = error.TestNAError if is_pcap else error.TestError
raise e('%s not found / is down.' % device_hostname)
# Use CrosHost for all router/pcap hosts and avoid host detection.
# Host detection would use JetstreamHost for Whirlwind routers.
# JetstreamHost assumes ap-daemons are running.
# Testbed routers run the testbed-ap profile with no ap-daemons.
# TODO(ecgh): Fix testbed-ap JetstreamHost detection.
device_host = hosts.create_host(device_hostname,
def update_device(self, device_host):
"""Update router and pcap associated with host.
@param device_host: router / pcap host object
@param device_board: router / pcap board name
device_board = device_host.get_board().split(':', 1)[1]
desired = self.STABLE_VERSIONS.get(device_board, None)
if desired is None:
raise error.TestFail('No stable version found for %s with board=%s.'
% (device_host.hostname, device_board))'Checking whether %s is at the latest stable version: %s',
device_host.hostname, desired.release_version)
current_release_version = self.get_release_version(device_host)
if desired.release_version == current_release_version:
raise error.TestNAError('%s is already at latest version %s.' %
desired.release_version))'Updating %s to image %s from %s',
device_host.hostname, desired.release_version,
current_release_version)'Staging artifacts.')
ds = dev_server.ImageServer.resolve(desired.builder_version,
['full_payload', 'stateful'])
except dev_server.DevServerException as e:
raise error.TestFail(str(e))
url = self.get_update_url(ds.url(), desired.builder_version)
autoupdater.ChromiumOSUpdater(url, host=device_host).run_update()