Add entity tests