scripts/fetch_cipd_unittest.py

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

"""Test fetch_cipd."""

from chromite.lib import cros_test_lib
from chromite.scripts import fetch_cipd


class FetchCipdTest(cros_test_lib.TestCase):
    """Tests for fetch_cipd script."""

    def testParseCipdUri(self) -> None:
        uri = "cipd://chromiumos/infra/tclint/linux-amd64:abcdefghijklm"
        path, version = fetch_cipd.ParseCipdUri(uri)
        self.assertEqual("chromiumos/infra/tclint/linux-amd64", path)
        self.assertEqual("abcdefghijklm", version)

    def testParseCipdUriWrongScheme(self) -> None:
        uri = "gs://chromiumos/infra/tclint/linux-amd64:abcdefghijklm"
        with self.assertRaises(ValueError):
            fetch_cipd.ParseCipdUri(uri)

    def testParseCipdUriNoVersion(self) -> None:
        uri = "cipd://chromiumos/infra/tclint/linux-amd64"
        with self.assertRaises(ValueError):
            fetch_cipd.ParseCipdUri(uri)

    def testParseCipdUriGitVersion(self) -> None:
        uri = "cipd://chromiumos/infra/tclint/linux-amd64:git_version:1234"
        path, version = fetch_cipd.ParseCipdUri(uri)
        self.assertEqual("chromiumos/infra/tclint/linux-amd64", path)
        self.assertEqual("git_version:1234", version)