Fix support for document references in IFC2X3

src/ifcopenshell-python/ifcopenshell/api/document/add_reference.py

@@ -26,5 +26,10 @@ class Usecase:
 
     def execute(self):
         if self.file.schema == "IFC2X3":
-            return self.file.create_entity("IfcDocumentReference")
+            reference = self.file.create_entity("IfcDocumentReference")
+            if self.settings["information"]:
+                references = list(self.settings["information"].DocumentReferences or [])
+                references.append(reference)
+                self.settings["information"].DocumentReferences = references
+            return reference
         return self.file.create_entity("IfcDocumentReference", ReferencedDocument=self.settings["information"])

src/ifcopenshell-python/test/api/document/test_add_reference_.py

@@ -33,3 +33,18 @@ class TestAddReference(test.bootstrap.IFC4):
         assert element.is_a("IfcDocumentReference")
         assert len(self.file.by_type("IfcDocumentReference")) == 1
         assert element.ReferencedDocument == information
+
+
+class TestAddReferenceIFC2X3(test.bootstrap.IFC2X3):
+    def test_adding_a_reference(self):
+        element = ifcopenshell.api.run("document.add_reference", self.file, information=None)
+        assert element.is_a("IfcDocumentReference")
+        assert len(self.file.by_type("IfcDocumentReference")) == 1
+
+    def test_adding_a_reference_to_an_information(self):
+        self.file.createIfcProject()
+        information = ifcopenshell.api.run("document.add_information", self.file, parent=None)
+        element = ifcopenshell.api.run("document.add_reference", self.file, information=information)
+        assert element.is_a("IfcDocumentReference")
+        assert len(self.file.by_type("IfcDocumentReference")) == 1
+        assert element.ReferenceToDocument[0] == information