1 files changed, 96 insertions, 0 deletions
diff --git a/python/mozbuild/mozbuild/test/action/test_buildlist.py b/python/mozbuild/mozbuild/test/action/test_buildlist.py
new file mode 100644
index 0000000000..9a1d2738ed
--- /dev/null
+++ b/python/mozbuild/mozbuild/test/action/test_buildlist.py
@@ -0,0 +1,96 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import os
+import os.path
+import unittest
+from shutil import rmtree
+from tempfile import mkdtemp
+
+import mozunit
+
+from mozbuild.action.buildlist import addEntriesToListFile
+
+
+class TestBuildList(unittest.TestCase):
+    """
+    Unit tests for buildlist.py
+    """
+
+    def setUp(self):
+        self.tmpdir = mkdtemp()
+
+    def tearDown(self):
+        rmtree(self.tmpdir)
+
+    # utility methods for tests
+    def touch(self, file, dir=None):
+        if dir is None:
+            dir = self.tmpdir
+        f = os.path.join(dir, file)
+        open(f, "w").close()
+        return f
+
+    def assertFileContains(self, filename, l):
+        """Assert that the lines in the file |filename| are equal
+        to the contents of the list |l|, in order."""
+        l = l[:]
+        f = open(filename, "r")
+        lines = [line.rstrip() for line in f.readlines()]
+        f.close()
+        for line in lines:
+            self.assertTrue(
+                len(l) > 0,
+                "ran out of expected lines! (expected '{0}', got '{1}')".format(
+                    l, lines
+                ),
+            )
+            self.assertEqual(line, l.pop(0))
+        self.assertTrue(
+            len(l) == 0,
+            "not enough lines in file! (expected '{0}'," " got '{1}'".format(l, lines),
+        )
+
+    def test_basic(self):
+        "Test that addEntriesToListFile works when file doesn't exist."
+        testfile = os.path.join(self.tmpdir, "test.list")
+        l = ["a", "b", "c"]
+        addEntriesToListFile(testfile, l)
+        self.assertFileContains(testfile, l)
+        # ensure that attempting to add the same entries again doesn't change it
+        addEntriesToListFile(testfile, l)
+        self.assertFileContains(testfile, l)
+
+    def test_append(self):
+        "Test adding new entries."
+        testfile = os.path.join(self.tmpdir, "test.list")
+        l = ["a", "b", "c"]
+        addEntriesToListFile(testfile, l)
+        self.assertFileContains(testfile, l)
+        l2 = ["x", "y", "z"]
+        addEntriesToListFile(testfile, l2)
+        l.extend(l2)
+        self.assertFileContains(testfile, l)
+
+    def test_append_some(self):
+        "Test adding new entries mixed with existing entries."
+        testfile = os.path.join(self.tmpdir, "test.list")
+        l = ["a", "b", "c"]
+        addEntriesToListFile(testfile, l)
+        self.assertFileContains(testfile, l)
+        addEntriesToListFile(testfile, ["a", "x", "c", "z"])
+        self.assertFileContains(testfile, ["a", "b", "c", "x", "z"])
+
+    def test_add_multiple(self):
+        """Test that attempting to add the same entry multiple times results in
+        only one entry being added."""
+        testfile = os.path.join(self.tmpdir, "test.list")
+        addEntriesToListFile(testfile, ["a", "b", "a", "a", "b"])
+        self.assertFileContains(testfile, ["a", "b"])
+        addEntriesToListFile(testfile, ["c", "a", "c", "b", "c"])
+        self.assertFileContains(testfile, ["a", "b", "c"])
+
+
+if __name__ == "__main__":
+    mozunit.main()