feat: add user.updateLimits

src/resource_clients/user.ts

@@ -64,6 +64,19 @@ export class UserClient extends ResourceClient {
 
         return undefined;
     }
+
+    /**
+     * https://docs.apify.com/api/v2/#/reference/users/account-and-usage-limits
+     */
+    async updateLimits(options: LimitsUpdateOptions): Promise<void> {
+        const requestOpts: ApifyRequestConfig = {
+            url: this._url('limits'),
+            method: 'PUT',
+            params: this._params(),
+            data: options,
+        };
+        await this.httpClient.call(requestOpts);
+    }
 }
 
 //
@@ -207,6 +220,8 @@ export interface Limits {
     dataRetentionDays: number;
 }
 
+export type LimitsUpdateOptions = Pick<Limits, 'maxMonthlyUsageUsd' | 'dataRetentionDays'>
+
 export interface Current {
     monthlyUsageUsd: number;
     monthlyActorComputeUnits: number;

test/mock_server/routes/users.js

@@ -8,6 +8,7 @@ const ROUTES = [
     { id: 'get-user', method: 'GET', path: '/:userId' },
     { id: 'get-monthly-usage', method: 'GET', path: '/:userId/usage/monthly' },
     { id: 'get-limits', method: 'GET', path: '/:userId/limits' },
+    { id: 'update-limits', method: 'PUT', path: '/:userId/limits' },
 ];
 
 addRoutes(users, ROUTES);

test/users.test.js

@@ -82,5 +82,17 @@ describe('User methods', () => {
             expect(browserRes).toEqual(res);
             validateRequest({}, { userId });
         });
+
+        test('updateLimits() works', async () => {
+            const userId = 'me';
+
+            const res = await client.user(userId).updateLimits({ maxMonthlyUsageUsd: 1000 });
+            expect(res).toBeUndefined();
+            validateRequest({}, { userId }, { maxMonthlyUsageUsd: 1000 });
+
+            const browserRes = await page.evaluate((id) => client.user(id).updateLimits({ maxMonthlyUsageUsd: 1000 }), userId);
+            expect(browserRes).toBeUndefined();
+            validateRequest({}, { userId }, { maxMonthlyUsageUsd: 1000 });
+        });
     });
 });