Add client tests