Merge pull request #276 from greydaemon/DOT-5128-prod

DOT-5128 : Changes for custom user agent and bumped cli version to 8

Author: sushobhit-lt

package.json

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

src/lib/ctx.ts

@@ -100,7 +100,8 @@ export default (options: Record<string, string>): Context => {
             ignoreHTTPSErrors: config.ignoreHTTPSErrors ?? false,
             skipBuildCreation: config.skipBuildCreation ?? false,
             tunnel: config.tunnel ?? false,
-            tunnelName: config.tunnelName || ''
+            tunnelName: config.tunnelName || '',
+            userAgent: config.userAgent || ''
         },
         uploadFilePath: '',
         webStaticConfig: [],

src/lib/schemaValidation.ts

@@ -211,6 +211,10 @@ const WebStaticConfigSchema: JSONSchemaType<WebStaticConfig> = {
                 maximum: 30000,
                 errorMessage: "waitForTimeout must be > 0 and <= 30000"
             },
+            userAgent: {
+                type: "string",
+                errorMessage: "User Agent value must be a valid string"
+            },
             execute: {
                 type: "object",
                 properties: {

src/lib/screenshot.ts

@@ -16,7 +16,7 @@ async function captureScreenshotsForConfig(
 ): Promise<void> {
     ctx.log.debug(`*** urlConfig  ${JSON.stringify(urlConfig)}`);
 
-    let {name, url, waitForTimeout, execute, pageEvent} = urlConfig;
+    let {name, url, waitForTimeout, execute, pageEvent, userAgent} = urlConfig;
     let afterNavigationScript = execute?.afterNavigation;
     let beforeSnapshotScript = execute?.beforeSnapshot;
     let waitUntilEvent = pageEvent || process.env.SMARTUI_PAGE_WAIT_UNTIL_EVENT || 'load';
@@ -31,6 +31,14 @@ async function captureScreenshotsForConfig(
     else if (browserName == constants.FIREFOX) contextOptions.userAgent = constants.FIREFOX_USER_AGENT;
     else if (browserName == constants.SAFARI) contextOptions.userAgent = constants.SAFARI_USER_AGENT;
     else if (browserName == constants.EDGE) contextOptions.userAgent = constants.EDGE_USER_AGENT;
+    if (ctx.config.userAgent || userAgent) {
+        if(ctx.config.userAgent !== ""){
+        contextOptions.userAgent = ctx.config.userAgent;
+        }
+        if (userAgent && userAgent !== "") {
+            contextOptions.userAgent = userAgent;
+        }
+    }
 
     try {
         const browser = browsers[browserName];

src/types.ts

@@ -34,6 +34,7 @@ export interface Context {
         skipBuildCreation?: boolean;
         tunnel?: boolean;
         tunnelName?: string;
+        userAgent?: string;
     };
     uploadFilePath: string;
     webStaticConfig: WebStaticConfig;
@@ -173,7 +174,8 @@ export interface MobileConfig {
 export type WebStaticConfig = Array<{
     name: string;
     url: string;
-    waitForTimeout?: number
+    waitForTimeout?: number;
+    userAgent?: string;
 }>;
 
 export type FigmaConfigItem = {