Fix bug where you could assign a product twice to a process.

This commit is contained in:
Dion Moult
2022-01-14 16:04:55 +11:00
parent ec38fe2edc
commit 258ca7fb8c
4 changed files with 32 additions and 2 deletions
@@ -16,7 +16,7 @@ class Usecase:
if self.settings["related_object"].HasAssignments:
for assignment in self.settings["related_object"].HasAssignments:
if (
assignment.is_a("IfclRelAssignsToProduct")
assignment.is_a("IfcRelAssignsToProduct")
and assignment.RelatingProduct == self.settings["relating_product"]
):
return
@@ -6,7 +6,6 @@ class TestUnassignReference(test.bootstrap.IFC4):
def test_unassigning_a_reference(self):
reference = self.file.createIfcLibraryReference()
product = self.file.createIfcWall()
product2 = self.file.createIfcWall()
ifcopenshell.api.run("library.assign_reference", self.file, product=product, reference=reference)
ifcopenshell.api.run("library.unassign_reference", self.file, product=product, reference=reference)
assert len(self.file.by_type("IfcRelAssociatesLibrary")) == 0
@@ -0,0 +1,20 @@
import test.bootstrap
import ifcopenshell.api
class TestAssignProduct(test.bootstrap.IFC4):
def test_assigning_a_product(self):
wall = self.file.createIfcWall()
task = self.file.createIfcTask()
task2 = self.file.createIfcTask()
ifcopenshell.api.run("sequence.assign_product", self.file, relating_product=wall, related_object=task)
assert wall.ReferencedBy[0].RelatedObjects == (task,)
ifcopenshell.api.run("sequence.assign_product", self.file, relating_product=wall, related_object=task2)
assert wall.ReferencedBy[0].RelatedObjects == (task, task2)
def test_not_assigning_twice(self):
wall = self.file.createIfcWall()
task = self.file.createIfcTask()
ifcopenshell.api.run("sequence.assign_product", self.file, relating_product=wall, related_object=task)
ifcopenshell.api.run("sequence.assign_product", self.file, relating_product=wall, related_object=task)
assert wall.ReferencedBy[0].RelatedObjects == (task,)
@@ -0,0 +1,11 @@
import test.bootstrap
import ifcopenshell.api
class TestUnassignProduct(test.bootstrap.IFC4):
def test_unassigning_a_product(self):
wall = self.file.createIfcWall()
task = self.file.createIfcTask()
ifcopenshell.api.run("sequence.assign_product", self.file, relating_product=wall, related_object=task)
ifcopenshell.api.run("sequence.unassign_product", self.file, relating_product=wall, related_object=task)
assert len(self.file.by_type("IfcRelAssignsToProduct")) == 0