cokit-cli 1.0.1 → 1.0.3

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

@@ -0,0 +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"])

package/skills/ck-media-processing/scripts/tests/test_media_convert.py

@@ -0,0 +1,259 @@
+#!/usr/bin/env python3
+"""Tests for media_convert.py"""
+
+import sys
+from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+# Add parent directory to path
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+from media_convert import (
+    build_audio_command,
+    build_image_command,
+    build_video_command,
+    check_dependencies,
+    convert_file,
+    detect_media_type,
+)
+
+
+class TestMediaTypeDetection:
+    """Test media type detection."""
+
+    def test_detect_video_formats(self):
+        """Test video format detection."""
+        assert detect_media_type(Path("test.mp4")) == "video"
+        assert detect_media_type(Path("test.mkv")) == "video"
+        assert detect_media_type(Path("test.avi")) == "video"
+        assert detect_media_type(Path("test.mov")) == "video"
+
+    def test_detect_audio_formats(self):
+        """Test audio format detection."""
+        assert detect_media_type(Path("test.mp3")) == "audio"
+        assert detect_media_type(Path("test.aac")) == "audio"
+        assert detect_media_type(Path("test.flac")) == "audio"
+        assert detect_media_type(Path("test.wav")) == "audio"
+
+    def test_detect_image_formats(self):
+        """Test image format detection."""
+        assert detect_media_type(Path("test.jpg")) == "image"
+        assert detect_media_type(Path("test.png")) == "image"
+        assert detect_media_type(Path("test.gif")) == "image"
+        assert detect_media_type(Path("test.webp")) == "image"
+
+    def test_detect_unknown_format(self):
+        """Test unknown format detection."""
+        assert detect_media_type(Path("test.txt")) == "unknown"
+        assert detect_media_type(Path("test.doc")) == "unknown"
+
+    def test_case_insensitive(self):
+        """Test case-insensitive detection."""
+        assert detect_media_type(Path("TEST.MP4")) == "video"
+        assert detect_media_type(Path("TEST.JPG")) == "image"
+
+
+class TestCommandBuilding:
+    """Test command building functions."""
+
+    def test_build_video_command_web_preset(self):
+        """Test video command with web preset."""
+        cmd = build_video_command(
+            Path("input.mp4"),
+            Path("output.mp4"),
+            preset="web"
+        )
+
+        assert "ffmpeg" in cmd
+        assert "-i" in cmd
+        assert str(Path("input.mp4")) in cmd
+        assert "-c:v" in cmd
+        assert "libx264" in cmd
+        assert "-crf" in cmd
+        assert "23" in cmd
+        assert "-preset" in cmd
+        assert "medium" in cmd
+        assert str(Path("output.mp4")) in cmd
+
+    def test_build_video_command_archive_preset(self):
+        """Test video command with archive preset."""
+        cmd = build_video_command(
+            Path("input.mp4"),
+            Path("output.mp4"),
+            preset="archive"
+        )
+
+        assert "18" in cmd  # CRF for archive
+        assert "slow" in cmd  # Preset for archive
+
+    def test_build_audio_command_mp3(self):
+        """Test audio command for MP3 output."""
+        cmd = build_audio_command(
+            Path("input.wav"),
+            Path("output.mp3"),
+            preset="web"
+        )
+
+        assert "ffmpeg" in cmd
+        assert "-c:a" in cmd
+        assert "libmp3lame" in cmd
+        assert "-b:a" in cmd
+
+    def test_build_audio_command_flac(self):
+        """Test audio command for FLAC (lossless)."""
+        cmd = build_audio_command(
+            Path("input.wav"),
+            Path("output.flac"),
+            preset="web"
+        )
+
+        assert "flac" in cmd
+        assert "-b:a" not in cmd  # No bitrate for lossless
+
+    def test_build_image_command(self):
+        """Test image command building."""
+        cmd = build_image_command(
+            Path("input.png"),
+            Path("output.jpg"),
+            preset="web"
+        )
+
+        assert "magick" in cmd
+        assert str(Path("input.png")) in cmd
+        assert "-quality" in cmd
+        assert "85" in cmd
+        assert "-strip" in cmd
+        assert str(Path("output.jpg")) in cmd
+
+
+class TestDependencyCheck:
+    """Test dependency checking."""
+
+    @patch("subprocess.run")
+    def test_check_dependencies_both_available(self, mock_run):
+        """Test when both tools are available."""
+        mock_run.return_value = MagicMock(returncode=0)
+        ffmpeg_ok, magick_ok = check_dependencies()
+        assert ffmpeg_ok is True
+        assert magick_ok is True
+
+    @patch("subprocess.run")
+    def test_check_dependencies_ffmpeg_only(self, mock_run):
+        """Test when only FFmpeg is available."""
+        def side_effect(*args, **kwargs):
+            if "ffmpeg" in args[0]:
+                return MagicMock(returncode=0)
+            return MagicMock(returncode=1)
+
+        mock_run.side_effect = side_effect
+        ffmpeg_ok, magick_ok = check_dependencies()
+        assert ffmpeg_ok is True
+        assert magick_ok is False
+
+
+class TestFileConversion:
+    """Test file conversion functionality."""
+
+    @patch("subprocess.run")
+    @patch("media_convert.detect_media_type")
+    def test_convert_video_file_dry_run(self, mock_detect, mock_run):
+        """Test video conversion in dry-run mode."""
+        mock_detect.return_value = "video"
+
+        result = convert_file(
+            Path("input.mp4"),
+            Path("output.mp4"),
+            preset="web",
+            dry_run=True
+        )
+
+        assert result is True
+        mock_run.assert_not_called()
+
+    @patch("subprocess.run")
+    @patch("media_convert.detect_media_type")
+    def test_convert_image_file_success(self, mock_detect, mock_run):
+        """Test successful image conversion."""
+        mock_detect.return_value = "image"
+        mock_run.return_value = MagicMock(returncode=0)
+
+        result = convert_file(
+            Path("input.png"),
+            Path("output.jpg"),
+            preset="web"
+        )
+
+        assert result is True
+        mock_run.assert_called_once()
+
+    @patch("subprocess.run")
+    @patch("media_convert.detect_media_type")
+    def test_convert_file_error(self, mock_detect, mock_run):
+        """Test conversion error handling."""
+        mock_detect.return_value = "video"
+        mock_run.side_effect = Exception("Conversion failed")
+
+        result = convert_file(
+            Path("input.mp4"),
+            Path("output.mp4")
+        )
+
+        assert result is False
+
+    @patch("media_convert.detect_media_type")
+    def test_convert_unknown_format(self, mock_detect):
+        """Test conversion with unknown format."""
+        mock_detect.return_value = "unknown"
+
+        result = convert_file(
+            Path("input.txt"),
+            Path("output.txt")
+        )
+
+        assert result is False
+
+
+class TestQualityPresets:
+    """Test quality preset functionality."""
+
+    def test_web_preset_settings(self):
+        """Test web preset values."""
+        cmd = build_video_command(
+            Path("input.mp4"),
+            Path("output.mp4"),
+            preset="web"
+        )
+
+        cmd_str = " ".join(cmd)
+        assert "23" in cmd_str  # CRF
+        assert "128k" in cmd_str  # Audio bitrate
+
+    def test_archive_preset_settings(self):
+        """Test archive preset values."""
+        cmd = build_video_command(
+            Path("input.mp4"),
+            Path("output.mp4"),
+            preset="archive"
+        )
+
+        cmd_str = " ".join(cmd)
+        assert "18" in cmd_str  # Higher quality CRF
+        assert "192k" in cmd_str  # Higher audio bitrate
+
+    def test_mobile_preset_settings(self):
+        """Test mobile preset values."""
+        cmd = build_video_command(
+            Path("input.mp4"),
+            Path("output.mp4"),
+            preset="mobile"
+        )
+
+        cmd_str = " ".join(cmd)
+        assert "26" in cmd_str  # Lower quality CRF
+        assert "96k" in cmd_str  # Lower audio bitrate
+
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v"])