Add client model tests