@@ -88,8 +88,8 @@ func TestAddOauth(t *testing.T) {
8888				"--restricted-group" , "restricted" ,
8989				"--group-team-map" , `{"group1": [1,2]}` ,
9090				"--group-team-map-removal=true" ,
91- 				"--attribute- ssh-public-key" , "attr_ssh_pub_key" ,
92- 				"--attribute- full-name" , "attr_full_name" ,
91+ 				"--ssh-public-key-claim-name " , "attr_ssh_pub_key" ,
92+ 				"--full-name-claim -name" , "attr_full_name" ,
9393			},
9494			source : & auth_model.Source {
9595				Type :     auth_model .OAuth2 ,
@@ -115,8 +115,8 @@ func TestAddOauth(t *testing.T) {
115115					RestrictedGroup :       "restricted" ,
116116					GroupTeamMap :          `{"group1": [1,2]}` ,
117117					GroupTeamMapRemoval :   true ,
118- 					AttributeSSHPublicKey : "attr_ssh_pub_key" ,
119- 					AttributeFullName :     "attr_full_name" ,
118+ 					SSHPublicKeyClaimName : "attr_ssh_pub_key" ,
119+ 					FullNameClaimName :     "attr_full_name" ,
120120				},
121121				TwoFactorPolicy : "skip" ,
122122			},
@@ -236,8 +236,8 @@ func TestUpdateOauth(t *testing.T) {
236236					RestrictedGroup :       "old_restricted" ,
237237					GroupTeamMap :          `{"old_group1": [1,2]}` ,
238238					GroupTeamMapRemoval :   true ,
239- 					AttributeSSHPublicKey : "old_ssh_pub_key" ,
240- 					AttributeFullName :     "old_full_name" ,
239+ 					SSHPublicKeyClaimName : "old_ssh_pub_key" ,
240+ 					FullNameClaimName :     "old_full_name" ,
241241				},
242242				TwoFactorPolicy : "" ,
243243			},
@@ -263,8 +263,8 @@ func TestUpdateOauth(t *testing.T) {
263263				"--restricted-group" , "restricted" ,
264264				"--group-team-map" , `{"group1": [1,2]}` ,
265265				"--group-team-map-removal=false" ,
266- 				"--attribute- ssh-public-key" , "new_ssh_pub_key" ,
267- 				"--attribute- full-name" , "new_full_name" ,
266+ 				"--ssh-public-key-claim-name " , "new_ssh_pub_key" ,
267+ 				"--full-name-claim -name" , "new_full_name" ,
268268			},
269269			authSource : & auth_model.Source {
270270				ID :       1 ,
@@ -291,8 +291,8 @@ func TestUpdateOauth(t *testing.T) {
291291					RestrictedGroup :       "restricted" ,
292292					GroupTeamMap :          `{"group1": [1,2]}` ,
293293					GroupTeamMapRemoval :   false ,
294- 					AttributeSSHPublicKey : "new_ssh_pub_key" ,
295- 					AttributeFullName :     "new_full_name" ,
294+ 					SSHPublicKeyClaimName : "new_ssh_pub_key" ,
295+ 					FullNameClaimName :     "new_full_name" ,
296296				},
297297				TwoFactorPolicy : "skip" ,
298298			},
0 commit comments