import (
	"crypto/ecdsa"
	"crypto/elliptic"
	"crypto/rand"
	"crypto/rsa"
	"testing"
)

func BenchmarkRSA2048(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_, err := rsa.GenerateKey(rand.Reader, 2048)
		if err != nil {
			b.Fatalf("error occurred")
		}
	}
}

func BenchmarkRSA3072(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_, err := rsa.GenerateKey(rand.Reader, 3072)
		if err != nil {
			b.Fatalf("error occurred")
		}
	}
}

func BenchmarkRSA4096(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_, err := rsa.GenerateKey(rand.Reader, 4096)
		if err != nil {
			b.Fatalf("error occurred")
		}
	}
}

func BenchmarkECDSA256(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
		if err != nil {
			b.Fatalf("error occurred")
		}
	}
}