up

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