diff --git a/unittests/test_apiutils.py b/unittests/test_apiutils.py index 43d4955e4ffb9c3646760475a4ba664fd8e657db..f957b48ddbd61031cf7e351d566a130febb50732 100644 --- a/unittests/test_apiutils.py +++ b/unittests/test_apiutils.py @@ -78,20 +78,22 @@ def test_resolve_reference(): prop.is_valid = lambda: True items = [200, 300, 400] prop_list = db.Property(datatype=db.LIST(db.REFERENCE), value=items) - prop_list2 = db.Property(datatype=db.LIST(db.REFERENCE), value=db.Record(id=500)) + prop_list2 = db.Property(datatype=db.LIST(db.REFERENCE), value=[db.Record(id=500)]) resolve_reference(prop) resolve_reference(prop_list) resolve_reference(prop_list2) assert prop.value.id == 100 - assert isinstance(prop.value, db.Entity) == True + assert isinstance(prop.value, db.Entity) prop_list_ids = [] for i in prop_list.value: prop_list_ids.append(i.id) + assert isinstance(i, db.Entity) assert prop_list_ids == items for i in prop_list2.value: assert i.id == 500 + assert isinstance(i, db.Entity) assert resolve_reference(db.Property(id=5000, datatype=db.INTEGER, value=2)) == None