FIX: implement tmp folder and headless chrome to test print pdf

Author: kayqueGovetri

botcity/web/bot.py

@@ -25,6 +25,7 @@
 from selenium.webdriver.remote.webelement import WebElement
 from selenium.webdriver.support.wait import WebDriverWait, TimeoutException, NoSuchElementException
 from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.print_page_options import PrintOptions
 
 from . import config, cv2find, compat
 from .browsers import BROWSER_CONFIGS, Browser, PageLoadStrategy
@@ -266,7 +267,7 @@ def check_driver():
     def _get_parameters_to_driver(self):
         if self.browser == Browser.UNDETECTED_CHROME:
             return {"driver_executable_path": self.driver_path, "options": self.options,
-                    "desired_capabilities": self.capabilities}
+                    "desired_capabilities": self.capabilities, "headless": True}
         if compat.version_selenium_is_larger_than_four():
             return {"options": self.options, "service": self._get_service()}
 
@@ -1137,7 +1138,13 @@ def print_pdf(self, path=None, print_options=None):
                 'marginTop': 0,
                 'marginBottom': 0
             }
-        data = self._webdriver_command("print", print_options)
+        print_options = PrintOptions()
+        print_options.page_ranges = ['1-2']
+        print_options.margin_top = 0
+        print_options.margin_bottom = 0
+        print_options.background = True
+        print_options.orientation = 'landscape'
+        data = self._driver.print_page(print_options)
         bytes_file = base64.b64decode(data)
         if not path:
             path = default_path

botcity/web/browsers/chrome.py

@@ -54,10 +54,11 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
     # Disable banner for Browser being remote-controlled
     chrome_options.add_experimental_option("excludeSwitches", ["enable-automation"])
     chrome_options.add_experimental_option('useAutomationExtension', False)
-
+    chrome_options.binary_location = '/usr/bin/google-chrome-beta'
     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

@@ -284,10 +284,10 @@ def test_set_current_element(web: WebBot):
     assert result['data'] == ['Left2'] or result['data'] == ['Left']
 
 
-@pytest.mark.xfail(reason="Unknown chrome/undetected error to be investigated.")
-def test_print_pdf(web: WebBot):
+# @pytest.mark.xfail(reason="Unknown chrome/undetected error to be investigated.")
+def test_print_pdf(web: WebBot, tmp_folder):
     web.browse(conftest.INDEX_PAGE)
-    pdf = web.print_pdf(path=os.path.join(conftest.PROJECT_DIR, 'page.pdf'))
+    pdf = web.print_pdf(path=os.path.join(tmp_folder, 'page.pdf'))
 
     assert os.path.exists(pdf)
     os.remove(pdf)