lib/auto_update_util_unittest.py

# Copyright 2016 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for the auto_update_util module."""

from chromite.lib import auto_update_util
from chromite.lib import cros_test_lib


class VersionMatchUnittest(cros_test_lib.TestCase):
    """Test version_match function."""

    def testVersionMatch(self):
        """Test version_match function."""
        canary_build = "lumpy-release/R43-6803.0.0"
        canary_release = "6803.0.0"
        cq_build = "lumpy-release/R43-6803.0.0-rc1"
        cq_release = "6803.0.0-rc1"
        trybot_paladin_build = "trybot-lumpy-paladin/R43-6803.0.0-b123"
        trybot_paladin_release = "6803.0.2015_03_12_2103"
        trybot_pre_cq_build = "trybot-wifi-pre-cq/R43-7000.0.0-b36"
        trybot_pre_cq_release = "7000.0.2016_03_12_2103"
        trybot_toolchain_build = (
            "trybot-sentry-llvm-toolchain/R56-8885.0.0-b943"
        )
        trybot_toolchain_release = "8885.0.2016_10_10_1432"

        builds = [
            canary_build,
            cq_build,
            trybot_paladin_build,
            trybot_pre_cq_build,
            trybot_toolchain_build,
        ]
        releases = [
            canary_release,
            cq_release,
            trybot_paladin_release,
            trybot_pre_cq_release,
            trybot_toolchain_release,
        ]
        for i, build in enumerate(builds):
            for j, release in enumerate(releases):
                self.assertEqual(
                    auto_update_util.VersionMatch(build, release),
                    i == j,
                    "Build version %s should%s match release version %s."
                    % (build, "" if i == j else " not", release),
                )