Add test for ServiceManager

mojo/shell/service_manager_unittest.cc

Index: mojo/shell/service_manager_unittest.cc
diff --git a/mojo/shell/service_manager_unittest.cc b/mojo/shell/service_manager_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..d3545d59f08eb3e09046ecf73592cec5a5df8049
--- /dev/null
+++ b/mojo/shell/service_manager_unittest.cc
@@ -0,0 +1,125 @@
+// 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/common/bindings_support_impl.h"
+#include "mojo/public/bindings/lib/remote_ptr.h"
+#include "mojo/shell/service_manager.h"
+#include "mojom/shell.h"
+#include "mojom/test.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace mojo {
+namespace shell {
+namespace {
+
+class TestApp : public ShellClientStub {
+ public:
+  TestApp(ScopedMessagePipeHandle shell_handle)
+      : shell_(shell_handle.Pass()) {
+    shell_.SetPeer(this);
+  }
+  virtual ~TestApp() {
+  }
+  virtual void AcceptConnection(ScopedMessagePipeHandle client_handle)
+      MOJO_OVERRIDE {
+    service_.reset(new TestServiceImpl(this, client_handle.Pass()));
+  }
+  std::string GetLastTestString() {
+    return service_->last_test_string_;
+  }
+
+ private:
+  class TestServiceImpl : public TestServiceStub {
+   public:
+    TestServiceImpl(TestApp* service, ScopedMessagePipeHandle client_handle)
+        : service_(service),
+          client_(client_handle.Pass()) {
+      client_.SetPeer(this);
+    }
+    virtual ~TestServiceImpl() {
+    }
+    virtual void Test(const mojo::String& test_string) OVERRIDE {
+      last_test_string_ = test_string.To<std::string>();
+      client_->AckTest();
+    }
+    TestApp* service_;
+    RemotePtr<TestClient> client_;
+    std::string last_test_string_;
+  };
+  RemotePtr<Shell> shell_;
+  scoped_ptr<TestServiceImpl> service_;
+};
+
+class TestClientImpl : public TestClientStub {
+ public:
+  explicit TestClientImpl(ScopedMessagePipeHandle service_handle)
+      : service_(service_handle.Pass()),
+        quit_after_ack_(false) {
+    service_.SetPeer(this);
+  }
+  virtual ~TestClientImpl() {
+  }
+  virtual void AckTest() OVERRIDE {
+    if (quit_after_ack_)
+      base::MessageLoop::current()->QuitNow();
+  }
+  void Test(std::string test_string) {
+    AllocationScope scope;
+    quit_after_ack_ = true;
+    service_->Test(mojo::String(test_string));
+  }
+  RemotePtr<TestService> service_;
+  bool quit_after_ack_;
+};
+
+class ServiceManagerTest : public testing::Test,
+                           public ServiceManager::Loader {
+ public:
+  ServiceManagerTest() {
+  }
+
+  virtual ~ServiceManagerTest() {
+  }
+
+  virtual void SetUp() OVERRIDE {
+    mojo::BindingsSupport::Set(&support_);
+    GURL test_url("test:testService");
+    service_manager_.reset(new ServiceManager);
+    service_manager_->SetLoaderForURL(this, test_url);
+    MessagePipe pipe;
+    test_client_.reset(new TestClientImpl(pipe.handle0.Pass()));
+    service_manager_->Connect(test_url, pipe.handle1.Pass());
+  }
+
+  virtual void TearDown() OVERRIDE {
+    test_client_.reset(NULL);
+    test_app_.reset(NULL);
+    service_manager_.reset(NULL);
+    mojo::BindingsSupport::Set(NULL);
+  }
+
+  virtual void Load(const GURL& url,
+                    ScopedMessagePipeHandle shell_handle) OVERRIDE {
+    test_app_.reset(new TestApp(shell_handle.Pass()));
+  }
+
+ protected:
+  common::BindingsSupportImpl support_;
+  base::MessageLoop loop_;
+  scoped_ptr<TestApp> test_app_;
+  scoped_ptr<TestClientImpl> test_client_;
+  scoped_ptr<ServiceManager> service_manager_;
+  DISALLOW_COPY_AND_ASSIGN(ServiceManagerTest);
+};
+
+TEST_F(ServiceManagerTest, Basic) {
+  test_client_->Test("test");
+  loop_.Run();
+  EXPECT_EQ(std::string("test"), test_app_->GetLastTestString());
+}
+
+}  // namespace
+}  // namespace shell
+}  // namespace mojo