FIX: Implement headless chrome and refactoring test

Author: kayqueGovetri

botcity/web/browsers/undetected_chrome.py

@@ -55,13 +55,10 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
 
     chrome_options.add_argument("--disable-blink-features=AutomationControlled")
 
-    # Disable banner for Browser being remote-controlled
-    # chrome_options.add_experimental_option("excludeSwitches", ["enable-automation"])
-    # chrome_options.add_experimental_option('useAutomationExtension', False)
-
     if headless:
         chrome_options.add_argument('--headless')
         chrome_options.add_argument('--headless=new')
+        chrome_options.add_argument('--headless=chrome')
         chrome_options.add_argument("--disable-gpu")
         chrome_options.add_argument("--hide-scrollbars")
         chrome_options.add_argument("--mute-audio")

tests/test_browser.py

@@ -43,7 +43,7 @@ def test_display_size(web: WebBot):
     web.set_screen_resolution(1280, 720)
     (w, h) = web.display_size()
 
-    assert w == 1280 or w == 1233 or w == 1223 or w == 1028
+    assert w in [1280, 1233, 1223, 1028]
 
 
 def test_javascript(web: WebBot):