1 files changed, 126 insertions, 0 deletions
diff --git a/testing/mozbase/mozfile/tests/test_copycontents.py b/testing/mozbase/mozfile/tests/test_copycontents.py
new file mode 100644
index 0000000000..b829d7b3a4
--- /dev/null
+++ b/testing/mozbase/mozfile/tests/test_copycontents.py
@@ -0,0 +1,126 @@
+#!/usr/bin/env python
+
+import os
+import shutil
+import unittest
+
+import mozfile
+import mozunit
+import stubs
+
+
+class MozfileCopyContentsTestCase(unittest.TestCase):
+    """Test our ability to copy the contents of directories"""
+
+    def _directory_is_subset(self, set_, subset_):
+        """
+        Confirm that all the contents of 'subset_' are contained in 'set_'
+        """
+        names = os.listdir(subset_)
+        for name in names:
+            full_set_path = os.path.join(set_, name)
+            full_subset_path = os.path.join(subset_, name)
+            if os.path.isdir(full_subset_path):
+                self.assertTrue(os.path.isdir(full_set_path))
+                self._directory_is_subset(full_set_path, full_subset_path)
+            elif os.path.islink(full_subset_path):
+                self.assertTrue(os.path.islink(full_set_path))
+            else:
+                self.assertTrue(os.stat(full_set_path))
+
+    def _directories_are_equal(self, dir1, dir2):
+        """
+        Confirm that the contents of 'dir1' are the same as 'dir2'
+        """
+        names1 = os.listdir(dir1)
+        names2 = os.listdir(dir2)
+        self.assertTrue(len(names1) == len(names2))
+        for name in names1:
+            self.assertTrue(name in names2)
+            dir1_path = os.path.join(dir1, name)
+            dir2_path = os.path.join(dir2, name)
+            if os.path.isdir(dir1_path):
+                self.assertTrue(os.path.isdir(dir2_path))
+                self._directories_are_equal(dir1_path, dir2_path)
+            elif os.path.islink(dir1_path):
+                self.assertTrue(os.path.islink(dir2_path))
+            else:
+                self.assertTrue(os.stat(dir2_path))
+
+    def test_copy_empty_directory(self):
+        tempdir = stubs.create_empty_stub()
+        dstdir = stubs.create_empty_stub()
+        self.assertTrue(os.path.isdir(tempdir))
+
+        mozfile.copy_contents(tempdir, dstdir)
+        self._directories_are_equal(dstdir, tempdir)
+
+        if os.path.isdir(tempdir):
+            shutil.rmtree(tempdir)
+        if os.path.isdir(dstdir):
+            shutil.rmtree(dstdir)
+
+    def test_copy_full_directory(self):
+        tempdir = stubs.create_stub()
+        dstdir = stubs.create_empty_stub()
+        self.assertTrue(os.path.isdir(tempdir))
+
+        mozfile.copy_contents(tempdir, dstdir)
+        self._directories_are_equal(dstdir, tempdir)
+
+        if os.path.isdir(tempdir):
+            shutil.rmtree(tempdir)
+        if os.path.isdir(dstdir):
+            shutil.rmtree(dstdir)
+
+    def test_copy_full_directory_with_existing_file(self):
+        tempdir = stubs.create_stub()
+        dstdir = stubs.create_empty_stub()
+
+        filename = "i_dont_exist_in_tempdir"
+        f = open(os.path.join(dstdir, filename), "w")
+        f.write("Hello World")
+        f.close()
+
+        self.assertTrue(os.path.isdir(tempdir))
+
+        mozfile.copy_contents(tempdir, dstdir)
+        self._directory_is_subset(dstdir, tempdir)
+        self.assertTrue(os.path.exists(os.path.join(dstdir, filename)))
+
+        if os.path.isdir(tempdir):
+            shutil.rmtree(tempdir)
+        if os.path.isdir(dstdir):
+            shutil.rmtree(dstdir)
+
+    def test_copy_full_directory_with_overlapping_file(self):
+        tempdir = stubs.create_stub()
+        dstdir = stubs.create_empty_stub()
+
+        filename = "i_do_exist_in_tempdir"
+        for d in [tempdir, dstdir]:
+            f = open(os.path.join(d, filename), "w")
+            f.write("Hello " + d)
+            f.close()
+
+        self.assertTrue(os.path.isdir(tempdir))
+        self.assertTrue(os.path.exists(os.path.join(tempdir, filename)))
+        self.assertTrue(os.path.exists(os.path.join(dstdir, filename)))
+
+        line = open(os.path.join(dstdir, filename), "r").readlines()[0]
+        self.assertTrue(line == "Hello " + dstdir)
+
+        mozfile.copy_contents(tempdir, dstdir)
+
+        line = open(os.path.join(dstdir, filename), "r").readlines()[0]
+        self.assertTrue(line == "Hello " + tempdir)
+        self._directories_are_equal(tempdir, dstdir)
+
+        if os.path.isdir(tempdir):
+            shutil.rmtree(tempdir)
+        if os.path.isdir(dstdir):
+            shutil.rmtree(dstdir)
+
+
+if __name__ == "__main__":
+    mozunit.main()