SousChef

"""Unit tests for migration configuration public API functions."""

import json

import pytest

from souschef.assessment import calculate_activity_breakdown
from souschef.migration_config import (
    DeploymentTarget,
    MigrationConfig,
    MigrationStandard,
    ValidationTool,
)


@pytest.fixture
def sample_migration_config():
    """Create a sample migration configuration."""
    return MigrationConfig(
        deployment_target=DeploymentTarget.AWX,
        migration_standard=MigrationStandard.STANDARD,
        inventory_source="hosts.ini",
        validation_tools=[ValidationTool.ANSIBLE_LINT, ValidationTool.MOLECULE],
        target_python_version="3.10",
        target_ansible_version="2.15+",
    )


def test_migration_config_export_to_json(sample_migration_config):
    """Test that migration config can be exported to valid JSON."""
    config_dict = sample_migration_config.to_dict()
    json_str = json.dumps(config_dict, indent=2)

    # Verify JSON is valid
    parsed = json.loads(json_str)
    assert parsed["deployment_target"] == "awx"
    assert parsed["target_python_version"] == "3.10"
    assert len(parsed["validation_tools"]) == 2


def test_migration_config_all_fields(sample_migration_config):
    """Test that all configuration fields are accessible."""
    assert sample_migration_config.deployment_target == DeploymentTarget.AWX
    assert sample_migration_config.migration_standard == MigrationStandard.STANDARD
    assert sample_migration_config.inventory_source == "hosts.ini"
    assert ValidationTool.ANSIBLE_LINT in sample_migration_config.validation_tools
    assert sample_migration_config.target_python_version == "3.10"
    assert sample_migration_config.target_ansible_version == "2.15+"


def test_calculate_activity_breakdown_with_valid_cookbook(tmp_path):
    """Test activity breakdown calculation with a valid cookbook."""
    # Create a minimal cookbook structure
    cookbook_dir = tmp_path / "test_cookbook"
    cookbook_dir.mkdir()

    # Create metadata.rb
    metadata_file = cookbook_dir / "metadata.rb"
    metadata_file.write_text("""name 'test_cookbook'
version '1.0.0'
maintainer 'Test'
""")

    # Create a recipes directory with a recipe
    recipes_dir = cookbook_dir / "recipes"
    recipes_dir.mkdir()
    recipe_file = recipes_dir / "default.rb"
    recipe_file.write_text("""package 'nginx' do
  action :install
end
""")

    # Call the function
    result = calculate_activity_breakdown(str(cookbook_dir), "phased")

    # Verify structure
    assert "summary" in result
    assert "activities" in result
    assert "total_manual_hours" in result["summary"]
    assert "total_ai_assisted_hours" in result["summary"]

    if result["activities"]:
        activity = result["activities"][0]
        assert "count" in activity
        assert "description" in activity


def test_calculate_activity_breakdown_with_invalid_path():
    """Test activity breakdown with invalid cookbook path."""
    result = calculate_activity_breakdown("/nonexistent/path", "phased")

    # Should return error
    assert "error" in result


def test_migration_config_json_structure():
    """Test that exported JSON has all required fields."""
    config = MigrationConfig(
        deployment_target=DeploymentTarget.NATIVE,
        migration_standard=MigrationStandard.FLAT,
        inventory_source="inventory.yml",
        validation_tools=[ValidationTool.ANSIBLE_LINT],
        target_python_version="3.11",
        target_ansible_version="2.16+",
    )

    config_dict = config.to_dict()

    # Verify all fields present
    assert "deployment_target" in config_dict
    assert "migration_standard" in config_dict
    assert "inventory_source" in config_dict
    assert "validation_tools" in config_dict
    assert "target_python_version" in config_dict
    assert "target_ansible_version" in config_dict

    assert config_dict["deployment_target"] == "native"
    assert config_dict["migration_standard"] == "flat"
    assert isinstance(config_dict["validation_tools"], list)