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.infra.prusaslicer_profiles.models import ConfigWriter
from NodeSlicer.infra._3mf_writer._3mf import write_model_xml
from NodeSlicer.infra.blender.blender_mesh_capture import SlicingGroup

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)

if __name__ == "__main__":
    unittest.main()