diff --git a/pymisp/mispevent.py b/pymisp/mispevent.py index 16f66ea..97b6c0e 100644 --- a/pymisp/mispevent.py +++ b/pymisp/mispevent.py @@ -1489,6 +1489,7 @@ class MISPObject(AbstractMISP): relationship_type=relationship_type, comment=comment, **kwargs) self.ObjectReference.append(reference) self.edited = True + return reference def get_attributes_by_relation(self, object_relation): '''Returns the list of attributes with the given object relation in the object''' diff --git a/tests/test_mispevent.py b/tests/test_mispevent.py index 10cb18d..6a27130 100644 --- a/tests/test_mispevent.py +++ b/tests/test_mispevent.py @@ -87,7 +87,8 @@ class TestMISPEvent(unittest.TestCase): del a.uuid self.mispevent.objects[0].uuid = 'a' self.mispevent.objects[1].uuid = 'b' - self.mispevent.objects[0].add_reference(self.mispevent.objects[1], 'baz', comment='foo') + reference = self.mispevent.objects[0].add_reference(self.mispevent.objects[1], 'baz', comment='foo') + del reference.uuid self.assertEqual(self.mispevent.objects[0].references[0].relationship_type, 'baz') with open('tests/mispevent_testfiles/event_obj_attr_tag.json', 'r') as f: ref_json = json.load(f)