You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
2.0 KiB
61 lines
2.0 KiB
#!/usr/bin/python2.7
|
|
# 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.cros import provision
|
|
|
|
_CROS_VERSION_SAMPLES = [
|
|
'cave-release/R57-9030.0.0',
|
|
'grunt-llvm-next-toolchain-tryjob/R69-10851.0.0-b2726174'
|
|
'eve-tot-chrome-pfq-informational/R69-10822.0.0-b2700960',
|
|
]
|
|
_CROS_ANDROID_VERSION_SAMPLES = [
|
|
'git_nyc-mr1-arc/cheets_arm-user/4866647',
|
|
'git_nyc-mr1-arc/cheets_arm-user/P6244267',
|
|
'git_nyc-mr1-arc/cheets_x86-user/P6256537',
|
|
]
|
|
|
|
|
|
class ActionTestCase(unittest.TestCase):
|
|
"""Tests for Action functions."""
|
|
#pylint:disable=missing-docstring
|
|
|
|
def test__get_label_action_with_keyval_label(self):
|
|
got = provision._get_label_action('cros-version:foo')
|
|
self.assertEqual(got, provision._Action('cros-version', 'foo'))
|
|
|
|
def test__get_label_action_with_plain_label(self):
|
|
got = provision._get_label_action('webcam')
|
|
self.assertEqual(got, provision._Action('webcam', None))
|
|
|
|
def test__get_label_action_with_empty_string(self):
|
|
got = provision._get_label_action('')
|
|
self.assertEqual(got, provision._Action('', None))
|
|
|
|
|
|
class ImageParsingTests(unittest.TestCase):
|
|
"""Unit tests for `provision.get_version_label_prefix()`."""
|
|
|
|
def _do_test_prefixes(self, expected, version_samples):
|
|
for v in version_samples:
|
|
prefix = provision.get_version_label_prefix(v)
|
|
self.assertEqual(prefix, expected)
|
|
|
|
def test_cros_prefix(self):
|
|
"""Test handling of Chrome OS version strings."""
|
|
self._do_test_prefixes(provision.CROS_VERSION_PREFIX,
|
|
_CROS_VERSION_SAMPLES)
|
|
|
|
def test_cros_android_prefix(self):
|
|
"""Test handling of Chrome OS version strings."""
|
|
self._do_test_prefixes(provision.CROS_ANDROID_VERSION_PREFIX,
|
|
_CROS_ANDROID_VERSION_SAMPLES)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|