@@ -542,10 +542,11 @@ func TestMutationBulkUpdateOrgMemberRole(t *testing.T) {
542542 Role : & adminRole ,
543543 }
544544
545- for _ , id := range ids {
546- _ , err = suite .client .api .UpdateUserRoleInOrg (org .admin .UserCtx , id , input )
547- assert .NilError (t , err )
548- }
545+ resp , err := suite .client .api .UpdateBulkOrgMemberRoles (org .admin .UserCtx , ids , input )
546+ assert .NilError (t , err )
547+ assert .Assert (t , resp != nil )
548+ assert .Check (t , is .Len (resp .UpdateBulkOrgMembership .UpdatedIDs , len (ids )))
549+ assert .Check (t , is .Len (resp .UpdateBulkOrgMembership .OrgMemberships , len (ids )))
549550
550551 updatedMembers , err := suite .client .db .OrgMembership .Query ().
551552 Where (orgmembership .IDIn (ids ... )).
@@ -567,8 +568,17 @@ func TestMutationBulkUpdateOrgMemberRole(t *testing.T) {
567568 memberRole := enums .RoleMember
568569 input .Role = & memberRole
569570
570- _ , err = suite .client .api .UpdateUserRoleInOrg (org .admin .UserCtx , ownerMember .ID , input )
571- assert .ErrorContains (t , err , hooks .ErrOrgOwnerCannotBeUpdated .Error ())
571+ ownerResp , err := suite .client .api .UpdateBulkOrgMemberRoles (org .admin .UserCtx , []string {ownerMember .ID }, input )
572+ assert .NilError (t , err )
573+ assert .Assert (t , ownerResp != nil )
574+ assert .Check (t , is .Len (ownerResp .UpdateBulkOrgMembership .UpdatedIDs , 0 ))
575+ assert .Check (t , is .Len (ownerResp .UpdateBulkOrgMembership .OrgMemberships , 0 ))
576+
577+ ownerMember , err = suite .client .db .OrgMembership .Query ().
578+ Where (orgmembership .ID (ownerMember .ID )).
579+ Only (allowCtx )
580+ assert .NilError (t , err )
581+ assert .Check (t , is .Equal (enums .RoleOwner , ownerMember .Role ))
572582
573583 cleanupOrganizationDataWithContext (org .owner .UserCtx , t )
574584}
0 commit comments