commit 193ab2e6e20dceb59d368b51d59bc22e98897de1
author Bob Haarman <inglorion@chromium.org> Mon Jul 10 20:02:09 2023 +0000
committer Chromeos LUCI <chromeos-scoped@luci-project-accounts.iam.gserviceaccount.com> Tue Jul 11 22:04:35 2023 +0000
tree 7f5acb0460bf2a64be788e633684da6ee2fb0e55
parent 41f0ecfe23a62584090374792fc0b43508e3a6fa

scripts: upload_prebuilts: Handle version prefixes other than "chroot-"

Previously, upload_prebuilts assumed that the version prefix was "chroot-",
and chopped off the first 7 characters of the prefix+version string to
get the version alone. This updates the code to be a bit more robust,
instead looking for the expected format of the version and stripping
the prefix that precedes it.

BUG=b:289990183
TEST=./run_tests scripts/upload_prebuilts_unittest.py

Change-Id: Id36dd69c0e7529885151f26f41d86b5ce573de5b
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/4669938
Commit-Queue: Bob Haarman <inglorion@chromium.org>
Tested-by: Bob Haarman <inglorion@chromium.org>
Reviewed-by: Lizzy Presland <zland@google.com>

scripts/upload_prebuilts_unittest.py

diff --git a/scripts/upload_prebuilts_unittest.py b/scripts/upload_prebuilts_unittest.py
index 47dbaf8..09bb2b1 100644
--- a/scripts/upload_prebuilts_unittest.py
+++ b/scripts/upload_prebuilts_unittest.py
@@ -667,6 +667,8 @@
 class TestSdk(cros_test_lib.MockTestCase):
     """Test logic related to uploading SDK binaries"""
 
+    VERSION_PREFIX = "cros-"
+
     def setUp(self):
         self.PatchObject(
             prebuilt,
@@ -706,7 +708,7 @@
             False,
             "x86-foo",
             [],
-            "chroot-1234",
+            f"{self.VERSION_PREFIX}-1234.08.01.5678",
             report={},
         )
 
@@ -719,7 +721,7 @@
     ):
         """Make sure we can upload just an SDK tarball"""
         tar = "sdk.tar.xz"
-        ver = "1234"
+        ver = "1234.08.01.5678"
         vtar = "cros-sdk-%s.tar.xz" % ver
 
         upload_calls = [
@@ -805,6 +807,12 @@
         )
 
 
+class TestSdkBuildToolchain(TestSdk):
+    """Like TestSdk, but uses a different version prefix."""
+
+    VERSION_PREFIX = "build_toolchain-"
+
+
 @pytest.mark.parametrize(
     "extra_args,expected_sync",
     [