1 files changed, 157 insertions, 0 deletions
diff --git a/mojo/public/cpp/bindings/tests/request_response_unittest.cc b/mojo/public/cpp/bindings/tests/request_response_unittest.cc
new file mode 100644
index 0000000000..43b8f0dc90
--- /dev/null
+++ b/mojo/public/cpp/bindings/tests/request_response_unittest.cc
@@ -0,0 +1,157 @@
+// 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 <stdint.h>
+#include <utility>
+
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "mojo/public/cpp/bindings/binding.h"
+#include "mojo/public/cpp/test_support/test_utils.h"
+#include "mojo/public/interfaces/bindings/tests/sample_import.mojom.h"
+#include "mojo/public/interfaces/bindings/tests/sample_interfaces.mojom.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace mojo {
+namespace test {
+namespace {
+
+class ProviderImpl : public sample::Provider {
+ public:
+  explicit ProviderImpl(InterfaceRequest<sample::Provider> request)
+      : binding_(this, std::move(request)) {}
+
+  void EchoString(const std::string& a,
+                  const EchoStringCallback& callback) override {
+    EchoStringCallback callback_copy;
+    // Make sure operator= is used.
+    callback_copy = callback;
+    callback_copy.Run(a);
+  }
+
+  void EchoStrings(const std::string& a,
+                   const std::string& b,
+                   const EchoStringsCallback& callback) override {
+    callback.Run(a, b);
+  }
+
+  void EchoMessagePipeHandle(
+      ScopedMessagePipeHandle a,
+      const EchoMessagePipeHandleCallback& callback) override {
+    callback.Run(std::move(a));
+  }
+
+  void EchoEnum(sample::Enum a, const EchoEnumCallback& callback) override {
+    callback.Run(a);
+  }
+
+  void EchoInt(int32_t a, const EchoIntCallback& callback) override {
+    callback.Run(a);
+  }
+
+  Binding<sample::Provider> binding_;
+};
+
+void RecordString(std::string* storage,
+                  const base::Closure& closure,
+                  const std::string& str) {
+  *storage = str;
+  closure.Run();
+}
+
+void RecordStrings(std::string* storage,
+                   const base::Closure& closure,
+                   const std::string& a,
+                   const std::string& b) {
+  *storage = a + b;
+  closure.Run();
+}
+
+void WriteToMessagePipe(const char* text,
+                        const base::Closure& closure,
+                        ScopedMessagePipeHandle handle) {
+  WriteTextMessage(handle.get(), text);
+  closure.Run();
+}
+
+void RecordEnum(sample::Enum* storage,
+                const base::Closure& closure,
+                sample::Enum value) {
+  *storage = value;
+  closure.Run();
+}
+
+class RequestResponseTest : public testing::Test {
+ public:
+  RequestResponseTest() {}
+  ~RequestResponseTest() override { base::RunLoop().RunUntilIdle(); }
+
+  void PumpMessages() { base::RunLoop().RunUntilIdle(); }
+
+ private:
+  base::MessageLoop loop_;
+};
+
+TEST_F(RequestResponseTest, EchoString) {
+  sample::ProviderPtr provider;
+  ProviderImpl provider_impl(MakeRequest(&provider));
+
+  std::string buf;
+  base::RunLoop run_loop;
+  provider->EchoString("hello",
+                       base::Bind(&RecordString, &buf, run_loop.QuitClosure()));
+
+  run_loop.Run();
+
+  EXPECT_EQ(std::string("hello"), buf);
+}
+
+TEST_F(RequestResponseTest, EchoStrings) {
+  sample::ProviderPtr provider;
+  ProviderImpl provider_impl(MakeRequest(&provider));
+
+  std::string buf;
+  base::RunLoop run_loop;
+  provider->EchoStrings("hello", " world", base::Bind(&RecordStrings, &buf,
+                                                      run_loop.QuitClosure()));
+
+  run_loop.Run();
+
+  EXPECT_EQ(std::string("hello world"), buf);
+}
+
+TEST_F(RequestResponseTest, EchoMessagePipeHandle) {
+  sample::ProviderPtr provider;
+  ProviderImpl provider_impl(MakeRequest(&provider));
+
+  MessagePipe pipe2;
+  base::RunLoop run_loop;
+  provider->EchoMessagePipeHandle(
+      std::move(pipe2.handle1),
+      base::Bind(&WriteToMessagePipe, "hello", run_loop.QuitClosure()));
+
+  run_loop.Run();
+
+  std::string value;
+  ReadTextMessage(pipe2.handle0.get(), &value);
+
+  EXPECT_EQ(std::string("hello"), value);
+}
+
+TEST_F(RequestResponseTest, EchoEnum) {
+  sample::ProviderPtr provider;
+  ProviderImpl provider_impl(MakeRequest(&provider));
+
+  sample::Enum value;
+  base::RunLoop run_loop;
+  provider->EchoEnum(sample::Enum::VALUE,
+                     base::Bind(&RecordEnum, &value, run_loop.QuitClosure()));
+  run_loop.Run();
+
+  EXPECT_EQ(sample::Enum::VALUE, value);
+}
+
+}  // namespace
+}  // namespace test
+}  // namespace mojo