blob: be88484bdaf98f7e802b714a294e93f94fe4fedb [file] [log] [blame]
Tatsuhisa Yamaguchic439ce62020-07-08 10:58:26 +00001# Copyright 2020 The Chromium OS Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5"""Test fetch_cipd."""
6
7from chromite.lib import cros_test_lib
8from chromite.scripts import fetch_cipd
9
10
11class FetchCipdTest(cros_test_lib.TestCase):
Alex Klein1699fab2022-09-08 08:46:06 -060012 """Tests for fetch_cipd script."""
Tatsuhisa Yamaguchic439ce62020-07-08 10:58:26 +000013
Alex Klein1699fab2022-09-08 08:46:06 -060014 def testParseCipdUri(self):
15 uri = "cipd://chromiumos/infra/tclint/linux-amd64:abcdefghijklm"
16 path, version = fetch_cipd.ParseCipdUri(uri)
17 self.assertEqual("chromiumos/infra/tclint/linux-amd64", path)
18 self.assertEqual("abcdefghijklm", version)
Tatsuhisa Yamaguchic439ce62020-07-08 10:58:26 +000019
Alex Klein1699fab2022-09-08 08:46:06 -060020 def testParseCipdUriWrongScheme(self):
21 uri = "gs://chromiumos/infra/tclint/linux-amd64:abcdefghijklm"
22 with self.assertRaises(ValueError):
23 fetch_cipd.ParseCipdUri(uri)
Tatsuhisa Yamaguchic439ce62020-07-08 10:58:26 +000024
Alex Klein1699fab2022-09-08 08:46:06 -060025 def testParseCipdUriNoVersion(self):
26 uri = "cipd://chromiumos/infra/tclint/linux-amd64"
27 with self.assertRaises(ValueError):
28 fetch_cipd.ParseCipdUri(uri)