added support for custom userAgent in cli capture

Author: lt-aditya

src/lib/constants.ts

@@ -31,7 +31,8 @@ export default {
         {
             "name": "lambdatest-home-page",
             "url": "https://www.lambdatest.com",
-            "waitForTimeout": 1000
+            "waitForTimeout": 1000,
+            "userAgent" : ""
         },
         {
             "name": "example-page",

src/lib/schemaValidation.ts

@@ -230,6 +230,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

@@ -31,10 +31,11 @@ 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;
-    else if (ctx.config.userAgent || urlConfig.userAgent) {
-        contextOptions.userAgent = ctx.config.userAgent;
-        if (urlConfig.userAgent !== null) {
-            contextOptions.userAgent = urlConfig.userAgent;
+    if (ctx.config.userAgent || urlConfig.userAgent) {
+        if(ctx.config.userAgent !== ""){
+        contextOptions.userAgent = ctx.config.userAgent;}
+        if (urlConfig.userAgent !== "") {
+            contextOptions.userAgent = userAgent;
         }
     }
 

src/types.ts

@@ -175,7 +175,8 @@ export interface MobileConfig {
 export type WebStaticConfig = Array<{
     name: string;
     url: string;
-    waitForTimeout?: number
+    waitForTimeout?: number;
+    userAgent?: string;
 }>;
 
 export type FigmaConfigItem = {