@lambdatest/smartui-cli 4.1.36 → 4.1.37-beta.0

package/dist/index.cjs

@@ -2145,7 +2145,7 @@ var server_default = (ctx) => __async(void 0, null, function* () {
         ctx.log.debug(`Initialized empty context mapping map for contextId: ${contextId}`);
       }
       if (contextId && ctx.contextToSnapshotMap) {
-        ctx.contextToSnapshotMap.set(contextId, 0);
+        ctx.contextToSnapshotMap.set(contextId, "0");
         ctx.log.debug(`Marking contextId as captured and added to queue: ${contextId}`);
       }
       if (contextId) {
@@ -2258,19 +2258,29 @@ var server_default = (ctx) => __async(void 0, null, function* () {
       const timeoutDuration = pollTimeout * 1e3 || 3e4;
       if ((_a = ctx.contextToSnapshotMap) == null ? void 0 : _a.has(contextId)) {
         let contextStatus = ctx.contextToSnapshotMap.get(contextId);
-        while (contextStatus == 0) {
-          yield new Promise((resolve) => setTimeout(resolve, 5e3));
+        let counter = 60;
+        while (contextStatus === "0") {
+          if (counter <= 0) {
+            throw new Error("Snapshot processing failed");
+          }
           contextStatus = ctx.contextToSnapshotMap.get(contextId);
+          yield new Promise((resolve) => setTimeout(resolve, 5e3));
+          counter--;
         }
-        if (contextStatus == 2) {
+        if (contextStatus === "2") {
           throw new Error("Snapshot Failed");
         }
         ctx.log.debug("Snapshot uploaded successfully");
+        const buildId = contextStatus;
+        if (!buildId) {
+          throw new Error(`No buildId found for contextId: ${contextId}`);
+        }
         let lastExternalResponse = null;
         const startTime = Date.now();
         while (true) {
           try {
             const externalResponse = yield ctx.client.getSnapshotStatus(
+              buildId,
               snapshotName,
               contextId,
               ctx
@@ -2488,7 +2498,7 @@ var authExec_default = (ctx) => {
 };
 
 // package.json
-var version = "4.1.36";
+var version = "4.1.37-beta.0";
 var package_default = {
   name: "@lambdatest/smartui-cli",
   version,
@@ -2885,7 +2895,7 @@ var httpClient = class {
       data: requestData
     }, ctx.log);
   }
-  processSnapshotCaps(ctx, snapshot, snapshotUuid, capsBuildId, capsProjectToken, discoveryErrors) {
+  processSnapshotCaps(ctx, snapshot, snapshotUuid, capsBuildId, capsProjectToken, discoveryErrors, variantCount, sync = false) {
     return this.request({
       url: `/build/${capsBuildId}/snapshot`,
       method: "POST",
@@ -2897,16 +2907,18 @@ var httpClient = class {
         name: snapshot.name,
         url: snapshot.url,
         snapshotUuid,
+        variantCount,
         test: {
           type: ctx.testType,
           source: "cli"
         },
         doRemoteDiscovery: snapshot.options.doRemoteDiscovery,
-        discoveryErrors
+        discoveryErrors,
+        sync
       }
     }, ctx.log);
   }
-  uploadSnapshotForCaps(ctx, snapshot, capsBuildId, capsProjectToken, discoveryErrors) {
+  uploadSnapshotForCaps(ctx, snapshot, capsBuildId, capsProjectToken, discoveryErrors, variantCount, sync = false) {
     const buildId = capsBuildId !== "" ? capsBuildId : ctx.build.id;
     return this.request({
       url: `/builds/${buildId}/snapshot`,
@@ -2922,7 +2934,9 @@ var httpClient = class {
           type: ctx.testType,
           source: "cli"
         },
-        discoveryErrors
+        discoveryErrors,
+        variantCount,
+        sync
       }
     }, ctx.log);
   }
@@ -3146,9 +3160,9 @@ var httpClient = class {
       data: requestData
     }, ctx.log);
   }
-  getSnapshotStatus(snapshotName, snapshotUuid, ctx) {
+  getSnapshotStatus(buildId, snapshotName, snapshotUuid, ctx) {
     return this.request({
-      url: `/snapshot/status?buildId=${ctx.build.id}&snapshotName=${snapshotName}&snapshotUUID=${snapshotUuid}`,
+      url: `/snapshot/status?buildId=${buildId}&snapshotName=${snapshotName}&snapshotUUID=${snapshotUuid}`,
       method: "GET",
       headers: {
         "Content-Type": "application/json"
@@ -4788,7 +4802,7 @@ var Queue = class {
   }
   processNext() {
     return __async(this, null, function* () {
-      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o;
+      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t;
       if (!this.isEmpty()) {
         let snapshot;
         if (this.ctx.config.delayedUpload) {
@@ -4862,7 +4876,10 @@ var Queue = class {
             if (useCapsBuildId) {
               this.ctx.log.info(`Using cached buildId: ${capsBuildId}`);
               if (useKafkaFlowCaps) {
-                const snapshotUuid = uuid.v4();
+                let snapshotUuid = uuid.v4();
+                if (((_c = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _c.contextId) && ((_d = this.ctx.contextToSnapshotMap) == null ? void 0 : _d.has(snapshot.options.contextId))) {
+                  snapshotUuid = snapshot.options.contextId;
+                }
                 let uploadDomToS3 = this.ctx.config.useLambdaInternal || uploadDomToS3ViaEnv3;
                 if (!uploadDomToS3) {
                   this.ctx.log.debug(`Uploading dom to S3 for snapshot using presigned URL for CAPS`);
@@ -4873,16 +4890,19 @@ var Queue = class {
                   this.ctx.log.debug(`Uploading dom to S3 for snapshot using LSRS`);
                   yield this.ctx.client.sendDomToLSRSForCaps(this.ctx, processedSnapshot, snapshotUuid, capsBuildId, capsProjectToken);
                 }
-                yield this.ctx.client.processSnapshotCaps(this.ctx, processedSnapshot, snapshotUuid, capsBuildId, capsProjectToken, discoveryErrors);
+                yield this.ctx.client.processSnapshotCaps(this.ctx, processedSnapshot, snapshotUuid, capsBuildId, capsProjectToken, discoveryErrors, calculateVariantCountFromSnapshot(processedSnapshot, this.ctx.config), (_e = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _e.sync);
               } else {
-                yield this.ctx.client.uploadSnapshotForCaps(this.ctx, processedSnapshot, capsBuildId, capsProjectToken, discoveryErrors);
+                yield this.ctx.client.uploadSnapshotForCaps(this.ctx, processedSnapshot, capsBuildId, capsProjectToken, discoveryErrors, calculateVariantCountFromSnapshot(processedSnapshot, this.ctx.config), (_f = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _f.sync);
               }
               const cachedCapabilities = this.ctx.sessionCapabilitiesMap.get(sessionId);
               const currentCount = (cachedCapabilities == null ? void 0 : cachedCapabilities.snapshotCount) || 0;
               cachedCapabilities.snapshotCount = currentCount + 1;
               this.ctx.sessionCapabilitiesMap.set(sessionId, cachedCapabilities);
+              if (((_g = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _g.contextId) && this.ctx.contextToSnapshotMap) {
+                this.ctx.contextToSnapshotMap.set(snapshot.options.contextId, capsBuildId);
+              }
             } else {
-              if (!((_c = this.ctx.build) == null ? void 0 : _c.id)) {
+              if (!((_h = this.ctx.build) == null ? void 0 : _h.id)) {
                 if (this.ctx.authenticatedInitially) {
                   let resp = yield this.ctx.client.createBuild(this.ctx.git, this.ctx.config, this.ctx.log, this.ctx.build.name, false, false, false, "");
                   this.ctx.build = {
@@ -4907,7 +4927,7 @@ var Queue = class {
               if (this.ctx.build && this.ctx.build.useKafkaFlow) {
                 let snapshotUuid = uuid.v4();
                 let snapshotUploadResponse;
-                if (((_d = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _d.contextId) && ((_e = this.ctx.contextToSnapshotMap) == null ? void 0 : _e.has(snapshot.options.contextId))) {
+                if (((_i = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _i.contextId) && ((_j = this.ctx.contextToSnapshotMap) == null ? void 0 : _j.has(snapshot.options.contextId))) {
                   snapshotUuid = snapshot.options.contextId;
                 }
                 let uploadDomToS3 = this.ctx.config.useLambdaInternal || uploadDomToS3ViaEnv3;
@@ -4933,18 +4953,18 @@ var Queue = class {
                       this.ctx.log.debug(`Closed browser context for snapshot ${snapshot.name}`);
                     }
                   }
-                  if ((_f = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _f.contextId) {
-                    (_h = this.ctx.contextToSnapshotMap) == null ? void 0 : _h.set((_g = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _g.contextId, 2);
+                  if ((_k = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _k.contextId) {
+                    (_m = this.ctx.contextToSnapshotMap) == null ? void 0 : _m.set((_l = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _l.contextId, "2");
                   }
                   this.processNext();
                 } else {
-                  let approvalThreshold = ((_i = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _i.approvalThreshold) || this.ctx.config.approvalThreshold;
-                  let rejectionThreshold = ((_j = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _j.rejectionThreshold) || this.ctx.config.rejectionThreshold;
-                  yield this.ctx.client.processSnapshot(this.ctx, processedSnapshot, snapshotUuid, discoveryErrors, calculateVariantCountFromSnapshot(processedSnapshot, this.ctx.config), (_k = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _k.sync, approvalThreshold, rejectionThreshold);
-                  if (((_l = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _l.contextId) && ((_m = this.ctx.contextToSnapshotMap) == null ? void 0 : _m.has(snapshot.options.contextId))) {
-                    this.ctx.contextToSnapshotMap.set(snapshot.options.contextId, 1);
+                  let approvalThreshold = ((_n = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _n.approvalThreshold) || this.ctx.config.approvalThreshold;
+                  let rejectionThreshold = ((_o = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _o.rejectionThreshold) || this.ctx.config.rejectionThreshold;
+                  yield this.ctx.client.processSnapshot(this.ctx, processedSnapshot, snapshotUuid, discoveryErrors, calculateVariantCountFromSnapshot(processedSnapshot, this.ctx.config), (_p = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _p.sync, approvalThreshold, rejectionThreshold);
+                  if (((_q = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _q.contextId) && ((_r = this.ctx.contextToSnapshotMap) == null ? void 0 : _r.has(snapshot.options.contextId))) {
+                    this.ctx.contextToSnapshotMap.set(snapshot.options.contextId, this.ctx.build.id);
                   }
-                  this.ctx.log.debug(`ContextId: ${(_n = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _n.contextId} status set to uploaded`);
+                  this.ctx.log.debug(`ContextId: ${(_s = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _s.contextId} status set to uploaded`);
                 }
               } else {
                 this.ctx.log.info(`Uploading snapshot to S3`);
@@ -4957,8 +4977,8 @@ var Queue = class {
         } catch (error) {
           this.ctx.log.debug(`snapshot failed; ${error}`);
           this.processedSnapshots.push({ name: snapshot == null ? void 0 : snapshot.name, error: error.message });
-          if (((_o = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _o.contextId) && this.ctx.contextToSnapshotMap) {
-            this.ctx.contextToSnapshotMap.set(snapshot.options.contextId, 2);
+          if (((_t = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _t.contextId) && this.ctx.contextToSnapshotMap) {
+            this.ctx.contextToSnapshotMap.set(snapshot.options.contextId, "2");
           }
         }
         if (this.ctx.browser) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lambdatest/smartui-cli",
-  "version": "4.1.36",
+  "version": "4.1.37-beta.0",
   "description": "A command line interface (CLI) to run SmartUI tests on LambdaTest",
   "files": [
     "dist/**/*"