Add SetClient method implementation to InterfaceImpl.

mojo/service_manager/service_manager.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <stdio.h>
 
 #include "mojo/service_manager/service_manager.h"
 
 #include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/stl_util.h"
 #include "mojo/public/cpp/bindings/allocation_scope.h"
 #include "mojo/service_manager/service_loader.h"
 
 namespace mojo {
 
 namespace {
 // Used by TestAPI.
 bool has_created_instance = false;
 }
 
 class ServiceManager::ServiceFactory : public InterfaceImpl<Shell> {
  public:
   ServiceFactory(ServiceManager* manager, const GURL& url)
       : manager_(manager),
-        url_(url),
-        client_(NULL) {
+        url_(url) {
   }
 
   virtual ~ServiceFactory() {
   }
 
   void ConnectToClient(ScopedMessagePipeHandle handle) {
     if (!handle.is_valid()) {
       assert(false);
       return;
     }
     AllocationScope scope;
-    client_->AcceptConnection(url_.spec(), handle.Pass());
+    client()->AcceptConnection(url_.spec(), handle.Pass());
   }
 
   // Shell implementation:
-  virtual void SetClient(ShellClient* client) OVERRIDE {
-    client_ = client;
-  }
   virtual void Connect(const String& url,
                        ScopedMessagePipeHandle client_pipe) OVERRIDE {
     manager_->Connect(GURL(url.To<std::string>()), client_pipe.Pass());
   }
 
   const GURL& url() const { return url_; }
 
  private:
   virtual void OnConnectionError() OVERRIDE {
     manager_->OnServiceFactoryError(this);
   }
 
   ServiceManager* const manager_;
   const GURL url_;
-  ShellClient* client_;
 
   DISALLOW_COPY_AND_ASSIGN(ServiceFactory);
 };
 
 class ServiceManager::TestAPI::TestShellConnection
     : public InterfaceImpl<Shell> {
  public:
-  explicit TestShellConnection(ServiceManager* manager)
-      : manager_(manager),
-        client_(NULL) {
-  }
+  explicit TestShellConnection(ServiceManager* manager) : manager_(manager) {}
   virtual ~TestShellConnection() {}
 
   virtual void OnConnectionError() OVERRIDE {
     // TODO(darin): How should we handle this error?
   }
 
   // Shell:
-  virtual void SetClient(ShellClient* client) OVERRIDE {
-    client_ = client;
-  }
   virtual void Connect(const String& url,
                        ScopedMessagePipeHandle client_pipe) OVERRIDE {
     manager_->Connect(GURL(url.To<std::string>()), client_pipe.Pass());
   }
 
  private:
   ServiceManager* manager_;
-  ShellClient* client_;
 
   DISALLOW_COPY_AND_ASSIGN(TestShellConnection);
 };
 
 // static
 ServiceManager::TestAPI::TestAPI(ServiceManager* manager) : manager_(manager) {
 }
 
 ServiceManager::TestAPI::~TestAPI() {
 }
@@ skipping 92 matching lines @@
   // Called from ~ServiceFactory, so we do not need to call Destroy here.
   const GURL url = service_factory->url();
   URLToServiceFactoryMap::iterator it = url_to_service_factory_.find(url);
   DCHECK(it != url_to_service_factory_.end());
   delete it->second;
   url_to_service_factory_.erase(it);
   GetLoaderForURL(url)->OnServiceError(this, url);
 }
 
 }  // namespace mojo