import copy

import pytest
from pydantic import ValidationError

from app.schemas.requests import GenerateRoadmapRequest
from app.schemas.roadmap import Roadmap
from app.schemas.roadmap import ResourceType
from app.services.id_generator import assign_backend_ids


def valid_payload():
    resource_a = {
        "title": "Python for Beginners",
        "url": "https://www.youtube.com/watch?v=kqtD5dpn9C8",
        "type": "youtube_video",
        "source": "YouTube",
        "is_free": True,
    }
    resource_b = {
        "title": "Python Tutorial",
        "url": "https://www.w3schools.com/python/",
        "type": "article",
        "source": "Tavily",
        "is_free": True,
    }
    phases = []
    for phase_index in range(1, 4):
        nodes = []
        for node_index in range(1, 4):
            nodes.append(
                {
                    "title": f"Node {node_index}",
                    "description": "Learn the core idea with simple examples and short practice tasks.",
                    "estimated_completion_time": "2-3 hours",
                    "resources": [resource_a, resource_b],
                }
            )
        phases.append(
            {
                "title": f"Phase {phase_index}",
                "level": "beginner",
                "goal": "Build practical beginner confidence.",
                "estimated_duration": "2 weeks",
                "nodes": nodes,
                "project": {
                    "title": f"Project {phase_index}",
                    "brief": "Build a small practical project that demonstrates the phase skills.",
                    "tools_needed": ["Browser"],
                    "resources": [resource_a, resource_b],
                },
            }
        )
    payload = {
        "skill": "Python for Beginners",
        "overview": "A beginner roadmap with practical examples and projects for learning step by step.",
        "estimated_total_duration": "6-8 weeks",
        "phases": phases,
        "metadata": {"model_used": "test", "resource_sources": ["test"], "cached": False},
    }
    return assign_backend_ids(payload, "python")


def test_valid_roadmap_passes():
    roadmap = Roadmap.model_validate(valid_payload())
    assert len(roadmap.phases) == 3
    assert len(roadmap.phases[0].nodes) == 3


def test_fewer_than_three_phases_fails():
    payload = valid_payload()
    payload["phases"] = payload["phases"][:2]
    payload["projects"] = payload["projects"][:2]
    with pytest.raises(ValidationError):
        Roadmap.model_validate(payload)


def test_node_with_fewer_than_two_resources_fails():
    payload = valid_payload()
    payload["phases"][0]["nodes"][0]["resources"] = payload["phases"][0]["nodes"][0]["resources"][:1]
    with pytest.raises(ValidationError):
        Roadmap.model_validate(payload)


def test_paid_resource_with_is_free_false_fails():
    payload = valid_payload()
    payload["phases"][0]["nodes"][0]["resources"][0]["is_free"] = False
    with pytest.raises(ValidationError):
        Roadmap.model_validate(payload)


def test_invalid_resource_type_fails():
    payload = copy.deepcopy(valid_payload())
    payload["phases"][0]["nodes"][0]["resources"][0]["type"] = "paid_course"
    with pytest.raises(ValidationError):
        Roadmap.model_validate(payload)


def test_request_defaults_include_all_resource_types():
    request = GenerateRoadmapRequest(skill="Learn Python")
    assert request.preferred_resource_types == list(ResourceType)