Add tests for fs.hardlink_dir()

requirements-dev.txt

@@ -0,0 +1 @@
+pytest

tests/test_fs.py

@@ -0,0 +1,53 @@
+import os
+import os.path
+import shutil
+import string
+import tempfile
+import unittest
+
+from spqr.curateipsum import fs
+
+
+class TestHardlinkDir(unittest.TestCase):
+    @classmethod
+    def setUpClass(self):
+        self.tmp_dir = tempfile.TemporaryDirectory()
+        self.src_dir = self.tmp_dir.name
+        self.dst_dir = self.src_dir + ".copy"
+
+        # common_file
+        self.cf_name = "common_file"
+        self.cf = os.path.join(self.src_dir, self.cf_name)
+        with open(self.cf, "w") as f:
+            f.write(string.printable)
+
+        # symlink to common file
+        self.sl2cf_name = "symlink_to_common_file"
+        self.sl2cf = os.path.join(self.src_dir, self.sl2cf_name)
+        os.symlink(self.cf, self.sl2cf)
+
+        # hardlink to common file
+        self.hl2cf_name = "hardlink_to_common_file"
+        self.hl2cf = os.path.join(self.src_dir, self.hl2cf_name)
+        os.link(self.cf, self.hl2cf)
+
+        fs.hardlink_dir(self.src_dir, self.dst_dir)
+
+    def test_common_file(self):
+        src_stat = os.lstat(self.cf)
+        dst_stat = os.lstat(os.path.join(self.dst_dir, self.cf_name))
+        self.assertTrue(os.path.samestat(src_stat, dst_stat))
+
+    def test_symlink_to_common_file(self):
+        dst_sl2cf_path = os.path.join(self.dst_dir, self.sl2cf_name)
+        self.assertEqual(os.readlink(dst_sl2cf_path), self.cf)
+
+    def test_hardlink_to_common_file(self):
+        src_stat = os.lstat(self.hl2cf)
+        dst_stat = os.lstat(os.path.join(self.dst_dir, self.hl2cf_name))
+        self.assertTrue(os.path.samestat(src_stat, dst_stat))
+
+    @classmethod
+    def tearDownClass(self):
+        self.tmp_dir.cleanup()
+        shutil.rmtree(self.dst_dir, ignore_errors=True)