Add tests for client model