fix: test cases

Author: aman19K

packages/contentstack-config/test/unit/commands/rate-limit.test.ts

@@ -54,33 +54,49 @@ describe('Rate Limit Commands', () => {
     });
 
     it('Set Rate Limit: should handle invalid utilization percentages', async () => {
+      const exitStub = stub(SetRateLimitCommand.prototype, 'exit'); // Stub the exit method
+
       const args = ['--org', 'test-org-id', '--utilize', '150', '--limit-name', 'getLimit'];
-      try {
-        await SetRateLimitCommand.run(args);
-        expect.fail('Expected command to exit with error');
-      } catch (error) {
-        expect(error.oclif.exit).to.equal(1);
-      }
+      await SetRateLimitCommand.run(args);
+
+      expect(errorMessage).to.equal('Utilize percentages must be numbers between 0 and 100.');
+
+      expect(exitStub.calledWith(1)).to.be.true;
+
+      // Restore the stub after the test
+      exitStub.restore();
     });
 
     it('Set Rate Limit: should handle mismatch between utilize percentages and limit names', async () => {
+      const exitStub = stub(SetRateLimitCommand.prototype, 'exit'); // Stub the exit method
+
       const args = ['--org', 'test-org-id', '--utilize', '70', '--limit-name', 'getLimit,postLimit'];
-      try {
-        await SetRateLimitCommand.run(args);
-        expect.fail('Expected command to exit with error');
-      } catch (error) {
-        expect(error.oclif.exit).to.equal(1);
-      }
+      await SetRateLimitCommand.run(args);
+
+      expect(errorMessage).to.equal(
+        'The number of utilization percentages must match the number of limit names provided.',
+      );
+
+      expect(exitStub.calledWith(1)).to.be.true;
+
+      // Restore the stub after the test
+      exitStub.restore();
     });
 
     it('Set Rate Limit: should handle invalid number of limit names', async () => {
+      const exitStub = stub(SetRateLimitCommand.prototype, 'exit'); // Stub the exit method
+
       const args = ['--org', 'test-org-id', '--utilize', '70,80', '--limit-name', 'getLimit'];
-      try {
-        await SetRateLimitCommand.run(args);
-        expect.fail('Expected command to exit with error');
-      } catch (error) {
-        expect(error.oclif.exit).to.equal(1);
-      }
+      await SetRateLimitCommand.run(args);
+
+      expect(errorMessage).to.equal(
+        'The number of utilization percentages must match the number of limit names provided.',
+      );
+
+      expect(exitStub.calledWith(1)).to.be.true;
+
+      // Restore the stub after the test
+      exitStub.restore();
     });
 
     it('Set Rate Limit: should prompt for the organization UID', async () => {
@@ -107,17 +123,22 @@ describe('Rate Limit Commands', () => {
       }
     });
 
-    it.skip('Set Rate Limit: should handle unauthenticated user', async () => {
-      // Mock isAuthenticated to return false
+    it('Set Rate Limit: should handle unauthenticated user', async () => {
       const isAuthenticatedStub = stub().returns(false);
-      Object.defineProperty(require('@contentstack/cli-utilities'), 'isAuthenticated', {
-        value: isAuthenticatedStub,
-        writable: true,
-      });
-
+      authenticated = isAuthenticatedStub;
+      // Stub the exit method to prevent process exit
+      const exitStub = stub(SetRateLimitCommand.prototype, 'exit');
       const args = ['--org', 'test-org-id', '--utilize', '70,80', '--limit-name', 'getLimit,bulkLimit'];
       await SetRateLimitCommand.run(args);
+
+      // Assert that the correct error message was printed
       expect(printMessage).to.equal('You are not logged in. Please login with command $ csdx auth:login');
+
+      // Ensure exit was called with code 1
+      expect(exitStub.calledWith(1)).to.be.true;
+
+      // Restore the stub
+      exitStub.restore();
     });
 
     it('should set default rate limit for organization', async () => {
@@ -179,6 +200,7 @@ describe('Rate Limit Commands', () => {
       await RemoveRateLimitCommand.run(['--org', 'test-org-id']);
       const updatedRateLimit = configHandler.get('rateLimit');
       expect(updatedRateLimit['test-org-id']).to.be.undefined;
+      expect(printMessage).to.equal('Rate limit entry for organization UID test-org-id has been removed.');
     });
 
     it('Remove Rate Limit: should throw an error if the organization is not found', async () => {