Interceptor test for the mojo spy.

mojo/service_manager/service_manager_unittest.cc

 // Copyright 2014 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 "base/message_loop/message_loop.h"
 #include "mojo/public/cpp/bindings/allocation_scope.h"
 #include "mojo/public/cpp/bindings/remote_ptr.h"
 #include "mojo/public/cpp/environment/environment.h"
 #include "mojo/public/cpp/shell/application.h"
 #include "mojo/public/interfaces/shell/shell.mojom.h"
@@ skipping 124 matching lines @@
                            ScopedShellHandle service_handle) OVERRIDE {
     ++num_loads_;
   }
   virtual void OnServiceError(ServiceManager* manager, const GURL& url)
       OVERRIDE {}
 
   int num_loads_;
   DISALLOW_COPY_AND_ASSIGN(TestServiceLoader);
 };
 
+class TestServiceInterceptor : public ServiceManager::Interceptor {
+ public:
+  TestServiceInterceptor() : call_count_(0) {}
+
+  virtual ScopedMessagePipeHandle OnConnectToClient(
+      const GURL& url, ScopedMessagePipeHandle handle) OVERRIDE {
+    ++call_count_;
+    url_ = url;
+    return handle.Pass();
+  }
+
+  std::string url_spec() const {
+    if (!url_.is_valid())
+      return "invalid url";
+    return url_.spec();
+  }
+
+  int call_count() const {
+    return call_count_;
+  }
+
+ private:
+  int call_count_;
+  GURL url_;
+  DISALLOW_COPY_AND_ASSIGN(TestServiceInterceptor);
+};
+
 TEST_F(ServiceManagerTest, Basic) {
   test_client_->Test("test");
   loop_.Run();
   EXPECT_EQ(std::string("test"), context_.last_test_string);
 }
 
 TEST_F(ServiceManagerTest, ClientError) {
   test_client_->Test("test");
   EXPECT_TRUE(HasFactoryForTestURL());
   loop_.Run();
@@ skipping 29 matching lines @@
   EXPECT_EQ(0, default_loader.num_loads());
 
   // http::test1 should go to default loader.
   InterfacePipe<TestService, AnyInterface> pipe3;
   sm.Connect(GURL("http:test1"), pipe3.handle_to_peer.Pass());
   EXPECT_EQ(1, url_loader.num_loads());
   EXPECT_EQ(1, scheme_loader.num_loads());
   EXPECT_EQ(1, default_loader.num_loads());
 }
 
+TEST_F(ServiceManagerTest, Interceptor) {
+  ServiceManager sm;
+  TestServiceLoader default_loader;
+  TestServiceInterceptor interceptor;
+  sm.set_default_loader(&default_loader);
+  sm.SetInterceptor(&interceptor);
+
+  std::string url("test:test3");
+  InterfacePipe<TestService, AnyInterface> pipe1;
+  sm.Connect(GURL(url), pipe1.handle_to_peer.Pass());
+  EXPECT_EQ(1, interceptor.call_count());
+  EXPECT_EQ(url, interceptor.url_spec());
+  EXPECT_EQ(1, default_loader.num_loads());
+}
+
 }  // namespace mojo