| #!/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() |