Merge pull request #69 from pinanks/DOT-3168

Add user agents for browser context options

Author: pinanks

src/lib/constants.ts

@@ -46,6 +46,12 @@ export default {
     EDGE_CHANNEL: 'msedge',
     WEBKIT: 'webkit',
 
+    // user agents
+    CHROME_USER_AGENT: 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.6312.107 Safari/537.3',
+    FIREFOX_USER_AGENT: 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:112.0) Gecko/20100101 Firefox/112.0',
+    SAFARI_USER_AGENT: 'Mozilla/5.0 (Macintosh; Intel Mac OS X 13_3) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.4 Safari/605.1.15',
+    EDGE_USER_AGENT: 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Safari/537.36 Edg/113.0.1774.35',
+
     // viewports
     MIN_VIEWPORT_HEIGHT: 1080,
 

src/lib/processSnapshot.ts

@@ -1,6 +1,7 @@
 import { Snapshot, Context, ProcessedSnapshot } from "../types.js";
 import { scrollToBottomAndBackToTop, getRenderViewports } from "./utils.js"
 import { chromium, Locator } from "@playwright/test"
+import constants from "./constants.js";
 
 const MAX_RESOURCE_SIZE = 5 * (1024 ** 2); // 5MB
 var ALLOWED_RESOURCES = ['document', 'stylesheet', 'image', 'media', 'font', 'other'];
@@ -12,7 +13,7 @@ export default async (snapshot: Snapshot, ctx: Context): Promise<Record<string,
     ctx.log.debug(`Processing snapshot ${snapshot.name}`);
 
     if (!ctx.browser) ctx.browser = await chromium.launch({ headless: true });
-    const context = await ctx.browser.newContext()
+    const context = await ctx.browser.newContext({userAgent: constants.CHROME_USER_AGENT})
     const page = await context.newPage();
     let cache: Record<string, any> = {};
 

src/lib/screenshot.ts

@@ -14,11 +14,16 @@ async function captureScreenshotsForConfig(
     let pageOptions = { waitUntil: process.env.SMARTUI_PAGE_WAIT_UNTIL_EVENT || 'load' };
     let ssId = name.toLowerCase().replace(/\s/g, '_');
     let context: BrowserContext;
+    let contextOptions: Record<string, any> = {};
     let page: Page;
+    if (browserName == constants.CHROME) contextOptions.userAgent = constants.CHROME_USER_AGENT;
+    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;
 
     try {
         const browser = browsers[browserName];
-        context = await browser?.newContext();
+        context = await browser?.newContext(contextOptions);
         page = await context?.newPage();
 
         await page?.goto(url.trim(), pageOptions);