# This file was generated with the assistance of an AI coding tool.
from __future__ import annotations

import ifcopenshell
import ifcopenshell.api.cost
import ifcopenshell.api.owner.settings
import ifcopenshell.api.project
import ifcopenshell.api.root
import ifcopenshell.api.unit
import pytest

from ifcquery.cost import cost


@pytest.fixture
def cost_model():
    """Create an IFC4 model with a cost schedule, a top-level item, and one nested subitem."""
    f = ifcopenshell.api.project.create_file()
    ifcopenshell.api.owner.settings.get_user = lambda ifc: (ifc.by_type("IfcPersonAndOrganization") or [None])[0]
    ifcopenshell.api.owner.settings.get_application = lambda ifc: (ifc.by_type("IfcApplication") or [None])[0]

    project = ifcopenshell.api.root.create_entity(f, ifc_class="IfcProject", name="TestProject")
    ifcopenshell.api.unit.assign_unit(f)

    cs = ifcopenshell.api.cost.add_cost_schedule(f, name="Bill of Quantities")
    item = ifcopenshell.api.cost.add_cost_item(f, cost_schedule=cs)
    ifcopenshell.api.cost.edit_cost_item(f, cost_item=item, attributes={"Name": "Concrete Works"})
    cv = ifcopenshell.api.cost.add_cost_value(f, parent=item)
    ifcopenshell.api.cost.edit_cost_value(f, cost_value=cv, attributes={"AppliedValue": 1200.0, "Category": "material"})

    # Add a nested subitem
    subitem = ifcopenshell.api.cost.add_cost_item(f, cost_item=item)
    ifcopenshell.api.cost.edit_cost_item(f, cost_item=subitem, attributes={"Name": "Formwork"})

    return f


class TestCost:
    def test_returns_list(self, cost_model):
        result = cost(cost_model)
        assert isinstance(result, list)

    def test_finds_cost_schedule(self, cost_model):
        result = cost(cost_model)
        assert len(result) == 1

    def test_schedule_has_name(self, cost_model):
        result = cost(cost_model)
        assert result[0]["name"] == "Bill of Quantities"

    def test_schedule_has_id(self, cost_model):
        result = cost(cost_model)
        assert isinstance(result[0]["id"], int)
        assert result[0]["id"] > 0

    def test_schedule_has_items(self, cost_model):
        result = cost(cost_model)
        assert len(result[0]["items"]) == 1

    def test_item_has_required_fields(self, cost_model):
        result = cost(cost_model)
        item = result[0]["items"][0]
        assert "id" in item
        assert "name" in item
        assert "values" in item
        assert "subitems" in item

    def test_item_name(self, cost_model):
        result = cost(cost_model)
        assert result[0]["items"][0]["name"] == "Concrete Works"

    def test_item_has_values(self, cost_model):
        result = cost(cost_model)
        values = result[0]["items"][0]["values"]
        assert len(values) == 1
        assert "formula" in values[0]
        assert "category" in values[0]

    def test_item_value_category(self, cost_model):
        result = cost(cost_model)
        values = result[0]["items"][0]["values"]
        assert values[0]["category"] == "material"

    def test_empty_model_returns_empty_list(self, model):
        result = cost(model)
        assert result == []

    def test_max_depth_none_returns_full_tree(self, cost_model):
        result = cost(cost_model, max_depth=None)
        item = result[0]["items"][0]
        assert isinstance(item["subitems"], list)
        assert len(item["subitems"]) == 1
        assert item["subitems"][0]["name"] == "Formwork"

    def test_max_depth_1_truncates_subitems(self, cost_model):
        result = cost(cost_model, max_depth=1)
        item = result[0]["items"][0]
        assert isinstance(item["subitems"], dict)
        assert item["subitems"]["truncated"] is True
        assert item["subitems"]["count"] == 1

    def test_max_depth_2_expands_to_depth_2(self, cost_model):
        result = cost(cost_model, max_depth=2)
        item = result[0]["items"][0]
        assert isinstance(item["subitems"], list)
        assert item["subitems"][0]["name"] == "Formwork"
        # subitem has no children, so subitems should be empty list
        assert item["subitems"][0]["subitems"] == []