From 11c8197dbafa49fc659733a88fa2d26bc6abd78b Mon Sep 17 00:00:00 2001
From: Maks Snegov <snegov@spqr.link>
Date: Tue, 15 Jun 2021 23:05:09 +0300
Subject: [PATCH] Add tests for fs.hardlink_dir()

---
 requirements-dev.txt |  1 +
 tests/test_fs.py     | 53 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)
 create mode 100644 requirements-dev.txt
 create mode 100644 tests/test_fs.py

diff --git a/requirements-dev.txt b/requirements-dev.txt
new file mode 100644
index 0000000..e079f8a
--- /dev/null
+++ b/requirements-dev.txt
@@ -0,0 +1 @@
+pytest
diff --git a/tests/test_fs.py b/tests/test_fs.py
new file mode 100644
index 0000000..172d305
--- /dev/null
+++ b/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)