From 012a151b46b167d061134a4731473641e9be8ec2 Mon Sep 17 00:00:00 2001
From: Mike Fridman <mf192@icloud.com>
Date: Fri, 2 Aug 2024 08:58:24 -0400
Subject: [PATCH] fix tests and instantiate

---
 parser.go      |  2 +-
 parser_test.go | 14 +++++++++-----
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/parser.go b/parser.go
index ecf99af..0723509 100644
--- a/parser.go
+++ b/parser.go
@@ -93,7 +93,7 @@ func (p *Parser) ParseWithClaims(tokenString string, claims Claims, keyFunc Keyf
 	}
 
 	switch have := got.(type) {
-	case VerificationKeySet:
+	case VerificationKeySet[VerificationKey]:
 		if len(have.Keys) == 0 {
 			return token, newError("keyfunc returned empty verification key set", ErrTokenUnverifiable)
 		}
diff --git a/parser_test.go b/parser_test.go
index c0f8171..d3f6ddf 100644
--- a/parser_test.go
+++ b/parser_test.go
@@ -30,22 +30,26 @@ var (
 	nilKeyFunc             jwt.Keyfunc = nil
 	multipleZeroKeyFunc    jwt.Keyfunc = func(t *jwt.Token) (interface{}, error) { return []interface{}{}, nil }
 	multipleEmptyKeyFunc   jwt.Keyfunc = func(t *jwt.Token) (interface{}, error) {
-		return jwt.VerificationKeySet{Keys: []jwt.VerificationKey{nil, nil}}, nil
+		keys := []jwt.VerificationKey{nil, nil}
+		return jwt.VerificationKeySet[jwt.VerificationKey]{Keys: keys}, nil
 	}
 	multipleVerificationKeysFunc jwt.Keyfunc = func(t *jwt.Token) (interface{}, error) {
 		return []jwt.VerificationKey{jwtTestDefaultKey, jwtTestEC256PublicKey}, nil
 	}
 	multipleLastKeyFunc jwt.Keyfunc = func(t *jwt.Token) (interface{}, error) {
-		return jwt.VerificationKeySet{Keys: []jwt.VerificationKey{jwtTestEC256PublicKey, jwtTestDefaultKey}}, nil
+		keys := []jwt.VerificationKey{jwtTestEC256PublicKey, jwtTestDefaultKey}
+		return jwt.VerificationKeySet[jwt.VerificationKey]{Keys: keys}, nil
 	}
 	multipleFirstKeyFunc jwt.Keyfunc = func(t *jwt.Token) (interface{}, error) {
-		return jwt.VerificationKeySet{Keys: []jwt.VerificationKey{jwtTestDefaultKey, jwtTestEC256PublicKey}}, nil
+		keys := []jwt.VerificationKey{jwtTestDefaultKey, jwtTestEC256PublicKey}
+		return jwt.VerificationKeySet[jwt.VerificationKey]{Keys: keys}, nil
 	}
 	multipleAltTypedKeyFunc jwt.Keyfunc = func(t *jwt.Token) (interface{}, error) {
-		return jwt.VerificationKeySet{Keys: []jwt.VerificationKey{jwtTestDefaultKey, jwtTestDefaultKey}}, nil
+		keys := []jwt.VerificationKey{jwtTestDefaultKey, jwtTestDefaultKey}
+		return jwt.VerificationKeySet[jwt.VerificationKey]{Keys: keys}, nil
 	}
 	emptyVerificationKeySetFunc jwt.Keyfunc = func(t *jwt.Token) (interface{}, error) {
-		return jwt.VerificationKeySet{}, nil
+		return jwt.VerificationKeySet[jwt.VerificationKey]{}, nil
 	}
 )