New go generator shakedown.

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
         isEnum              bool
+        isInterface         bool
+        isInterfaceRequest  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) IsEnum() bool                                 { return m.isEnum }
+func (m mockEncodingInfo) IsInterface() bool                            { return m.isInterface }
+func (m mockEncodingInfo) IsInterfaceRequest() bool                     { return m.isInterfaceRequest }
 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 331 matching lines @@
         return err
 }`
 
         encodingInfo := mockEncodingInfo{
                 isEnum:     true,
                 identifier: "s.EnumField",
         }
 
         check(t, expected, "FieldEncodingTmpl", encodingInfo)
 }
+
+func TestEncodingInterfaceFieldEncoding(t *testing.T) {
+        expected := `if err := encoder.WriteInterface(s.IntField.PassMessagePipe()); err != nil {
+        return err
+}`
+
+        encodingInfo := mockEncodingInfo{
+                isInterface:   true,
+                identifier:    "s.IntField",
+                writeFunction: "WriteInterface",
+        }
+
+        check(t, expected, "FieldEncodingTmpl", encodingInfo)
+}
+
+func TestEncodingNullableInterfaceFieldEncoding(t *testing.T) {
+        expected := `if s.IntField == nil {
+        encoder.WriteInvalidInterface()
+} else {
+        if err := encoder.WriteInterface(s.IntField.PassMessagePipe()); err != nil {
+                return err
+        }
+}`
+
+        encodingInfo := mockEncodingInfo{
+                isInterface:   true,
+                isNullable:    true,
+                identifier:    "s.IntField",
+                writeFunction: "WriteInterface",
+        }
+
+        check(t, expected, "FieldEncodingTmpl", encodingInfo)
+}
+
+func TestEncodingInterfaceRequestFieldEncoding(t *testing.T) {
+        expected := `if err := encoder.WriteHandle(s.IntField.PassMessagePipe()); err != nil {
+        return err
+}`
+
+        encodingInfo := mockEncodingInfo{
+                isInterface:        true,
+                isInterfaceRequest: true,
+                identifier:         "s.IntField",
+                writeFunction:      "WriteHandle",
+        }
+
+        check(t, expected, "FieldEncodingTmpl", encodingInfo)
+}
+
+func TestEncodingNullableInterfaceRequestFieldEncoding(t *testing.T) {
+        expected := `if s.IntField == nil {
+        encoder.WriteInvalidHandle()
+} else {
+        if err := encoder.WriteHandle(s.IntField.PassMessagePipe()); err != nil {
+                return err
+        }
+}`
+
+        encodingInfo := mockEncodingInfo{
+                isInterface:        true,
+                isInterfaceRequest: true,
+                isNullable:         true,
+                identifier:         "s.IntField",
+                writeFunction:      "WriteHandle",
+        }
+
+        check(t, expected, "FieldEncodingTmpl", encodingInfo)
+}