otko/tests/gui/test_section_forms.py

"""Regression: clicking a FiberSection / SectionAggregator row in the
Section Library must not crash ``form_for`` with KeyError."""

from __future__ import annotations

import pytest

pytest.importorskip("PySide6")

from otko.core import (  # noqa: E402
    AggregatorDOF,
    ElasticSection,
    FiberSection,
    RectangularPatch,
    SectionAggregator,
)
from otko.views.dialogs.section_forms import (  # noqa: E402
    FORM_REGISTRY,
    form_for,
)


def test_fiber_section_has_registered_form() -> None:
    assert "FiberSection" in FORM_REGISTRY


def test_section_aggregator_has_registered_form() -> None:
    assert "SectionAggregator" in FORM_REGISTRY


@pytest.mark.gui
def test_form_for_elastic_section_returns_editable_form(qtbot) -> None:  # type: ignore[no-untyped-def]
    s = ElasticSection(
        id=1,
        name="Rect",
        E=200e9,
        A=0.01,
        Iz=1e-5,
        Iy=1e-5,
        G=80e9,
        J=1e-6,
    )
    f = form_for(s)
    qtbot.addWidget(f)
    assert f.type_label == "Elastic Section"


@pytest.mark.gui
def test_form_for_fiber_section_returns_summary(qtbot) -> None:  # type: ignore[no-untyped-def]
    s = FiberSection(
        id=1,
        name="RC",
        patches=[
            RectangularPatch(
                material_id=1,
                n_fib_y=4,
                n_fib_z=4,
                y_i=-0.1,
                z_i=-0.15,
                y_j=0.1,
                z_j=0.15,
            )
        ],
    )
    f = form_for(s)
    qtbot.addWidget(f)
    assert f.type_label == "Fiber Section"
    # read() round-trips without crashing.
    out = f.read(1)
    assert isinstance(out, FiberSection)
    assert len(out.patches) == 1


@pytest.mark.gui
def test_form_for_section_aggregator_returns_summary(qtbot) -> None:  # type: ignore[no-untyped-def]
    s = SectionAggregator(
        id=2,
        name="Agg",
        section_id=1,
        pairings=[AggregatorDOF(material_id=3, dof="T")],
    )
    f = form_for(s)
    qtbot.addWidget(f)
    assert f.type_label == "Section Aggregator"


@pytest.mark.gui
def test_form_for_unknown_type_falls_back(qtbot) -> None:  # type: ignore[no-untyped-def]
    """An unknown section type must NOT raise — the dialog gets a
    placeholder form so the whole UI doesn't go down."""

    class _Mystery:
        type = "NotRegistered"
        id = 99
        name = "???"

    f = form_for(_Mystery())
    qtbot.addWidget(f)
    # Placeholder form exists; no exception raised.
    assert f is not None