commit c439ce6767a0d8252a41b284530f35a4b2582664
author Tatsuhisa Yamaguchi <yamaguchi@chromium.org> Wed Jul 08 10:58:26 2020 +0000
committer Commit Bot <commit-bot@chromium.org> Mon Jul 13 20:05:24 2020 +0000
tree 713a907cdc9b2a75228bbc9b7ce4a3cd4c466348
parent 1ecb643b284347131c100c1a4267f10a66cdff05

Add script to fetch a file from CIPD.

BUG=chromium:1096407
TEST=manually verified with crrev.com/c/2251682
TEST=run_pytest scripts/fetch_cipd_unittest.py

Change-Id: I7a725f4c789f9b4935c3be91ec0e901209731770
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/2287020
Tested-by: Tatsuhisa Yamaguchi <yamaguchi@chromium.org>
Commit-Queue: Mike Frysinger <vapier@chromium.org>
Reviewed-by: Mike Frysinger <vapier@chromium.org>

scripts/fetch_cipd_unittest.py

diff --git a/scripts/fetch_cipd_unittest.py b/scripts/fetch_cipd_unittest.py
new file mode 100644
index 0000000..6652f4a
--- /dev/null
+++ b/scripts/fetch_cipd_unittest.py
@@ -0,0 +1,28 @@
+# Copyright 2020 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.
+
+"""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):
+    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):
+    uri = 'gs://chromiumos/infra/tclint/linux-amd64:abcdefghijklm'
+    with self.assertRaises(ValueError):
+      fetch_cipd.ParseCipdUri(uri)
+
+  def testParseCipdUriNoVersion(self):
+    uri = 'cipd://chromiumos/infra/tclint/linux-amd64'
+    with self.assertRaises(ValueError):
+      fetch_cipd.ParseCipdUri(uri)