"""Tests for cli.folderzip module."""

from __future__ import annotations

from pathlib import Path
from unittest.mock import patch

import pyflowx as px
from pyflowx.cli import folderzip
from pyflowx.cli._ops import files


# ---------------------------------------------------------------------- #
# archive_folder
# ---------------------------------------------------------------------- #
class TestArchiveFolder:
    """Test archive_folder function."""

    def test_archive_folder(self, tmp_path: Path) -> None:
        """Should archive a folder."""
        folder = tmp_path / "test_folder"
        folder.mkdir()
        (folder / "test.txt").write_text("test content")

        with patch("shutil.make_archive") as mock_archive:
            files.archive_folder(folder)
            assert mock_archive.called


# ---------------------------------------------------------------------- #
# zip_folders
# ---------------------------------------------------------------------- #
class TestZipFolders:
    """Test zip_folders function."""

    def test_zip_folders_with_cwd(self, tmp_path: Path) -> None:
        """Should zip folders in cwd."""
        # Create some folders
        (tmp_path / "folder1").mkdir()
        (tmp_path / "folder2").mkdir()
        (tmp_path / ".git").mkdir()  # Should be ignored

        with patch.object(files, "archive_folder") as mock_archive:
            files.zip_folders(str(tmp_path))
            # Should archive folder1 and folder2, but not .git
            assert mock_archive.call_count == 2

    def test_zip_folders_nonexistent_cwd(self, tmp_path: Path) -> None:
        """Should handle nonexistent cwd."""
        files.zip_folders(str(tmp_path / "nonexistent"))
        # Should print error message and return


# ---------------------------------------------------------------------- #
# 函数注册
# ---------------------------------------------------------------------- #
class TestRegisteredFunctions:
    """Test that folderzip functions are registered."""

    def test_folderzip_default_spec(self) -> None:
        """folderzip_default should be a registered callable."""
        # folderzip_default 现在是通过 @px.register_fn 注册的普通函数, 不是 TaskSpec
        assert callable(files.folderzip_default)


# ---------------------------------------------------------------------- #
# main function
# ---------------------------------------------------------------------- #
class TestMain:
    """Test main function."""

    def test_main_calls_run_yaml(self) -> None:
        """main() should parse args and call px.run_yaml()."""
        with patch("sys.argv", ["folderzip", "--cwd", "."]), patch.object(px, "run_yaml") as mock_run_yaml:
            folderzip.main()
            assert mock_run_yaml.called
            # 验证 jobs 参数为 "zip" (YAML 中的 job 名)
            assert mock_run_yaml.call_args.kwargs["jobs"] == "zip"