update oidc_test.go

Signed-off-by: Atif Ali <[email protected]>
argoproj · Nov 8, 2024 · 9c0e980 · 9c0e980
1 parent d168329
commit 9c0e980
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/util/oidc/oidc_test.go b/util/oidc/oidc_test.go
@@ -639,7 +639,7 @@ func TestGetUserInfo(t *testing.T) {
 				expectError     bool
 			}{
 				{
-					key:         formatUserInfoResponseCacheKey("randomUser"),
+					key:         formatUserInfoResponseCacheKey(jwt.MapClaims{"sub": "randomUser"}),
 					expectError: true,
 				},
 			},
@@ -654,7 +654,7 @@ func TestGetUserInfo(t *testing.T) {
 				encrypt bool
 			}{
 				{
-					key:     formatAccessTokenCacheKey("randomUser"),
+					key:     formatUserInfoResponseCacheKey(jwt.MapClaims{"sub": "randomUser"}),
 					value:   "FakeAccessToken",
 					encrypt: true,
 				},
@@ -673,7 +673,7 @@ func TestGetUserInfo(t *testing.T) {
 				expectError     bool
 			}{
 				{
-					key:         formatUserInfoResponseCacheKey("randomUser"),
+					key:         formatUserInfoResponseCacheKey(jwt.MapClaims{"sub": "randomUser"}),
 					expectError: true,
 				},
 			},
@@ -688,7 +688,7 @@ func TestGetUserInfo(t *testing.T) {
 				encrypt bool
 			}{
 				{
-					key:     formatAccessTokenCacheKey("randomUser"),
+					key:     formatUserInfoResponseCacheKey(jwt.MapClaims{"sub": "randomUser"}),
 					value:   "FakeAccessToken",
 					encrypt: true,
 				},
@@ -707,7 +707,7 @@ func TestGetUserInfo(t *testing.T) {
 				expectError     bool
 			}{
 				{
-					key:         formatUserInfoResponseCacheKey("randomUser"),
+					key:         formatUserInfoResponseCacheKey(jwt.MapClaims{"sub": "randomUser"}),
 					expectError: true,
 				},
 			},
@@ -730,7 +730,7 @@ func TestGetUserInfo(t *testing.T) {
 				encrypt bool
 			}{
 				{
-					key:     formatAccessTokenCacheKey("randomUser"),
+					key:     formatUserInfoResponseCacheKey(jwt.MapClaims{"sub": "randomUser"}),
 					value:   "FakeAccessToken",
 					encrypt: true,
 				},
@@ -749,7 +749,7 @@ func TestGetUserInfo(t *testing.T) {
 				expectError     bool
 			}{
 				{
-					key:         formatUserInfoResponseCacheKey("randomUser"),
+					key:         formatUserInfoResponseCacheKey(jwt.MapClaims{"sub": "randomUser"}),
 					expectError: true,
 				},
 			},
@@ -782,7 +782,7 @@ func TestGetUserInfo(t *testing.T) {
 				expectError     bool
 			}{
 				{
-					key:             formatUserInfoResponseCacheKey("randomUser"),
+					key:             formatUserInfoResponseCacheKey(jwt.MapClaims{"sub": "randomUser"}),
 					value:           "{\"groups\":[\"githubOrg:engineers\"]}",
 					expectEncrypted: true,
 					expectError:     false,
@@ -809,7 +809,7 @@ func TestGetUserInfo(t *testing.T) {
 				encrypt bool
 			}{
 				{
-					key:     formatAccessTokenCacheKey("randomUser"),
+					key:     formatUserInfoResponseCacheKey(jwt.MapClaims{"sub": "randomUser"}),
 					value:   "FakeAccessToken",
 					encrypt: true,
 				},