1 files changed, 69 insertions, 0 deletions
diff --git a/llvm_tools/upload_llvm_testing_helper_cl_test.py b/llvm_tools/upload_llvm_testing_helper_cl_test.py
new file mode 100755
index 00000000..63951a59
--- /dev/null
+++ b/llvm_tools/upload_llvm_testing_helper_cl_test.py
@@ -0,0 +1,69 @@
+#!/usr/bin/env python3
+# Copyright 2024 The ChromiumOS Authors
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""Tests for upload_llvm_testing_helper_cl"""
+
+from pathlib import Path
+import shutil
+import tempfile
+import unittest
+
+import upload_llvm_testing_helper_cl
+
+
+class Test(unittest.TestCase):
+    """Tests for upload_llvm_testing_helper_cl"""
+
+    def make_tempdir(self) -> Path:
+        tempdir = tempfile.mkdtemp(
+            os.path.basename("upload_llvm_testing_helper_cl_test")
+        )
+        self.addCleanup(shutil.rmtree, tempdir)
+        return Path(tempdir)
+
+    def test_force_rebuild_marker_addition(self):
+        chromiumos_overlay = self.make_tempdir()
+        filesdir = chromiumos_overlay / "sys-devel" / "llvm" / "files"
+        filesdir.mkdir(parents=True)
+        upload_llvm_testing_helper_cl.add_force_rebuild_marker(
+            chromiumos_overlay
+        )
+        self.assertTrue((filesdir / "force_rebuild").exists())
+
+    def test_use_force_block_addition(self):
+        chromiumos_overlay = self.make_tempdir()
+        use_force_file = chromiumos_overlay / "profiles" / "base" / "use.force"
+        use_force_file.parent.mkdir(parents=True)
+        use_force_file.write_text("# Whee", encoding="utf-8")
+
+        upload_llvm_testing_helper_cl.add_use_force_block(chromiumos_overlay)
+        new_contents = use_force_file.read_text(encoding="utf-8")
+
+        self.assertIn("# Whee\n", new_contents)
+        self.assertIn(
+            upload_llvm_testing_helper_cl.USE_FORCE_BLOCK, new_contents
+        )
+
+    def test_warning_disable_block_addition(self):
+        chromiumos_overlay = self.make_tempdir()
+        profile_bashrc = (
+            chromiumos_overlay / "profiles" / "base" / "profile.bashrc"
+        )
+        profile_bashrc.parent.mkdir(parents=True)
+        profile_bashrc.write_text("# Whee", encoding="utf-8")
+
+        upload_llvm_testing_helper_cl.add_disable_warnings_block(
+            chromiumos_overlay
+        )
+        new_contents = profile_bashrc.read_text(encoding="utf-8")
+
+        self.assertIn("# Whee\n", new_contents)
+        self.assertIn(
+            upload_llvm_testing_helper_cl.DISABLE_WARNINGS_BLOCK, new_contents
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()