parent
e4b7a95355
commit
ad3e17234a
@ -1,158 +0,0 @@
|
|||||||
import importlib
|
|
||||||
import json
|
|
||||||
import sys
|
|
||||||
import tempfile
|
|
||||||
import types
|
|
||||||
import unittest
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
|
|
||||||
REPO_ROOT = Path(__file__).resolve().parents[2]
|
|
||||||
MODULE_DIR = REPO_ROOT / "src" / "Mod" / "FreeCADExchange"
|
|
||||||
if str(MODULE_DIR) not in sys.path:
|
|
||||||
sys.path.insert(0, str(MODULE_DIR))
|
|
||||||
|
|
||||||
|
|
||||||
def _install_fake_freecad():
|
|
||||||
class Vector:
|
|
||||||
def __init__(self, x=0.0, y=0.0, z=0.0):
|
|
||||||
self.x = float(x)
|
|
||||||
self.y = float(y)
|
|
||||||
self.z = float(z)
|
|
||||||
|
|
||||||
fake_freecad = types.ModuleType("FreeCAD")
|
|
||||||
fake_freecad.Vector = Vector
|
|
||||||
fake_freecad.ActiveDocument = None
|
|
||||||
fake_freecad.Console = types.SimpleNamespace(
|
|
||||||
PrintMessage=lambda *args, **kwargs: None,
|
|
||||||
PrintWarning=lambda *args, **kwargs: None,
|
|
||||||
PrintError=lambda *args, **kwargs: None,
|
|
||||||
)
|
|
||||||
fake_freecad.addDocumentObserver = lambda observer: None
|
|
||||||
sys.modules["FreeCAD"] = fake_freecad
|
|
||||||
|
|
||||||
fake_freecadgui = types.ModuleType("FreeCADGui")
|
|
||||||
fake_freecadgui.addCommand = lambda *args, **kwargs: None
|
|
||||||
sys.modules["FreeCADGui"] = fake_freecadgui
|
|
||||||
|
|
||||||
fake_importgui = types.ModuleType("ImportGui")
|
|
||||||
fake_importgui.insert = lambda *args, **kwargs: None
|
|
||||||
sys.modules["ImportGui"] = fake_importgui
|
|
||||||
|
|
||||||
fake_device_preview = types.ModuleType("DevicePreview")
|
|
||||||
fake_device_preview.find_main_exchange_document = lambda preferred_name="": None
|
|
||||||
sys.modules["DevicePreview"] = fake_device_preview
|
|
||||||
|
|
||||||
|
|
||||||
class FakeObject:
|
|
||||||
def __init__(self, name, type_id="App::DocumentObjectGroup"):
|
|
||||||
self.Name = name
|
|
||||||
self.Label = name
|
|
||||||
self.TypeId = type_id
|
|
||||||
self.PropertiesList = []
|
|
||||||
self.Group = []
|
|
||||||
self.InList = []
|
|
||||||
|
|
||||||
def isDerivedFrom(self, type_name):
|
|
||||||
if self.TypeId == type_name:
|
|
||||||
return True
|
|
||||||
if type_name == "App::DocumentObjectGroup":
|
|
||||||
return self.TypeId in {"App::DocumentObjectGroup", "App::Part"}
|
|
||||||
return False
|
|
||||||
|
|
||||||
def addProperty(self, prop_type, prop_name, group_name, description):
|
|
||||||
if prop_name not in self.PropertiesList:
|
|
||||||
self.PropertiesList.append(prop_name)
|
|
||||||
|
|
||||||
def addObject(self, child):
|
|
||||||
if child not in self.Group:
|
|
||||||
self.Group.append(child)
|
|
||||||
if self not in child.InList:
|
|
||||||
child.InList.append(self)
|
|
||||||
|
|
||||||
|
|
||||||
class FakeDocument:
|
|
||||||
def __init__(self):
|
|
||||||
self.Name = "QETScene"
|
|
||||||
self.FileName = ""
|
|
||||||
self.Objects = []
|
|
||||||
|
|
||||||
def addObject(self, type_name, name):
|
|
||||||
obj = FakeObject(name, type_name)
|
|
||||||
self.Objects.append(obj)
|
|
||||||
return obj
|
|
||||||
|
|
||||||
def getObject(self, name):
|
|
||||||
for obj in self.Objects:
|
|
||||||
if obj.Name == name:
|
|
||||||
return obj
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
def _reload_writeback():
|
|
||||||
for name in ["DeviceImport", "TerminalObjects", "ExchangeWriteBack"]:
|
|
||||||
sys.modules.pop(name, None)
|
|
||||||
return importlib.import_module("ExchangeWriteBack"), importlib.import_module("TerminalObjects")
|
|
||||||
|
|
||||||
|
|
||||||
class ExchangeWriteBackManualWireTest(unittest.TestCase):
|
|
||||||
def test_write_back_includes_manual_wires_with_route_points(self):
|
|
||||||
_install_fake_freecad()
|
|
||||||
exchange_writeback, terminal_objects = _reload_writeback()
|
|
||||||
app = sys.modules["FreeCAD"]
|
|
||||||
doc = FakeDocument()
|
|
||||||
|
|
||||||
root = terminal_objects.ensure_root_group(doc, "project-1")
|
|
||||||
device = doc.addObject("App::Part", "QETDevice_device_1")
|
|
||||||
root.addObject(device)
|
|
||||||
terminal_objects.ensure_string_property(device, "QetElementUuid", "QET Exchange", "", "device-1")
|
|
||||||
terminal_objects.ensure_string_property(device, "QetInstanceId", "QET Exchange", "", "instance-1")
|
|
||||||
|
|
||||||
wire_group = terminal_objects.ensure_wire_group(
|
|
||||||
doc,
|
|
||||||
device,
|
|
||||||
project_uuid="project-1",
|
|
||||||
instance_id="instance-1",
|
|
||||||
)
|
|
||||||
wire = doc.addObject("Part::Feature", "QETWire_terminal_1_terminal_2")
|
|
||||||
wire_group.addObject(wire)
|
|
||||||
terminal_objects.ensure_string_property(wire, "QetProjectUuid", "QET Exchange", "", "project-1")
|
|
||||||
terminal_objects.ensure_string_property(wire, "QetStartTerminalUuid", "QET Exchange", "", "terminal-1")
|
|
||||||
terminal_objects.ensure_string_property(wire, "QetEndTerminalUuid", "QET Exchange", "", "terminal-2")
|
|
||||||
terminal_objects.ensure_string_property(wire, "QetStartInstanceId", "QET Exchange", "", "instance-1")
|
|
||||||
terminal_objects.ensure_string_property(wire, "QetEndInstanceId", "QET Exchange", "", "instance-2")
|
|
||||||
terminal_objects.ensure_string_property(wire, "RouteType", "QET Exchange", "", "Manual")
|
|
||||||
wire.Shape = (
|
|
||||||
app.Vector(1, 2, 3),
|
|
||||||
app.Vector(4, 5, 6),
|
|
||||||
)
|
|
||||||
|
|
||||||
with tempfile.TemporaryDirectory() as temp_dir:
|
|
||||||
scene_path = str(Path(temp_dir) / "scene.FCStd")
|
|
||||||
report = exchange_writeback.write_back_document(
|
|
||||||
doc,
|
|
||||||
scene_path=scene_path,
|
|
||||||
payload={"project_uuid": "project-1"},
|
|
||||||
)
|
|
||||||
payload = json.loads(Path(report["output_path"]).read_text(encoding="utf-8"))
|
|
||||||
|
|
||||||
self.assertEqual(1, len(report["manual_wires"]))
|
|
||||||
self.assertEqual(1, len(payload["manual_wires"]))
|
|
||||||
self.assertEqual(
|
|
||||||
{
|
|
||||||
"start_terminal_uuid": "terminal-1",
|
|
||||||
"end_terminal_uuid": "terminal-2",
|
|
||||||
"start_instance_id": "instance-1",
|
|
||||||
"end_instance_id": "instance-2",
|
|
||||||
"route_type": "Manual",
|
|
||||||
"points": [
|
|
||||||
{"x": 1.0, "y": 2.0, "z": 3.0},
|
|
||||||
{"x": 4.0, "y": 5.0, "z": 6.0},
|
|
||||||
],
|
|
||||||
},
|
|
||||||
payload["manual_wires"][0],
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
unittest.main()
|
|
||||||
Loading…
Reference in New Issue