mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-30 08:33:10 +00:00
Bruno Postle
060afb94b5
Model Context Protocol server implementation for AI agent access to IFC projects
100 lines
3.4 KiB
Python
100 lines
3.4 KiB
Python
import json
|
|
|
|
import pytest
|
|
|
|
from ifcmcp.server import ifc_docs, ifc_edit, ifc_list
|
|
|
|
|
|
class TestNoModel:
|
|
def test_edit_no_model(self):
|
|
with pytest.raises(ValueError, match="No model loaded"):
|
|
ifc_edit("root.create_entity")
|
|
|
|
|
|
class TestList:
|
|
def test_list_all_modules(self, loaded_model):
|
|
result = ifc_list()
|
|
assert isinstance(result, list)
|
|
assert len(result) > 0
|
|
modules = [m["module"] for m in result]
|
|
assert "root" in modules
|
|
assert "spatial" in modules
|
|
|
|
def test_list_module_functions(self, loaded_model):
|
|
result = ifc_list(module="root")
|
|
assert isinstance(result, list)
|
|
names = [f["name"] for f in result]
|
|
assert "create_entity" in names
|
|
|
|
def test_list_empty_string_returns_modules(self, loaded_model):
|
|
result = ifc_list(module="")
|
|
assert isinstance(result, list)
|
|
assert any(m["module"] == "root" for m in result)
|
|
|
|
|
|
class TestDocs:
|
|
def test_docs_create_entity(self, loaded_model):
|
|
result = ifc_docs("root.create_entity")
|
|
assert result["module"] == "root"
|
|
assert result["function"] == "create_entity"
|
|
assert "params" in result
|
|
|
|
def test_docs_bad_format(self, loaded_model):
|
|
with pytest.raises(ValueError):
|
|
ifc_docs("no_dot_here")
|
|
|
|
|
|
class TestEdit:
|
|
def test_create_entity(self, loaded_model):
|
|
result = ifc_edit("root.create_entity", json.dumps({"ifc_class": "IfcWall", "name": "NewWall"}))
|
|
assert result["ok"] is True
|
|
assert result["result"]["type"] == "IfcWall"
|
|
assert result["result"]["name"] == "NewWall"
|
|
|
|
def test_create_entity_default_params(self, loaded_model):
|
|
result = ifc_edit("root.create_entity", "{}")
|
|
assert result["ok"] is True
|
|
|
|
def test_unknown_function(self, loaded_model):
|
|
result = ifc_edit("root.nonexistent", "{}")
|
|
assert result["ok"] is False
|
|
assert "Cannot find" in result["error"]
|
|
|
|
def test_unknown_parameter(self, loaded_model):
|
|
result = ifc_edit("root.create_entity", json.dumps({"bogus": "value"}))
|
|
assert result["ok"] is False
|
|
assert "Unknown parameter" in result["error"]
|
|
|
|
def test_bad_json(self, loaded_model):
|
|
with pytest.raises(json.JSONDecodeError):
|
|
ifc_edit("root.create_entity", "not json")
|
|
|
|
def test_edit_does_not_save(self, loaded_model, tmp_path):
|
|
"""Verify that ifc_edit mutates the in-memory model but does not write to disk."""
|
|
import ifcmcp.server as server_mod
|
|
|
|
path = str(tmp_path / "test.ifc")
|
|
loaded_model.write(path)
|
|
server_mod._model_path = path
|
|
|
|
before_count = sum(1 for _ in loaded_model)
|
|
ifc_edit("root.create_entity", json.dumps({"ifc_class": "IfcWall", "name": "Unsaved"}))
|
|
after_count = sum(1 for _ in loaded_model)
|
|
assert after_count == before_count + 1
|
|
|
|
# Re-read the file — it should not have the new entity
|
|
import ifcopenshell
|
|
|
|
on_disk = ifcopenshell.open(path)
|
|
disk_count = sum(1 for _ in on_disk)
|
|
assert disk_count == before_count
|
|
|
|
def test_assign_container(self, loaded_model):
|
|
wall = loaded_model.by_type("IfcWall")[0]
|
|
storey = loaded_model.by_type("IfcBuildingStorey")[0]
|
|
result = ifc_edit(
|
|
"spatial.assign_container",
|
|
json.dumps({"products": str(wall.id()), "relating_structure": str(storey.id())}),
|
|
)
|
|
assert result["ok"] is True
|