Keep the Content Security Policy on headless Chrome

Author: mdmintz

seleniumbase/core/browser_launcher.py

@@ -94,7 +94,7 @@ def _add_chrome_disable_csp_extension(chrome_options):
 
 
 def _set_chrome_options(
-        downloads_path, proxy_string, proxy_auth,
+        downloads_path, headless, proxy_string, proxy_auth,
         proxy_user, proxy_pass, user_agent):
     chrome_options = webdriver.ChromeOptions()
     prefs = {
@@ -119,7 +119,9 @@ def _set_chrome_options(
     chrome_options.add_argument("--disable-single-click-autofill")
     chrome_options.add_argument("--disable-translate")
     chrome_options.add_argument("--disable-web-security")
-    if settings.DISABLE_CONTENT_SECURITY_POLICY:
+    if settings.DISABLE_CONTENT_SECURITY_POLICY and not headless:
+        # Headless Chrome doesn't support extensions, which are required
+        # for disabling the Content Security Policy on Chrome
         chrome_options = _add_chrome_disable_csp_extension(chrome_options)
     if proxy_string:
         if proxy_auth:
@@ -261,7 +263,7 @@ def get_remote_driver(
         desired_caps = capabilities_parser.get_desired_capabilities(cap_file)
     if browser_name == constants.Browser.GOOGLE_CHROME:
         chrome_options = _set_chrome_options(
-            downloads_path, proxy_string, proxy_auth,
+            downloads_path, headless, proxy_string, proxy_auth,
             proxy_user, proxy_pass, user_agent)
         if headless:
             if not proxy_auth:
@@ -472,7 +474,7 @@ def get_local_driver(
     elif browser_name == constants.Browser.GOOGLE_CHROME:
         try:
             chrome_options = _set_chrome_options(
-                downloads_path, proxy_string, proxy_auth,
+                downloads_path, headless, proxy_string, proxy_auth,
                 proxy_user, proxy_pass, user_agent)
             if headless:
                 # Headless Chrome doesn't support extensions, which are