plugin/testgen/testgen.go - external/github.com/gogo/protobuf - Git at Google

 // Protocol Buffers for Go with Gadgets
 //
 // Copyright (c) 2013, The GoGo Authors. All rights reserved.
 // http://github.com/gogo/protobuf
 //
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are
 // met:
 //
 //     * Redistributions of source code must retain the above copyright
 // notice, this list of conditions and the following disclaimer.
 //     * Redistributions in binary form must reproduce the above
 // copyright notice, this list of conditions and the following disclaimer
 // in the documentation and/or other materials provided with the
 // distribution.
 //
 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 /*
 The testgen plugin generates Test and Benchmark functions for each message.

 Tests are enabled using the following extensions:

   - testgen
   - testgen_all

 Benchmarks are enabled using the following extensions:

   - benchgen
   - benchgen_all

 Let us look at:

   github.com/gogo/protobuf/test/example/example.proto

 Btw all the output can be seen at:

   github.com/gogo/protobuf/test/example/*

 The following message:

   option (gogoproto.testgen_all) = true;
   option (gogoproto.benchgen_all) = true;

   message A {
 	optional string Description = 1 [(gogoproto.nullable) = false];
 	optional int64 Number = 2 [(gogoproto.nullable) = false];
 	optional bytes Id = 3 [(gogoproto.customtype) = "github.com/gogo/protobuf/test/custom.Uuid", (gogoproto.nullable) = false];
   }

 given to the testgen plugin, will generate the following test code:

 	func TestAProto(t *testing.T) {
 		popr := math_rand.New(math_rand.NewSource(time.Now().UnixNano()))
 		p := NewPopulatedA(popr, false)
 		dAtA, err := github_com_gogo_protobuf_proto.Marshal(p)
 		if err != nil {
 			panic(err)
 		}
 		msg := &A{}
 		if err := github_com_gogo_protobuf_proto.Unmarshal(dAtA, msg); err != nil {
 			panic(err)
 		}
 		for i := range dAtA {
 			dAtA[i] = byte(popr.Intn(256))
 		}
 		if err := p.VerboseEqual(msg); err != nil {
 			t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)
 		}
 		if !p.Equal(msg) {
 			t.Fatalf("%#v !Proto %#v", msg, p)
 		}
 	}

 	func BenchmarkAProtoMarshal(b *testing.B) {
 		popr := math_rand.New(math_rand.NewSource(616))
 		total := 0
 		pops := make([]*A, 10000)
 		for i := 0; i < 10000; i++ {
 			pops[i] = NewPopulatedA(popr, false)
 		}
 		b.ResetTimer()
 		for i := 0; i < b.N; i++ {
 			dAtA, err := github_com_gogo_protobuf_proto.Marshal(pops[i%10000])
 			if err != nil {
 				panic(err)
 			}
 			total += len(dAtA)
 		}
 		b.SetBytes(int64(total / b.N))
 	}

 	func BenchmarkAProtoUnmarshal(b *testing.B) {
 		popr := math_rand.New(math_rand.NewSource(616))
 		total := 0
 		datas := make([][]byte, 10000)
 		for i := 0; i < 10000; i++ {
 			dAtA, err := github_com_gogo_protobuf_proto.Marshal(NewPopulatedA(popr, false))
 			if err != nil {
 				panic(err)
 			}
 			datas[i] = dAtA
 		}
 		msg := &A{}
 		b.ResetTimer()
 		for i := 0; i < b.N; i++ {
 			total += len(datas[i%10000])
 			if err := github_com_gogo_protobuf_proto.Unmarshal(datas[i%10000], msg); err != nil {
 				panic(err)
 			}
 		}
 		b.SetBytes(int64(total / b.N))
 	}


 	func TestAJSON(t *testing1.T) {
 		popr := math_rand1.New(math_rand1.NewSource(time1.Now().UnixNano()))
 		p := NewPopulatedA(popr, true)
 		jsondata, err := encoding_json.Marshal(p)
 		if err != nil {
 			panic(err)
 		}
 		msg := &A{}
 		err = encoding_json.Unmarshal(jsondata, msg)
 		if err != nil {
 			panic(err)
 		}
 		if err := p.VerboseEqual(msg); err != nil {
 			t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)
 		}
 		if !p.Equal(msg) {
 			t.Fatalf("%#v !Json Equal %#v", msg, p)
 		}
 	}

 	func TestAProtoText(t *testing2.T) {
 		popr := math_rand2.New(math_rand2.NewSource(time2.Now().UnixNano()))
 		p := NewPopulatedA(popr, true)
 		dAtA := github_com_gogo_protobuf_proto1.MarshalTextString(p)
 		msg := &A{}
 		if err := github_com_gogo_protobuf_proto1.UnmarshalText(dAtA, msg); err != nil {
 			panic(err)
 		}
 		if err := p.VerboseEqual(msg); err != nil {
 			t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)
 		}
 		if !p.Equal(msg) {
 			t.Fatalf("%#v !Proto %#v", msg, p)
 		}
 	}

 	func TestAProtoCompactText(t *testing2.T) {
 		popr := math_rand2.New(math_rand2.NewSource(time2.Now().UnixNano()))
 		p := NewPopulatedA(popr, true)
 		dAtA := github_com_gogo_protobuf_proto1.CompactTextString(p)
 		msg := &A{}
 		if err := github_com_gogo_protobuf_proto1.UnmarshalText(dAtA, msg); err != nil {
 			panic(err)
 		}
 		if err := p.VerboseEqual(msg); err != nil {
 			t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)
 		}
 		if !p.Equal(msg) {
 			t.Fatalf("%#v !Proto %#v", msg, p)
 		}
 	}

 Other registered tests are also generated.
 Tests are registered to this test plugin by calling the following function.

   func RegisterTestPlugin(newFunc NewTestPlugin)

 where NewTestPlugin is:

   type NewTestPlugin func(g *generator.Generator) TestPlugin

 and TestPlugin is an interface:

   type TestPlugin interface {
 	Generate(imports generator.PluginImports, file *generator.FileDescriptor) (used bool)
   }

 Plugins that use this interface include:

   - populate
   - gostring
   - equal
   - union
   - and more

 Please look at these plugins as examples of how to create your own.
 A good idea is to let each plugin generate its own tests.

 */
 package testgen

 import (
 	"github.com/gogo/protobuf/gogoproto"
 	"github.com/gogo/protobuf/protoc-gen-gogo/generator"
 )

 type TestPlugin interface {
 	Generate(imports generator.PluginImports, file *generator.FileDescriptor) (used bool)
 }

 type NewTestPlugin func(g *generator.Generator) TestPlugin

 var testplugins = make([]NewTestPlugin, 0)

 func RegisterTestPlugin(newFunc NewTestPlugin) {
 	testplugins = append(testplugins, newFunc)
 }

 type plugin struct {
 	*generator.Generator
 	generator.PluginImports
 	tests []TestPlugin
 }

 func NewPlugin() *plugin {
 	return &plugin{}
 }

 func (p *plugin) Name() string {
 	return "testgen"
 }

 func (p *plugin) Init(g *generator.Generator) {
 	p.Generator = g
 	p.tests = make([]TestPlugin, 0, len(testplugins))
 	for i := range testplugins {
 		p.tests = append(p.tests, testplugins[i](g))
 	}
 }

 func (p *plugin) Generate(file *generator.FileDescriptor) {
 	p.PluginImports = generator.NewPluginImports(p.Generator)
 	atLeastOne := false
 	for i := range p.tests {
 		used := p.tests[i].Generate(p.PluginImports, file)
 		if used {
 			atLeastOne = true
 		}
 	}
 	if atLeastOne {
 		p.P(`//These tests are generated by github.com/gogo/protobuf/plugin/testgen`)
 	}
 }

 type testProto struct {
 	*generator.Generator
 }

 func newProto(g *generator.Generator) TestPlugin {
 	return &testProto{g}
 }

 func (p *testProto) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool {
 	used := false
 	testingPkg := imports.NewImport("testing")
 	randPkg := imports.NewImport("math/rand")
 	timePkg := imports.NewImport("time")
 	protoPkg := imports.NewImport("github.com/gogo/protobuf/proto")
 	if !gogoproto.ImportsGoGoProto(file.FileDescriptorProto) {
 		protoPkg = imports.NewImport("github.com/golang/protobuf/proto")
 	}
 	for _, message := range file.Messages() {
 		ccTypeName := generator.CamelCaseSlice(message.TypeName())
 		if message.DescriptorProto.GetOptions().GetMapEntry() {
 			continue
 		}
 		if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) {
 			used = true

 			p.P(`func Test`, ccTypeName, `Proto(t *`, testingPkg.Use(), `.T) {`)
 			p.In()
 			p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
 			p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
 			p.P(`p := NewPopulated`, ccTypeName, `(popr, false)`)
 			p.P(`dAtA, err := `, protoPkg.Use(), `.Marshal(p)`)
 			p.P(`if err != nil {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
 			p.Out()
 			p.P(`}`)
 			p.P(`msg := &`, ccTypeName, `{}`)
 			p.P(`if err := `, protoPkg.Use(), `.Unmarshal(dAtA, msg); err != nil {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
 			p.Out()
 			p.P(`}`)
 			p.P(`littlefuzz := make([]byte, len(dAtA))`)
 			p.P(`copy(littlefuzz, dAtA)`)
 			p.P(`for i := range dAtA {`)
 			p.In()
 			p.P(`dAtA[i] = byte(popr.Intn(256))`)
 			p.Out()
 			p.P(`}`)
 			if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
 				p.P(`if err := p.VerboseEqual(msg); err != nil {`)
 				p.In()
 				p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
 				p.Out()
 				p.P(`}`)
 			}
 			p.P(`if !p.Equal(msg) {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, %#v !Proto %#v", seed, msg, p)`)
 			p.Out()
 			p.P(`}`)
 			p.P(`if len(littlefuzz) > 0 {`)
 			p.In()
 			p.P(`fuzzamount := 100`)
 			p.P(`for i := 0; i < fuzzamount; i++ {`)
 			p.In()
 			p.P(`littlefuzz[popr.Intn(len(littlefuzz))] = byte(popr.Intn(256))`)
 			p.P(`littlefuzz = append(littlefuzz, byte(popr.Intn(256)))`)
 			p.Out()
 			p.P(`}`)
 			p.P(`// shouldn't panic`)
 			p.P(`_ = `, protoPkg.Use(), `.Unmarshal(littlefuzz, msg)`)
 			p.Out()
 			p.P(`}`)
 			p.Out()
 			p.P(`}`)
 			p.P()
 		}

 		if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) {
 			if gogoproto.IsMarshaler(file.FileDescriptorProto, message.DescriptorProto) || gogoproto.IsUnsafeMarshaler(file.FileDescriptorProto, message.DescriptorProto) {
 				p.P(`func Test`, ccTypeName, `MarshalTo(t *`, testingPkg.Use(), `.T) {`)
 				p.In()
 				p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
 				p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
 				p.P(`p := NewPopulated`, ccTypeName, `(popr, false)`)
 				if gogoproto.IsProtoSizer(file.FileDescriptorProto, message.DescriptorProto) {
 					p.P(`size := p.ProtoSize()`)
 				} else {
 					p.P(`size := p.Size()`)
 				}
 				p.P(`dAtA := make([]byte, size)`)
 				p.P(`for i := range dAtA {`)
 				p.In()
 				p.P(`dAtA[i] = byte(popr.Intn(256))`)
 				p.Out()
 				p.P(`}`)
 				p.P(`_, err := p.MarshalTo(dAtA)`)
 				p.P(`if err != nil {`)
 				p.In()
 				p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
 				p.Out()
 				p.P(`}`)
 				p.P(`msg := &`, ccTypeName, `{}`)
 				p.P(`if err := `, protoPkg.Use(), `.Unmarshal(dAtA, msg); err != nil {`)
 				p.In()
 				p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
 				p.Out()
 				p.P(`}`)
 				p.P(`for i := range dAtA {`)
 				p.In()
 				p.P(`dAtA[i] = byte(popr.Intn(256))`)
 				p.Out()
 				p.P(`}`)
 				if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
 					p.P(`if err := p.VerboseEqual(msg); err != nil {`)
 					p.In()
 					p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
 					p.Out()
 					p.P(`}`)
 				}
 				p.P(`if !p.Equal(msg) {`)
 				p.In()
 				p.P(`t.Fatalf("seed = %d, %#v !Proto %#v", seed, msg, p)`)
 				p.Out()
 				p.P(`}`)
 				p.Out()
 				p.P(`}`)
 				p.P()
 			}
 		}

 		if gogoproto.HasBenchGen(file.FileDescriptorProto, message.DescriptorProto) {
 			used = true
 			p.P(`func Benchmark`, ccTypeName, `ProtoMarshal(b *`, testingPkg.Use(), `.B) {`)
 			p.In()
 			p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(616))`)
 			p.P(`total := 0`)
 			p.P(`pops := make([]*`, ccTypeName, `, 10000)`)
 			p.P(`for i := 0; i < 10000; i++ {`)
 			p.In()
 			p.P(`pops[i] = NewPopulated`, ccTypeName, `(popr, false)`)
 			p.Out()
 			p.P(`}`)
 			p.P(`b.ResetTimer()`)
 			p.P(`for i := 0; i < b.N; i++ {`)
 			p.In()
 			p.P(`dAtA, err := `, protoPkg.Use(), `.Marshal(pops[i%10000])`)
 			p.P(`if err != nil {`)
 			p.In()
 			p.P(`panic(err)`)
 			p.Out()
 			p.P(`}`)
 			p.P(`total += len(dAtA)`)
 			p.Out()
 			p.P(`}`)
 			p.P(`b.SetBytes(int64(total / b.N))`)
 			p.Out()
 			p.P(`}`)
 			p.P()

 			p.P(`func Benchmark`, ccTypeName, `ProtoUnmarshal(b *`, testingPkg.Use(), `.B) {`)
 			p.In()
 			p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(616))`)
 			p.P(`total := 0`)
 			p.P(`datas := make([][]byte, 10000)`)
 			p.P(`for i := 0; i < 10000; i++ {`)
 			p.In()
 			p.P(`dAtA, err := `, protoPkg.Use(), `.Marshal(NewPopulated`, ccTypeName, `(popr, false))`)
 			p.P(`if err != nil {`)
 			p.In()
 			p.P(`panic(err)`)
 			p.Out()
 			p.P(`}`)
 			p.P(`datas[i] = dAtA`)
 			p.Out()
 			p.P(`}`)
 			p.P(`msg := &`, ccTypeName, `{}`)
 			p.P(`b.ResetTimer()`)
 			p.P(`for i := 0; i < b.N; i++ {`)
 			p.In()
 			p.P(`total += len(datas[i%10000])`)
 			p.P(`if err := `, protoPkg.Use(), `.Unmarshal(datas[i%10000], msg); err != nil {`)
 			p.In()
 			p.P(`panic(err)`)
 			p.Out()
 			p.P(`}`)
 			p.Out()
 			p.P(`}`)
 			p.P(`b.SetBytes(int64(total / b.N))`)
 			p.Out()
 			p.P(`}`)
 			p.P()
 		}
 	}
 	return used
 }

 type testJson struct {
 	*generator.Generator
 }

 func newJson(g *generator.Generator) TestPlugin {
 	return &testJson{g}
 }

 func (p *testJson) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool {
 	used := false
 	testingPkg := imports.NewImport("testing")
 	randPkg := imports.NewImport("math/rand")
 	timePkg := imports.NewImport("time")
 	jsonPkg := imports.NewImport("github.com/gogo/protobuf/jsonpb")
 	for _, message := range file.Messages() {
 		ccTypeName := generator.CamelCaseSlice(message.TypeName())
 		if message.DescriptorProto.GetOptions().GetMapEntry() {
 			continue
 		}
 		if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) {
 			used = true
 			p.P(`func Test`, ccTypeName, `JSON(t *`, testingPkg.Use(), `.T) {`)
 			p.In()
 			p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
 			p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
 			p.P(`p := NewPopulated`, ccTypeName, `(popr, true)`)
 			p.P(`marshaler := `, jsonPkg.Use(), `.Marshaler{}`)
 			p.P(`jsondata, err := marshaler.MarshalToString(p)`)
 			p.P(`if err != nil {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
 			p.Out()
 			p.P(`}`)
 			p.P(`msg := &`, ccTypeName, `{}`)
 			p.P(`err = `, jsonPkg.Use(), `.UnmarshalString(jsondata, msg)`)
 			p.P(`if err != nil {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
 			p.Out()
 			p.P(`}`)
 			if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
 				p.P(`if err := p.VerboseEqual(msg); err != nil {`)
 				p.In()
 				p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
 				p.Out()
 				p.P(`}`)
 			}
 			p.P(`if !p.Equal(msg) {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, %#v !Json Equal %#v", seed, msg, p)`)
 			p.Out()
 			p.P(`}`)
 			p.Out()
 			p.P(`}`)
 		}
 	}
 	return used
 }

 type testText struct {
 	*generator.Generator
 }

 func newText(g *generator.Generator) TestPlugin {
 	return &testText{g}
 }

 func (p *testText) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool {
 	used := false
 	testingPkg := imports.NewImport("testing")
 	randPkg := imports.NewImport("math/rand")
 	timePkg := imports.NewImport("time")
 	protoPkg := imports.NewImport("github.com/gogo/protobuf/proto")
 	if !gogoproto.ImportsGoGoProto(file.FileDescriptorProto) {
 		protoPkg = imports.NewImport("github.com/golang/protobuf/proto")
 	}
 	//fmtPkg := imports.NewImport("fmt")
 	for _, message := range file.Messages() {
 		ccTypeName := generator.CamelCaseSlice(message.TypeName())
 		if message.DescriptorProto.GetOptions().GetMapEntry() {
 			continue
 		}
 		if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) {
 			used = true

 			p.P(`func Test`, ccTypeName, `ProtoText(t *`, testingPkg.Use(), `.T) {`)
 			p.In()
 			p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
 			p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
 			p.P(`p := NewPopulated`, ccTypeName, `(popr, true)`)
 			p.P(`dAtA := `, protoPkg.Use(), `.MarshalTextString(p)`)
 			p.P(`msg := &`, ccTypeName, `{}`)
 			p.P(`if err := `, protoPkg.Use(), `.UnmarshalText(dAtA, msg); err != nil {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
 			p.Out()
 			p.P(`}`)
 			if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
 				p.P(`if err := p.VerboseEqual(msg); err != nil {`)
 				p.In()
 				p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
 				p.Out()
 				p.P(`}`)
 			}
 			p.P(`if !p.Equal(msg) {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, %#v !Proto %#v", seed, msg, p)`)
 			p.Out()
 			p.P(`}`)
 			p.Out()
 			p.P(`}`)
 			p.P()

 			p.P(`func Test`, ccTypeName, `ProtoCompactText(t *`, testingPkg.Use(), `.T) {`)
 			p.In()
 			p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
 			p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
 			p.P(`p := NewPopulated`, ccTypeName, `(popr, true)`)
 			p.P(`dAtA := `, protoPkg.Use(), `.CompactTextString(p)`)
 			p.P(`msg := &`, ccTypeName, `{}`)
 			p.P(`if err := `, protoPkg.Use(), `.UnmarshalText(dAtA, msg); err != nil {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
 			p.Out()
 			p.P(`}`)
 			if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
 				p.P(`if err := p.VerboseEqual(msg); err != nil {`)
 				p.In()
 				p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
 				p.Out()
 				p.P(`}`)
 			}
 			p.P(`if !p.Equal(msg) {`)
 			p.In()
 			p.P(`t.Fatalf("seed = %d, %#v !Proto %#v", seed, msg, p)`)
 			p.Out()
 			p.P(`}`)
 			p.Out()
 			p.P(`}`)
 			p.P()

 		}
 	}
 	return used
 }

 func init() {
 	RegisterTestPlugin(newProto)
 	RegisterTestPlugin(newJson)
 	RegisterTestPlugin(newText)
 }
	// Protocol Buffers for Go with Gadgets
	//
	// Copyright (c) 2013, The GoGo Authors. All rights reserved.
	// http://github.com/gogo/protobuf
	//
	// Redistribution and use in source and binary forms, with or without
	// modification, are permitted provided that the following conditions are
	// met:
	//
	// * Redistributions of source code must retain the above copyright
	// notice, this list of conditions and the following disclaimer.
	// * Redistributions in binary form must reproduce the above
	// copyright notice, this list of conditions and the following disclaimer
	// in the documentation and/or other materials provided with the
	// distribution.
	//
	// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
	// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
	// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
	// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
	// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
	// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
	// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
	// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
	// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
	// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
	// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

	/*
	The testgen plugin generates Test and Benchmark functions for each message.

	Tests are enabled using the following extensions:

	- testgen
	- testgen_all

	Benchmarks are enabled using the following extensions:

	- benchgen
	- benchgen_all

	Let us look at:

	github.com/gogo/protobuf/test/example/example.proto

	Btw all the output can be seen at:

	github.com/gogo/protobuf/test/example/*

	The following message:

	option (gogoproto.testgen_all) = true;
	option (gogoproto.benchgen_all) = true;

	message A {
	optional string Description = 1 [(gogoproto.nullable) = false];
	optional int64 Number = 2 [(gogoproto.nullable) = false];
	optional bytes Id = 3 [(gogoproto.customtype) = "github.com/gogo/protobuf/test/custom.Uuid", (gogoproto.nullable) = false];
	}

	given to the testgen plugin, will generate the following test code:

	func TestAProto(t *testing.T) {
	popr := math_rand.New(math_rand.NewSource(time.Now().UnixNano()))
	p := NewPopulatedA(popr, false)
	dAtA, err := github_com_gogo_protobuf_proto.Marshal(p)
	if err != nil {
	panic(err)
	}
	msg := &A{}
	if err := github_com_gogo_protobuf_proto.Unmarshal(dAtA, msg); err != nil {
	panic(err)
	}
	for i := range dAtA {
	dAtA[i] = byte(popr.Intn(256))
	}
	if err := p.VerboseEqual(msg); err != nil {
	t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)
	}
	if !p.Equal(msg) {
	t.Fatalf("%#v !Proto %#v", msg, p)
	}
	}

	func BenchmarkAProtoMarshal(b *testing.B) {
	popr := math_rand.New(math_rand.NewSource(616))
	total := 0
	pops := make([]*A, 10000)
	for i := 0; i < 10000; i++ {
	pops[i] = NewPopulatedA(popr, false)
	}
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
	dAtA, err := github_com_gogo_protobuf_proto.Marshal(pops[i%10000])
	if err != nil {
	panic(err)
	}
	total += len(dAtA)
	}
	b.SetBytes(int64(total / b.N))
	}

	func BenchmarkAProtoUnmarshal(b *testing.B) {
	popr := math_rand.New(math_rand.NewSource(616))
	total := 0
	datas := make([][]byte, 10000)
	for i := 0; i < 10000; i++ {
	dAtA, err := github_com_gogo_protobuf_proto.Marshal(NewPopulatedA(popr, false))
	if err != nil {
	panic(err)
	}
	datas[i] = dAtA
	}
	msg := &A{}
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
	total += len(datas[i%10000])
	if err := github_com_gogo_protobuf_proto.Unmarshal(datas[i%10000], msg); err != nil {
	panic(err)
	}
	}
	b.SetBytes(int64(total / b.N))
	}


	func TestAJSON(t *testing1.T) {
	popr := math_rand1.New(math_rand1.NewSource(time1.Now().UnixNano()))
	p := NewPopulatedA(popr, true)
	jsondata, err := encoding_json.Marshal(p)
	if err != nil {
	panic(err)
	}
	msg := &A{}
	err = encoding_json.Unmarshal(jsondata, msg)
	if err != nil {
	panic(err)
	}
	if err := p.VerboseEqual(msg); err != nil {
	t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)
	}
	if !p.Equal(msg) {
	t.Fatalf("%#v !Json Equal %#v", msg, p)
	}
	}

	func TestAProtoText(t *testing2.T) {
	popr := math_rand2.New(math_rand2.NewSource(time2.Now().UnixNano()))
	p := NewPopulatedA(popr, true)
	dAtA := github_com_gogo_protobuf_proto1.MarshalTextString(p)
	msg := &A{}
	if err := github_com_gogo_protobuf_proto1.UnmarshalText(dAtA, msg); err != nil {
	panic(err)
	}
	if err := p.VerboseEqual(msg); err != nil {
	t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)
	}
	if !p.Equal(msg) {
	t.Fatalf("%#v !Proto %#v", msg, p)
	}
	}

	func TestAProtoCompactText(t *testing2.T) {
	popr := math_rand2.New(math_rand2.NewSource(time2.Now().UnixNano()))
	p := NewPopulatedA(popr, true)
	dAtA := github_com_gogo_protobuf_proto1.CompactTextString(p)
	msg := &A{}
	if err := github_com_gogo_protobuf_proto1.UnmarshalText(dAtA, msg); err != nil {
	panic(err)
	}
	if err := p.VerboseEqual(msg); err != nil {
	t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)
	}
	if !p.Equal(msg) {
	t.Fatalf("%#v !Proto %#v", msg, p)
	}
	}

	Other registered tests are also generated.
	Tests are registered to this test plugin by calling the following function.

	func RegisterTestPlugin(newFunc NewTestPlugin)

	where NewTestPlugin is:

	type NewTestPlugin func(g *generator.Generator) TestPlugin

	and TestPlugin is an interface:

	type TestPlugin interface {
	Generate(imports generator.PluginImports, file *generator.FileDescriptor) (used bool)
	}

	Plugins that use this interface include:

	- populate
	- gostring
	- equal
	- union
	- and more

	Please look at these plugins as examples of how to create your own.
	A good idea is to let each plugin generate its own tests.

	*/
	package testgen

	import (
	"github.com/gogo/protobuf/gogoproto"
	"github.com/gogo/protobuf/protoc-gen-gogo/generator"
	)

	type TestPlugin interface {
	Generate(imports generator.PluginImports, file *generator.FileDescriptor) (used bool)
	}

	type NewTestPlugin func(g *generator.Generator) TestPlugin

	var testplugins = make([]NewTestPlugin, 0)

	func RegisterTestPlugin(newFunc NewTestPlugin) {
	testplugins = append(testplugins, newFunc)
	}

	type plugin struct {
	*generator.Generator
	generator.PluginImports
	tests []TestPlugin
	}

	func NewPlugin() *plugin {
	return &plugin{}
	}

	func (p *plugin) Name() string {
	return "testgen"
	}

	func (p plugin) Init(g generator.Generator) {
	p.Generator = g
	p.tests = make([]TestPlugin, 0, len(testplugins))
	for i := range testplugins {
	p.tests = append(p.tests, testplugins[i](g))
	}
	}

	func (p plugin) Generate(file generator.FileDescriptor) {
	p.PluginImports = generator.NewPluginImports(p.Generator)
	atLeastOne := false
	for i := range p.tests {
	used := p.tests[i].Generate(p.PluginImports, file)
	if used {
	atLeastOne = true
	}
	}
	if atLeastOne {
	p.P(`//These tests are generated by github.com/gogo/protobuf/plugin/testgen`)
	}
	}

	type testProto struct {
	*generator.Generator
	}

	func newProto(g *generator.Generator) TestPlugin {
	return &testProto{g}
	}

	func (p testProto) Generate(imports generator.PluginImports, file generator.FileDescriptor) bool {
	used := false
	testingPkg := imports.NewImport("testing")
	randPkg := imports.NewImport("math/rand")
	timePkg := imports.NewImport("time")
	protoPkg := imports.NewImport("github.com/gogo/protobuf/proto")
	if !gogoproto.ImportsGoGoProto(file.FileDescriptorProto) {
	protoPkg = imports.NewImport("github.com/golang/protobuf/proto")
	}
	for _, message := range file.Messages() {
	ccTypeName := generator.CamelCaseSlice(message.TypeName())
	if message.DescriptorProto.GetOptions().GetMapEntry() {
	continue
	}
	if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) {
	used = true

	p.P(`func Test`, ccTypeName, `Proto(t *`, testingPkg.Use(), `.T) {`)
	p.In()
	p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
	p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
	p.P(`p := NewPopulated`, ccTypeName, `(popr, false)`)
	p.P(`dAtA, err := `, protoPkg.Use(), `.Marshal(p)`)
	p.P(`if err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
	p.Out()
	p.P(`}`)
	p.P(`msg := &`, ccTypeName, `{}`)
	p.P(`if err := `, protoPkg.Use(), `.Unmarshal(dAtA, msg); err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
	p.Out()
	p.P(`}`)
	p.P(`littlefuzz := make([]byte, len(dAtA))`)
	p.P(`copy(littlefuzz, dAtA)`)
	p.P(`for i := range dAtA {`)
	p.In()
	p.P(`dAtA[i] = byte(popr.Intn(256))`)
	p.Out()
	p.P(`}`)
	if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
	p.P(`if err := p.VerboseEqual(msg); err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
	p.Out()
	p.P(`}`)
	}
	p.P(`if !p.Equal(msg) {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !Proto %#v", seed, msg, p)`)
	p.Out()
	p.P(`}`)
	p.P(`if len(littlefuzz) > 0 {`)
	p.In()
	p.P(`fuzzamount := 100`)
	p.P(`for i := 0; i < fuzzamount; i++ {`)
	p.In()
	p.P(`littlefuzz[popr.Intn(len(littlefuzz))] = byte(popr.Intn(256))`)
	p.P(`littlefuzz = append(littlefuzz, byte(popr.Intn(256)))`)
	p.Out()
	p.P(`}`)
	p.P(`// shouldn't panic`)
	p.P(`_ = `, protoPkg.Use(), `.Unmarshal(littlefuzz, msg)`)
	p.Out()
	p.P(`}`)
	p.Out()
	p.P(`}`)
	p.P()
	}

	if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) {
	if gogoproto.IsMarshaler(file.FileDescriptorProto, message.DescriptorProto) \|\| gogoproto.IsUnsafeMarshaler(file.FileDescriptorProto, message.DescriptorProto) {
	p.P(`func Test`, ccTypeName, `MarshalTo(t *`, testingPkg.Use(), `.T) {`)
	p.In()
	p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
	p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
	p.P(`p := NewPopulated`, ccTypeName, `(popr, false)`)
	if gogoproto.IsProtoSizer(file.FileDescriptorProto, message.DescriptorProto) {
	p.P(`size := p.ProtoSize()`)
	} else {
	p.P(`size := p.Size()`)
	}
	p.P(`dAtA := make([]byte, size)`)
	p.P(`for i := range dAtA {`)
	p.In()
	p.P(`dAtA[i] = byte(popr.Intn(256))`)
	p.Out()
	p.P(`}`)
	p.P(`_, err := p.MarshalTo(dAtA)`)
	p.P(`if err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
	p.Out()
	p.P(`}`)
	p.P(`msg := &`, ccTypeName, `{}`)
	p.P(`if err := `, protoPkg.Use(), `.Unmarshal(dAtA, msg); err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
	p.Out()
	p.P(`}`)
	p.P(`for i := range dAtA {`)
	p.In()
	p.P(`dAtA[i] = byte(popr.Intn(256))`)
	p.Out()
	p.P(`}`)
	if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
	p.P(`if err := p.VerboseEqual(msg); err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
	p.Out()
	p.P(`}`)
	}
	p.P(`if !p.Equal(msg) {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !Proto %#v", seed, msg, p)`)
	p.Out()
	p.P(`}`)
	p.Out()
	p.P(`}`)
	p.P()
	}
	}

	if gogoproto.HasBenchGen(file.FileDescriptorProto, message.DescriptorProto) {
	used = true
	p.P(`func Benchmark`, ccTypeName, `ProtoMarshal(b *`, testingPkg.Use(), `.B) {`)
	p.In()
	p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(616))`)
	p.P(`total := 0`)
	p.P(`pops := make([]*`, ccTypeName, `, 10000)`)
	p.P(`for i := 0; i < 10000; i++ {`)
	p.In()
	p.P(`pops[i] = NewPopulated`, ccTypeName, `(popr, false)`)
	p.Out()
	p.P(`}`)
	p.P(`b.ResetTimer()`)
	p.P(`for i := 0; i < b.N; i++ {`)
	p.In()
	p.P(`dAtA, err := `, protoPkg.Use(), `.Marshal(pops[i%10000])`)
	p.P(`if err != nil {`)
	p.In()
	p.P(`panic(err)`)
	p.Out()
	p.P(`}`)
	p.P(`total += len(dAtA)`)
	p.Out()
	p.P(`}`)
	p.P(`b.SetBytes(int64(total / b.N))`)
	p.Out()
	p.P(`}`)
	p.P()

	p.P(`func Benchmark`, ccTypeName, `ProtoUnmarshal(b *`, testingPkg.Use(), `.B) {`)
	p.In()
	p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(616))`)
	p.P(`total := 0`)
	p.P(`datas := make([][]byte, 10000)`)
	p.P(`for i := 0; i < 10000; i++ {`)
	p.In()
	p.P(`dAtA, err := `, protoPkg.Use(), `.Marshal(NewPopulated`, ccTypeName, `(popr, false))`)
	p.P(`if err != nil {`)
	p.In()
	p.P(`panic(err)`)
	p.Out()
	p.P(`}`)
	p.P(`datas[i] = dAtA`)
	p.Out()
	p.P(`}`)
	p.P(`msg := &`, ccTypeName, `{}`)
	p.P(`b.ResetTimer()`)
	p.P(`for i := 0; i < b.N; i++ {`)
	p.In()
	p.P(`total += len(datas[i%10000])`)
	p.P(`if err := `, protoPkg.Use(), `.Unmarshal(datas[i%10000], msg); err != nil {`)
	p.In()
	p.P(`panic(err)`)
	p.Out()
	p.P(`}`)
	p.Out()
	p.P(`}`)
	p.P(`b.SetBytes(int64(total / b.N))`)
	p.Out()
	p.P(`}`)
	p.P()
	}
	}
	return used
	}

	type testJson struct {
	*generator.Generator
	}

	func newJson(g *generator.Generator) TestPlugin {
	return &testJson{g}
	}

	func (p testJson) Generate(imports generator.PluginImports, file generator.FileDescriptor) bool {
	used := false
	testingPkg := imports.NewImport("testing")
	randPkg := imports.NewImport("math/rand")
	timePkg := imports.NewImport("time")
	jsonPkg := imports.NewImport("github.com/gogo/protobuf/jsonpb")
	for _, message := range file.Messages() {
	ccTypeName := generator.CamelCaseSlice(message.TypeName())
	if message.DescriptorProto.GetOptions().GetMapEntry() {
	continue
	}
	if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) {
	used = true
	p.P(`func Test`, ccTypeName, `JSON(t *`, testingPkg.Use(), `.T) {`)
	p.In()
	p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
	p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
	p.P(`p := NewPopulated`, ccTypeName, `(popr, true)`)
	p.P(`marshaler := `, jsonPkg.Use(), `.Marshaler{}`)
	p.P(`jsondata, err := marshaler.MarshalToString(p)`)
	p.P(`if err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
	p.Out()
	p.P(`}`)
	p.P(`msg := &`, ccTypeName, `{}`)
	p.P(`err = `, jsonPkg.Use(), `.UnmarshalString(jsondata, msg)`)
	p.P(`if err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
	p.Out()
	p.P(`}`)
	if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
	p.P(`if err := p.VerboseEqual(msg); err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
	p.Out()
	p.P(`}`)
	}
	p.P(`if !p.Equal(msg) {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !Json Equal %#v", seed, msg, p)`)
	p.Out()
	p.P(`}`)
	p.Out()
	p.P(`}`)
	}
	}
	return used
	}

	type testText struct {
	*generator.Generator
	}

	func newText(g *generator.Generator) TestPlugin {
	return &testText{g}
	}

	func (p testText) Generate(imports generator.PluginImports, file generator.FileDescriptor) bool {
	used := false
	testingPkg := imports.NewImport("testing")
	randPkg := imports.NewImport("math/rand")
	timePkg := imports.NewImport("time")
	protoPkg := imports.NewImport("github.com/gogo/protobuf/proto")
	if !gogoproto.ImportsGoGoProto(file.FileDescriptorProto) {
	protoPkg = imports.NewImport("github.com/golang/protobuf/proto")
	}
	//fmtPkg := imports.NewImport("fmt")
	for _, message := range file.Messages() {
	ccTypeName := generator.CamelCaseSlice(message.TypeName())
	if message.DescriptorProto.GetOptions().GetMapEntry() {
	continue
	}
	if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) {
	used = true

	p.P(`func Test`, ccTypeName, `ProtoText(t *`, testingPkg.Use(), `.T) {`)
	p.In()
	p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
	p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
	p.P(`p := NewPopulated`, ccTypeName, `(popr, true)`)
	p.P(`dAtA := `, protoPkg.Use(), `.MarshalTextString(p)`)
	p.P(`msg := &`, ccTypeName, `{}`)
	p.P(`if err := `, protoPkg.Use(), `.UnmarshalText(dAtA, msg); err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
	p.Out()
	p.P(`}`)
	if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
	p.P(`if err := p.VerboseEqual(msg); err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
	p.Out()
	p.P(`}`)
	}
	p.P(`if !p.Equal(msg) {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !Proto %#v", seed, msg, p)`)
	p.Out()
	p.P(`}`)
	p.Out()
	p.P(`}`)
	p.P()

	p.P(`func Test`, ccTypeName, `ProtoCompactText(t *`, testingPkg.Use(), `.T) {`)
	p.In()
	p.P(`seed := `, timePkg.Use(), `.Now().UnixNano()`)
	p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(seed))`)
	p.P(`p := NewPopulated`, ccTypeName, `(popr, true)`)
	p.P(`dAtA := `, protoPkg.Use(), `.CompactTextString(p)`)
	p.P(`msg := &`, ccTypeName, `{}`)
	p.P(`if err := `, protoPkg.Use(), `.UnmarshalText(dAtA, msg); err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, err = %v", seed, err)`)
	p.Out()
	p.P(`}`)
	if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) {
	p.P(`if err := p.VerboseEqual(msg); err != nil {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)`)
	p.Out()
	p.P(`}`)
	}
	p.P(`if !p.Equal(msg) {`)
	p.In()
	p.P(`t.Fatalf("seed = %d, %#v !Proto %#v", seed, msg, p)`)
	p.Out()
	p.P(`}`)
	p.Out()
	p.P(`}`)
	p.P()

	}
	}
	return used
	}

	func init() {
	RegisterTestPlugin(newProto)
	RegisterTestPlugin(newJson)
	RegisterTestPlugin(newText)
	}