from unittest.mock import Mock

from django.db import models

from servala.core.crd import generate_custom_form_class
from servala.core.models import ControlPlaneCRD


def test_custom_model_form_class_is_none_when_no_form_config():
    crd = Mock(spec=ControlPlaneCRD)
    service_def = Mock()
    service_def.form_config = None
    crd.service_definition = service_def
    crd.django_model = Mock()

    if not (
        crd.django_model
        and crd.service_definition
        and crd.service_definition.form_config
        and crd.service_definition.form_config.get("fieldsets")
    ):
        result = None
    else:
        result = generate_custom_form_class(
            crd.service_definition.form_config, crd.django_model
        )

    assert result is None


def test_custom_model_form_class_returns_class_when_form_config_exists():

    crd = Mock(spec=ControlPlaneCRD)
    service_def = Mock()
    service_def.form_config = {
        "fieldsets": [
            {
                "title": "General",
                "fields": [
                    {
                        "type": "text",
                        "label": "Name",
                        "controlplane_field_mapping": "name",
                        "required": True,
                    }
                ],
            }
        ]
    }
    crd.service_definition = service_def

    class TestModel(models.Model):
        name = models.CharField(max_length=100)

        class Meta:
            app_label = "test"

    crd.django_model = TestModel

    if not (
        crd.django_model
        and crd.service_definition
        and crd.service_definition.form_config
        and crd.service_definition.form_config.get("fieldsets")
    ):
        result = None
    else:
        result = generate_custom_form_class(
            crd.service_definition.form_config, crd.django_model
        )

    assert result is not None
    assert hasattr(result, "form_config")