Encode and decode fields of union type.

mojom/generators/go/templates/encoding_test.go

 // Copyright 2016 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.
 
 package templates
 
 import (
         "testing"
 
         "mojom/generators/go/translator"
 )
 
 type mockEncodingInfo struct {
         translator.EncodingInfo
         isSimple            bool
         isPointer           bool
         isHandle            bool
         isArray             bool
         isMap               bool
         isNullable          bool
         isStruct            bool
+        isUnion             bool
         elementEncodingInfo *mockEncodingInfo
         keyEncodingInfo     *mockEncodingInfo
         valueEncodingInfo   *mockEncodingInfo
         bitSize             uint32
         writeFunction       string
         readFunction        string
         identifier          string
         goType              string
 }
 
 func (m mockEncodingInfo) IsSimple() bool                               { return m.isSimple }
 func (m mockEncodingInfo) IsPointer() bool                              { return m.isPointer }
 func (m mockEncodingInfo) IsHandle() bool                               { return m.isHandle }
 func (m mockEncodingInfo) IsArray() bool                                { return m.isArray }
 func (m mockEncodingInfo) IsMap() bool                                  { return m.isMap }
 func (m mockEncodingInfo) IsNullable() bool                             { return m.isNullable }
 func (m mockEncodingInfo) IsStruct() bool                               { return m.isStruct }
+func (m mockEncodingInfo) IsUnion() bool                                { return m.isUnion }
 func (m mockEncodingInfo) ElementEncodingInfo() translator.EncodingInfo { return m.elementEncodingInfo }
 func (m mockEncodingInfo) KeyEncodingInfo() translator.EncodingInfo     { return m.keyEncodingInfo }
 func (m mockEncodingInfo) ValueEncodingInfo() translator.EncodingInfo   { return m.valueEncodingInfo }
 func (m mockEncodingInfo) BitSize() uint32                              { return m.bitSize }
 func (m mockEncodingInfo) WriteFunction() string                        { return m.writeFunction }
 func (m mockEncodingInfo) ReadFunction() string                         { return m.readFunction }
 func (m mockEncodingInfo) Identifier() string                           { return m.identifier }
 func (m mockEncodingInfo) GoType() string                               { return m.goType }
 
 func TestEncodingSimpleFieldEncoding(t *testing.T) {
@@ skipping 238 matching lines @@
 
         encodingInfo := mockEncodingInfo{
                 isPointer:  true,
                 isStruct:   true,
                 isNullable: true,
                 identifier: "s.FNullableStruct",
         }
 
         check(t, expected, "FieldEncodingTmpl", encodingInfo)
 }
+
+func TestEncodingUnionFieldEncoding(t *testing.T) {
+        expected := `if s.FUnion == nil {
+        return &bindings.ValidationError{bindings.UnexpectedNullUnion, "unexpected null union"}
+} else {
+        if err := s.FUnion.Encode(encoder); err != nil {
+                return err
+        }
+}`
+
+        encodingInfo := mockEncodingInfo{
+                isUnion:    true,
+                identifier: "s.FUnion",
+        }
+
+        check(t, expected, "FieldEncodingTmpl", encodingInfo)
+}
+
+func TestEncodingNullableUnionFieldEncoding(t *testing.T) {
+        expected := `if s.FUnion == nil {
+        encoder.WriteNullUnion()
+} else {
+        if err := s.FUnion.Encode(encoder); err != nil {
+                return err
+        }
+}`
+
+        encodingInfo := mockEncodingInfo{
+                isNullable: true,
+                isUnion:    true,
+                identifier: "s.FUnion",
+        }
+
+        check(t, expected, "FieldEncodingTmpl", encodingInfo)
+}
+
+func TestEncodingNestedUnionFieldEncoding(t *testing.T) {
+        expected := `if err := encoder.WritePointer(); err != nil {
+                return err
+}
+encoder.StartNestedUnion()
+if s.FUnion == nil {
+        return &bindings.ValidationError{bindings.UnexpectedNullUnion, "unexpected null union"}
+} else {
+        if err := s.FUnion.Encode(encoder); err != nil {
+                return err
+        }
+}`
+
+        encodingInfo := mockEncodingInfo{
+                isPointer:  true,
+                isUnion:    true,
+                identifier: "s.FUnion",
+        }
+
+        check(t, expected, "FieldEncodingTmpl", encodingInfo)
+}
+
+func TestEncodingNestedNullableUnionFieldEncoding(t *testing.T) {
+        expected := `if err := encoder.WritePointer(); err != nil {
+                return err
+}
+encoder.StartNestedUnion()
+if s.FUnion == nil {
+        encoder.WriteNullUnion()
+} else {
+        if err := s.FUnion.Encode(encoder); err != nil {
+                return err
+        }
+}`
+
+        encodingInfo := mockEncodingInfo{
+                isNullable: true,
+                isPointer:  true,
+                isUnion:    true,
+                identifier: "s.FUnion",
+        }
+
+        check(t, expected, "FieldEncodingTmpl", encodingInfo)
+}