mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2024-12-28 18:25:46 +00:00
d0b074956d
Move the test inputs for the wire marshaler and unmarshaler out of decode_test.go and into a new file. Consolidate some tests for invalid messages (UTF-8 validation failures, field numbers out of range) into a single list of invalid messages. Break out the no-enforce-utf8 test into a separate file, since it is both complicated and conditional on legacy support. Change-Id: Ide80fa9d3aec2b6d42a57e6f9265358aa5e661a7 Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/211557 Reviewed-by: Joe Tsai <joetsai@google.com>
76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
// Copyright 2019 The Go 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 proto_test
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
|
|
protoV1 "github.com/golang/protobuf/proto"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
// The results of these microbenchmarks are unlikely to correspond well
|
|
// to real world peformance. They are mainly useful as a quick check to
|
|
// detect unexpected regressions and for profiling specific cases.
|
|
|
|
var (
|
|
benchV1 = flag.Bool("v1", false, "benchmark the v1 implementation")
|
|
allowPartial = flag.Bool("allow_partial", false, "set AllowPartial")
|
|
)
|
|
|
|
// BenchmarkEncode benchmarks encoding all the test messages.
|
|
func BenchmarkEncode(b *testing.B) {
|
|
for _, test := range testValidMessages {
|
|
for _, want := range test.decodeTo {
|
|
v1 := want.(protoV1.Message)
|
|
opts := proto.MarshalOptions{AllowPartial: *allowPartial}
|
|
b.Run(fmt.Sprintf("%s (%T)", test.desc, want), func(b *testing.B) {
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
var err error
|
|
if *benchV1 {
|
|
_, err = protoV1.Marshal(v1)
|
|
} else {
|
|
_, err = opts.Marshal(want)
|
|
}
|
|
if err != nil && !test.partial {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
})
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
// BenchmarkDecode benchmarks decoding all the test messages.
|
|
func BenchmarkDecode(b *testing.B) {
|
|
for _, test := range testValidMessages {
|
|
for _, want := range test.decodeTo {
|
|
opts := proto.UnmarshalOptions{AllowPartial: *allowPartial}
|
|
b.Run(fmt.Sprintf("%s (%T)", test.desc, want), func(b *testing.B) {
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
m := reflect.New(reflect.TypeOf(want).Elem()).Interface().(proto.Message)
|
|
v1 := m.(protoV1.Message)
|
|
var err error
|
|
if *benchV1 {
|
|
err = protoV1.Unmarshal(test.wire, v1)
|
|
} else {
|
|
err = opts.Unmarshal(test.wire, m)
|
|
}
|
|
if err != nil && !test.partial {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
})
|
|
})
|
|
}
|
|
}
|
|
}
|