@vodailoc/kilo-kit-mcp 1.1.0 → 1.2.0

package/skills/ai-media/media-processing/scripts/tests/test_batch_resize.py

@@ -1,372 +1,372 @@
-#!/usr/bin/env python3
-"""Tests for batch_resize.py"""
-
-import sys
-from pathlib import Path
-from unittest.mock import MagicMock, call, patch
-
-import pytest
-
-# Add parent directory to path
-sys.path.insert(0, str(Path(__file__).parent.parent))
-
-from batch_resize import ImageResizer, collect_images
-
-
-class TestImageResizer:
-    """Test ImageResizer class."""
-
-    def setup_method(self):
-        """Set up test fixtures."""
-        self.resizer = ImageResizer(verbose=False, dry_run=False)
-
-    @patch("subprocess.run")
-    def test_check_imagemagick_available(self, mock_run):
-        """Test ImageMagick availability check."""
-        mock_run.return_value = MagicMock(returncode=0)
-        assert self.resizer.check_imagemagick() is True
-
-    @patch("subprocess.run")
-    def test_check_imagemagick_unavailable(self, mock_run):
-        """Test when ImageMagick is not available."""
-        mock_run.side_effect = FileNotFoundError()
-        assert self.resizer.check_imagemagick() is False
-
-    def test_build_resize_command_fit_strategy(self):
-        """Test command building for 'fit' strategy."""
-        cmd = self.resizer.build_resize_command(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=800,
-            height=600,
-            strategy="fit",
-            quality=85
-        )
-
-        assert "magick" in cmd
-        assert str(Path("input.jpg")) in cmd
-        assert "-resize" in cmd
-        assert "800x600" in cmd
-        assert "-quality" in cmd
-        assert "85" in cmd
-        assert "-strip" in cmd
-
-    def test_build_resize_command_fill_strategy(self):
-        """Test command building for 'fill' strategy."""
-        cmd = self.resizer.build_resize_command(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=800,
-            height=600,
-            strategy="fill",
-            quality=85
-        )
-
-        assert "-resize" in cmd
-        assert "800x600^" in cmd
-        assert "-gravity" in cmd
-        assert "center" in cmd
-        assert "-extent" in cmd
-
-    def test_build_resize_command_thumbnail_strategy(self):
-        """Test command building for 'thumbnail' strategy."""
-        cmd = self.resizer.build_resize_command(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=200,
-            height=None,
-            strategy="thumbnail",
-            quality=85
-        )
-
-        assert "200x200^" in cmd
-        assert "-gravity" in cmd
-        assert "center" in cmd
-
-    def test_build_resize_command_with_watermark(self):
-        """Test command building with watermark."""
-        watermark = Path("watermark.png")
-        cmd = self.resizer.build_resize_command(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=800,
-            height=None,
-            strategy="fit",
-            quality=85,
-            watermark=watermark
-        )
-
-        assert str(watermark) in cmd
-        assert "-gravity" in cmd
-        assert "southeast" in cmd
-        assert "-composite" in cmd
-
-    def test_build_resize_command_exact_strategy(self):
-        """Test command building for 'exact' strategy."""
-        cmd = self.resizer.build_resize_command(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=800,
-            height=600,
-            strategy="exact",
-            quality=85
-        )
-
-        assert "800x600!" in cmd
-
-    def test_build_resize_command_fill_requires_dimensions(self):
-        """Test that 'fill' strategy requires both dimensions."""
-        with pytest.raises(ValueError):
-            self.resizer.build_resize_command(
-                Path("input.jpg"),
-                Path("output.jpg"),
-                width=800,
-                height=None,
-                strategy="fill",
-                quality=85
-            )
-
-    @patch("subprocess.run")
-    def test_resize_image_success(self, mock_run):
-        """Test successful image resize."""
-        mock_run.return_value = MagicMock(returncode=0)
-
-        result = self.resizer.resize_image(
-            Path("input.jpg"),
-            Path("output/output.jpg"),
-            width=800,
-            height=None,
-            strategy="fit",
-            quality=85
-        )
-
-        assert result is True
-        mock_run.assert_called_once()
-
-    @patch("subprocess.run")
-    def test_resize_image_dry_run(self, mock_run):
-        """Test resize in dry-run mode."""
-        resizer = ImageResizer(dry_run=True)
-
-        result = resizer.resize_image(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=800,
-            height=None
-        )
-
-        assert result is True
-        mock_run.assert_not_called()
-
-    @patch("subprocess.run")
-    def test_resize_image_failure(self, mock_run):
-        """Test resize failure handling."""
-        mock_run.side_effect = Exception("Resize failed")
-
-        result = self.resizer.resize_image(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=800,
-            height=None
-        )
-
-        assert result is False
-
-
-class TestCollectImages:
-    """Test image collection functionality."""
-
-    def test_collect_images_from_file(self, tmp_path):
-        """Test collecting a single image file."""
-        img_file = tmp_path / "test.jpg"
-        img_file.touch()
-
-        images = collect_images([img_file])
-        assert len(images) == 1
-        assert images[0] == img_file
-
-    def test_collect_images_from_directory(self, tmp_path):
-        """Test collecting images from directory."""
-        (tmp_path / "image1.jpg").touch()
-        (tmp_path / "image2.png").touch()
-        (tmp_path / "text.txt").touch()
-
-        images = collect_images([tmp_path])
-        assert len(images) == 2
-        assert all(img.suffix.lower() in {'.jpg', '.png'} for img in images)
-
-    def test_collect_images_recursive(self, tmp_path):
-        """Test recursive image collection."""
-        subdir = tmp_path / "subdir"
-        subdir.mkdir()
-        (tmp_path / "image1.jpg").touch()
-        (subdir / "image2.jpg").touch()
-
-        images = collect_images([tmp_path], recursive=True)
-        assert len(images) == 2
-
-        images_non_recursive = collect_images([tmp_path], recursive=False)
-        assert len(images_non_recursive) == 1
-
-    def test_collect_images_filters_extensions(self, tmp_path):
-        """Test that only image files are collected."""
-        (tmp_path / "image.jpg").touch()
-        (tmp_path / "doc.pdf").touch()
-        (tmp_path / "text.txt").touch()
-
-        images = collect_images([tmp_path])
-        assert len(images) == 1
-        assert images[0].suffix.lower() == '.jpg'
-
-    def test_collect_images_multiple_paths(self, tmp_path):
-        """Test collecting from multiple paths."""
-        dir1 = tmp_path / "dir1"
-        dir2 = tmp_path / "dir2"
-        dir1.mkdir()
-        dir2.mkdir()
-
-        (dir1 / "image1.jpg").touch()
-        (dir2 / "image2.png").touch()
-
-        images = collect_images([dir1, dir2])
-        assert len(images) == 2
-
-
-class TestBatchResize:
-    """Test batch resize functionality."""
-
-    def setup_method(self):
-        """Set up test fixtures."""
-        self.resizer = ImageResizer(verbose=False, dry_run=False)
-
-    @patch.object(ImageResizer, "resize_image")
-    def test_batch_resize_success(self, mock_resize, tmp_path):
-        """Test successful batch resize."""
-        mock_resize.return_value = True
-
-        input_images = [
-            tmp_path / "image1.jpg",
-            tmp_path / "image2.jpg"
-        ]
-        for img in input_images:
-            img.touch()
-
-        output_dir = tmp_path / "output"
-
-        success, fail = self.resizer.batch_resize(
-            input_images,
-            output_dir,
-            width=800,
-            height=None,
-            strategy="fit"
-        )
-
-        assert success == 2
-        assert fail == 0
-        assert mock_resize.call_count == 2
-
-    @patch.object(ImageResizer, "resize_image")
-    def test_batch_resize_with_failures(self, mock_resize, tmp_path):
-        """Test batch resize with some failures."""
-        mock_resize.side_effect = [True, False, True]
-
-        input_images = [
-            tmp_path / "image1.jpg",
-            tmp_path / "image2.jpg",
-            tmp_path / "image3.jpg"
-        ]
-        for img in input_images:
-            img.touch()
-
-        output_dir = tmp_path / "output"
-
-        success, fail = self.resizer.batch_resize(
-            input_images,
-            output_dir,
-            width=800,
-            height=None
-        )
-
-        assert success == 2
-        assert fail == 1
-
-    @patch.object(ImageResizer, "resize_image")
-    def test_batch_resize_format_conversion(self, mock_resize, tmp_path):
-        """Test batch resize with format conversion."""
-        mock_resize.return_value = True
-
-        input_images = [tmp_path / "image.png"]
-        input_images[0].touch()
-
-        output_dir = tmp_path / "output"
-
-        self.resizer.batch_resize(
-            input_images,
-            output_dir,
-            width=800,
-            height=None,
-            format_ext="jpg"
-        )
-
-        # Check that resize_image was called with .jpg extension
-        call_args = mock_resize.call_args[0]
-        assert call_args[1].suffix == ".jpg"
-
-
-class TestResizeStrategies:
-    """Test different resize strategies."""
-
-    def setup_method(self):
-        """Set up test fixtures."""
-        self.resizer = ImageResizer()
-
-    def test_fit_strategy_maintains_aspect(self):
-        """Test that 'fit' strategy maintains aspect ratio."""
-        cmd = self.resizer.build_resize_command(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=800,
-            height=600,
-            strategy="fit",
-            quality=85
-        )
-
-        # Should have resize without ^ or !
-        resize_idx = cmd.index("-resize")
-        geometry = cmd[resize_idx + 1]
-        assert "^" not in geometry
-        assert "!" not in geometry
-
-    def test_cover_strategy_fills_dimensions(self):
-        """Test that 'cover' strategy fills dimensions."""
-        cmd = self.resizer.build_resize_command(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=800,
-            height=600,
-            strategy="cover",
-            quality=85
-        )
-
-        resize_idx = cmd.index("-resize")
-        geometry = cmd[resize_idx + 1]
-        assert "^" in geometry
-
-    def test_exact_strategy_ignores_aspect(self):
-        """Test that 'exact' strategy ignores aspect ratio."""
-        cmd = self.resizer.build_resize_command(
-            Path("input.jpg"),
-            Path("output.jpg"),
-            width=800,
-            height=600,
-            strategy="exact",
-            quality=85
-        )
-
-        resize_idx = cmd.index("-resize")
-        geometry = cmd[resize_idx + 1]
-        assert "!" in geometry
-
-
-if __name__ == "__main__":
-    pytest.main([__file__, "-v"])
+#!/usr/bin/env python3
+"""Tests for batch_resize.py"""
+
+import sys
+from pathlib import Path
+from unittest.mock import MagicMock, call, patch
+
+import pytest
+
+# Add parent directory to path
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+from batch_resize import ImageResizer, collect_images
+
+
+class TestImageResizer:
+    """Test ImageResizer class."""
+
+    def setup_method(self):
+        """Set up test fixtures."""
+        self.resizer = ImageResizer(verbose=False, dry_run=False)
+
+    @patch("subprocess.run")
+    def test_check_imagemagick_available(self, mock_run):
+        """Test ImageMagick availability check."""
+        mock_run.return_value = MagicMock(returncode=0)
+        assert self.resizer.check_imagemagick() is True
+
+    @patch("subprocess.run")
+    def test_check_imagemagick_unavailable(self, mock_run):
+        """Test when ImageMagick is not available."""
+        mock_run.side_effect = FileNotFoundError()
+        assert self.resizer.check_imagemagick() is False
+
+    def test_build_resize_command_fit_strategy(self):
+        """Test command building for 'fit' strategy."""
+        cmd = self.resizer.build_resize_command(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=800,
+            height=600,
+            strategy="fit",
+            quality=85
+        )
+
+        assert "magick" in cmd
+        assert str(Path("input.jpg")) in cmd
+        assert "-resize" in cmd
+        assert "800x600" in cmd
+        assert "-quality" in cmd
+        assert "85" in cmd
+        assert "-strip" in cmd
+
+    def test_build_resize_command_fill_strategy(self):
+        """Test command building for 'fill' strategy."""
+        cmd = self.resizer.build_resize_command(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=800,
+            height=600,
+            strategy="fill",
+            quality=85
+        )
+
+        assert "-resize" in cmd
+        assert "800x600^" in cmd
+        assert "-gravity" in cmd
+        assert "center" in cmd
+        assert "-extent" in cmd
+
+    def test_build_resize_command_thumbnail_strategy(self):
+        """Test command building for 'thumbnail' strategy."""
+        cmd = self.resizer.build_resize_command(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=200,
+            height=None,
+            strategy="thumbnail",
+            quality=85
+        )
+
+        assert "200x200^" in cmd
+        assert "-gravity" in cmd
+        assert "center" in cmd
+
+    def test_build_resize_command_with_watermark(self):
+        """Test command building with watermark."""
+        watermark = Path("watermark.png")
+        cmd = self.resizer.build_resize_command(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=800,
+            height=None,
+            strategy="fit",
+            quality=85,
+            watermark=watermark
+        )
+
+        assert str(watermark) in cmd
+        assert "-gravity" in cmd
+        assert "southeast" in cmd
+        assert "-composite" in cmd
+
+    def test_build_resize_command_exact_strategy(self):
+        """Test command building for 'exact' strategy."""
+        cmd = self.resizer.build_resize_command(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=800,
+            height=600,
+            strategy="exact",
+            quality=85
+        )
+
+        assert "800x600!" in cmd
+
+    def test_build_resize_command_fill_requires_dimensions(self):
+        """Test that 'fill' strategy requires both dimensions."""
+        with pytest.raises(ValueError):
+            self.resizer.build_resize_command(
+                Path("input.jpg"),
+                Path("output.jpg"),
+                width=800,
+                height=None,
+                strategy="fill",
+                quality=85
+            )
+
+    @patch("subprocess.run")
+    def test_resize_image_success(self, mock_run):
+        """Test successful image resize."""
+        mock_run.return_value = MagicMock(returncode=0)
+
+        result = self.resizer.resize_image(
+            Path("input.jpg"),
+            Path("output/output.jpg"),
+            width=800,
+            height=None,
+            strategy="fit",
+            quality=85
+        )
+
+        assert result is True
+        mock_run.assert_called_once()
+
+    @patch("subprocess.run")
+    def test_resize_image_dry_run(self, mock_run):
+        """Test resize in dry-run mode."""
+        resizer = ImageResizer(dry_run=True)
+
+        result = resizer.resize_image(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=800,
+            height=None
+        )
+
+        assert result is True
+        mock_run.assert_not_called()
+
+    @patch("subprocess.run")
+    def test_resize_image_failure(self, mock_run):
+        """Test resize failure handling."""
+        mock_run.side_effect = Exception("Resize failed")
+
+        result = self.resizer.resize_image(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=800,
+            height=None
+        )
+
+        assert result is False
+
+
+class TestCollectImages:
+    """Test image collection functionality."""
+
+    def test_collect_images_from_file(self, tmp_path):
+        """Test collecting a single image file."""
+        img_file = tmp_path / "test.jpg"
+        img_file.touch()
+
+        images = collect_images([img_file])
+        assert len(images) == 1
+        assert images[0] == img_file
+
+    def test_collect_images_from_directory(self, tmp_path):
+        """Test collecting images from directory."""
+        (tmp_path / "image1.jpg").touch()
+        (tmp_path / "image2.png").touch()
+        (tmp_path / "text.txt").touch()
+
+        images = collect_images([tmp_path])
+        assert len(images) == 2
+        assert all(img.suffix.lower() in {'.jpg', '.png'} for img in images)
+
+    def test_collect_images_recursive(self, tmp_path):
+        """Test recursive image collection."""
+        subdir = tmp_path / "subdir"
+        subdir.mkdir()
+        (tmp_path / "image1.jpg").touch()
+        (subdir / "image2.jpg").touch()
+
+        images = collect_images([tmp_path], recursive=True)
+        assert len(images) == 2
+
+        images_non_recursive = collect_images([tmp_path], recursive=False)
+        assert len(images_non_recursive) == 1
+
+    def test_collect_images_filters_extensions(self, tmp_path):
+        """Test that only image files are collected."""
+        (tmp_path / "image.jpg").touch()
+        (tmp_path / "doc.pdf").touch()
+        (tmp_path / "text.txt").touch()
+
+        images = collect_images([tmp_path])
+        assert len(images) == 1
+        assert images[0].suffix.lower() == '.jpg'
+
+    def test_collect_images_multiple_paths(self, tmp_path):
+        """Test collecting from multiple paths."""
+        dir1 = tmp_path / "dir1"
+        dir2 = tmp_path / "dir2"
+        dir1.mkdir()
+        dir2.mkdir()
+
+        (dir1 / "image1.jpg").touch()
+        (dir2 / "image2.png").touch()
+
+        images = collect_images([dir1, dir2])
+        assert len(images) == 2
+
+
+class TestBatchResize:
+    """Test batch resize functionality."""
+
+    def setup_method(self):
+        """Set up test fixtures."""
+        self.resizer = ImageResizer(verbose=False, dry_run=False)
+
+    @patch.object(ImageResizer, "resize_image")
+    def test_batch_resize_success(self, mock_resize, tmp_path):
+        """Test successful batch resize."""
+        mock_resize.return_value = True
+
+        input_images = [
+            tmp_path / "image1.jpg",
+            tmp_path / "image2.jpg"
+        ]
+        for img in input_images:
+            img.touch()
+
+        output_dir = tmp_path / "output"
+
+        success, fail = self.resizer.batch_resize(
+            input_images,
+            output_dir,
+            width=800,
+            height=None,
+            strategy="fit"
+        )
+
+        assert success == 2
+        assert fail == 0
+        assert mock_resize.call_count == 2
+
+    @patch.object(ImageResizer, "resize_image")
+    def test_batch_resize_with_failures(self, mock_resize, tmp_path):
+        """Test batch resize with some failures."""
+        mock_resize.side_effect = [True, False, True]
+
+        input_images = [
+            tmp_path / "image1.jpg",
+            tmp_path / "image2.jpg",
+            tmp_path / "image3.jpg"
+        ]
+        for img in input_images:
+            img.touch()
+
+        output_dir = tmp_path / "output"
+
+        success, fail = self.resizer.batch_resize(
+            input_images,
+            output_dir,
+            width=800,
+            height=None
+        )
+
+        assert success == 2
+        assert fail == 1
+
+    @patch.object(ImageResizer, "resize_image")
+    def test_batch_resize_format_conversion(self, mock_resize, tmp_path):
+        """Test batch resize with format conversion."""
+        mock_resize.return_value = True
+
+        input_images = [tmp_path / "image.png"]
+        input_images[0].touch()
+
+        output_dir = tmp_path / "output"
+
+        self.resizer.batch_resize(
+            input_images,
+            output_dir,
+            width=800,
+            height=None,
+            format_ext="jpg"
+        )
+
+        # Check that resize_image was called with .jpg extension
+        call_args = mock_resize.call_args[0]
+        assert call_args[1].suffix == ".jpg"
+
+
+class TestResizeStrategies:
+    """Test different resize strategies."""
+
+    def setup_method(self):
+        """Set up test fixtures."""
+        self.resizer = ImageResizer()
+
+    def test_fit_strategy_maintains_aspect(self):
+        """Test that 'fit' strategy maintains aspect ratio."""
+        cmd = self.resizer.build_resize_command(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=800,
+            height=600,
+            strategy="fit",
+            quality=85
+        )
+
+        # Should have resize without ^ or !
+        resize_idx = cmd.index("-resize")
+        geometry = cmd[resize_idx + 1]
+        assert "^" not in geometry
+        assert "!" not in geometry
+
+    def test_cover_strategy_fills_dimensions(self):
+        """Test that 'cover' strategy fills dimensions."""
+        cmd = self.resizer.build_resize_command(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=800,
+            height=600,
+            strategy="cover",
+            quality=85
+        )
+
+        resize_idx = cmd.index("-resize")
+        geometry = cmd[resize_idx + 1]
+        assert "^" in geometry
+
+    def test_exact_strategy_ignores_aspect(self):
+        """Test that 'exact' strategy ignores aspect ratio."""
+        cmd = self.resizer.build_resize_command(
+            Path("input.jpg"),
+            Path("output.jpg"),
+            width=800,
+            height=600,
+            strategy="exact",
+            quality=85
+        )
+
+        resize_idx = cmd.index("-resize")
+        geometry = cmd[resize_idx + 1]
+        assert "!" in geometry
+
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v"])