Implement unions as members of structs.

mojo/public/cpp/bindings/tests/array_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 "mojo/public/cpp/bindings/array.h"
 #include "mojo/public/cpp/bindings/lib/array_internal.h"
 #include "mojo/public/cpp/bindings/lib/array_serialization.h"
 #include "mojo/public/cpp/bindings/lib/fixed_buffer.h"
 #include "mojo/public/cpp/bindings/tests/container_test_util.h"
 #include "mojo/public/cpp/environment/environment.h"
 #include "mojo/public/interfaces/bindings/tests/test_structs.mojom.h"
+#include "mojo/public/interfaces/bindings/tests/test_unions.mojom.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace mojo {
 namespace test {
 namespace {
 
 using mojo::internal::Array_Data;
 using mojo::internal::ArrayValidateParams;
 using mojo::internal::FixedBuffer;
 using mojo::internal::NoValidateParams;
@@ skipping 416 matching lines @@
   EXPECT_EQ(array.size(), MoveOnlyType::num_instances());
 
   for (size_t i = 0; i < array.size(); i++) {
     EXPECT_TRUE(array[i].moved());
     EXPECT_EQ(value_ptrs[i], array[i].ptr());
   }
   array.reset();
   EXPECT_EQ(0u, MoveOnlyType::num_instances());
 }
 
+TEST_F(ArrayTest, PodUnionInArray) {
+  SmallStructPtr small_struct(SmallStruct::New());
+  small_struct->pod_union_array = Array<PodUnionPtr>(2);
+  small_struct->pod_union_array[0] = PodUnion::New();
+  small_struct->pod_union_array[1] = PodUnion::New();
+
+  small_struct->pod_union_array[0]->set_f_int8(10);
+  small_struct->pod_union_array[1]->set_f_int16(12);
+
+  EXPECT_EQ(10, small_struct->pod_union_array[0]->get_f_int8());
+  EXPECT_EQ(12, small_struct->pod_union_array[1]->get_f_int16());
+}
+
+TEST_F(ArrayTest, PodUnionInArraySerialization) {
+  Array<PodUnionPtr> array(2);
+  array[0] = PodUnion::New();
+  array[1] = PodUnion::New();
+
+  array[0]->set_f_int8(10);
+  array[1]->set_f_int16(12);
+  EXPECT_EQ(2U, array.size());
+
+  size_t size = GetSerializedSize_(array);
+  EXPECT_EQ(40U, size);
+
+  mojo::internal::FixedBuffer buf(size);
+  mojo::internal::Array_Data<internal::PodUnion_Data>* data;
+  SerializeArray_<mojo::internal::ArrayValidateParams<
+      0, false, mojo::internal::NoValidateParams>>(array.Pass(), &buf, &data);
+
+  Array<PodUnionPtr> array2;
+  Deserialize_(data, &array2);
+
+  EXPECT_EQ(2U, array2.size());
+
+  EXPECT_EQ(10, array2[0]->get_f_int8());
+  EXPECT_EQ(12, array2[1]->get_f_int16());
+}
+
+TEST_F(ArrayTest, PodUnionInArraySerializationWithNull) {
+  Array<PodUnionPtr> array(2);
+  array[0] = PodUnion::New();
+
+  array[0]->set_f_int8(10);
+  EXPECT_EQ(2U, array.size());
+
+  size_t size = GetSerializedSize_(array);
+  EXPECT_EQ(40U, size);
+
+  mojo::internal::FixedBuffer buf(size);
+  mojo::internal::Array_Data<internal::PodUnion_Data>* data;
+  SerializeArray_<mojo::internal::ArrayValidateParams<
+      0, true, mojo::internal::NoValidateParams>>(array.Pass(), &buf, &data);
+
+  Array<PodUnionPtr> array2;
+  Deserialize_(data, &array2);
+
+  EXPECT_EQ(2U, array2.size());
+
+  EXPECT_EQ(10, array2[0]->get_f_int8());
+  EXPECT_TRUE(array2[1].is_null());
+}
+
 }  // namespace
 }  // namespace test
 }  // namespace mojo