Refactord the Browser Management and created Screenshots Utils class.

iamcharankumar · iamcharankumar · commit 48745346293b · 2025-08-09T16:59:07.000+05:30
diff --git a/src/main/java/io/swaglabs/portal/qa/browsermanager/BrowserFactory.java b/src/main/java/io/swaglabs/portal/qa/browsermanager/BrowserFactory.java
@@ -1,9 +1,11 @@
 package io.swaglabs.portal.qa.browsermanager;
 
+import io.swaglabs.portal.qa.constants.WebPortalConstants;
+
 public class BrowserFactory {
 
-    public IBrowser createBrowser(String browserName) {
-        return switch (BrowserName.fromString(browserName)) {
+    public IBrowser createBrowser() {
+        return switch (BrowserName.fromString(WebPortalConstants.BROWSER)) {
             case FIREFOX -> new FirefoxBrowser();
             case WEBKIT -> new WebkitBrowser();
             case MS_EDGE -> new MsEdgeBrowser();
diff --git a/src/main/java/io/swaglabs/portal/qa/browsermanager/BrowserManager.java b/src/main/java/io/swaglabs/portal/qa/browsermanager/BrowserManager.java
@@ -11,10 +11,8 @@ public class BrowserManager implements IBrowserManager<Page> {
 
     @Override
     public Page getBrowserPage(Playwright playwright) {
-        String browserName = WebPortalConstants.BROWSER;
-        String runMode = WebPortalConstants.RUN_MODE;
-        boolean isHeadless = runMode.equals("headless");
-        BrowserContext browserContext = new BrowserFactory().createBrowser(browserName).createSession(playwright, isHeadless);
+        boolean isHeadless = "headless".equals(WebPortalConstants.RUN_MODE);
+        BrowserContext browserContext = new BrowserFactory().createBrowser().createSession(playwright, isHeadless);
         Objects.requireNonNull(browserContext, "Playwright Browser Context is null!");
         return browserContext.newPage();
     }
diff --git a/src/main/java/io/swaglabs/portal/qa/browsermanager/BrowserName.java b/src/main/java/io/swaglabs/portal/qa/browsermanager/BrowserName.java
@@ -1,6 +1,6 @@
 package io.swaglabs.portal.qa.browsermanager;
 
-import io.swaglabs.portal.qa.exceptions.UtilsException;
+import io.swaglabs.portal.qa.exceptions.WebUtilsException;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 
@@ -20,6 +20,6 @@ public enum BrowserName {
     public static BrowserName fromString(String browserName) {
         return Arrays.stream(BrowserName.values())
                 .filter(browserType -> browserType.getBrowserType().equalsIgnoreCase(browserName))
-                .findFirst().orElseThrow(() -> new UtilsException("Unknown browser: " + browserName));
+                .findFirst().orElseThrow(() -> new WebUtilsException("Unknown browser: " + browserName));
     }
 }
diff --git a/src/main/java/io/swaglabs/portal/qa/browsermanager/ChromeBrowser.java b/src/main/java/io/swaglabs/portal/qa/browsermanager/ChromeBrowser.java
@@ -10,10 +10,9 @@ public class ChromeBrowser implements IBrowser {
 
     @Override
     public BrowserContext createSession(Playwright playwright, boolean isHeadless) {
-        return playwright.chromium().launch(new BrowserType.LaunchOptions()
-                        .setHeadless(isHeadless))
-                .newContext(new Browser.NewContextOptions()
-                        .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)
-                        .setTimezoneId(WebPortalConstants.TIME_ZONE));
+        BrowserType.LaunchOptions chromeLaunchOptions = WebPortalConstants.BROWSER_LAUNCH_OPTIONS.setHeadless(isHeadless);
+        return playwright.chromium().launch(chromeLaunchOptions).newContext(new Browser.NewContextOptions()
+                .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)
+                .setTimezoneId(WebPortalConstants.TIME_ZONE));
     }
 }
diff --git a/src/main/java/io/swaglabs/portal/qa/browsermanager/FirefoxBrowser.java b/src/main/java/io/swaglabs/portal/qa/browsermanager/FirefoxBrowser.java
@@ -9,10 +9,9 @@
 public class FirefoxBrowser implements IBrowser {
     @Override
     public BrowserContext createSession(Playwright playwright, boolean isHeadless) {
-        return playwright.firefox().launch(new BrowserType.LaunchOptions()
-                        .setHeadless(isHeadless))
-                .newContext(new Browser.NewContextOptions()
-                        .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)
-                        .setTimezoneId(WebPortalConstants.TIME_ZONE));
+        BrowserType.LaunchOptions firefoxLaunchOptions = WebPortalConstants.BROWSER_LAUNCH_OPTIONS.setHeadless(isHeadless);
+        return playwright.firefox().launch(firefoxLaunchOptions).newContext(new Browser.NewContextOptions()
+                .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)
+                .setTimezoneId(WebPortalConstants.TIME_ZONE));
     }
 }
diff --git a/src/main/java/io/swaglabs/portal/qa/browsermanager/MsEdgeBrowser.java b/src/main/java/io/swaglabs/portal/qa/browsermanager/MsEdgeBrowser.java
@@ -9,10 +9,11 @@
 public class MsEdgeBrowser implements IBrowser {
     @Override
     public BrowserContext createSession(Playwright playwright, boolean isHeadless) {
-        return playwright.chromium().launch(new BrowserType.LaunchOptions()
-                        .setHeadless(isHeadless).setChannel(BrowserName.MS_EDGE.getBrowserType()))
-                .newContext(new Browser.NewContextOptions()
-                        .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)
-                        .setTimezoneId(WebPortalConstants.TIME_ZONE));
+        BrowserType.LaunchOptions msedgeLaunchOptions = WebPortalConstants.BROWSER_LAUNCH_OPTIONS
+                .setHeadless(isHeadless)
+                .setChannel(BrowserName.MS_EDGE.getBrowserType());
+        return playwright.chromium().launch(msedgeLaunchOptions).newContext(new Browser.NewContextOptions()
+                .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)
+                .setTimezoneId(WebPortalConstants.TIME_ZONE));
     }
 }
diff --git a/src/main/java/io/swaglabs/portal/qa/browsermanager/WebkitBrowser.java b/src/main/java/io/swaglabs/portal/qa/browsermanager/WebkitBrowser.java
@@ -9,10 +9,9 @@
 public class WebkitBrowser implements IBrowser {
     @Override
     public BrowserContext createSession(Playwright playwright, boolean isHeadless) {
-        return playwright.webkit().launch(new BrowserType.LaunchOptions()
-                        .setHeadless(isHeadless))
-                .newContext(new Browser.NewContextOptions()
-                        .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)
-                        .setTimezoneId(WebPortalConstants.TIME_ZONE));
+        BrowserType.LaunchOptions webkitLaunchOptions = new BrowserType.LaunchOptions().setHeadless(isHeadless);
+        return playwright.webkit().launch(webkitLaunchOptions).newContext(new Browser.NewContextOptions()
+                .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)
+                .setTimezoneId(WebPortalConstants.TIME_ZONE));
     }
 }
diff --git a/src/main/java/io/swaglabs/portal/qa/commons/WebBasePage.java b/src/main/java/io/swaglabs/portal/qa/commons/WebBasePage.java
@@ -2,18 +2,13 @@
 
 import com.microsoft.playwright.Locator;
 import com.microsoft.playwright.Page;
-import io.swaglabs.portal.qa.constants.WebPortalConstants;
 import io.swaglabs.portal.qa.exceptions.WebPageException;
 import io.swaglabs.portal.qa.locators.Locators;
-import io.swaglabs.portal.qa.screenshotsmanager.ElementScreenshotStrategy;
-import io.swaglabs.portal.qa.screenshotsmanager.ScreenshotContext;
 
 import java.util.Objects;
 
 public abstract class WebBasePage {
 
-    private static final String ELEMENT_SCREENSHOT_FILE_LOCATION = WebPortalConstants.SCREENSHOT_FILE_LOCATION + "/elements/"
-            + WebPortalConstants.BROWSER + "_" + WebPortalConstants.RUN_MODE + "_Element_";
     protected Page basePage;
     protected Locators locators;
 
@@ -53,9 +48,4 @@ private String extractText(Locator locator, String errorMessage) {
         Objects.requireNonNull(locator, errorMessage);
         return locator.textContent().trim();
     }
-
-    protected void takeElementScreenshot(Locator locator, String fileName) {
-        ScreenshotContext screenshotContext = new ScreenshotContext(new ElementScreenshotStrategy(locator));
-        screenshotContext.captureScreenshot(basePage, ELEMENT_SCREENSHOT_FILE_LOCATION + fileName + WebPortalConstants.IMAGE_FORMAT);
-    }
 }
diff --git a/src/main/java/io/swaglabs/portal/qa/constants/WebPortalConstants.java b/src/main/java/io/swaglabs/portal/qa/constants/WebPortalConstants.java
@@ -1,9 +1,14 @@
 package io.swaglabs.portal.qa.constants;
 
+import com.microsoft.playwright.BrowserType;
+import io.swaglabs.portal.qa.screenshotsmanager.ScreenshotContext;
+import io.swaglabs.portal.qa.screenshotsmanager.ScreenshotsUtils;
 import io.swaglabs.portal.qa.utils.WebConfigLoader;
 import lombok.AccessLevel;
 import lombok.NoArgsConstructor;
 
+import java.util.List;
+
 @NoArgsConstructor(access = AccessLevel.PRIVATE)
 public final class WebPortalConstants {
 
@@ -18,9 +23,18 @@ public final class WebPortalConstants {
     public static final String BROWSER = System.getProperty("browser");
     public static final String TIME_ZONE = "Asia/Kolkata";
 
+    // Web Browser resolution
     public static final int SCREEN_WIDTH = 1920;
     public static final int SCREEN_HEIGHT = 1080;
 
     public static final String SCREENSHOT_FILE_LOCATION = "./target/screenshots/";
     public static final String IMAGE_FORMAT = ".png";
+
+    // Browser Type Launch Options Config
+    public static final BrowserType.LaunchOptions BROWSER_LAUNCH_OPTIONS = new BrowserType.LaunchOptions()
+            .setArgs(List.of("--window-position=0,0"));
+
+    // SCREENSHOT STRATEGIES
+    public static final ScreenshotsUtils SCREENSHOTS_UTILS = ScreenshotsUtils.getInstance();
+    public static final ScreenshotContext FULL_PAGE_SCREENSHOT = SCREENSHOTS_UTILS.getFullPageScreenshotContext();
 }
diff --git a/src/main/java/io/swaglabs/portal/qa/exceptions/UtilsException.java b/src/main/java/io/swaglabs/portal/qa/exceptions/UtilsException.java
diff --git a/src/main/java/io/swaglabs/portal/qa/exceptions/WebUtilsException.java b/src/main/java/io/swaglabs/portal/qa/exceptions/WebUtilsException.java
@@ -0,0 +1,8 @@
+package io.swaglabs.portal.qa.exceptions;
+
+public class WebUtilsException extends RuntimeException {
+
+    public WebUtilsException(String errorMessage) {
+        super(errorMessage);
+    }
+}
diff --git a/src/main/java/io/swaglabs/portal/qa/pages/SwagLabsProductPage.java b/src/main/java/io/swaglabs/portal/qa/pages/SwagLabsProductPage.java
@@ -3,6 +3,7 @@
 import com.microsoft.playwright.Locator;
 import com.microsoft.playwright.Page;
 import com.microsoft.playwright.options.AriaRole;
+import io.swaglabs.portal.qa.constants.WebPortalConstants;
 
 public final class SwagLabsProductPage extends SwagLabsBasePage {
 
@@ -17,7 +18,7 @@ public String getProductNameText() {
     public String getProductPriceText() {
         Locator productPrice = locators.getPageLocator(".inventory_details_price");
         String productPriceText = getTextContent(productPrice);
-        takeElementScreenshot(productPrice, productPriceText);
+        WebPortalConstants.SCREENSHOTS_UTILS.takeElementScreenshot(basePage, productPrice, productPriceText);
         return productPriceText;
     }
 
@@ -32,7 +33,7 @@ public boolean isProductAddedToCart() {
 
     public boolean isShoppingCartClicked() {
         Locator shoppingCart = locators.getPageLocator(".shopping_cart_link");
-        takeElementScreenshot(shoppingCart, "shoppingCart");
+        WebPortalConstants.SCREENSHOTS_UTILS.takeElementScreenshot(basePage, shoppingCart, "shoppingCart");
         clickElement(shoppingCart);
         return true;
     }
diff --git a/src/main/java/io/swaglabs/portal/qa/screenshotsmanager/ScreenshotsUtils.java b/src/main/java/io/swaglabs/portal/qa/screenshotsmanager/ScreenshotsUtils.java
@@ -0,0 +1,42 @@
+package io.swaglabs.portal.qa.screenshotsmanager;
+
+import com.microsoft.playwright.Locator;
+import com.microsoft.playwright.Page;
+import io.swaglabs.portal.qa.constants.WebPortalConstants;
+import lombok.AccessLevel;
+import lombok.NoArgsConstructor;
+
+import java.util.Objects;
+
+@NoArgsConstructor(access = AccessLevel.PRIVATE)
+public class ScreenshotsUtils {
+
+    private static final String ELEMENT_SCREENSHOT_FILE_LOCATION = WebPortalConstants.SCREENSHOT_FILE_LOCATION
+            + "/elements/" + WebPortalConstants.BROWSER + "_" + WebPortalConstants.RUN_MODE + "_Element_";
+    private static ScreenshotsUtils instance;
+
+    public static ScreenshotsUtils getInstance() {
+        if (instance == null) {
+            synchronized (ScreenshotsUtils.class) {
+                if (instance == null) {
+                    instance = new ScreenshotsUtils();
+                }
+            }
+        }
+        return instance;
+    }
+
+    public ScreenshotContext getFullPageScreenshotContext() {
+        return new ScreenshotContext(new FullPageScreenshotStrategy());
+    }
+
+    public ScreenshotContext getElementScreenshotContext(Locator locator) {
+        Objects.requireNonNull(locator, "Locator cannot be null for taking element screenshot");
+        return new ScreenshotContext(new ElementScreenshotStrategy(locator));
+    }
+
+    public void takeElementScreenshot(Page page, Locator locator, String fileName) {
+        getElementScreenshotContext(locator).captureScreenshot(page,
+                ELEMENT_SCREENSHOT_FILE_LOCATION + fileName + WebPortalConstants.IMAGE_FORMAT);
+    }
+}
diff --git a/src/main/java/io/swaglabs/portal/qa/utils/PropertiesUtils.java b/src/main/java/io/swaglabs/portal/qa/utils/PropertiesUtils.java
@@ -1,6 +1,6 @@
 package io.swaglabs.portal.qa.utils;
 
-import io.swaglabs.portal.qa.exceptions.UtilsException;
+import io.swaglabs.portal.qa.exceptions.WebUtilsException;
 import lombok.AccessLevel;
 import lombok.NoArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -23,7 +23,7 @@ public static Properties loadProperties(String fileName) {
             properties = new Properties();
             properties.load(fileInputStream);
         } catch (IOException e) {
-            throw new UtilsException("Properties File failed loading..." + e.getMessage());
+            throw new WebUtilsException("Properties File failed loading..." + e.getMessage());
         }
         return properties;
     }
diff --git a/src/test/java/io/swaglabs/portal/qa/commons/WebBaseTest.java b/src/test/java/io/swaglabs/portal/qa/commons/WebBaseTest.java
@@ -24,7 +24,6 @@ public abstract class WebBaseTest {
     private static final ThreadLocal<Playwright> PLAYWRIGHT = new ThreadLocal<>();
     private static BrowserManager browserManager;
 
-
     @BeforeSuite(alwaysRun = true)
     public void setUp() {
         browserManager = new BrowserManager();
@@ -37,6 +36,13 @@ public void init(Method method) {
         page.set(browserManager.getBrowserPage(PLAYWRIGHT.get()));
         log.info("Browser has been set.");
         WebTestListeners.setPage(page.get());
+        if (isChromiumBrowser()) {
+            initializeCdpSession();
+        }
+        trackPerformanceMetrics(method.getName());
+    }
+
+    private void initializeCdpSession() {
         if (isChromiumBrowser()) {
             CdpUtils.initializeCdpSession(page.get());
             CdpUtils.enableCdpSession();
@@ -51,8 +57,11 @@ public void init(Method method) {
             CdpUtils.logPageLoadCompletion();
             CdpUtils.logPageNavigatedWithinDocument();
         }
-        PerformanceUtils.evaluatePageLoadTime(page.get(), method.getName());
-        PerformanceUtils.evaluateDomContentLoadTime(page.get(), method.getName());
+    }
+
+    private void trackPerformanceMetrics(String methodName) {
+        PerformanceUtils.evaluatePageLoadTime(page.get(), methodName);
+        PerformanceUtils.evaluateDomContentLoadTime(page.get(), methodName);
     }
 
     @AfterMethod(alwaysRun = true)
diff --git a/src/test/java/io/swaglabs/portal/qa/listeners/WebTestListeners.java b/src/test/java/io/swaglabs/portal/qa/listeners/WebTestListeners.java
@@ -2,8 +2,6 @@
 
 import com.microsoft.playwright.Page;
 import io.swaglabs.portal.qa.constants.WebPortalConstants;
-import io.swaglabs.portal.qa.screenshotsmanager.FullPageScreenshotStrategy;
-import io.swaglabs.portal.qa.screenshotsmanager.ScreenshotContext;
 import lombok.extern.slf4j.Slf4j;
 import org.testng.*;
 
@@ -73,8 +71,7 @@ private void takeScreenshot(ITestResult testResult) {
         String filePath = Paths.get(dirPath, fileName).toString();
         try {
             Files.createDirectories(Paths.get(dirPath));
-            ScreenshotContext screenshotContext = new ScreenshotContext(new FullPageScreenshotStrategy());
-            screenshotContext.captureScreenshot(currentPage, filePath);
+            WebPortalConstants.FULL_PAGE_SCREENSHOT.captureScreenshot(currentPage, filePath);
         } catch (IOException e) {
             log.error("Screenshot failed for {}: {}", testName, e.getMessage());
         }

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`package io.swaglabs.portal.qa.browsermanager;`
`2`	`2`
`3`		`-import io.swaglabs.portal.qa.exceptions.UtilsException;`
	`3`	`+import io.swaglabs.portal.qa.exceptions.WebUtilsException;`
`4`	`4`	`import lombok.AllArgsConstructor;`
`5`	`5`	`import lombok.Getter;`
`6`	`6`
`@@ -20,6 +20,6 @@ public enum BrowserName {`
`20`	`20`	`public static BrowserName fromString(String browserName) {`
`21`	`21`	`return Arrays.stream(BrowserName.values())`
`22`	`22`	`.filter(browserType -> browserType.getBrowserType().equalsIgnoreCase(browserName))`
`23`		`- .findFirst().orElseThrow(() -> new UtilsException("Unknown browser: " + browserName));`
	`23`	`+ .findFirst().orElseThrow(() -> new WebUtilsException("Unknown browser: " + browserName));`
`24`	`24`	`}`
`25`	`25`	`}`
Original file line number	Diff line number	Diff line change
`@@ -10,10 +10,9 @@ public class ChromeBrowser implements IBrowser {`
`10`	`10`
`11`	`11`	`@Override`
`12`	`12`	`public BrowserContext createSession(Playwright playwright, boolean isHeadless) {`
`13`		`- return playwright.chromium().launch(new BrowserType.LaunchOptions()`
`14`		`- .setHeadless(isHeadless))`
`15`		`- .newContext(new Browser.NewContextOptions()`
`16`		`- .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)`
`17`		`- .setTimezoneId(WebPortalConstants.TIME_ZONE));`
	`13`	`+ BrowserType.LaunchOptions chromeLaunchOptions = WebPortalConstants.BROWSER_LAUNCH_OPTIONS.setHeadless(isHeadless);`
	`14`	`+ return playwright.chromium().launch(chromeLaunchOptions).newContext(new Browser.NewContextOptions()`
	`15`	`+ .setViewportSize(WebPortalConstants.SCREEN_WIDTH, WebPortalConstants.SCREEN_HEIGHT)`
	`16`	`+ .setTimezoneId(WebPortalConstants.TIME_ZONE));`
`18`	`17`	`}`
`19`	`18`	`}`