import unittest import numpy as np from pathlib import Path import tempfile import os import sys PROJECT_ROOT = Path(__file__).resolve().parent.parent sys.path.append(str(PROJECT_ROOT.parent)) from unittest.mock import MagicMock class DummyBase: pass mock_types = MagicMock() mock_types.NodeSocket = type("NodeSocket", (DummyBase,), {}) mock_types.Node = type("Node", (DummyBase,), {}) mock_types.NodeCustomGroup = type("NodeCustomGroup", (DummyBase,), {}) mock_types.NodeTree = type("NodeTree", (DummyBase,), {}) mock_types.PropertyGroup = type("PropertyGroup", (DummyBase,), {}) mock_types.Operator = type("Operator", (DummyBase,), {}) mock_types.Panel = type("Panel", (DummyBase,), {}) mock_types.Object = MagicMock mock_types.Mesh = MagicMock mock_bpy = MagicMock() mock_bpy.types = mock_types mock_bpy.path.abspath = lambda path: path.replace("//", "/abs/") sys.modules['bpy'] = mock_bpy sys.modules['bpy.types'] = mock_types sys.modules['bpy.props'] = MagicMock() sys.modules['bpy.utils'] = MagicMock() sys.modules['bpy.utils.previews'] = MagicMock() sys.modules['bpy.app'] = MagicMock() sys.modules['bpy.app.handlers'] = MagicMock() from NodeSlicer.core.nodeslicer.nodes.inputs import SLICENODE_WipeTower, SLICENODE_OffsetObject from NodeSlicer.core.nodeslicer.nodes.mesh import SLICENODE_AssignModifiers from NodeSlicer.infra.prusaslicer_profiles.models import ConfigWriter from NodeSlicer.infra._3mf_writer._3mf import write_model_xml, write_metadata_xml from NodeSlicer.infra.blender.blender_mesh_capture import SlicingObject, SlicingCollection, SlicingGroup import xml.etree.ElementTree as ET class TestWipeTowerNode(unittest.TestCase): def test_wipe_tower_node_output(self): node = SLICENODE_WipeTower() node.resolve = lambda key: 180.5 if key == "X Position" else 135.2 res = node.evaluate_output("Settings") self.assertIn("wipe_tower_position", res) self.assertEqual(res["wipe_tower_position"], (180.5, 135.2)) def test_offset_object_node_output(self): node = SLICENODE_OffsetObject() node.resolve = lambda key: {"X Offset": 12.5, "Y Offset": -5.0, "Z Offset": 2.0}.get(key, 0.0) res = node.evaluate_output("Settings") self.assertIn("offset_object", res) self.assertEqual(res["offset_object"], (12.5, -5.0, 2.0)) def test_config_writer_excludes_special_keys(self): conf = { "layer_height": "0.2", "wipe_tower_position": "(180.5, 135.2)", "offset_object": "(12.5, -5.0, 2.0)", "nozzle_diameter": "0.4" } writer = ConfigWriter(conf) with tempfile.TemporaryDirectory() as temp_dir: ini_path = Path(temp_dir) / "Slic3r_PE.config" writer.write_ini_3mf(ini_path) with open(ini_path, 'r', encoding='utf-8') as f: content = f.read() self.assertIn("; layer_height = 0.2", content) self.assertIn("; nozzle_diameter = 0.4", content) self.assertNotIn("wipe_tower_position", content) self.assertNotIn("offset_object", content) class TestAssignModifiersNode(unittest.TestCase): def setUp(self): self.so = SlicingObject() self.so.name = "TestPart" self.so.object_type = "ModelPart" self.so.extruder = "1" self.so.modifiers = [{"param_id": "initial_param", "param_value": "123"}] self.so.mesh = np.array([[[0.0, 0.0, 0.0], [1.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 0.0, 1.0]]], dtype=np.float64) self.sc = SlicingCollection([], "TestColl") self.sc.objects = [self.so] self.sc.mesh_start_ids = [0] self.sc.mesh_end_ids = [1] self.sg = SlicingGroup([]) self.sg.collections = {"TestColl": self.sc} def test_assign_modifiers_to_slicing_object(self): node = SLICENODE_AssignModifiers() node.part_type = "ParameterModifier" node.extruder = "3" node.resolve = lambda f: { "Geometry": self.so, "Modifiers": {"infill_density": "20%", "layer_height": "0.15"} }.get(f) result_so = node.evaluate_output("Geometry") self.assertIsNotNone(result_so) self.assertEqual(result_so.object_type, "ParameterModifier") self.assertEqual(result_so.extruder, "3") mod_dict = {m['param_id']: m['param_value'] for m in result_so.modifiers} self.assertEqual(mod_dict["initial_param"], "123") self.assertEqual(mod_dict["infill_density"], "20%") self.assertEqual(mod_dict["layer_height"], "0.15") def test_assign_modifiers_to_slicing_group(self): node = SLICENODE_AssignModifiers() node.part_type = "NegativeVolume" node.extruder = "2" node.resolve = lambda f: { "Geometry": self.sg, "Modifiers": {"perimeters": "4"} }.get(f) result_sg = node.evaluate_output("Geometry") self.assertIsNotNone(result_sg) res_so = result_sg.collections["TestColl"].objects[0] self.assertEqual(res_so.object_type, "NegativeVolume") self.assertEqual(res_so.extruder, "2") mod_dict = {m['param_id']: m['param_value'] for m in res_so.modifiers} self.assertEqual(mod_dict["perimeters"], "4") def test_3mf_metadata_xml_output(self): node = SLICENODE_AssignModifiers() node.part_type = "ParameterModifier" node.extruder = "2" node.resolve = lambda f: { "Geometry": self.sg, "Modifiers": {"fill_density": "30%"} }.get(f) modified_sg = node.evaluate_output("Geometry") with tempfile.TemporaryDirectory() as temp_dir: xml_path = Path(temp_dir) / "Slic3r_PE_model.config" write_metadata_xml(modified_sg, xml_path) tree = ET.parse(xml_path) root = tree.getroot() volumes = root.findall(".//volume") self.assertEqual(len(volumes), 1) vol = volumes[0] meta_entries = {m.get("key"): m.get("value") for m in vol.findall("metadata")} self.assertEqual(meta_entries.get("volume_type"), "ParameterModifier") self.assertEqual(meta_entries.get("modifier"), "1") self.assertEqual(meta_entries.get("extruder"), "2") self.assertEqual(meta_entries.get("fill_density"), "30%") if __name__ == "__main__": unittest.main()