api-snapshot-cli/app/tests/test_cli.py

"""Tests for the CLI module."""

from click.testing import CliRunner

from api_snapshot.cli.cli import (
    delete_snapshot,
    list_snapshots,
    main,
    snapshot_info,
)
from api_snapshot.cli.serve import serve_command, serve_info_command
from api_snapshot.snapshot.manager import SnapshotManager


class TestMainCommand:
    """Tests for main command group."""

    def test_main_help(self):
        """Test main command help."""
        runner = CliRunner()

        result = runner.invoke(main, ["--help"])

        assert result.exit_code == 0
        assert "API Snapshot CLI" in result.output
        assert "record" in result.output
        assert "serve" in result.output
        assert "list" in result.output

    def test_main_version(self):
        """Test main command version."""
        runner = CliRunner()

        result = runner.invoke(main, ["--version"])

        assert result.exit_code == 0
        assert "0.1.0" in result.output


class TestListCommand:
    """Tests for list command."""

    def test_list_empty(self, temp_dir):
        """Test listing with no snapshots."""
        runner = CliRunner()

        result = runner.invoke(list_snapshots, obj={"snapshot_dir": temp_dir})

        assert result.exit_code == 0
        assert "No snapshots found" in result.output

    def test_list_with_snapshots(self, temp_dir):
        """Test listing with snapshots."""
        from api_snapshot.recorder.recorder import (
            RecordedRequest,
            RecordedResponse,
            RequestResponsePair,
        )

        manager = SnapshotManager(temp_dir)

        req = RecordedRequest(
            method="GET",
            url="https://api.example.com",
            headers={},
            body=None,
        )
        resp = RecordedResponse(
            status_code=200,
            headers={},
            body="{}",
            latency_ms=100,
        )
        pair = RequestResponsePair(request=req, response=resp)

        manager.save_snapshot("test-snap", requests=[pair])

        runner = CliRunner()

        result = runner.invoke(list_snapshots, obj={"snapshot_dir": temp_dir})

        assert result.exit_code == 0
        assert "test-snap" in result.output


class TestSnapshotInfoCommand:
    """Tests for info command."""

    def test_info_nonexistent(self, temp_dir):
        """Test info for non-existent snapshot."""
        runner = CliRunner()

        result = runner.invoke(snapshot_info, ["nonexistent"],
                               obj={"snapshot_dir": temp_dir})

        assert result.exit_code != 0
        assert "not found" in result.output

    def test_info_existing(self, temp_dir):
        """Test info for existing snapshot."""
        from api_snapshot.recorder.recorder import (
            RecordedRequest,
            RecordedResponse,
            RequestResponsePair,
        )

        manager = SnapshotManager(temp_dir)

        req = RecordedRequest(
            method="GET",
            url="https://api.example.com/users",
            headers={},
            body=None,
        )
        resp = RecordedResponse(
            status_code=200,
            headers={"Content-Type": "application/json"},
            body='{"users": []}',
            latency_ms=100,
        )
        pair = RequestResponsePair(request=req, response=resp)

        manager.save_snapshot(
            "my-snapshot",
            requests=[pair],
            description="Test description",
        )

        runner = CliRunner()

        result = runner.invoke(snapshot_info, ["my-snapshot"],
                               obj={"snapshot_dir": temp_dir})

        assert result.exit_code == 0
        assert "my-snapshot" in result.output
        assert "Test description" in result.output


class TestDeleteCommand:
    """Tests for delete command."""

    def test_delete_nonexistent(self, temp_dir):
        """Test deleting non-existent snapshot."""
        runner = CliRunner()

        result = runner.invoke(
            delete_snapshot,
            ["nonexistent", "--force"],
            obj={"snapshot_dir": temp_dir},
        )

        assert result.exit_code != 0
        assert "not found" in result.output

    def test_delete_with_confirmation(self, temp_dir):
        """Test deleting with confirmation."""
        from api_snapshot.recorder.recorder import (
            RecordedRequest,
            RecordedResponse,
            RequestResponsePair,
        )

        manager = SnapshotManager(temp_dir)

        req = RecordedRequest(
            method="GET",
            url="https://api.example.com",
            headers={},
            body=None,
        )
        resp = RecordedResponse(
            status_code=200,
            headers={},
            body="{}",
            latency_ms=50,
        )
        pair = RequestResponsePair(request=req, response=resp)

        manager.save_snapshot("to-delete", requests=[pair])

        runner = CliRunner()

        result = runner.invoke(
            delete_snapshot,
            ["to-delete", "--force"],
            obj={"snapshot_dir": temp_dir},
        )

        assert result.exit_code == 0
        assert "Deleted" in result.output

        assert not manager.snapshot_exists("to-delete")


class TestServeCommand:
    """Tests for serve command."""

    def test_serve_nonexistent_snapshot(self, temp_dir):
        """Test serving non-existent snapshot."""
        runner = CliRunner()

        result = runner.invoke(
            serve_command,
            ["nonexistent"],
            obj={"snapshot_dir": temp_dir},
        )

        assert result.exit_code != 0
        assert "not found" in result.output

    def test_serve_info_nonexistent(self, temp_dir):
        """Test serve-info for non-existent snapshot."""
        runner = CliRunner()

        result = runner.invoke(
            serve_info_command,
            ["nonexistent"],
            obj={"snapshot_dir": temp_dir},
        )

        assert result.exit_code != 0
        assert "not found" in result.output


class TestCLIIntegration:
    """Integration tests for CLI."""

    def test_cli_commands_chain(self, temp_dir):
        """Test chaining CLI commands together."""
        from api_snapshot.recorder.recorder import (
            RecordedRequest,
            RecordedResponse,
            RequestResponsePair,
        )

        runner = CliRunner()

        req = RecordedRequest(
            method="GET",
            url="https://api.example.com/test",
            headers={},
            body=None,
        )
        resp = RecordedResponse(
            status_code=200,
            headers={"Content-Type": "application/json"},
            body='{"message": "hello"}',
            latency_ms=50,
        )
        pair = RequestResponsePair(request=req, response=resp)

        manager = SnapshotManager(temp_dir)
        manager.save_snapshot("test-chain", requests=[pair])

        list_result = runner.invoke(
            list_snapshots,
            obj={"snapshot_dir": temp_dir},
        )

        assert list_result.exit_code == 0
        assert "test-chain" in list_result.output

        info_result = runner.invoke(
            snapshot_info,
            ["test-chain"],
            obj={"snapshot_dir": temp_dir},
        )

        assert info_result.exit_code == 0
        assert "test-chain" in info_result.output