Merge pull request #1602 from seleniumbase/mac-m1-and-m2-compatibility

Mac M1 and M2 compatibility

Author: mdmintz

examples/boilerplates/samples/google_test.py

@@ -1,4 +1,4 @@
-""" google.com example test that uses page objects """
+"""google.com example test that uses page objects"""
 
 from seleniumbase import BaseCase
 from .google_objects import HomePage, ResultsPage
@@ -7,7 +7,9 @@
 class GoogleTests(BaseCase):
     def test_google_dot_com(self):
         self.open("https://google.com/ncr")
-        self.type(HomePage.search_box, "github")
+        self.sleep(0.1)
+        self.save_screenshot_to_logs()  # In "./latest_logs/" folder.
+        self.type(HomePage.search_box, "github.com")
         self.assert_element(HomePage.search_button)
         self.assert_element(HomePage.feeling_lucky_button)
         self.click(HomePage.search_button)

examples/boilerplates/samples/test_page_objects.py

@@ -15,6 +15,7 @@ def click_search_result(self, sb, content):
 
 class SeleniumBaseIOPage:
     def do_search_and_click(self, sb, search_term):
+        sb.sleep(0.05)
         sb.type('form[name="search"] input', search_term)
         sb.click("li.md-search-result__item h1:contains(%s)" % search_term)
 

examples/swag_labs_user_tests.py

@@ -7,6 +7,8 @@ def login_to_swag_labs(self, username="standard_user"):
         """Login to Swag Labs and verify success."""
         url = "https://www.saucedemo.com"
         self.open(url)
+        self.wait_for_element("div.login_logo")
+        self.wait_for_element("div.bot_column")
         if username not in self.get_text("#login_credentials"):
             self.fail("Invalid user for login: %s" % username)
         self.type("#user-name", username)

examples/test_login.py

@@ -5,6 +5,8 @@
 class SwagLabsLoginTests(BaseCase):
     def login_to_swag_labs(self):
         self.open("https://www.saucedemo.com")
+        self.wait_for_element("div.login_logo")
+        self.wait_for_element("div.bot_column")
         self.type("#user-name", "standard_user")
         self.type("#password", "secret_sauce")
         self.click('input[type="submit"]')

examples/visual_testing/layout_test.py

@@ -6,11 +6,13 @@ def test_applitools_layout_change(self):
         self.open("https://applitools.com/helloworld?diff1")
         self.wait_for_element('a[href="?diff1"]')
         print('\nCreating baseline in "visual_baseline" folder.')
+        self.sleep(0.05)
         self.check_window(name="helloworld", baseline=True)
         # Click a button that changes the text of an element
         # (Text changes do not impact visual comparisons)
+        self.sleep(0.05)
         self.click('a[href="?diff1"]')
-        self.wait_for_ready_state_complete()
+        self.sleep(0.05)
         # Verify html tags match the baseline
         self.check_window(name="helloworld", level=1)
         # Verify html tags and attribute names match the baseline

requirements.txt

@@ -10,7 +10,7 @@ tomli>=1.2.3;python_version>="3.6" and python_version<"3.7"
 tomli>=2.0.1;python_version>="3.7"
 tqdm>=4.64.1
 wheel>=0.37.1;python_version<"3.7"
-wheel>=0.38.2;python_version>="3.7"
+wheel>=0.38.3;python_version>="3.7"
 attrs>=21.4.0;python_version<"3.6"
 attrs>=22.1.0;python_version>="3.6"
 PyYAML>=6.0;python_version>="3.6"

seleniumbase/__version__.py

@@ -1,2 +1,2 @@
 # seleniumbase package
-__version__ = "4.8.1"
+__version__ = "4.8.2"

seleniumbase/console_scripts/sb_install.py

@@ -117,7 +117,7 @@ def requests_get_with_retry(url):
     return response
 
 
-def main(override=None):
+def main(override=None, intel_for_uc=None):
     if override:
         if override == "chromedriver":
             sys.argv = ["seleniumbase", "get", "chromedriver"]
@@ -229,7 +229,10 @@ def main(override=None):
             else:
                 invalid_run_command()
         if "darwin" in sys_plat:
-            file_name = "chromedriver_mac64.zip"
+            if "arm" in platform.processor().lower() and not intel_for_uc:
+                file_name = "chromedriver_mac_arm64.zip"
+            else:
+                file_name = "chromedriver_mac64.zip"
         elif "linux" in sys_plat:
             file_name = "chromedriver_linux64.zip"
         elif "win32" in sys_plat or "win64" in sys_plat or "x64" in sys_plat:
@@ -393,7 +396,10 @@ def main(override=None):
             file_name = "edgedriver_win32.zip"
             suffix = "WINDOWS"
         elif "darwin" in sys_plat:
-            file_name = "edgedriver_mac64.zip"
+            if "arm" in platform.processor().lower():
+                file_name = "edgedriver_mac64_m1.zip"
+            else:
+                file_name = "edgedriver_mac64.zip"
             suffix = "MACOS"
         elif "linux" in sys_plat:
             file_name = "edgedriver_linux64.zip"
@@ -631,7 +637,23 @@ def main(override=None):
                     if os.path.exists(new_file):
                         os.remove(new_file)  # Technically the old file now
             print("Extracting %s from %s ..." % (contents, file_name))
-            zip_ref.extractall(downloads_folder)
+            if (
+                intel_for_uc
+                and "darwin" in sys_plat
+                and "arm" in platform.processor().lower()
+            ):
+                f_name = "uc_driver"
+                new_file = os.path.join(downloads_folder, f_name)
+                if os.path.exists(new_file):
+                    os.remove(new_file)
+                zipinfos = zip_ref.infolist()
+                for zipinfo in zipinfos:
+                    if zipinfo.filename == "chromedriver":
+                        zipinfo.filename = "uc_driver"
+                        zip_ref.extract(zipinfo, downloads_folder)
+                contents = zip_ref.namelist()
+            else:
+                zip_ref.extractall(downloads_folder)
             zip_ref.close()
             os.remove(zip_file_path)
             print("%sUnzip Complete!%s\n" % (c2, cr))

seleniumbase/console_scripts/sb_mkdir.py

@@ -612,7 +612,9 @@ def main():
     data.append("class GoogleTests(BaseCase):")
     data.append("    def test_google_dot_com(self):")
     data.append('        self.open("https://google.com/ncr")')
-    data.append('        self.type(HomePage.search_box, "github")')
+    data.append("        self.sleep(0.1)")
+    data.append("        self.save_screenshot_to_logs()")
+    data.append('        self.type(HomePage.search_box, "github.com")')
     data.append("        self.assert_element(HomePage.search_button)")
     data.append("        self.assert_element(HomePage.feeling_lucky_button)")
     data.append("        self.click(HomePage.search_button)")

seleniumbase/core/browser_launcher.py

@@ -1,6 +1,7 @@
 import fasteners
 import logging
 import os
+import platform
 import re
 import shutil
 import subprocess
@@ -49,6 +50,9 @@
 PROXY_DIR_PATH = proxy_helper.PROXY_DIR_PATH
 PROXY_DIR_LOCK = proxy_helper.PROXY_DIR_LOCK
 PLATFORM = sys.platform
+IS_ARM_MAC = False
+if PLATFORM.endswith("darwin") and "arm" in platform.processor().lower():
+    IS_ARM_MAC = True
 IS_WINDOWS = False
 LOCAL_CHROMEDRIVER = None
 LOCAL_GECKODRIVER = None
@@ -130,6 +134,25 @@ def chromedriver_on_path():
     return None
 
 
+def get_uc_driver_version():
+    uc_driver_version = None
+    if os.path.exists(LOCAL_UC_DRIVER):
+        try:
+            output = subprocess.check_output(
+                "%s --version" % LOCAL_UC_DRIVER, shell=True
+            )
+            if IS_WINDOWS:
+                output = output.decode("latin1")
+            else:
+                output = output.decode("utf-8")
+            output = output.split(" ")[1].split(".")[0]
+            if int(output) >= 72:
+                uc_driver_version = output
+        except Exception:
+            pass
+    return uc_driver_version
+
+
 def edgedriver_on_path():
     return os.path.exists(LOCAL_EDGEDRIVER)
 
@@ -2501,13 +2524,25 @@ def get_local_driver(
                 except Exception:
                     pass
             disable_build_check = False
+            uc_driver_version = None
+            if IS_ARM_MAC:
+                uc_driver_version = get_uc_driver_version()
             if (
                 LOCAL_CHROMEDRIVER
                 and os.path.exists(LOCAL_CHROMEDRIVER)
                 and (
                     use_version == driver_version
                     or use_version == "latest"
                 )
+                and (
+                    not is_using_uc(undetectable, browser_name)
+                    or not IS_ARM_MAC
+                )
+                or (
+                    is_using_uc(undetectable, browser_name)
+                    and IS_ARM_MAC
+                    and uc_driver_version == use_version
+                )
             ):
                 try:
                     make_driver_executable_if_not(LOCAL_CHROMEDRIVER)
@@ -2527,6 +2562,16 @@ def get_local_driver(
                         or driver_version != "72"
                     )
                 )
+                or (
+                    is_using_uc(undetectable, browser_name)
+                    and not os.path.exists(LOCAL_UC_DRIVER)
+                )
+                or (
+                    is_using_uc(undetectable, browser_name)
+                    and IS_ARM_MAC
+                    and use_version != "latest"
+                    and uc_driver_version != use_version
+                )
             ):
                 # chromedriver download needed in the seleniumbase/drivers dir
                 from seleniumbase.console_scripts import sb_install
@@ -2538,11 +2583,14 @@ def get_local_driver(
                     msg = "chromedriver update needed. Getting it now:"
                     if not path_chromedriver:
                         msg = "chromedriver not found. Getting it now:"
-
                     print("\nWarning: %s" % msg)
+                    intel_for_uc = False
+                    if IS_ARM_MAC and is_using_uc(undetectable, browser_name):
+                        intel_for_uc = True  # Use Intel's driver for UC Mode
                     try:
                         sb_install.main(
-                            override="chromedriver %s" % use_version
+                            override="chromedriver %s" % use_version,
+                            intel_for_uc=intel_for_uc,
                         )
                     except Exception:
                         d_latest = get_latest_chromedriver_version()
@@ -2613,27 +2661,15 @@ def get_local_driver(
                 uc_lock = fasteners.InterProcessLock(
                     constants.MultiBrowser.DRIVER_FIXING_LOCK
                 )
-                with uc_lock:  # No UC multithreaded tests
-                    uc_driver_version = None
-                    if os.path.exists(LOCAL_UC_DRIVER):
-                        try:
-                            output = subprocess.check_output(
-                                "%s --version" % LOCAL_UC_DRIVER, shell=True
-                            )
-                            if IS_WINDOWS:
-                                output = output.decode("latin1")
-                            else:
-                                output = output.decode("utf-8")
-                            output = output.split(" ")[1].split(".")[0]
-                            if int(output) >= 72:
-                                uc_driver_version = output
-                        except Exception:
-                            pass
+                with uc_lock:  # Avoid multithreaded issues
+                    uc_driver_version = get_uc_driver_version()
                     if (
                         uc_driver_version != use_version
                         and use_version != "latest"
                     ):
-                        if os.path.exists(LOCAL_CHROMEDRIVER):
+                        if IS_ARM_MAC:
+                            pass  # Already taken care of in sb_install
+                        elif os.path.exists(LOCAL_CHROMEDRIVER):
                             shutil.copyfile(
                                 LOCAL_CHROMEDRIVER, LOCAL_UC_DRIVER
                             )
@@ -2683,7 +2719,7 @@ def get_local_driver(
                                 uc_lock = fasteners.InterProcessLock(
                                     constants.MultiBrowser.DRIVER_FIXING_LOCK
                                 )
-                                with uc_lock:
+                                with uc_lock:  # Avoid multithreaded issues
                                     try:
                                         uc_path = None
                                         if os.path.exists(LOCAL_UC_DRIVER):