1 // Protocol Buffers for Go with Gadgets 2 // 3 // Copyright (c) 2013, The GoGo Authors. All rights reserved. 4 // http://github.com/gogo/protobuf 5 // 6 // Redistribution and use in source and binary forms, with or without 7 // modification, are permitted provided that the following conditions are 8 // met: 9 // 10 // * Redistributions of source code must retain the above copyright 11 // notice, this list of conditions and the following disclaimer. 12 // * Redistributions in binary form must reproduce the above 13 // copyright notice, this list of conditions and the following disclaimer 14 // in the documentation and/or other materials provided with the 15 // distribution. 16 // 17 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 29 package description 30 31 import ( 32 "github.com/gogo/protobuf/gogoproto" 33 "github.com/gogo/protobuf/plugin/testgen" 34 "github.com/gogo/protobuf/protoc-gen-gogo/generator" 35 ) 36 37 type test struct { 38 *generator.Generator 39 } 40 41 func NewTest(g *generator.Generator) testgen.TestPlugin { 42 return &test{g} 43 } 44 45 func (p *test) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool { 46 used := false 47 testingPkg := imports.NewImport("testing") 48 for _, message := range file.Messages() { 49 if !gogoproto.HasDescription(file.FileDescriptorProto, message.DescriptorProto) || 50 !gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) { 51 continue 52 } 53 if message.DescriptorProto.GetOptions().GetMapEntry() { 54 continue 55 } 56 used = true 57 } 58 59 if used { 60 localName := generator.FileName(file) 61 p.P(`func Test`, localName, `Description(t *`, testingPkg.Use(), `.T) {`) 62 p.In() 63 p.P(localName, `Description()`) 64 p.Out() 65 p.P(`}`) 66 67 } 68 return used 69 } 70 71 func init() { 72 testgen.RegisterTestPlugin(NewTest) 73 } 74