server/cros/provision_unittest.py

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