xref: /aosp_15_r20/external/toolchain-utils/crosperf/download_images_buildid_test.py (revision 760c253c1ed00ce9abd48f8546f08516e57485fe)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2014 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Test translation of xbuddy names."""


import argparse
import sys

import download_images


# On May 1, 2014:
# latest         : lumpy-release/R34-5500.132.0
# latest-beta    : lumpy-release/R35-5712.43.0
# latest-official: lumpy-release/R36-5814.0.0
# latest-dev     : lumpy-release/R36-5814.0.0
# latest-canary  : lumpy-release/R36-5814.0.0


class ImageDownloaderBuildIDTest(object):
    """Test translation of xbuddy names."""

    def __init__(self):
        parser = argparse.ArgumentParser()
        parser.add_argument(
            "-c",
            "--chromeos_root",
            dest="chromeos_root",
            help="Directory containing ChromeOS root.",
        )

        options = parser.parse_known_args(sys.argv[1:])[0]
        if options.chromeos_root is None:
            self._usage(parser, "--chromeos_root must be set")
        self.chromeos_root = options.chromeos_root
        self.tests_passed = 0
        self.tests_run = 0
        self.tests_failed = 0

    def _usage(self, parser, message):
        print("ERROR: " + message)
        parser.print_help()
        sys.exit(0)

    def print_test_status(self):
        print("----------------------------------------\n")
        print("Tests attempted: %d" % self.tests_run)
        print("Tests passed:    %d" % self.tests_passed)
        print("Tests failed:    %d" % self.tests_failed)
        print("\n----------------------------------------")

    def assert_failure(self, msg):
        print("Assert failure: %s" % msg)
        self.print_test_status()
        sys.exit(1)

    def assertIsNotNone(self, arg, arg_name):
        if arg is None:
            self.tests_failed = self.tests_failed + 1
            self.assert_failure("%s is not None" % arg_name)

    def assertNotEqual(self, arg1, arg2, arg1_name, arg2_name):
        if arg1 == arg2:
            self.tests_failed = self.tests_failed + 1
            self.assert_failure(
                "%s is not NotEqual to %s" % (arg1_name, arg2_name)
            )

    def assertEqual(self, arg1, arg2, arg1_name, arg2_name):
        if arg1 != arg2:
            self.tests_failed = self.tests_failed + 1
            self.assert_failure(
                "%s is not Equal to %s" % (arg1_name, arg2_name)
            )

    def test_one_id(self, downloader, test_id, result_string, exact_match):
        print("Translating '%s'" % test_id)
        self.tests_run = self.tests_run + 1

        result = downloader.GetBuildID(self.chromeos_root, test_id)
        # Verify that we got a build id back.
        self.assertIsNotNone(result, "result")

        # Verify that the result either contains or exactly matches the
        # result_string, depending on the exact_match argument.
        if exact_match:
            self.assertEqual(result, result_string, "result", result_string)
        else:
            self.assertNotEqual(
                result.find(result_string), -1, "result.find", "-1"
            )
        self.tests_passed = self.tests_passed + 1

    def test_get_build_id(self):
        """Test that the actual translating of xbuddy names is working properly."""
        downloader = download_images.ImageDownloader(log_level="quiet")

        self.test_one_id(
            downloader, "remote/lumpy/latest-dev", "lumpy-release/R", False
        )
        self.test_one_id(
            downloader,
            "remote/trybot-lumpy-release-afdo-use/R35-5672.0.0-b86",
            "trybot-lumpy-release-afdo-use/R35-5672.0.0-b86",
            True,
        )
        self.test_one_id(
            downloader,
            "remote/lumpy-release/R35-5672.0.0",
            "lumpy-release/R35-5672.0.0",
            True,
        )
        self.test_one_id(
            downloader, "remote/lumpy/latest-dev", "lumpy-release/R", False
        )
        self.test_one_id(
            downloader, "remote/lumpy/latest-official", "lumpy-release/R", False
        )
        self.test_one_id(
            downloader, "remote/lumpy/latest-beta", "lumpy-release/R", False
        )

        self.print_test_status()


if __name__ == "__main__":
    tester = ImageDownloaderBuildIDTest()
    tester.test_get_build_id()