~tests

tests/cli/test_folderback.py

@@ -9,6 +9,95 @@ import pyflowx as px
 from pyflowx.cli import folderback
 
 
+# ---------------------------------------------------------------------- #
+# remove_dump
+# ---------------------------------------------------------------------- #
+class TestRemoveDump:
+    """Test remove_dump function."""
+
+    def test_remove_dump_no_files(self, tmp_path: Path) -> None:
+        """Should handle no zip files."""
+        src = tmp_path / "source"
+        src.mkdir()
+        dst = tmp_path / "backup"
+        dst.mkdir()
+
+        folderback.remove_dump(src, dst, 5)
+        # Should not raise error
+
+    def test_remove_dump_within_limit(self, tmp_path: Path) -> None:
+        """Should not remove files within limit."""
+        src = tmp_path / "source"
+        src.mkdir()
+        dst = tmp_path / "backup"
+        dst.mkdir()
+
+        # Create some zip files
+        for i in range(3):
+            zip_file = dst / f"source_20240101_12000{i}.zip"
+            zip_file.write_bytes(b"ZIP content")
+
+        folderback.remove_dump(src, dst, 5)
+        # All files should remain
+        assert len(list(dst.glob("*.zip"))) == 3
+
+    def test_remove_dump_exceeds_limit(self, tmp_path: Path) -> None:
+        """Should remove oldest files when exceeds limit."""
+        src = tmp_path / "source"
+        src.mkdir()
+        dst = tmp_path / "backup"
+        dst.mkdir()
+
+        # Create more zip files than limit
+        for i in range(7):
+            zip_file = dst / f"source_20240101_12000{i}.zip"
+            zip_file.write_bytes(b"ZIP content")
+
+        folderback.remove_dump(src, dst, 5)
+        # Should have only 5 files
+        assert len(list(dst.glob("*.zip"))) == 5
+
+
+# ---------------------------------------------------------------------- #
+# zip_target
+# ---------------------------------------------------------------------- #
+class TestZipTarget:
+    """Test zip_target function."""
+
+    def test_zip_target_creates_zip(self, tmp_path: Path) -> None:
+        """Should create zip file."""
+        src = tmp_path / "source"
+        src.mkdir()
+        (src / "test.txt").write_text("test content")
+        dst = tmp_path / "backup"
+        dst.mkdir()
+
+        with patch("time.strftime", return_value="_20240101_120000"):
+            folderback.zip_target(src, dst, 5)
+
+        # Should create zip file
+        zip_files = list(dst.glob("*.zip"))
+        assert len(zip_files) == 1
+
+    def test_zip_target_with_subdirectories(self, tmp_path: Path) -> None:
+        """Should zip files in subdirectories."""
+        src = tmp_path / "source"
+        src.mkdir()
+        subdir = src / "subdir"
+        subdir.mkdir()
+        (src / "test.txt").write_text("test content")
+        (subdir / "nested.txt").write_text("nested content")
+        dst = tmp_path / "backup"
+        dst.mkdir()
+
+        with patch("time.strftime", return_value="_20240101_120000"):
+            folderback.zip_target(src, dst, 5)
+
+        # Should create zip file
+        zip_files = list(dst.glob("*.zip"))
+        assert len(zip_files) == 1
+
+
 # ---------------------------------------------------------------------- #
 # backup_folder
 # ---------------------------------------------------------------------- #
@@ -46,6 +135,18 @@ class TestBackupFolder:
         folderback.backup_folder(str(source_dir), str(backup_dir), 5)
         # Should print error message and return
 
+    def test_backup_folder_creates_dst(self, tmp_path: Path) -> None:
+        """Should create destination directory."""
+        source_dir = tmp_path / "source"
+        source_dir.mkdir()
+        (source_dir / "test.txt").write_text("test content")
+        backup_dir = tmp_path / "backup"
+
+        with patch.object(folderback, "zip_target") as mock_zip:
+            folderback.backup_folder(str(source_dir), str(backup_dir), 5)
+            assert backup_dir.exists()
+            assert mock_zip.called
+
 
 # ---------------------------------------------------------------------- #
 # TaskSpec definitions