cosmos-sdk/codec/bench_test.go
Julien Robert c452721442
style: add gci linter (#16758)
Co-authored-by: Jacob Gadikian <faddat@users.noreply.github.com>
2023-06-29 09:46:37 +00:00

98 lines
2.3 KiB
Go

package codec_test
import (
"testing"
"github.com/stretchr/testify/require"
protov2 "google.golang.org/protobuf/proto"
"google.golang.org/protobuf/types/dynamicpb"
bankv1beta1 "cosmossdk.io/api/cosmos/bank/v1beta1"
codectestutil "github.com/cosmos/cosmos-sdk/codec/testutil"
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
"github.com/cosmos/cosmos-sdk/testutil/testdata"
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
)
func BenchmarkLegacyGetSigners(b *testing.B) {
_, _, addr := testdata.KeyTestPubAddr()
msg := &banktypes.MsgSend{
FromAddress: addr.String(),
ToAddress: "",
Amount: nil,
}
b.ResetTimer()
for i := 0; i < b.N; i++ {
_ = msg.GetSigners()
}
}
func BenchmarkProtoreflectGetSigners(b *testing.B) {
cdc := codectestutil.CodecOptions{}.NewCodec()
signingCtx := cdc.InterfaceRegistry().SigningContext()
_, _, addr := testdata.KeyTestPubAddr()
// use a pulsar message
msg := &bankv1beta1.MsgSend{
FromAddress: addr.String(),
ToAddress: "",
Amount: nil,
}
b.ResetTimer()
for i := 0; i < b.N; i++ {
_, err := signingCtx.GetSigners(msg)
if err != nil {
panic(err)
}
}
}
func BenchmarkProtoreflectGetSignersWithUnmarshal(b *testing.B) {
cdc := codectestutil.CodecOptions{}.NewCodec()
_, _, addr := testdata.KeyTestPubAddr()
// start with a protoreflect message
msg := &banktypes.MsgSend{
FromAddress: addr.String(),
ToAddress: "",
Amount: nil,
}
// marshal to an any first because this is what we get from the wire
a, err := codectypes.NewAnyWithValue(msg)
require.NoError(b, err)
b.ResetTimer()
for i := 0; i < b.N; i++ {
_, _, err := cdc.GetMsgAnySigners(a)
if err != nil {
panic(err)
}
}
}
func BenchmarkProtoreflectGetSignersDynamicpb(b *testing.B) {
cdc := codectestutil.CodecOptions{}.NewCodec()
signingCtx := cdc.InterfaceRegistry().SigningContext()
_, _, addr := testdata.KeyTestPubAddr()
msg := &bankv1beta1.MsgSend{
FromAddress: addr.String(),
ToAddress: "",
Amount: nil,
}
bz, err := protov2.Marshal(msg)
require.NoError(b, err)
dynamicmsg := dynamicpb.NewMessage(msg.ProtoReflect().Descriptor())
err = protov2.Unmarshal(bz, dynamicmsg)
require.NoError(b, err)
b.ResetTimer()
for i := 0; i < b.N; i++ {
_, err := signingCtx.GetSigners(dynamicmsg)
if err != nil {
panic(err)
}
}
}