#!/usr/bin/python # Copyright 2017 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 unittest import common from autotest_lib.server import utils from autotest_lib.server.hosts.cros_label import BoardLabel from autotest_lib.server.hosts.cros_label import ModelLabel from autotest_lib.server.hosts.cros_label import SparseCoverageLabel # pylint: disable=missing-docstring NON_UNI_LSB_RELEASE_OUTPUT = """ CHROMEOS_RELEASE_APPID={63A9F698-C1CA-4A75-95E7-6B90181B3718} CHROMEOS_BOARD_APPID={63A9F698-C1CA-4A75-95E7-6B90181B3718} CHROMEOS_CANARY_APPID={90F229CE-83E2-4FAF-8479-E368A34938B1} DEVICETYPE=CHROMEBOOK CHROMEOS_ARC_VERSION=4234098 CHROMEOS_ARC_ANDROID_SDK_VERSION=25 GOOGLE_RELEASE=9798.0.2017_08_02_1022 CHROMEOS_DEVSERVER=http://shapiroc3.bld.corp.google.com:8080 CHROMEOS_RELEASE_BOARD=pyro CHROMEOS_RELEASE_BUILD_NUMBER=9798 CHROMEOS_RELEASE_BRANCH_NUMBER=0 CHROMEOS_RELEASE_CHROME_MILESTONE=62 CHROMEOS_RELEASE_PATCH_NUMBER=2017_08_02_1022 CHROMEOS_RELEASE_TRACK=testimage-channel CHROMEOS_RELEASE_DESCRIPTION=9798.0.2017_08_02_1022 (Test Build) CHROMEOS_RELEASE_BUILD_TYPE=Test Build CHROMEOS_RELEASE_NAME=Chromium OS CHROMEOS_RELEASE_VERSION=9798.0.2017_08_02_1022 CHROMEOS_AUSERVER=http://someserver.bld.corp.google.com:8080/update """ UNI_LSB_RELEASE_OUTPUT = """ CHROMEOS_RELEASE_APPID={5A3AB642-2A67-470A-8F37-37E737A53CFC} CHROMEOS_BOARD_APPID={5A3AB642-2A67-470A-8F37-37E737A53CFC} CHROMEOS_CANARY_APPID={90F229CE-83E2-4FAF-8479-E368A34938B1} DEVICETYPE=CHROMEBOOK CHROMEOS_ARC_VERSION=4340813 CHROMEOS_ARC_ANDROID_SDK_VERSION=25 GOOGLE_RELEASE=9953.0.2017_09_18_1334 CHROMEOS_DEVSERVER=http://server.bld.corp.google.com:8080 CHROMEOS_RELEASE_BOARD=coral CHROMEOS_RELEASE_BUILD_NUMBER=9953 CHROMEOS_RELEASE_BRANCH_NUMBER=0 CHROMEOS_RELEASE_CHROME_MILESTONE=63 CHROMEOS_RELEASE_PATCH_NUMBER=2017_09_18_1334 CHROMEOS_RELEASE_TRACK=testimage-channel CHROMEOS_RELEASE_DESCRIPTION=9953.0.2017_09_18_1334 (Test Build) CHROMEOS_RELEASE_BUILD_TYPE=Test Build CHROMEOS_RELEASE_NAME=Chromium OS CHROMEOS_RELEASE_UNIBUILD=1 CHROMEOS_RELEASE_VERSION=9953.0.2017_09_18_1334 CHROMEOS_AUSERVER=http://server.bld.corp.google.com:8080/update CHROMEOS_RELEASE_MODELS=coral astronaut blue bruce lava nasher """ class MockCmd(object): """Simple mock command with base command and results""" def __init__(self, cmd, exit_status, stdout): self.cmd = cmd self.stdout = stdout self.exit_status = exit_status class MockAFEHost(utils.EmptyAFEHost): def __init__(self, labels=[], attributes={}): self.labels = labels self.attributes = attributes class MockHost(object): """Simple host for running mock'd host commands""" def __init__(self, labels, *args): self._afe_host = MockAFEHost(labels) self.mock_cmds = {c.cmd: c for c in args} def run(self, command, **kwargs): """Finds the matching result by command value""" return self.mock_cmds[command] class ModelLabelTests(unittest.TestCase): """Unit tests for ModelLabel""" def test_cros_config_succeeds(self): cat_lsb_release_output = """ CHROMEOS_RELEASE_BOARD=pyro CHROMEOS_RELEASE_UNIBUILD=1 """ host = MockHost([], MockCmd('cros_config / test-label', 0, 'coral\n'), MockCmd('cat /etc/lsb-release', 0, cat_lsb_release_output)) self.assertEqual(ModelLabel().generate_labels(host), ['coral']) def test_cros_config_fails_mosys_succeeds(self): cat_lsb_release_output = """ CHROMEOS_RELEASE_BOARD=pyro CHROMEOS_RELEASE_UNIBUILD=1 """ host = MockHost([], MockCmd('cros_config / test-label', 1, ''), MockCmd('mosys platform model', 0, 'coral\n'), MockCmd('cat /etc/lsb-release', 0, cat_lsb_release_output)) self.assertEqual(ModelLabel().generate_labels(host), ['coral']) def test_cros_config_fails_mosys_fails(self): cat_lsb_release_output = """ CHROMEOS_RELEASE_BOARD=pyro CHROMEOS_RELEASE_UNIBUILD=1 """ host = MockHost([], MockCmd('cros_config / test-label', 1, ''), MockCmd('mosys platform model', 1, ''), MockCmd('cat /etc/lsb-release', 0, cat_lsb_release_output)) self.assertEqual(ModelLabel().generate_labels(host), ['pyro']) def test_cros_config_only_used_for_unibuilds(self): cat_lsb_release_output = """ CHROMEOS_RELEASE_BOARD=pyro """ host = MockHost([], MockCmd('cat /etc/lsb-release', 0, cat_lsb_release_output)) self.assertEqual(ModelLabel().generate_labels(host), ['pyro']) def test_existing_label(self): host = MockHost(['model:existing']) self.assertEqual(ModelLabel().generate_labels(host), ['existing']) class BoardLabelTests(unittest.TestCase): """Unit tests for BoardLabel""" def test_new_label(self): cat_cmd = 'cat /etc/lsb-release' host = MockHost([], MockCmd(cat_cmd, 0, NON_UNI_LSB_RELEASE_OUTPUT)) self.assertEqual(BoardLabel().generate_labels(host), ['pyro']) def test_existing_label(self): host = MockHost(['board:existing']) self.assertEqual(BoardLabel().generate_labels(host), ['existing']) SPARSE_COVERAGE_TEMPLATE = """ CHROMEOS_RELEASE_APPID={{01906EA2-3EB2-41F1-8F62-F0B7120EFD2E}} CHROMEOS_BOARD_APPID={{01906EA2-3EB2-41F1-8F62-F0B7120EFD2E}} CHROMEOS_CANARY_APPID={{90F229CE-83E2-4FAF-8479-E368A34938B1}} DEVICETYPE=CHROMEBOOK CHROMEOS_ARC_VERSION=4473730 CHROMEOS_ARC_ANDROID_SDK_VERSION=25 GOOGLE_RELEASE={build}.{branch}.{patch} CHROMEOS_DEVSERVER= CHROMEOS_RELEASE_BUILDER_PATH=eve-{builder}/R64-{build}.{branch}.{patch} CHROMEOS_RELEASE_BUILD_NUMBER={build} CHROMEOS_RELEASE_BRANCH_NUMBER={branch} CHROMEOS_RELEASE_CHROME_MILESTONE=64 CHROMEOS_RELEASE_PATCH_NUMBER={patch} CHROMEOS_RELEASE_TRACK=testimage-channel CHROMEOS_RELEASE_DESCRIPTION={build}.{branch}.{patch} (Official Build) dev-channel eve test CHROMEOS_RELEASE_BUILD_TYPE=Official Build CHROMEOS_RELEASE_NAME=Chrome OS CHROMEOS_RELEASE_BOARD=eve CHROMEOS_RELEASE_VERSION={build}.{branch}.{patch} CHROMEOS_AUSERVER=https://tools.google.com/service/update2 """ def _cat_output(builder, build, branch, patch): return SPARSE_COVERAGE_TEMPLATE.format( builder=builder, build=build, branch=branch, patch=patch) class SparseCoverageLabelTests(unittest.TestCase): """Unit tests for SparseCoverageLabel""" _mock_data = [ # Master canary build - sparse. (('release', '60000', '0', '0'), {2, 3, 5}), (('release', '60001', '0', '0'), {}), (('release', '60002', '0', '0'), {2}), (('release', '60003', '0', '0'), {3}), (('release', '60004', '0', '0'), {2}), (('release', '60005', '0', '0'), {5}), (('release', '60006', '0', '0'), {2, 3}), # Branch canary build - not sparse. (('release', '60000', '1', '0'), {2, 3, 5}), (('release', '60001', '1', '0'), {2, 3, 5}), (('release', '60002', '1', '0'), {2, 3, 5}), (('release', '60003', '1', '0'), {2, 3, 5}), (('release', '60004', '1', '0'), {2, 3, 5}), (('release', '60005', '1', '0'), {2, 3, 5}), (('release', '60006', '1', '0'), {2, 3, 5}), # A CQ/PFQ like build - not sparse. (('release', '60000', '0', '0-rc4'), {2, 3, 5}), (('release', '60001', '0', '0-rc4'), {2, 3, 5}), (('release', '60002', '0', '0-rc4'), {2, 3, 5}), (('release', '60003', '0', '0-rc4'), {2, 3, 5}), (('release', '60004', '0', '0-rc4'), {2, 3, 5}), (('release', '60005', '0', '0-rc4'), {2, 3, 5}), (('release', '60006', '0', '0-rc4'), {2, 3, 5}), # Not a release build - not sparse. (('chrome-pfq', '60000', '0', '0'), {2, 3, 5}), (('chrome-pfq', '60001', '0', '0'), {2, 3, 5}), (('chrome-pfq', '60002', '0', '0'), {2, 3, 5}), (('chrome-pfq', '60003', '0', '0'), {2, 3, 5}), (('chrome-pfq', '60004', '0', '0'), {2, 3, 5}), (('chrome-pfq', '60005', '0', '0'), {2, 3, 5}), (('chrome-pfq', '60006', '0', '0'), {2, 3, 5}), ] def test_coverage_label(self): cat_cmd = 'cat /etc/lsb-release' for release, short_labels in self._mock_data: host = MockHost([], MockCmd(cat_cmd, 0, _cat_output(*release))) expected_labels = set( 'sparse_coverage_%d' % l for l in short_labels) generated_labels = set(SparseCoverageLabel().generate_labels(host)) self.assertEqual(expected_labels, generated_labels) if __name__ == '__main__': unittest.main()