func (p *test) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool { used := false randPkg := imports.NewImport("math/rand") timePkg := imports.NewImport("time") testingPkg := imports.NewImport("testing") for _, message := range file.Messages() { if !gogoproto.IsUnion(file.FileDescriptorProto, message.DescriptorProto) || !gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) { continue } if message.DescriptorProto.GetOptions().GetMapEntry() { continue } used = true ccTypeName := generator.CamelCaseSlice(message.TypeName()) p.P(`func Test`, ccTypeName, `OnlyOne(t *`, testingPkg.Use(), `.T) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(`, timePkg.Use(), `.Now().UnixNano()))`) p.P(`p := NewPopulated`, ccTypeName, `(popr, true)`) p.P(`v := p.GetValue()`) p.P(`msg := &`, ccTypeName, `{}`) p.P(`if !msg.SetValue(v) {`) p.In() p.P(`t.Fatalf("OnlyOne: Could not set Value")`) p.Out() p.P(`}`) p.P(`if !p.Equal(msg) {`) p.In() p.P(`t.Fatalf("%#v !OnlyOne Equal %#v", msg, p)`) p.Out() p.P(`}`) p.Out() p.P(`}`) } return used }
func (p *union) Generate(file *generator.FileDescriptor) { p.PluginImports = generator.NewPluginImports(p.Generator) for _, message := range file.Messages() { if !gogoproto.IsUnion(file.FileDescriptorProto, message.DescriptorProto) { continue } if message.DescriptorProto.HasExtension() { panic("onlyone does not currently support extensions") } if message.DescriptorProto.GetOptions().GetMapEntry() { continue } ccTypeName := generator.CamelCaseSlice(message.TypeName()) p.P(`func (this *`, ccTypeName, `) GetValue() interface{} {`) p.In() for _, field := range message.Field { fieldname := p.GetFieldName(message, field) if fieldname == "Value" { panic("cannot have a onlyone message " + ccTypeName + " with a field named Value") } p.P(`if this.`, fieldname, ` != nil {`) p.In() p.P(`return this.`, fieldname) p.Out() p.P(`}`) } p.P(`return nil`) p.Out() p.P(`}`) p.P(``) p.P(`func (this *`, ccTypeName, `) SetValue(value interface{}) bool {`) p.In() p.P(`switch vt := value.(type) {`) p.In() for _, field := range message.Field { fieldname := p.GetFieldName(message, field) goTyp, _ := p.GoType(message, field) p.P(`case `, goTyp, `:`) p.In() p.P(`this.`, fieldname, ` = vt`) p.Out() } p.P(`default:`) p.In() for _, field := range message.Field { fieldname := p.GetFieldName(message, field) if field.IsMessage() { goTyp, _ := p.GoType(message, field) obj := p.ObjectNamed(field.GetTypeName()).(*generator.Descriptor) if gogoproto.IsUnion(obj.File(), obj.DescriptorProto) { p.P(`this.`, fieldname, ` = new(`, generator.GoTypeToName(goTyp), `)`) p.P(`if set := this.`, fieldname, `.SetValue(value); set {`) p.In() p.P(`return true`) p.Out() p.P(`}`) p.P(`this.`, fieldname, ` = nil`) } } } p.P(`return false`) p.Out() p.P(`}`) p.P(`return true`) p.Out() p.P(`}`) } }