[py] Update tests and add aarch64

Author: cgoldberg

py/selenium/webdriver/common/selenium_manager.py

@@ -84,10 +84,13 @@ def _get_binary() -> Path:
                 ("cygwin", "arm64"): "windows/selenium-manager-arm64.exe",
                 ("linux", "x86_64"): "linux/selenium-manager",
                 ("linux", "arm64"): "linux/selenium-manager-arm64",
+                ("linux", "aarch64"): "linux/selenium-manager-arm64",
                 ("freebsd", "x86_64"): "linux/selenium-manager",
                 ("freebsd", "arm64"): "linux/selenium-manager-arm64",
+                ("freebsd", "aarch64"): "linux/selenium-manager-arm64",
                 ("openbsd", "x86_64"): "linux/selenium-manager",
                 ("openbsd", "arm64"): "linux/selenium-manager-arm64",
+                ("openbsd", "aarch64"): "linux/selenium-manager-arm64",
             }
 
             arch = "any" if sys.platform == "darwin" else platform.machine().lower()

py/test/selenium/webdriver/common/selenium_manager_tests.py

@@ -53,22 +53,38 @@ def test_uses_environment_variable(monkeypatch):
 
 def test_uses_windows(monkeypatch):
     monkeypatch.setattr(sys, "platform", "win32")
+    monkeypatch.setattr("platform.machine", lambda: "AMD64")
     binary = SeleniumManager()._get_binary()
 
     project_root = Path(selenium.__file__).parent.parent
     assert binary == project_root.joinpath("selenium/webdriver/common/windows/selenium-manager.exe")
 
 
+def test_uses_windows_arm(monkeypatch):
+    monkeypatch.setattr(sys, "platform", "win32")
+    monkeypatch.setattr("platform.machine", lambda: "arm64")
+    binary = SeleniumManager()._get_binary()
+
+    project_root = Path(selenium.__file__).parent.parent
+    assert binary == project_root.joinpath("selenium/webdriver/common/windows/selenium-manager-arm64.exe")
+
+
 def test_uses_linux(monkeypatch):
     monkeypatch.setattr(sys, "platform", "linux")
+    monkeypatch.setattr("platform.machine", lambda: "x86_64")
 
-    if platform.machine() == "arm64":
-        with pytest.raises(WebDriverException, match="Unsupported platform/architecture combination: linux/arm64"):
-            SeleniumManager()._get_binary()
-    else:
-        binary = SeleniumManager()._get_binary()
-        project_root = Path(selenium.__file__).parent.parent
-        assert binary == project_root.joinpath("selenium/webdriver/common/linux/selenium-manager")
+    binary = SeleniumManager()._get_binary()
+    project_root = Path(selenium.__file__).parent.parent
+    assert binary == project_root.joinpath("selenium/webdriver/common/linux/selenium-manager")
+
+
+def test_uses_linux(monkeypatch):
+    monkeypatch.setattr(sys, "platform", "linux")
+    monkeypatch.setattr("platform.machine", lambda: "arm64")
+
+    binary = SeleniumManager()._get_binary()
+    project_root = Path(selenium.__file__).parent.parent
+    assert binary == project_root.joinpath("selenium/webdriver/common/linux/selenium-manager-arm64")
 
 
 def test_uses_mac(monkeypatch):