blob: 41105bf7f51bd13a167ebea9f14d2498cf985a3b [file] [log] [blame]
# -*- coding: utf-8 -*-
# Copyright 2016 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.
"""Unit tests for the auto_update_util module."""
from __future__ import print_function
import sys
from chromite.lib import auto_update_util
from chromite.lib import cros_test_lib
assert sys.version_info >= (3, 6), 'This module requires Python 3.6+'
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):
auto_update_util.VersionMatch(build, release), i == j,
'Build version %s should%s match release version %s.' %
(build, '' if i == j else ' not', release))