Add unit tests for all classes in rbackup.hierarchy

tests/test_hierarchy.py

@@ -1,47 +1,110 @@
 import rbackup.hierarchy as hierarchy
 import unittest
 
-from unittest.mock import MagicMock, patch
+from unittest.mock import patch
 
+# ========== Constants ==========
+
+
+# ========== Test Cases ==========
 class TestHierarchy(unittest.TestCase):
-    def setUp(self):
+    @patch("rbackup.hierarchy.os.path.isdir")
-        self.patched_os_path = patch('rbackup.hierarchy.os.path')
+    def test_base_path(self, mocked_isdir):
-        self.mocked_os_path = self.patched_os_path.start()
+        mocked_isdir.return_value = True
 
-        self.mocked_os_path.isdir.return_value = True
+        self.assertEqual(hierarchy.Hierarchy("directory").base_path, "directory")
-
-        self.hier = hierarchy.Hierarchy('/mnt')
 
     @patch("rbackup.hierarchy.os.path.isdir")
-    def test_non_dir(self, patched_false_isdir):
+    def test_invalid_dir(self, mocked_isdir):
-        patched_false_isdir.return_value = False
+        mocked_isdir.return_value = False
 
-        with self.assertRaises(ValueError):
+        with self.assertRaises(NotADirectoryError):
-            h = hierarchy.Hierarchy('NotADir')
+            hierarchy.Hierarchy("notadirectory")
 
-    def test_base_path(self):
-        self.assertEqual(self.hier.base_dir, "/mnt")
 
-    def test_prev_snapshot(self):
+class TestRepository(unittest.TestCase):
-        raise NotImplementedError
+    def setUp(self):
+        self.patched_isdir = patch("rbackup.hierarchy.os.path.isdir")
+        self.mocked_isdir = self.patched_isdir.start()
 
-    def test_prev_snapshot_link(self):
+        self.mocked_isdir.return_value = True
-        raise NotImplementedError
+
+        self.patched_datetime = patch("rbackup.hierarchy.datetime")
+        self.mocked_datetime = self.patched_datetime.start()
+
+        self.mocked_datetime.datetime.utcnow.return_value.isoformat.return_value = (
+            "utcnow"
+        )
+
+        self.patched_glob = patch("rbackup.hierarchy.glob.glob")
+        self.mocked_glob = self.patched_glob.start()
+
+        self.mocked_glob.return_value = [
+            "backup/data/snapshot-first",
+            "backup/data/snapshot-second",
+            "backup/data/snapshot-last",
+        ]
+
+        self.repo = hierarchy.Repository("backup")
+
+    @patch("rbackup.hierarchy.glob.glob")
+    def test_first_snapshot(self, mocked_glob):
+        mocked_glob.return_value = []
+
+        empty_repo = hierarchy.Repository("backup")
+
+        self.assertIsNone(empty_repo.snapshots)
+
+    @patch("rbackup.hierarchy.os.path.realpath")
+    def test_prev_snapshot(self, mocked_realpath):
+        mocked_realpath.return_value = "backup/data/snapshot-last"
+
+        self.assertEqual(self.repo.prev_snapshot.path, "backup/data/snapshot-last")
+
+        self.assertIsInstance(self.repo.prev_snapshot, hierarchy.Snapshot)
 
     def test_curr_snapshot(self):
-        raise NotImplementedError
+        self.assertEqual(self.repo.curr_snapshot.path, "backup/data/snapshot-utcnow")
-
-    def test_boot_dir(self):
-        raise NotImplementedError
-
-    def test_etc_dir(self):
-        raise NotImplementedError
-
-    def test_home_dir(self):
-        raise NotImplementedError
-
-    def test_root_home_dir(self):
-        raise NotImplementedError
 
     def tearDown(self):
-        self.patched_os_path.stop()
+        self.patched_isdir.stop()
+        self.patched_datetime.stop()
+        self.patched_glob.stop()
+
+
+class TestSnapshot(unittest.TestCase):
+    def setUp(self):
+        self.patched_isdir = patch("rbackup.hierarchy.os.path.isdir")
+        self.mocked_isdir = self.patched_isdir.start()
+
+        self.mocked_isdir.return_value = True
+
+        self.patched_datetime = patch("rbackup.hierarchy.datetime")
+        self.mocked_datetime = self.patched_datetime.start()
+
+        self.mocked_datetime.datetime.utcnow.return_value.isoformat.return_value = (
+            "utcnow"
+        )
+
+        self.snapshot = hierarchy.Snapshot("backup/data/snapshot-utcnow")
+
+    def test_name(self):
+        self.assertEqual(self.snapshot.name, "snapshot-utcnow")
+
+    def test_boot_dir(self):
+        self.assertEqual(self.snapshot.boot_dir, "backup/data/snapshot-utcnow/boot")
+
+    def test_etc_dir(self):
+        self.assertEqual(self.snapshot.etc_dir, "backup/data/snapshot-utcnow/etc")
+
+    def test_home_dir(self):
+        self.assertEqual(self.snapshot.home_dir, "backup/data/snapshot-utcnow/home")
+
+    def test_root_home_dir(self):
+        self.assertEqual(
+            self.snapshot.root_home_dir, "backup/data/snapshot-utcnow/root"
+        )
+
+    def tearDown(self):
+        self.patched_isdir.stop()
+        self.patched_datetime.stop()