tests/backend/test_forge.py

import pytest
from animetix.models import CreativeFusion
from django.contrib.auth.models import User


@pytest.mark.django_db
def test_creative_fusion_model_creation():
    user = User.objects.create_user(username="forger", password="pwd")
    parent = CreativeFusion.objects.create(
        title_a="Naruto",
        title_b="Cyberpunk",
        media_type_a="Anime",
        media_type_b="Anime",
        scenario_text="Test",
        creator=user,
    )
    remix = CreativeFusion.objects.create(
        title_a="Naruto",
        title_b="Cyberpunk",
        media_type_a="Anime",
        media_type_b="Anime",
        scenario_text="Test Remix",
        creator=user,
        parent=parent,
        chaos_level=80,
    )
    assert remix.parent == parent
    assert remix.chaos_level == 80


from unittest.mock import patch  # noqa: E402

from animetix.tasks import generate_fusion_scenario_task  # noqa: E402


@patch("animetix.tasks.get_container")
def test_generate_fusion_scenario_with_params(mock_get_container):
    mock_service = mock_get_container.return_value.agentic.llm_service.return_value
    mock_service.generate_fusion_scenario.return_value = "Test Scenario"

    result = generate_fusion_scenario_task(
        "Anime",
        {"title": "A"},
        {"title": "B"},
        "Français",
        chaos_level=80,
        universe_balance=20,
        art_style="Ghibli",
    )

    assert result == "Test Scenario"
    mock_service.generate_fusion_scenario.assert_called_once()
    # Check if params are passed
    kwargs = mock_service.generate_fusion_scenario.call_args[1]
    assert kwargs.get("chaos_level") == 80