Implement python mojo bindings unions.

mojo/python/tests/bindings_unions_unittest.py

Index: mojo/python/tests/bindings_unions_unittest.py
diff --git a/mojo/python/tests/bindings_unions_unittest.py b/mojo/python/tests/bindings_unions_unittest.py
new file mode 100644
index 0000000000000000000000000000000000000000..2b74490e5b51293e81733a1522464bd9d3770cd6
--- /dev/null
+++ b/mojo/python/tests/bindings_unions_unittest.py
@@ -0,0 +1,172 @@
+import struct
+import unittest
+
+# Generated files
+# pylint: disable=F0401
+import test_unions_mojom
+import mojo_bindings.serialization as serialization
+
+class UnionBindingsTest(unittest.TestCase):
+
+  def testBasics(self):
+    u = test_unions_mojom.PodUnion()
+    u.f_uint32 = 32
+    self.assertEquals(u.f_uint32, 32)
+    self.assertEquals(u.data, 32)
+    self.assertEquals(test_unions_mojom.PodUnion.Tags.f_uint32, u.tag)
+
+    u = test_unions_mojom.PodUnion(f_uint8=8)
+    self.assertEquals(u.f_uint8, 8)
+    self.assertEquals(u.data, 8)
+    self.assertEquals(test_unions_mojom.PodUnion.Tags.f_uint8, u.tag)
+
+    with self.assertRaises(TypeError):
+      test_unions_mojom.PodUnion(f_uint8=8, f_int16=10)
+
+    with self.assertRaises(AttributeError):
+      test_unions_mojom.PodUnion(bad_field=10)
+
+    with self.assertRaises(AttributeError):
+      u = test_unions_mojom.PodUnion()
+      u.bad_field = 32
+
+    with self.assertRaises(AttributeError):
+      _ = u.f_uint16
+
+  def testPodUnionSerialization(self):
+    u = test_unions_mojom.PodUnion(f_uint32=32)
+    (data, handles) = u.Serialize()
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.PodUnion.Deserialize(context)
+
+    self.assertEquals(u, decoded)
+
+  def testObjectInUnionSerialization(self):
+    u = test_unions_mojom.ObjectUnion(
+        f_dummy=test_unions_mojom.DummyStruct())
+    u.f_dummy.f_int8 = 8
+    (data, handles) = u.Serialize()
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.ObjectUnion.Deserialize(context)
+
+    self.assertEquals(u, decoded)
+
+  def testObjectInUnionInObjectSerialization(self):
+    s = test_unions_mojom.SmallObjStruct()
+    s.obj_union = test_unions_mojom.ObjectUnion(
+        f_dummy=test_unions_mojom.DummyStruct())
+    s.obj_union.f_dummy.f_int8 = 25
+    (data, handles) = s.Serialize()
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.SmallObjStruct.Deserialize(context)
+
+    self.assertEquals(s, decoded)
+
+  def testNestedUnionSerialization(self):
+    u = test_unions_mojom.ObjectUnion(
+        f_pod_union=test_unions_mojom.PodUnion(f_int32=32))
+
+    (data, handles) = u.Serialize()
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.ObjectUnion.Deserialize(context)
+
+    self.assertEquals(u, decoded)
+
+  def testNullableNullObjectInUnionSerialization(self):
+    u =  test_unions_mojom.ObjectUnion(f_nullable=None)
+    (data, handles) = u.Serialize()
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.ObjectUnion.Deserialize(context)
+
+    self.assertEquals(u, decoded)
+
+  def testNonNullableNullObjectInUnionSerialization(self):
+    u =  test_unions_mojom.ObjectUnion(f_dummy=None)
+    with self.assertRaises(serialization.SerializationException):
+      u.Serialize()
+
+  def testArrayInUnionSerialization(self):
+    u = test_unions_mojom.ObjectUnion(
+        f_array_int8=[1, 2, 3, 4, 5])
+    (data, handles) = u.Serialize()
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.ObjectUnion.Deserialize(context)
+
+    self.assertEquals(u, decoded)
+
+  def testMapInUnionSerialization(self):
+    u = test_unions_mojom.ObjectUnion(
+        f_map_int8={'one': 1, 'two': 2, 'three': 3})
+    (data, handles) = u.Serialize()
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.ObjectUnion.Deserialize(context)
+
+    self.assertEquals(u, decoded)
+
+  def testUnionInObject(self):
+    s = test_unions_mojom.SmallStruct()
+    s.pod_union = test_unions_mojom.PodUnion(f_uint32=32)
+    (data, handles) = s.Serialize()
+
+    # This is where the data should be serialized to.
+    size, tag, value = struct.unpack_from('<IIQ', buffer(data), 16)
+    self.assertEquals(16, size)
+    self.assertEquals(6, tag)
+    self.assertEquals(32, value)
+
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.SmallStruct.Deserialize(context)
+
+    self.assertEquals(s, decoded)
+
+  def testUnionInArray(self):
+    s = test_unions_mojom.SmallStruct()
+    s.pod_union_array = [
+        test_unions_mojom.PodUnion(f_uint32=32),
+        test_unions_mojom.PodUnion(f_uint16=16),
+        test_unions_mojom.PodUnion(f_uint64=64),
+        ]
+    (data, handles) = s.Serialize()
+
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.SmallStruct.Deserialize(context)
+
+    self.assertEquals(s, decoded)
+
+  def testNonNullableNullUnionInArray(self):
+    s = test_unions_mojom.SmallStruct()
+    s.pod_union_array = [
+        test_unions_mojom.PodUnion(f_uint32=32),
+        None,
+        test_unions_mojom.PodUnion(f_uint64=64),
+        ]
+    with self.assertRaises(serialization.SerializationException):
+      s.Serialize()
+
+  def testNullableNullUnionInArray(self):
+    s = test_unions_mojom.SmallStruct()
+    s.nullable_pod_union_array = [
+        test_unions_mojom.PodUnion(f_uint32=32),
+        None,
+        test_unions_mojom.PodUnion(f_uint64=64),
+        ]
+    (data, handles) = s.Serialize()
+
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.SmallStruct.Deserialize(context)
+
+    self.assertEquals(s, decoded)
+
+  def testUnionInMap(self):
+    s = test_unions_mojom.SmallStruct()
+    s.pod_union_map = {
+        'f_uint32': test_unions_mojom.PodUnion(f_uint32=32),
+        'f_uint16': test_unions_mojom.PodUnion(f_uint16=16),
+        'f_uint64': test_unions_mojom.PodUnion(f_uint64=64),
+        }
+    (data, handles) = s.Serialize()
+
+    context = serialization.RootDeserializationContext(data, handles)
+    decoded = test_unions_mojom.SmallStruct.Deserialize(context)
+
+    self.assertEquals(s, decoded)