Fix bug where style colours were not auto synchronised.

src/blenderbim/test/tool/test_ifc.py

@@ -119,6 +119,26 @@ class TestLink(test.bim.bootstrap.NewFile):
         assert subject.get_entity(obj) == element
         assert subject.get_object(element) == obj
 
+    def test_link_a_style(self):
+        ifc = ifcopenshell.file()
+        subject.set(ifc)
+        element = ifc.createIfcSurfaceStyle()
+        obj = bpy.data.materials.new("Material")
+        subject.link(element, obj)
+        assert subject.get_object(element) == obj
+
+    def test_link_a_material_and_style(self):
+        ifc = ifcopenshell.file()
+        subject.set(ifc)
+        style = ifc.createIfcSurfaceStyle()
+        material = ifc.createIfcMaterial()
+        obj = bpy.data.materials.new("Material")
+        subject.link(style, obj)
+        subject.link(material, obj)
+        assert subject.get_entity(obj) == material
+        assert subject.get_object(style) == obj
+        assert subject.get_object(material) == obj
+
 
 class TestUnlink(test.bim.bootstrap.NewFile):
     def test_unlink_an_object(self):
@@ -141,6 +161,31 @@ class TestUnlink(test.bim.bootstrap.NewFile):
         assert subject.get_entity(obj) is None
         assert subject.get_object(element) is None
 
+    def test_unlink_a_style(self):
+        ifc = ifcopenshell.file()
+        subject.set(ifc)
+        element = ifc.createIfcSurfaceStyle()
+        obj = bpy.data.materials.new("Material")
+        subject.link(element, obj)
+        subject.unlink(element, obj)
+        assert subject.get_object(element) is None
+
+    def test_unlink_a_style_and_material(self):
+        ifc = ifcopenshell.file()
+        subject.set(ifc)
+        style = ifc.createIfcSurfaceStyle()
+        material = ifc.createIfcMaterial()
+        obj = bpy.data.materials.new("Material")
+        subject.link(style, obj)
+        subject.link(material, obj)
+        subject.unlink(element=style, obj=obj)
+        assert subject.get_entity(obj) == material
+        assert subject.get_object(material) == obj
+        assert subject.get_object(style) is None
+        subject.unlink(element=material, obj=obj)
+        assert subject.get_entity(obj) is None
+        assert subject.get_object(material) is None
+
     def test_unlinking_using_an_object(self):
         ifc = ifcopenshell.file()
         subject.set(ifc)

src/blenderbim/test/tool/test_style.py

@@ -189,22 +189,3 @@ class TestImportSurfaceAttributes(NewFile):
         assert len(obj.BIMStyleProperties.attributes) == 2
         assert obj.BIMStyleProperties.attributes.get("Name").string_value == "Name"
         assert obj.BIMStyleProperties.attributes.get("Side").enum_value == "BOTH"
-
-
-class TestLink(NewFile):
-    def test_run(self):
-        obj = bpy.data.materials.new("Material")
-        ifc = ifcopenshell.file()
-        style = ifc.createIfcSurfaceStyle()
-        subject.link(style, obj)
-        assert obj.BIMMaterialProperties.ifc_style_id == style.id()
-
-
-class TestUnlink(NewFile):
-    def test_run(self):
-        obj = bpy.data.materials.new("Material")
-        ifc = ifcopenshell.file()
-        style = ifc.createIfcSurfaceStyle()
-        subject.link(style, obj)
-        subject.unlink(obj)
-        assert obj.BIMMaterialProperties.ifc_style_id == 0