fix: issues in e2e tests

Signed-off-by: Ivo Yankov <[email protected]>

src/commands/cluster/handlers.ts

@@ -22,6 +22,7 @@ import * as ContextFlags from './flags.js';
 import {RemoteConfigTasks} from '../../core/config/remote/remote_config_tasks.js';
 import type {RemoteConfigManager} from '../../core/config/remote/remote_config_manager.js';
 import {connectConfigBuilder, resetConfigBuilder, setupConfigBuilder} from './configs.js';
+import {SoloError} from "../../core/errors.js";
 
 export class ClusterCommandHandlers implements CommandHandlers {
   readonly parent: BaseCommand;
@@ -110,7 +111,12 @@ export class ClusterCommandHandlers implements CommandHandlers {
       null,
     );
 
-    await action(argv, this);
+    try {
+        await action(argv, this);
+    } catch (e: Error | any) {
+        throw new SoloError('Error on cluster setup', e);
+    }
+
     return true;
   }
 
@@ -131,7 +137,11 @@ export class ClusterCommandHandlers implements CommandHandlers {
       null,
     );
 
-    await action(argv, this);
+    try {
+        await action(argv, this);
+    } catch (e: Error | any) {
+        throw new SoloError('Error on cluster reset', e);
+    }
     return true;
   }
 }

src/commands/cluster/tasks.ts

@@ -310,10 +310,10 @@ export class ClusterCommandTasks {
     );
   }
 
-  async acquireNewLease(argv) {
-    const lease = await this.parent.getLeaseManager().create();
+  acquireNewLease(argv) {
     return new Task('Acquire new lease', async (ctx: any, task: ListrTaskWrapper<any, any, any>) => {
-      ListrLease.newAcquireLeaseTask(lease, task);
+      const lease = await this.parent.getLeaseManager().create();
+      return ListrLease.newAcquireLeaseTask(lease, task);
     });
   }
 

test/e2e/commands/cluster.test.ts

@@ -111,7 +111,7 @@ describe('ClusterCommand', () => {
 
   it('function showInstalledChartList should return right true', async () => {
     // @ts-ignore
-    await expect(clusterCmd.showInstalledChartList()).to.eventually.be.undefined;
+    await expect(clusterCmd.handlers.tasks.showInstalledChartList()).to.eventually.be.undefined;
   }).timeout(Duration.ofMinutes(1).toMillis());
 
   // helm list would return an empty list if given invalid namespace