From eb72be0aabe65eed2f27d28540844d9d062be501 Mon Sep 17 00:00:00 2001 From: Pavindu Lakshan Date: Tue, 12 Aug 2025 07:58:34 +0530 Subject: [PATCH] Fix tests --- internal/util/jwks_test.go | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/internal/util/jwks_test.go b/internal/util/jwks_test.go index 3b00c68..19c506e 100644 --- a/internal/util/jwks_test.go +++ b/internal/util/jwks_test.go @@ -7,6 +7,7 @@ import ( "encoding/json" "net/http" "net/http/httptest" + "strings" "testing" "time" @@ -47,7 +48,14 @@ func TestValidateJWT(t *testing.T) { for _, tc := range tests { t.Run(tc.name, func(t *testing.T) { - err := ValidateJWT(tc.authHeader) + var accessToken string + parts := strings.Split(tc.authHeader, "Bearer ") + if len(parts) == 2 { + accessToken = parts[1] + } else { + accessToken = "" + } + err := ValidateJWT(true, accessToken, "test-audience") if tc.expectError && err == nil { t.Errorf("Expected error but got none") } @@ -128,6 +136,7 @@ func createValidJWT(t *testing.T) string { token := jwt.NewWithClaims(jwt.SigningMethodRS256, jwt.MapClaims{ "sub": "1234567890", "name": "Test User", + "aud": "test-audience", "iat": time.Now().Unix(), "exp": time.Now().Add(time.Hour).Unix(), })