Make cookie collection fail safe

percy · Sep 27, 2024 · e6006f8 · e6006f8
1 parent a1869b6
commit e6006f8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/main/java/io/percy/selenium/Percy.java b/src/main/java/io/percy/selenium/Percy.java
@@ -197,15 +197,20 @@ public JSONObject snapshot(String name, Map<String, Object> options) {
         try {
             JavascriptExecutor jse = (JavascriptExecutor) driver;
             jse.executeScript(fetchPercyDOM());
-            Set<Cookie> cookies = driver.manage().getCookies();
+            Set<Cookie> cookies = new HashSet<>();
+            try {
+                cookies = driver.manage().getCookies();
+            } catch(Exception e) {
+                log("Cookie collection failed " + e.getMessage(), "debug");
+            }
             if (isCaptureResponsiveDOM(options)) {
                 domSnapshot = captureResponsiveDom(driver, cookies, options);
             } else {
                 domSnapshot.add(getSerializedDOM(jse, cookies, options));
             }
         } catch (WebDriverException e) {
             // For some reason, the execution in the browser failed.
-            if (PERCY_DEBUG) { log(e.getMessage()); }
+            log(e.getMessage(), "debug");
         }
 
         return postSnapshot(domSnapshot, name, driver.getCurrentUrl(), options);