Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 1 | // Copyright 2019 The Go Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style |
| 3 | // license that can be found in the LICENSE file. |
| 4 | |
| 5 | // Package prototest exercises protobuf reflection. |
| 6 | package prototest |
| 7 | |
| 8 | import ( |
| 9 | "bytes" |
| 10 | "fmt" |
| 11 | "math" |
| 12 | "sort" |
| 13 | "testing" |
| 14 | |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 15 | "google.golang.org/protobuf/encoding/prototext" |
| 16 | "google.golang.org/protobuf/internal/encoding/wire" |
Damien Neil | e89e624 | 2019-05-13 23:55:40 -0700 | [diff] [blame] | 17 | "google.golang.org/protobuf/proto" |
| 18 | pref "google.golang.org/protobuf/reflect/protoreflect" |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 19 | preg "google.golang.org/protobuf/reflect/protoregistry" |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 20 | ) |
| 21 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 22 | // TODO: Test read-only properties of unpopulated composite values. |
| 23 | // TODO: Test invalid field descriptors or oneof descriptors. |
| 24 | // TODO: This should test the functionality that can be provided by fast-paths. |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 25 | |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 26 | // MessageOptions configure message tests. |
| 27 | type MessageOptions struct { |
| 28 | // ExtensionTypes is a list of types to test with. |
| 29 | // |
| 30 | // If nil, TestMessage will look for extension types in the global registry. |
| 31 | ExtensionTypes []pref.ExtensionType |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 32 | |
| 33 | // Resolver is used for looking up types when unmarshaling extension fields. |
| 34 | // If nil, this defaults to using protoregistry.GlobalTypes. |
| 35 | Resolver interface { |
| 36 | preg.ExtensionTypeResolver |
| 37 | } |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 38 | } |
| 39 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 40 | // TestMessage runs the provided m through a series of tests |
| 41 | // exercising the protobuf reflection API. |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 42 | func TestMessage(t testing.TB, m proto.Message, opts MessageOptions) { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 43 | md := m.ProtoReflect().Descriptor() |
| 44 | m1 := m.ProtoReflect().New() |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 45 | for i := 0; i < md.Fields().Len(); i++ { |
| 46 | fd := md.Fields().Get(i) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 47 | testField(t, m1, fd) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 48 | } |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 49 | if opts.ExtensionTypes == nil { |
| 50 | preg.GlobalTypes.RangeExtensionsByMessage(md.FullName(), func(e pref.ExtensionType) bool { |
| 51 | opts.ExtensionTypes = append(opts.ExtensionTypes, e) |
| 52 | return true |
| 53 | }) |
| 54 | } |
| 55 | for _, xt := range opts.ExtensionTypes { |
Damien Neil | 79bfdbe | 2019-08-28 11:08:22 -0700 | [diff] [blame] | 56 | testField(t, m1, xt.TypeDescriptor()) |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 57 | } |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 58 | for i := 0; i < md.Oneofs().Len(); i++ { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 59 | testOneof(t, m1, md.Oneofs().Get(i)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 60 | } |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 61 | testUnknown(t, m1) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 62 | |
| 63 | // Test round-trip marshal/unmarshal. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 64 | m2 := m.ProtoReflect().New().Interface() |
| 65 | populateMessage(m2.ProtoReflect(), 1, nil) |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 66 | for _, xt := range opts.ExtensionTypes { |
| 67 | m2.ProtoReflect().Set(xt.TypeDescriptor(), newValue(m2.ProtoReflect(), xt.TypeDescriptor(), 1, nil)) |
| 68 | } |
| 69 | b, err := proto.MarshalOptions{ |
| 70 | AllowPartial: true, |
| 71 | }.Marshal(m2) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 72 | if err != nil { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 73 | t.Errorf("Marshal() = %v, want nil\n%v", err, marshalText(m2)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 74 | } |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 75 | m3 := m.ProtoReflect().New().Interface() |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 76 | if err := (proto.UnmarshalOptions{ |
| 77 | AllowPartial: true, |
| 78 | Resolver: opts.Resolver, |
| 79 | }.Unmarshal(b, m3)); err != nil { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 80 | t.Errorf("Unmarshal() = %v, want nil\n%v", err, marshalText(m2)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 81 | } |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 82 | if !proto.Equal(m2, m3) { |
| 83 | t.Errorf("round-trip marshal/unmarshal did not preserve message\nOriginal:\n%v\nNew:\n%v", marshalText(m2), marshalText(m3)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 84 | } |
| 85 | } |
| 86 | |
| 87 | func marshalText(m proto.Message) string { |
Damien Neil | 5c5b531 | 2019-05-14 12:44:37 -0700 | [diff] [blame] | 88 | b, _ := prototext.MarshalOptions{Indent: " "}.Marshal(m) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 89 | return string(b) |
| 90 | } |
| 91 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 92 | // testField exercises set/get/has/clear of a field. |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 93 | func testField(t testing.TB, m pref.Message, fd pref.FieldDescriptor) { |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 94 | name := fd.FullName() |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 95 | num := fd.Number() |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 96 | |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 97 | switch { |
| 98 | case fd.IsList(): |
| 99 | testFieldList(t, m, fd) |
| 100 | case fd.IsMap(): |
| 101 | testFieldMap(t, m, fd) |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 102 | case fd.Message() != nil: |
| 103 | default: |
| 104 | if got, want := m.NewField(fd), fd.Default(); !valueEqual(got, want) { |
| 105 | t.Errorf("Message.NewField(%v) = %v, want default value %v", name, formatValue(got), formatValue(want)) |
| 106 | } |
| 107 | if fd.Kind() == pref.FloatKind || fd.Kind() == pref.DoubleKind { |
| 108 | testFieldFloat(t, m, fd) |
| 109 | } |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 110 | } |
| 111 | |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 112 | // Set to a non-zero value, the zero value, different non-zero values. |
| 113 | for _, n := range []seed{1, 0, minVal, maxVal} { |
| 114 | v := newValue(m, fd, n, nil) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 115 | m.Set(fd, v) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 116 | wantHas := true |
| 117 | if n == 0 { |
| 118 | if fd.Syntax() == pref.Proto3 && fd.Message() == nil { |
| 119 | wantHas = false |
| 120 | } |
| 121 | if fd.Cardinality() == pref.Repeated { |
| 122 | wantHas = false |
| 123 | } |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 124 | if fd.IsExtension() { |
| 125 | wantHas = true |
| 126 | } |
Joe Tsai | ac31a35 | 2019-05-13 14:32:56 -0700 | [diff] [blame] | 127 | if fd.ContainingOneof() != nil { |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 128 | wantHas = true |
| 129 | } |
| 130 | } |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 131 | if fd.Syntax() == pref.Proto3 && fd.Cardinality() != pref.Repeated && fd.ContainingOneof() == nil && fd.Kind() == pref.EnumKind && v.Enum() == 0 { |
| 132 | wantHas = false |
| 133 | } |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 134 | if got, want := m.Has(fd), wantHas; got != want { |
| 135 | t.Errorf("after setting %q to %v:\nMessage.Has(%v) = %v, want %v", name, formatValue(v), num, got, want) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 136 | } |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 137 | if got, want := m.Get(fd), v; !valueEqual(got, want) { |
| 138 | t.Errorf("after setting %q:\nMessage.Get(%v) = %v, want %v", name, num, formatValue(got), formatValue(want)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 139 | } |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 140 | found := false |
| 141 | m.Range(func(d pref.FieldDescriptor, got pref.Value) bool { |
| 142 | if fd != d { |
| 143 | return true |
| 144 | } |
| 145 | found = true |
| 146 | if want := v; !valueEqual(got, want) { |
| 147 | t.Errorf("after setting %q:\nMessage.Range got value %v, want %v", name, formatValue(got), formatValue(want)) |
| 148 | } |
| 149 | return true |
| 150 | }) |
| 151 | if got, want := wantHas, found; got != want { |
| 152 | t.Errorf("after setting %q:\nMessageRange saw field: %v, want %v", name, got, want) |
| 153 | } |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 154 | } |
| 155 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 156 | m.Clear(fd) |
| 157 | if got, want := m.Has(fd), false; got != want { |
| 158 | t.Errorf("after clearing %q:\nMessage.Has(%v) = %v, want %v", name, num, got, want) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 159 | } |
| 160 | switch { |
Joe Tsai | ac31a35 | 2019-05-13 14:32:56 -0700 | [diff] [blame] | 161 | case fd.IsList(): |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 162 | if got := m.Get(fd); got.List().Len() != 0 { |
| 163 | t.Errorf("after clearing %q:\nMessage.Get(%v) = %v, want empty list", name, num, formatValue(got)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 164 | } |
Joe Tsai | ac31a35 | 2019-05-13 14:32:56 -0700 | [diff] [blame] | 165 | case fd.IsMap(): |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 166 | if got := m.Get(fd); got.Map().Len() != 0 { |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 167 | t.Errorf("after clearing %q:\nMessage.Get(%v) = %v, want empty map", name, num, formatValue(got)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 168 | } |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 169 | case fd.Message() == nil: |
| 170 | if got, want := m.Get(fd), fd.Default(); !valueEqual(got, want) { |
| 171 | t.Errorf("after clearing %q:\nMessage.Get(%v) = %v, want default %v", name, num, formatValue(got), formatValue(want)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 172 | } |
| 173 | } |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 174 | |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 175 | // Set to the default value. |
| 176 | switch { |
| 177 | case fd.IsList() || fd.IsMap(): |
| 178 | m.Set(fd, m.Get(fd)) |
| 179 | if got, want := m.Has(fd), fd.IsExtension() || fd.ContainingOneof() != nil; got != want { |
| 180 | t.Errorf("after setting %q to default:\nMessage.Has(%v) = %v, want %v", name, num, got, want) |
| 181 | } |
| 182 | case fd.Message() == nil: |
| 183 | m.Set(fd, m.Get(fd)) |
| 184 | if got, want := m.Get(fd), fd.Default(); !valueEqual(got, want) { |
| 185 | t.Errorf("after setting %q to default:\nMessage.Get(%v) = %v, want default %v", name, num, formatValue(got), formatValue(want)) |
| 186 | } |
| 187 | } |
| 188 | m.Clear(fd) |
| 189 | |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 190 | // Set to the wrong type. |
| 191 | v := pref.ValueOf("") |
| 192 | if fd.Kind() == pref.StringKind { |
| 193 | v = pref.ValueOf(int32(0)) |
| 194 | } |
| 195 | if !panics(func() { |
| 196 | m.Set(fd, v) |
| 197 | }) { |
| 198 | t.Errorf("setting %v to %T succeeds, want panic", name, v.Interface()) |
| 199 | } |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 200 | } |
| 201 | |
| 202 | // testFieldMap tests set/get/has/clear of entries in a map field. |
| 203 | func testFieldMap(t testing.TB, m pref.Message, fd pref.FieldDescriptor) { |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 204 | name := fd.FullName() |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 205 | num := fd.Number() |
| 206 | |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 207 | // New values. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 208 | m.Clear(fd) // start with an empty map |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 209 | mapv := m.Get(fd).Map() |
| 210 | if got, want := mapv.NewValue(), newMapValue(fd, mapv, 0, nil); !valueEqual(got, want) { |
| 211 | t.Errorf("message.Get(%v).NewValue() = %v, want %v", name, formatValue(got), formatValue(want)) |
| 212 | } |
| 213 | mapv = m.Mutable(fd).Map() // mutable map |
| 214 | if got, want := mapv.NewValue(), newMapValue(fd, mapv, 0, nil); !valueEqual(got, want) { |
| 215 | t.Errorf("message.Mutable(%v).NewValue() = %v, want %v", name, formatValue(got), formatValue(want)) |
| 216 | } |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 217 | |
| 218 | // Add values. |
| 219 | want := make(testMap) |
| 220 | for i, n := range []seed{1, 0, minVal, maxVal} { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 221 | if got, want := m.Has(fd), i > 0; got != want { |
| 222 | t.Errorf("after inserting %d elements to %q:\nMessage.Has(%v) = %v, want %v", i, name, num, got, want) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 223 | } |
| 224 | |
| 225 | k := newMapKey(fd, n) |
| 226 | v := newMapValue(fd, mapv, n, nil) |
| 227 | mapv.Set(k, v) |
| 228 | want.Set(k, v) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 229 | if got, want := m.Get(fd), pref.ValueOf(want); !valueEqual(got, want) { |
| 230 | t.Errorf("after inserting %d elements to %q:\nMessage.Get(%v) = %v, want %v", i, name, num, formatValue(got), formatValue(want)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 231 | } |
| 232 | } |
| 233 | |
| 234 | // Set values. |
| 235 | want.Range(func(k pref.MapKey, v pref.Value) bool { |
| 236 | nv := newMapValue(fd, mapv, 10, nil) |
| 237 | mapv.Set(k, nv) |
| 238 | want.Set(k, nv) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 239 | if got, want := m.Get(fd), pref.ValueOf(want); !valueEqual(got, want) { |
| 240 | t.Errorf("after setting element %v of %q:\nMessage.Get(%v) = %v, want %v", formatValue(k.Value()), name, num, formatValue(got), formatValue(want)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 241 | } |
| 242 | return true |
| 243 | }) |
| 244 | |
| 245 | // Clear values. |
| 246 | want.Range(func(k pref.MapKey, v pref.Value) bool { |
| 247 | mapv.Clear(k) |
| 248 | want.Clear(k) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 249 | if got, want := m.Has(fd), want.Len() > 0; got != want { |
| 250 | t.Errorf("after clearing elements of %q:\nMessage.Has(%v) = %v, want %v", name, num, got, want) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 251 | } |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 252 | if got, want := m.Get(fd), pref.ValueOf(want); !valueEqual(got, want) { |
| 253 | t.Errorf("after clearing elements of %q:\nMessage.Get(%v) = %v, want %v", name, num, formatValue(got), formatValue(want)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 254 | } |
| 255 | return true |
| 256 | }) |
| 257 | |
| 258 | // Non-existent map keys. |
| 259 | missingKey := newMapKey(fd, 1) |
| 260 | if got, want := mapv.Has(missingKey), false; got != want { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 261 | t.Errorf("non-existent map key in %q: Map.Has(%v) = %v, want %v", name, formatValue(missingKey.Value()), got, want) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 262 | } |
| 263 | if got, want := mapv.Get(missingKey).IsValid(), false; got != want { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 264 | t.Errorf("non-existent map key in %q: Map.Get(%v).IsValid() = %v, want %v", name, formatValue(missingKey.Value()), got, want) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 265 | } |
| 266 | mapv.Clear(missingKey) // noop |
| 267 | } |
| 268 | |
| 269 | type testMap map[interface{}]pref.Value |
| 270 | |
| 271 | func (m testMap) Get(k pref.MapKey) pref.Value { return m[k.Interface()] } |
| 272 | func (m testMap) Set(k pref.MapKey, v pref.Value) { m[k.Interface()] = v } |
| 273 | func (m testMap) Has(k pref.MapKey) bool { return m.Get(k).IsValid() } |
| 274 | func (m testMap) Clear(k pref.MapKey) { delete(m, k.Interface()) } |
| 275 | func (m testMap) Len() int { return len(m) } |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 276 | func (m testMap) NewValue() pref.Value { panic("unimplemented") } |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 277 | func (m testMap) Range(f func(pref.MapKey, pref.Value) bool) { |
| 278 | for k, v := range m { |
| 279 | if !f(pref.ValueOf(k).MapKey(), v) { |
| 280 | return |
| 281 | } |
| 282 | } |
| 283 | } |
| 284 | |
| 285 | // testFieldList exercises set/get/append/truncate of values in a list. |
| 286 | func testFieldList(t testing.TB, m pref.Message, fd pref.FieldDescriptor) { |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 287 | name := fd.FullName() |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 288 | num := fd.Number() |
| 289 | |
| 290 | m.Clear(fd) // start with an empty list |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 291 | list := m.Get(fd).List() |
| 292 | if got, want := list.NewElement(), newListElement(fd, list, 0, nil); !valueEqual(got, want) { |
| 293 | t.Errorf("message.Get(%v).NewElement() = %v, want %v", name, formatValue(got), formatValue(want)) |
| 294 | } |
| 295 | list = m.Mutable(fd).List() // mutable list |
| 296 | if got, want := list.NewElement(), newListElement(fd, list, 0, nil); !valueEqual(got, want) { |
| 297 | t.Errorf("message.Mutable(%v).NewElement() = %v, want %v", name, formatValue(got), formatValue(want)) |
| 298 | } |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 299 | |
| 300 | // Append values. |
| 301 | var want pref.List = &testList{} |
| 302 | for i, n := range []seed{1, 0, minVal, maxVal} { |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 303 | if got, want := m.Has(fd), i > 0 || fd.IsExtension(); got != want { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 304 | t.Errorf("after appending %d elements to %q:\nMessage.Has(%v) = %v, want %v", i, name, num, got, want) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 305 | } |
| 306 | v := newListElement(fd, list, n, nil) |
| 307 | want.Append(v) |
| 308 | list.Append(v) |
| 309 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 310 | if got, want := m.Get(fd), pref.ValueOf(want); !valueEqual(got, want) { |
| 311 | t.Errorf("after appending %d elements to %q:\nMessage.Get(%v) = %v, want %v", i+1, name, num, formatValue(got), formatValue(want)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 312 | } |
| 313 | } |
| 314 | |
| 315 | // Set values. |
| 316 | for i := 0; i < want.Len(); i++ { |
| 317 | v := newListElement(fd, list, seed(i+10), nil) |
| 318 | want.Set(i, v) |
| 319 | list.Set(i, v) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 320 | if got, want := m.Get(fd), pref.ValueOf(want); !valueEqual(got, want) { |
| 321 | t.Errorf("after setting element %d of %q:\nMessage.Get(%v) = %v, want %v", i, name, num, formatValue(got), formatValue(want)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 322 | } |
| 323 | } |
| 324 | |
| 325 | // Truncate. |
| 326 | for want.Len() > 0 { |
| 327 | n := want.Len() - 1 |
| 328 | want.Truncate(n) |
| 329 | list.Truncate(n) |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 330 | if got, want := m.Has(fd), want.Len() > 0 || fd.IsExtension(); got != want { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 331 | t.Errorf("after truncating %q to %d:\nMessage.Has(%v) = %v, want %v", name, n, num, got, want) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 332 | } |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 333 | if got, want := m.Get(fd), pref.ValueOf(want); !valueEqual(got, want) { |
| 334 | t.Errorf("after truncating %q to %d:\nMessage.Get(%v) = %v, want %v", name, n, num, formatValue(got), formatValue(want)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 335 | } |
| 336 | } |
| 337 | } |
| 338 | |
| 339 | type testList struct { |
| 340 | a []pref.Value |
| 341 | } |
| 342 | |
Damien Neil | 2eaed36 | 2019-09-05 10:08:32 -0700 | [diff] [blame] | 343 | func (l *testList) Append(v pref.Value) { l.a = append(l.a, v) } |
| 344 | func (l *testList) Get(n int) pref.Value { return l.a[n] } |
| 345 | func (l *testList) Len() int { return len(l.a) } |
| 346 | func (l *testList) Set(n int, v pref.Value) { l.a[n] = v } |
| 347 | func (l *testList) Truncate(n int) { l.a = l.a[:n] } |
| 348 | func (l *testList) NewElement() pref.Value { panic("unimplemented") } |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 349 | |
| 350 | // testFieldFloat exercises some interesting floating-point scalar field values. |
| 351 | func testFieldFloat(t testing.TB, m pref.Message, fd pref.FieldDescriptor) { |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 352 | name := fd.FullName() |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 353 | num := fd.Number() |
| 354 | |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 355 | for _, v := range []float64{math.Inf(-1), math.Inf(1), math.NaN(), math.Copysign(0, -1)} { |
| 356 | var val pref.Value |
| 357 | if fd.Kind() == pref.FloatKind { |
| 358 | val = pref.ValueOf(float32(v)) |
| 359 | } else { |
| 360 | val = pref.ValueOf(v) |
| 361 | } |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 362 | m.Set(fd, val) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 363 | // Note that Has is true for -0. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 364 | if got, want := m.Has(fd), true; got != want { |
| 365 | t.Errorf("after setting %v to %v: Message.Has(%v) = %v, want %v", name, v, num, got, want) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 366 | } |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 367 | if got, want := m.Get(fd), val; !valueEqual(got, want) { |
| 368 | t.Errorf("after setting %v: Message.Get(%v) = %v, want %v", name, num, formatValue(got), formatValue(want)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 369 | } |
| 370 | } |
| 371 | } |
| 372 | |
| 373 | // testOneof tests the behavior of fields in a oneof. |
| 374 | func testOneof(t testing.TB, m pref.Message, od pref.OneofDescriptor) { |
Damien Neil | a3456c9 | 2019-04-30 11:36:49 -0700 | [diff] [blame] | 375 | for _, mutable := range []bool{false, true} { |
| 376 | for i := 0; i < od.Fields().Len(); i++ { |
| 377 | fda := od.Fields().Get(i) |
| 378 | if mutable { |
| 379 | // Set fields by requesting a mutable reference. |
| 380 | if !fda.IsMap() && !fda.IsList() && fda.Message() == nil { |
| 381 | continue |
| 382 | } |
| 383 | _ = m.Mutable(fda) |
| 384 | } else { |
| 385 | // Set fields explicitly. |
| 386 | m.Set(fda, newValue(m, fda, 1, nil)) |
| 387 | } |
| 388 | if got, want := m.WhichOneof(od), fda; got != want { |
| 389 | t.Errorf("after setting oneof field %q:\nWhichOneof(%q) = %v, want %v", fda.FullName(), fda.Name(), got, want) |
| 390 | } |
| 391 | for j := 0; j < od.Fields().Len(); j++ { |
| 392 | fdb := od.Fields().Get(j) |
| 393 | if got, want := m.Has(fdb), i == j; got != want { |
| 394 | t.Errorf("after setting oneof field %q:\nGet(%q) = %v, want %v", fda.FullName(), fdb.FullName(), got, want) |
| 395 | } |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 396 | } |
| 397 | } |
| 398 | } |
| 399 | } |
| 400 | |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 401 | // testUnknown tests the behavior of unknown fields. |
| 402 | func testUnknown(t testing.TB, m pref.Message) { |
| 403 | var b []byte |
| 404 | b = wire.AppendTag(b, 1000, wire.VarintType) |
| 405 | b = wire.AppendVarint(b, 1001) |
| 406 | m.SetUnknown(pref.RawFields(b)) |
| 407 | if got, want := []byte(m.GetUnknown()), b; !bytes.Equal(got, want) { |
| 408 | t.Errorf("after setting unknown fields:\nGetUnknown() = %v, want %v", got, want) |
| 409 | } |
| 410 | } |
| 411 | |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 412 | func formatValue(v pref.Value) string { |
| 413 | switch v := v.Interface().(type) { |
| 414 | case pref.List: |
| 415 | var buf bytes.Buffer |
| 416 | buf.WriteString("list[") |
| 417 | for i := 0; i < v.Len(); i++ { |
| 418 | if i > 0 { |
| 419 | buf.WriteString(" ") |
| 420 | } |
| 421 | buf.WriteString(formatValue(v.Get(i))) |
| 422 | } |
| 423 | buf.WriteString("]") |
| 424 | return buf.String() |
| 425 | case pref.Map: |
| 426 | var buf bytes.Buffer |
| 427 | buf.WriteString("map[") |
| 428 | var keys []pref.MapKey |
| 429 | v.Range(func(k pref.MapKey, v pref.Value) bool { |
| 430 | keys = append(keys, k) |
| 431 | return true |
| 432 | }) |
| 433 | sort.Slice(keys, func(i, j int) bool { |
| 434 | return keys[i].String() < keys[j].String() |
| 435 | }) |
| 436 | for i, k := range keys { |
| 437 | if i > 0 { |
| 438 | buf.WriteString(" ") |
| 439 | } |
| 440 | buf.WriteString(formatValue(k.Value())) |
| 441 | buf.WriteString(":") |
| 442 | buf.WriteString(formatValue(v.Get(k))) |
| 443 | } |
| 444 | buf.WriteString("]") |
| 445 | return buf.String() |
| 446 | case pref.Message: |
Damien Neil | 5c5b531 | 2019-05-14 12:44:37 -0700 | [diff] [blame] | 447 | b, err := prototext.Marshal(v.Interface()) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 448 | if err != nil { |
| 449 | return fmt.Sprintf("<%v>", err) |
| 450 | } |
Joe Tsai | 0fc49f8 | 2019-05-01 12:29:25 -0700 | [diff] [blame] | 451 | return fmt.Sprintf("%v{%v}", v.Descriptor().FullName(), string(b)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 452 | case string: |
| 453 | return fmt.Sprintf("%q", v) |
| 454 | default: |
| 455 | return fmt.Sprint(v) |
| 456 | } |
| 457 | } |
| 458 | |
| 459 | func valueEqual(a, b pref.Value) bool { |
| 460 | ai, bi := a.Interface(), b.Interface() |
| 461 | switch ai.(type) { |
| 462 | case pref.Message: |
| 463 | return proto.Equal( |
| 464 | a.Message().Interface(), |
| 465 | b.Message().Interface(), |
| 466 | ) |
| 467 | case pref.List: |
| 468 | lista, listb := a.List(), b.List() |
| 469 | if lista.Len() != listb.Len() { |
| 470 | return false |
| 471 | } |
| 472 | for i := 0; i < lista.Len(); i++ { |
| 473 | if !valueEqual(lista.Get(i), listb.Get(i)) { |
| 474 | return false |
| 475 | } |
| 476 | } |
| 477 | return true |
| 478 | case pref.Map: |
| 479 | mapa, mapb := a.Map(), b.Map() |
| 480 | if mapa.Len() != mapb.Len() { |
| 481 | return false |
| 482 | } |
| 483 | equal := true |
| 484 | mapa.Range(func(k pref.MapKey, v pref.Value) bool { |
| 485 | if !valueEqual(v, mapb.Get(k)) { |
| 486 | equal = false |
| 487 | return false |
| 488 | } |
| 489 | return true |
| 490 | }) |
| 491 | return equal |
| 492 | case []byte: |
| 493 | return bytes.Equal(a.Bytes(), b.Bytes()) |
Joe Tsai | fb30a40 | 2019-05-20 15:19:14 -0700 | [diff] [blame] | 494 | case float32: |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 495 | // NaNs are equal, but must be the same NaN. |
Joe Tsai | fb30a40 | 2019-05-20 15:19:14 -0700 | [diff] [blame] | 496 | return math.Float32bits(ai.(float32)) == math.Float32bits(bi.(float32)) |
| 497 | case float64: |
| 498 | // NaNs are equal, but must be the same NaN. |
| 499 | return math.Float64bits(ai.(float64)) == math.Float64bits(bi.(float64)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 500 | default: |
| 501 | return ai == bi |
| 502 | } |
| 503 | } |
| 504 | |
| 505 | // A seed is used to vary the content of a value. |
| 506 | // |
| 507 | // A seed of 0 is the zero value. Messages do not have a zero-value; a 0-seeded messages |
| 508 | // is unpopulated. |
| 509 | // |
| 510 | // A seed of minVal or maxVal is the least or greatest value of the value type. |
| 511 | type seed int |
| 512 | |
| 513 | const ( |
| 514 | minVal seed = -1 |
| 515 | maxVal seed = -2 |
| 516 | ) |
| 517 | |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 518 | // newSeed creates new seed values from a base, for example to create seeds for the |
| 519 | // elements in a list. If the input seed is minVal or maxVal, so is the output. |
| 520 | func newSeed(n seed, adjust ...int) seed { |
| 521 | switch n { |
| 522 | case minVal, maxVal: |
| 523 | return n |
| 524 | } |
| 525 | for _, a := range adjust { |
| 526 | n = 10*n + seed(a) |
| 527 | } |
| 528 | return n |
| 529 | } |
| 530 | |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 531 | // newValue returns a new value assignable to a field. |
| 532 | // |
| 533 | // The stack parameter is used to avoid infinite recursion when populating circular |
| 534 | // data structures. |
Joe Tsai | 0fc49f8 | 2019-05-01 12:29:25 -0700 | [diff] [blame] | 535 | func newValue(m pref.Message, fd pref.FieldDescriptor, n seed, stack []pref.MessageDescriptor) pref.Value { |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 536 | switch { |
Joe Tsai | ac31a35 | 2019-05-13 14:32:56 -0700 | [diff] [blame] | 537 | case fd.IsList(): |
Joe Tsai | ac31a35 | 2019-05-13 14:32:56 -0700 | [diff] [blame] | 538 | if n == 0 { |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 539 | return m.New().Get(fd) |
Joe Tsai | ac31a35 | 2019-05-13 14:32:56 -0700 | [diff] [blame] | 540 | } |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 541 | list := m.NewField(fd).List() |
Joe Tsai | ac31a35 | 2019-05-13 14:32:56 -0700 | [diff] [blame] | 542 | list.Append(newListElement(fd, list, 0, stack)) |
| 543 | list.Append(newListElement(fd, list, minVal, stack)) |
| 544 | list.Append(newListElement(fd, list, maxVal, stack)) |
| 545 | list.Append(newListElement(fd, list, n, stack)) |
| 546 | return pref.ValueOf(list) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 547 | case fd.IsMap(): |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 548 | if n == 0 { |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 549 | return m.New().Get(fd) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 550 | } |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 551 | mapv := m.NewField(fd).Map() |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 552 | mapv.Set(newMapKey(fd, 0), newMapValue(fd, mapv, 0, stack)) |
| 553 | mapv.Set(newMapKey(fd, minVal), newMapValue(fd, mapv, minVal, stack)) |
| 554 | mapv.Set(newMapKey(fd, maxVal), newMapValue(fd, mapv, maxVal, stack)) |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 555 | mapv.Set(newMapKey(fd, n), newMapValue(fd, mapv, newSeed(n, 0), stack)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 556 | return pref.ValueOf(mapv) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 557 | case fd.Message() != nil: |
Damien Neil | 290ceea | 2019-07-15 13:39:43 -0700 | [diff] [blame] | 558 | //if n == 0 { |
| 559 | // return m.New().Get(fd) |
| 560 | //} |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 561 | return populateMessage(m.NewField(fd).Message(), n, stack) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 562 | default: |
| 563 | return newScalarValue(fd, n) |
| 564 | } |
| 565 | } |
| 566 | |
Joe Tsai | 0fc49f8 | 2019-05-01 12:29:25 -0700 | [diff] [blame] | 567 | func newListElement(fd pref.FieldDescriptor, list pref.List, n seed, stack []pref.MessageDescriptor) pref.Value { |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 568 | if fd.Message() == nil { |
| 569 | return newScalarValue(fd, n) |
| 570 | } |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 571 | return populateMessage(list.NewElement().Message(), n, stack) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 572 | } |
| 573 | |
| 574 | func newMapKey(fd pref.FieldDescriptor, n seed) pref.MapKey { |
Joe Tsai | ac31a35 | 2019-05-13 14:32:56 -0700 | [diff] [blame] | 575 | kd := fd.MapKey() |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 576 | return newScalarValue(kd, n).MapKey() |
| 577 | } |
| 578 | |
Joe Tsai | 0fc49f8 | 2019-05-01 12:29:25 -0700 | [diff] [blame] | 579 | func newMapValue(fd pref.FieldDescriptor, mapv pref.Map, n seed, stack []pref.MessageDescriptor) pref.Value { |
Joe Tsai | ac31a35 | 2019-05-13 14:32:56 -0700 | [diff] [blame] | 580 | vd := fd.MapValue() |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 581 | if vd.Message() == nil { |
| 582 | return newScalarValue(vd, n) |
| 583 | } |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 584 | return populateMessage(mapv.NewValue().Message(), n, stack) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 585 | } |
| 586 | |
| 587 | func newScalarValue(fd pref.FieldDescriptor, n seed) pref.Value { |
| 588 | switch fd.Kind() { |
| 589 | case pref.BoolKind: |
| 590 | return pref.ValueOf(n != 0) |
| 591 | case pref.EnumKind: |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 592 | vals := fd.Enum().Values() |
| 593 | var i int |
| 594 | switch n { |
| 595 | case minVal: |
| 596 | i = 0 |
| 597 | case maxVal: |
| 598 | i = vals.Len() - 1 |
| 599 | default: |
| 600 | i = int(n) % vals.Len() |
| 601 | } |
| 602 | return pref.ValueOf(vals.Get(i).Number()) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 603 | case pref.Int32Kind, pref.Sint32Kind, pref.Sfixed32Kind: |
| 604 | switch n { |
| 605 | case minVal: |
| 606 | return pref.ValueOf(int32(math.MinInt32)) |
| 607 | case maxVal: |
| 608 | return pref.ValueOf(int32(math.MaxInt32)) |
| 609 | default: |
| 610 | return pref.ValueOf(int32(n)) |
| 611 | } |
| 612 | case pref.Uint32Kind, pref.Fixed32Kind: |
| 613 | switch n { |
| 614 | case minVal: |
| 615 | // Only use 0 for the zero value. |
| 616 | return pref.ValueOf(uint32(1)) |
| 617 | case maxVal: |
| 618 | return pref.ValueOf(uint32(math.MaxInt32)) |
| 619 | default: |
| 620 | return pref.ValueOf(uint32(n)) |
| 621 | } |
| 622 | case pref.Int64Kind, pref.Sint64Kind, pref.Sfixed64Kind: |
| 623 | switch n { |
| 624 | case minVal: |
| 625 | return pref.ValueOf(int64(math.MinInt64)) |
| 626 | case maxVal: |
| 627 | return pref.ValueOf(int64(math.MaxInt64)) |
| 628 | default: |
| 629 | return pref.ValueOf(int64(n)) |
| 630 | } |
| 631 | case pref.Uint64Kind, pref.Fixed64Kind: |
| 632 | switch n { |
| 633 | case minVal: |
| 634 | // Only use 0 for the zero value. |
| 635 | return pref.ValueOf(uint64(1)) |
| 636 | case maxVal: |
| 637 | return pref.ValueOf(uint64(math.MaxInt64)) |
| 638 | default: |
| 639 | return pref.ValueOf(uint64(n)) |
| 640 | } |
| 641 | case pref.FloatKind: |
| 642 | switch n { |
| 643 | case minVal: |
| 644 | return pref.ValueOf(float32(math.SmallestNonzeroFloat32)) |
| 645 | case maxVal: |
| 646 | return pref.ValueOf(float32(math.MaxFloat32)) |
| 647 | default: |
| 648 | return pref.ValueOf(1.5 * float32(n)) |
| 649 | } |
| 650 | case pref.DoubleKind: |
| 651 | switch n { |
| 652 | case minVal: |
| 653 | return pref.ValueOf(float64(math.SmallestNonzeroFloat64)) |
| 654 | case maxVal: |
| 655 | return pref.ValueOf(float64(math.MaxFloat64)) |
| 656 | default: |
| 657 | return pref.ValueOf(1.5 * float64(n)) |
| 658 | } |
| 659 | case pref.StringKind: |
| 660 | if n == 0 { |
| 661 | return pref.ValueOf("") |
| 662 | } |
| 663 | return pref.ValueOf(fmt.Sprintf("%d", n)) |
| 664 | case pref.BytesKind: |
| 665 | if n == 0 { |
| 666 | return pref.ValueOf([]byte(nil)) |
| 667 | } |
| 668 | return pref.ValueOf([]byte{byte(n >> 24), byte(n >> 16), byte(n >> 8), byte(n)}) |
| 669 | } |
| 670 | panic("unhandled kind") |
| 671 | } |
| 672 | |
Joe Tsai | 0fc49f8 | 2019-05-01 12:29:25 -0700 | [diff] [blame] | 673 | func populateMessage(m pref.Message, n seed, stack []pref.MessageDescriptor) pref.Value { |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 674 | if n == 0 { |
| 675 | return pref.ValueOf(m) |
| 676 | } |
Joe Tsai | 0fc49f8 | 2019-05-01 12:29:25 -0700 | [diff] [blame] | 677 | md := m.Descriptor() |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 678 | for _, x := range stack { |
| 679 | if md == x { |
| 680 | return pref.ValueOf(m) |
| 681 | } |
| 682 | } |
| 683 | stack = append(stack, md) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 684 | for i := 0; i < md.Fields().Len(); i++ { |
| 685 | fd := md.Fields().Get(i) |
| 686 | if fd.IsWeak() { |
| 687 | continue |
| 688 | } |
Damien Neil | f527451 | 2019-08-05 10:48:38 -0700 | [diff] [blame] | 689 | m.Set(fd, newValue(m, fd, newSeed(n, i), stack)) |
Damien Neil | 5ec58b9 | 2019-04-30 11:36:39 -0700 | [diff] [blame] | 690 | } |
| 691 | return pref.ValueOf(m) |
| 692 | } |
Damien Neil | d91246e | 2019-06-21 08:36:26 -0700 | [diff] [blame] | 693 | |
| 694 | func panics(f func()) (didPanic bool) { |
| 695 | defer func() { |
| 696 | if err := recover(); err != nil { |
| 697 | didPanic = true |
| 698 | } |
| 699 | }() |
| 700 | f() |
| 701 | return false |
| 702 | } |