package middleware_test

import (
	"testing"

	"github.com/stretchr/testify/require"
	tmcrypto "github.com/tendermint/tendermint/crypto"

	"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
	"github.com/cosmos/cosmos-sdk/crypto/keys/secp256r1"
)

// This benchmark is used to asses the middleware.Secp256k1ToR1GasFactor value
func BenchmarkSig(b *testing.B) {
	require := require.New(b)
	msg := tmcrypto.CRandBytes(1000)

	skK := secp256k1.GenPrivKey()
	pkK := skK.PubKey()
	skR, _ := secp256r1.GenPrivKey()
	pkR := skR.PubKey()

	sigK, err := skK.Sign(msg)
	require.NoError(err)
	sigR, err := skR.Sign(msg)
	require.NoError(err)
	b.ResetTimer()

	b.Run("secp256k1", func(b *testing.B) {
		b.ReportAllocs()
		for i := 0; i < b.N; i++ {
			ok := pkK.VerifySignature(msg, sigK)
			require.True(ok)
		}
	})

	b.Run("secp256r1", func(b *testing.B) {
		b.ReportAllocs()
		for i := 0; i < b.N; i++ {
			ok := pkR.VerifySignature(msg, sigR)
			require.True(ok)
		}
	})
}