Add tests for client context.