CAM: Replace complete tool management (PR 21425)

src/Mod/CAM/CAMTests/TestPathToolLibrarySerializer.py

@@ -0,0 +1,128 @@
+import unittest
+import json
+import FreeCAD
+from CAMTests.PathTestUtils import PathTestWithAssets
+from Path.Tool.library import Library
+from Path.Tool.toolbit import ToolBitEndmill
+from Path.Tool.shape import ToolBitShapeEndmill
+from Path.Tool.library.serializers import CamoticsLibrarySerializer, LinuxCNCSerializer
+
+
+class TestPathToolLibrarySerializerBase(PathTestWithAssets):
+    """Base class for Library serializer tests."""
+
+    def setUp(self):
+        super().setUp()
+        self.test_library_id = "test_library"
+        self.test_library_label = "Test Library"
+        self.test_library = Library(self.test_library_label, id=self.test_library_id)
+
+        # Create some dummy tool bits
+        shape1 = ToolBitShapeEndmill("endmill_1")
+        shape1.set_parameter("Diameter", FreeCAD.Units.Quantity("6.0 mm"))
+        shape1.set_parameter("Length", FreeCAD.Units.Quantity("20.0 mm"))
+        tool1 = ToolBitEndmill(shape1, id="tool_1")
+        tool1.label = "Endmill 6mm"
+
+        shape2 = ToolBitShapeEndmill("endmill_2")
+        shape2.set_parameter("Diameter", FreeCAD.Units.Quantity("3.0 mm"))
+        shape2.set_parameter("Length", FreeCAD.Units.Quantity("15.0 mm"))
+        tool2 = ToolBitEndmill(shape2, id="tool_2")
+        tool2.label = "Endmill 3mm"
+
+        self.test_library.add_bit(tool1, 1)
+        self.test_library.add_bit(tool2, 2)
+
+
+class TestCamoticsLibrarySerializer(TestPathToolLibrarySerializerBase):
+    """Tests for the CamoticsLibrarySerializer."""
+
+    def test_camotics_serialize(self):
+        serializer = CamoticsLibrarySerializer
+        serialized_data = serializer.serialize(self.test_library)
+        self.assertIsInstance(serialized_data, bytes)
+
+        # Verify the content structure (basic check)
+        data_dict = json.loads(serialized_data.decode("utf-8"))
+        self.assertIn("1", data_dict)
+        self.assertIn("2", data_dict)
+        self.assertEqual(data_dict["1"]["description"], self.test_library._bit_nos[1].label)
+        self.assertEqual(
+            data_dict["2"]["diameter"],
+            self.test_library._bit_nos[2]._tool_bit_shape.get_parameter("Diameter"),
+        )
+
+    def test_camotics_deserialize(self):
+        serializer = CamoticsLibrarySerializer
+        # Create a dummy serialized data matching the expected format
+        dummy_data = {
+            "10": {
+                "units": "metric",
+                "shape": "Ballnose",
+                "length": 25,
+                "diameter": 8,
+                "description": "Ballnose 8mm",
+            },
+            "20": {
+                "units": "metric",
+                "shape": "Cylindrical",
+                "length": 30,
+                "diameter": 10,
+                "description": "Endmill 10mm",
+            },
+        }
+        dummy_bytes = json.dumps(dummy_data, indent=2).encode("utf-8")
+
+        # Deserialize the data
+        deserialized_library = serializer.deserialize(dummy_bytes, "deserialized_lib", {})
+
+        self.assertIsInstance(deserialized_library, Library)
+        self.assertEqual(deserialized_library.get_id(), "deserialized_lib")
+        self.assertEqual(len(deserialized_library._bit_nos), 2)
+
+        tool_10 = deserialized_library._bit_nos.get(10)
+        assert tool_10 is not None, "tool not in the library"
+        self.assertEqual(tool_10.label, "Ballnose 8mm")
+        self.assertEqual(tool_10._tool_bit_shape.name, "Ballend")
+        self.assertEqual(
+            tool_10._tool_bit_shape.get_parameter("Diameter"), FreeCAD.Units.Quantity("8 mm")
+        )
+        self.assertEqual(
+            tool_10._tool_bit_shape.get_parameter("Length"), FreeCAD.Units.Quantity("25 mm")
+        )
+
+        tool_20 = deserialized_library._bit_nos.get(20)
+        assert tool_20 is not None, "tool not in the library"
+        self.assertEqual(tool_20.label, "Endmill 10mm")
+        self.assertEqual(tool_20._tool_bit_shape.name, "Endmill")
+        self.assertEqual(
+            tool_20._tool_bit_shape.get_parameter("Diameter"), FreeCAD.Units.Quantity("10 mm")
+        )
+        self.assertEqual(
+            tool_20._tool_bit_shape.get_parameter("Length"), FreeCAD.Units.Quantity("30 mm")
+        )
+
+
+class TestLinuxCNCLibrarySerializer(TestPathToolLibrarySerializerBase):
+    """Tests for the LinuxCNCLibrarySerializer."""
+
+    def test_linuxcnc_serialize(self):
+        serializer = LinuxCNCSerializer
+        serialized_data = serializer.serialize(self.test_library)
+        self.assertIsInstance(serialized_data, bytes)
+
+        # Verify the content format (basic check)
+        lines = serialized_data.decode("ascii", "ignore").strip().split("\n")
+        self.assertEqual(len(lines), 2)
+        self.assertTrue(lines[0].startswith("T1 P D6.0 ;Endmill 6mm"))
+        self.assertTrue(lines[1].startswith("T2 P D3.0 ;Endmill 3mm"))
+
+    def test_linuxcnc_deserialize_not_implemented(self):
+        serializer = LinuxCNCSerializer
+        dummy_data = b"T1 D6.0 ;Endmill 6mm\n"
+        with self.assertRaises(NotImplementedError):
+            serializer.deserialize(dummy_data, "dummy_id", {})
+
+
+if __name__ == "__main__":
+    unittest.main()