Mojo: more idiomatic C++ bindings

mojo/apps/js/test/js_to_cpp_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/file_util.h"
 #include "base/files/file_path.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "gin/public/isolate_holder.h"
 #include "mojo/apps/js/mojo_runner_delegate.h"
 #include "mojo/apps/js/test/js_to_cpp.mojom.h"
 #include "mojo/common/common_type_converters.h"
 #include "mojo/common/test/test_utils.h"
-#include "mojo/public/cpp/bindings/allocation_scope.h"
 #include "mojo/public/cpp/environment/environment.h"
 #include "mojo/public/cpp/system/core.h"
 #include "mojo/public/cpp/system/macros.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace mojo {
 namespace js {
 
 // Global value updated by some checks to prevent compilers from optimizing
 // reads out of existence.
@@ skipping 44 matching lines @@
   uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer));
   MojoResult result = MojoReadData(
       data_pipe_handle, buffer, &buffer_size, MOJO_READ_DATA_FLAG_NONE);
   EXPECT_EQ(MOJO_RESULT_OK, result);
   EXPECT_EQ(64u, buffer_size);
   for (int i = 0; i < 64; ++i) {
     EXPECT_EQ(i, buffer[i]);
   }
 }
 
-// NOTE: Callers will need to have established an AllocationScope, or you're
-// gonna have a bad time.
-js_to_cpp::EchoArgs BuildSampleEchoArgs() {
-  js_to_cpp::EchoArgs::Builder builder;
-  builder.set_si64(kExpectedInt64Value);
-  builder.set_si32(kExpectedInt32Value);
-  builder.set_si16(kExpectedInt16Value);
-  builder.set_si8(kExpectedInt8Value);
-  builder.set_ui64(kExpectedUInt64Value);
-  builder.set_ui32(kExpectedUInt32Value);
-  builder.set_ui16(kExpectedUInt16Value);
-  builder.set_ui8(kExpectedUInt8Value);
-  builder.set_float_val(kExpectedFloatVal);
-  builder.set_float_inf(kExpectedFloatInf);
-  builder.set_float_nan(kExpectedFloatNan);
-  builder.set_double_val(kExpectedDoubleVal);
-  builder.set_double_inf(kExpectedDoubleInf);
-  builder.set_double_nan(kExpectedDoubleNan);
-  builder.set_name("coming");
-  mojo::Array<mojo::String>::Builder string_array(3);
-  string_array[0] = "one";
-  string_array[1] = "two";
-  string_array[2] = "three";
-  builder.set_string_array(string_array.Finish());
-  return builder.Finish();
+js_to_cpp::EchoArgsPtr BuildSampleEchoArgs() {
+  js_to_cpp::EchoArgsPtr args;
+  args->si64 = kExpectedInt64Value;
+  args->si32 = kExpectedInt32Value;
+  args->si16 = kExpectedInt16Value;
+  args->si8 = kExpectedInt8Value;
+  args->ui64 = kExpectedUInt64Value;
+  args->ui32 = kExpectedUInt32Value;
+  args->ui16 = kExpectedUInt16Value;
+  args->ui8 = kExpectedUInt8Value;
+  args->float_val = kExpectedFloatVal;
+  args->float_inf = kExpectedFloatInf;
+  args->float_nan = kExpectedFloatNan;
+  args->double_val = kExpectedDoubleVal;
+  args->double_inf = kExpectedDoubleInf;
+  args->double_nan = kExpectedDoubleNan;
+  args->name = String::From("coming");
+  args->string_array = Array<String>::New(3);
+  args->string_array[0] = String::From("one");
+  args->string_array[1] = String::From("two");
+  args->string_array[2] = String::From("three");
+  return args.Pass();
 }
 
 void CheckSampleEchoArgs(const js_to_cpp::EchoArgs& arg) {
-  EXPECT_EQ(kExpectedInt64Value, arg.si64());
-  EXPECT_EQ(kExpectedInt32Value, arg.si32());
-  EXPECT_EQ(kExpectedInt16Value, arg.si16());
-  EXPECT_EQ(kExpectedInt8Value, arg.si8());
-  EXPECT_EQ(kExpectedUInt64Value, arg.ui64());
-  EXPECT_EQ(kExpectedUInt32Value, arg.ui32());
-  EXPECT_EQ(kExpectedUInt16Value, arg.ui16());
-  EXPECT_EQ(kExpectedUInt8Value, arg.ui8());
-  EXPECT_EQ(kExpectedFloatVal, arg.float_val());
-  EXPECT_EQ(kExpectedFloatInf, arg.float_inf());
-  EXPECT_NAN(arg.float_nan());
-  EXPECT_EQ(kExpectedDoubleVal, arg.double_val());
-  EXPECT_EQ(kExpectedDoubleInf, arg.double_inf());
-  EXPECT_NAN(arg.double_nan());
-  EXPECT_EQ(std::string("coming"), arg.name().To<std::string>());
-  EXPECT_EQ(std::string("one"), arg.string_array()[0].To<std::string>());
-  EXPECT_EQ(std::string("two"), arg.string_array()[1].To<std::string>());
-  EXPECT_EQ(std::string("three"), arg.string_array()[2].To<std::string>());
-  CheckDataPipe(arg.data_handle().get().value());
+  EXPECT_EQ(kExpectedInt64Value, arg.si64);
+  EXPECT_EQ(kExpectedInt32Value, arg.si32);
+  EXPECT_EQ(kExpectedInt16Value, arg.si16);
+  EXPECT_EQ(kExpectedInt8Value, arg.si8);
+  EXPECT_EQ(kExpectedUInt64Value, arg.ui64);
+  EXPECT_EQ(kExpectedUInt32Value, arg.ui32);
+  EXPECT_EQ(kExpectedUInt16Value, arg.ui16);
+  EXPECT_EQ(kExpectedUInt8Value, arg.ui8);
+  EXPECT_EQ(kExpectedFloatVal, arg.float_val);
+  EXPECT_EQ(kExpectedFloatInf, arg.float_inf);
+  EXPECT_NAN(arg.float_nan);
+  EXPECT_EQ(kExpectedDoubleVal, arg.double_val);
+  EXPECT_EQ(kExpectedDoubleInf, arg.double_inf);
+  EXPECT_NAN(arg.double_nan);
+  EXPECT_EQ(std::string("coming"), arg.name.To<std::string>());
+  EXPECT_EQ(std::string("one"), arg.string_array[0].To<std::string>());
+  EXPECT_EQ(std::string("two"), arg.string_array[1].To<std::string>());
+  EXPECT_EQ(std::string("three"), arg.string_array[2].To<std::string>());
+  CheckDataPipe(arg.data_handle.get().value());
 }
 
-js_to_cpp::EchoArgsList BuildSampleEchoArgsList() {
-  js_to_cpp::EchoArgsList::Builder builder;
-  builder.set_item(BuildSampleEchoArgs());
-  return builder.Finish();
+js_to_cpp::EchoArgsListPtr BuildSampleEchoArgsList() {
+  js_to_cpp::EchoArgsListPtr args_list(js_to_cpp::EchoArgsList::New());
+  args_list->item = BuildSampleEchoArgs();
+  return args_list.Pass();
 }
 
-void CheckSampleEchoArgsList(const js_to_cpp::EchoArgsList& list) {
+void CheckSampleEchoArgsList(const js_to_cpp::EchoArgsListPtr& list) {
   if (list.is_null())
     return;
-  CheckSampleEchoArgs(list.item());
-  CheckSampleEchoArgsList(list.next());
+  CheckSampleEchoArgs(*list->item);
+  CheckSampleEchoArgsList(list->next);
 }
 
 void CheckCorruptedString(const mojo::String& arg) {
   // The values don't matter so long as all accesses are within bounds.
   if (arg.is_null())
     return;
   for (size_t i = 0; i < arg.size(); ++i)
     g_waste_accumulator += arg[i];
 }
 
@@ skipping 25 matching lines @@
   }
 
   virtual void TestFinished() OVERRIDE {
     NOTREACHED();
   }
 
   virtual void PingResponse() OVERRIDE {
     NOTREACHED();
   }
 
-  virtual void EchoResponse(const js_to_cpp::EchoArgsList& list) OVERRIDE {
+  virtual void EchoResponse(js_to_cpp::EchoArgsListPtr list) OVERRIDE {
     NOTREACHED();
   }
 
-  virtual void BitFlipResponse(const js_to_cpp::EchoArgs& arg1) OVERRIDE {
+  virtual void BitFlipResponse(js_to_cpp::EchoArgsPtr arg1) OVERRIDE {
     NOTREACHED();
   }
 
  protected:
   base::RunLoop* run_loop_;
   js_to_cpp::JsSide* js_side_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(CppSideConnection);
 };
@@ skipping 27 matching lines @@
 class EchoCppSideConnection : public CppSideConnection {
  public:
   explicit EchoCppSideConnection() :
       message_count_(0),
       termination_seen_(false) {
   }
   virtual ~EchoCppSideConnection() {}
 
   // js_to_cpp::CppSide:
   virtual void StartTest() OVERRIDE {
-    AllocationScope scope;
     js_side_->Echo(kExpectedMessageCount, BuildSampleEchoArgsList());
   }
 
-  virtual void EchoResponse(const js_to_cpp::EchoArgsList& list) OVERRIDE {
-    const js_to_cpp::EchoArgs& special_arg = list.item();
+  virtual void EchoResponse(js_to_cpp::EchoArgsListPtr list) OVERRIDE {
+    const js_to_cpp::EchoArgsPtr& special_arg = list->item;
     message_count_ += 1;
-    EXPECT_EQ(-1, special_arg.si64());
-    EXPECT_EQ(-1, special_arg.si32());
-    EXPECT_EQ(-1, special_arg.si16());
-    EXPECT_EQ(-1, special_arg.si8());
-    EXPECT_EQ(std::string("going"), special_arg.name().To<std::string>());
-    CheckSampleEchoArgsList(list.next());
+    EXPECT_EQ(-1, special_arg->si64);
+    EXPECT_EQ(-1, special_arg->si32);
+    EXPECT_EQ(-1, special_arg->si16);
+    EXPECT_EQ(-1, special_arg->si8);
+    EXPECT_EQ(std::string("going"), special_arg->name.To<std::string>());
+    CheckSampleEchoArgsList(list->next);
   }
 
   virtual void TestFinished() OVERRIDE {
     termination_seen_ = true;
     run_loop()->Quit();
   }
 
   bool DidSucceed() {
     return termination_seen_ && message_count_ == kExpectedMessageCount;
   }
 
  private:
   static const int kExpectedMessageCount = 100;
   int message_count_;
   bool termination_seen_;
   DISALLOW_COPY_AND_ASSIGN(EchoCppSideConnection);
 };
 
 // Test that corrupted messages don't wreak havoc.
 class BitFlipCppSideConnection : public CppSideConnection {
  public:
   explicit BitFlipCppSideConnection() : termination_seen_(false) {}
   virtual ~BitFlipCppSideConnection() {}
 
   // js_to_cpp::CppSide:
   virtual void StartTest() OVERRIDE {
-    AllocationScope scope;
     js_side_->BitFlip(BuildSampleEchoArgs());
   }
 
-  virtual void BitFlipResponse(const js_to_cpp::EchoArgs& arg) OVERRIDE {
+  virtual void BitFlipResponse(js_to_cpp::EchoArgsPtr arg) OVERRIDE {
     if (arg.is_null())
       return;
-    CheckCorruptedString(arg.name());
-    CheckCorruptedStringArray(arg.string_array());
-    if (arg.data_handle().is_valid())
-      CheckDataPipe(arg.data_handle().get().value());
+    CheckCorruptedString(arg->name);
+    CheckCorruptedStringArray(arg->string_array);
+    if (arg->data_handle.is_valid())
+      CheckDataPipe(arg->data_handle.get().value());
   }
 
   virtual void TestFinished() OVERRIDE {
     termination_seen_ = true;
     run_loop()->Quit();
   }
 
   bool DidSucceed() {
     return termination_seen_;
   }
@@ skipping 61 matching lines @@
   if (IsRunningOnIsolatedBot())
     return;
 
   BitFlipCppSideConnection cpp_side_connection;
   RunTest("mojo/apps/js/test/js_to_cpp_unittest", &cpp_side_connection);
   EXPECT_TRUE(cpp_side_connection.DidSucceed());
 }
 
 }  // namespace js
 }  // namespace mojo