1 files changed, 84 insertions, 0 deletions
diff --git a/llvm_tools/manifest_utils_unittest.py b/llvm_tools/manifest_utils_unittest.py
new file mode 100644
index 00000000..9a21d990
--- /dev/null
+++ b/llvm_tools/manifest_utils_unittest.py
@@ -0,0 +1,84 @@
+# Copyright 2023 The ChromiumOS Authors
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import io
+from pathlib import Path
+import re
+import unittest
+from xml.etree import ElementTree
+
+import manifest_utils
+
+
+"""Provides utilities to read and edit the ChromiumOS Manifest entries.
+
+While this code reads and edits the internal manifest, it should only operate
+on toolchain projects (llvm-project, etc.) which are public.
+"""
+
+MANIFEST_FIXTURE = """<?xml version="1.0" encoding="UTF-8"?>
+<manifest>
+  <!-- Comment that should not be removed.
+       Multiple lines. -->
+  <include name="_remotes.xml" />
+  <default revision="refs/heads/main"
+           remote="cros"
+           sync-j="8" />
+
+  <include name="_kernel_upstream.xml" />
+
+  <!-- Common projects for developing CrOS. -->
+  <project path="src/repohooks"
+           name="chromiumos/repohooks"
+           groups="minilayout,paygen,firmware,buildtools,labtools,crosvm" />
+  <repo-hooks in-project="chromiumos/repohooks"
+              enabled-list="pre-upload" />
+  <project path="chromite"
+           name="chromiumos/chromite"
+           groups="minilayout,paygen,firmware,buildtools,chromeos-admin">
+    <copyfile src="AUTHORS" dest="AUTHORS" />
+    <copyfile src="LICENSE" dest="LICENSE" />
+  </project>
+  <project path="src/third_party/llvm-project"
+           name="external/github.com/llvm/llvm-project"
+           groups="notdefault,bazel"
+           revision="abcd" />
+  <project path="chromite/third_party/pyelftools"
+           name="chromiumos/third_party/pyelftools"
+           revision="refs/heads/chromeos-0.22"
+           groups="minilayout,paygen,firmware,buildtools" />
+</manifest>
+"""
+
+
+class TestManifestUtils(unittest.TestCase):
+    """Test manifest_utils."""
+
+    def test_update_chromeos_manifest(self):
+        root = ElementTree.fromstring(
+            MANIFEST_FIXTURE,
+            parser=manifest_utils.make_xmlparser(),
+        )
+        manifest_utils.update_chromeos_manifest_tree("wxyz", root)
+        string_root1 = ElementTree.tostring(root)
+        self.assertRegex(
+            str(string_root1, encoding="utf-8"),
+            r'revision="wxyz"',
+        )
+        self.assertRegex(
+            str(string_root1, encoding="utf-8"),
+            r"<!-- Comment that should not be removed.",
+        )
+        self.assertNotRegex(
+            str(string_root1, encoding="utf-8"),
+            r'revision="abcd"',
+        )
+        # Check idempotence.
+        manifest_utils.update_chromeos_manifest_tree("wxyz", root)
+        string_root2 = ElementTree.tostring(root)
+        self.assertEqual(string_root1, string_root2)
+
+
+if __name__ == "__main__":
+    unittest.main()