cleaned up benchmarks

hmac_test.go

@@ -78,25 +78,14 @@ func TestHMACSign(t *testing.T) {
 	}
 }
 
-func BenchmarkHMACSigning(b *testing.B) {
+func BenchmarkHS256Signing(b *testing.B) {
-
+	benchmarkSigning(b, jwt.SigningMethodHS256, hmacTestKey)
-	var preppedData = make([]struct {
+}
-		t      *jwt.Token
+
-		method jwt.SigningMethod
+func BenchmarkHS384Signing(b *testing.B) {
-		k      interface{}
+	benchmarkSigning(b, jwt.SigningMethodHS384, hmacTestKey)
-	}, len(hmacTestData))
+}
-
+
-	for i, data := range hmacTestData {
+func BenchmarkHS512Signing(b *testing.B) {
-		preppedData[i].t, _ = jwt.Parse(data.tokenString, func(*jwt.Token) (interface{}, error) { return nil, nil })
+	benchmarkSigning(b, jwt.SigningMethodHS512, hmacTestKey)
-		preppedData[i].method = jwt.GetSigningMethod(data.alg)
-	}
-
-	for _, data := range preppedData {
-		b.RunParallel(func(pb *testing.PB) {
-			for pb.Next() {
-				data.t.SignedString(hmacTestKey)
-			}
-		})
-	}
-
 }

jwt_test.go

@@ -172,3 +172,14 @@ func TestParseRequest(t *testing.T) {
 		}
 	}
 }
+
+// Helper method for benchmarking various methods
+func benchmarkSigning(b *testing.B, method jwt.SigningMethod, key interface{}) {
+	t := jwt.New(method)
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			t.SignedString(key)
+		}
+	})
+
+}

rsa_test.go

@@ -142,3 +142,15 @@ func TestRSAKeyParsing(t *testing.T) {
 	}
 
 }
+
+func BenchmarkRS256Signing(b *testing.B) {
+	benchmarkSigning(b, jwt.SigningMethodRS256, hmacTestKey)
+}
+
+func BenchmarkRS384Signing(b *testing.B) {
+	benchmarkSigning(b, jwt.SigningMethodRS384, hmacTestKey)
+}
+
+func BenchmarkRS512Signing(b *testing.B) {
+	benchmarkSigning(b, jwt.SigningMethodRS512, hmacTestKey)
+}