Set user roles

On top of the existing add, remove + test related to descope/etc#4886
descope · Dec 18, 2023 · d8984dc · d8984dc
1 parent 7f19cae
commit d8984dc
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/lib/management/paths.ts b/lib/management/paths.ts
@@ -17,6 +17,7 @@ export default {
     updateDisplayName: '/v1/mgmt/user/update/name',
     updatePicture: '/v1/mgmt/user/update/picture',
     updateCustomAttribute: '/v1/mgmt/user/update/customAttribute',
+    setRole: '/v1/mgmt/user/update/role/set',
     addRole: '/v1/mgmt/user/update/role/add',
     removeRole: '/v1/mgmt/user/update/role/remove',
     addTenant: '/v1/mgmt/user/update/tenant/add',

diff --git a/lib/management/user.test.ts b/lib/management/user.test.ts
@@ -813,6 +813,35 @@ describe('Management User', () => {
     });
   });
 
+  describe('setRoles', () => {
+    it('should send the correct request and receive correct response', async () => {
+      const httpResponse = {
+        ok: true,
+        json: () => mockMgmtUserResponse,
+        clone: () => ({
+          json: () => Promise.resolve(mockMgmtUserResponse),
+        }),
+        status: 200,
+      };
+      mockHttpClient.post.mockResolvedValue(httpResponse);
+
+      const resp: SdkResponse<UserResponse> = await management.user.setRoles('lid', ['foo', 'bar']);
+
+      expect(mockHttpClient.post).toHaveBeenCalledWith(
+        apiPaths.user.setRole,
+        { loginId: 'lid', roleNames: ['foo', 'bar'] },
+        { token: 'key' },
+      );
+
+      expect(resp).toEqual({
+        code: 200,
+        data: mockUserResponse,
+        ok: true,
+        response: httpResponse,
+      });
+    });
+  });
+
   describe('addRoles', () => {
     it('should send the correct request and receive correct response', async () => {
       const httpResponse = {

diff --git a/lib/management/user.ts b/lib/management/user.ts
@@ -420,6 +420,15 @@ const withUser = (sdk: CoreSdk, managementKey?: string) => ({
       ),
       (data) => data.user,
     ),
+  setRoles: (loginId: string, roles: string[]): Promise<SdkResponse<UserResponse>> =>
+    transformResponse<SingleUserResponse, UserResponse>(
+      sdk.httpClient.post(
+        apiPaths.user.setRole,
+        { loginId, roleNames: roles },
+        { token: managementKey },
+      ),
+      (data) => data.user,
+    ),
   addRoles: (loginId: string, roles: string[]): Promise<SdkResponse<UserResponse>> =>
     transformResponse<SingleUserResponse, UserResponse>(
       sdk.httpClient.post(