[rust] Selenium Manager: Update tests to work with Linux arm64

Currently, only Firefox and Geckodriver have official support
for Linux arm64. This commit ensures that the Selenium Manager
test suite passes on this platform, by skipping tests on non-
Firefox browsers.

Author: dennisameling

rust/src/chrome.rs

@@ -383,6 +383,8 @@ impl SeleniumManager for ChromeManager {
             } else {
                 "mac64"
             }
+        } else if LINUX.is(os) && ARM64.is(arch) {
+            return Err(anyhow!("Linux arm64 is not supported yet by Google Chrome. Please try another browser."));
         } else {
             "linux64"
         };

rust/src/edge.rs

@@ -28,6 +28,7 @@ use crate::{
     DASH_DASH_VERSION, DEV, ENV_PROGRAM_FILES_X86, NIGHTLY, OFFLINE_REQUEST_ERR_MSG, REG_PV_ARG,
     REG_VERSION_ARG, STABLE,
 };
+use anyhow::anyhow;
 use anyhow::Error;
 use reqwest::Client;
 use serde::{Deserialize, Serialize};
@@ -288,6 +289,8 @@ impl SeleniumManager for EdgeManager {
             } else {
                 "mac64"
             }
+        } else if LINUX.is(os) && ARM64.is(arch) {
+            return Err(anyhow!("Linux arm64 is not supported yet by Google Chrome. Please try another browser."));
         } else {
             "linux64"
         };

rust/tests/browser_download_tests.rs

@@ -18,6 +18,7 @@
 use crate::common::{assert_browser, assert_driver, get_selenium_manager};
 
 use rstest::rstest;
+use std::env::consts::ARCH;
 use std::env::consts::OS;
 
 mod common;
@@ -27,23 +28,27 @@ mod common;
 #[case("firefox")]
 #[case("edge")]
 fn browser_latest_download_test(#[case] browser: String) {
-    if !browser.eq("edge") || !OS.eq("windows") {
-        let mut cmd = get_selenium_manager();
-        cmd.args([
-            "--browser",
-            &browser,
-            "--force-browser-download",
-            "--output",
-            "json",
-            "--debug",
-        ])
-        .assert()
-        .success()
-        .code(0);
-
-        assert_driver(&mut cmd);
-        assert_browser(&mut cmd);
+    if browser.eq("edge") && OS.eq("windows") {
+      return
+    } else if OS.eq("linux") && ARCH.eq("aarch64") && !browser.eq("firefox") {
+      return
     }
+
+    let mut cmd = get_selenium_manager();
+    cmd.args([
+        "--browser",
+        &browser,
+        "--force-browser-download",
+        "--output",
+        "json",
+        "--debug",
+    ])
+    .assert()
+    .success()
+    .code(0);
+
+    assert_driver(&mut cmd);
+    assert_browser(&mut cmd);
 }
 
 #[rstest]
@@ -59,6 +64,8 @@ fn browser_latest_download_test(#[case] browser: String) {
 fn browser_version_download_test(#[case] browser: String, #[case] browser_version: String) {
     if OS.eq("windows") && browser.eq("edge") {
         println!("Skipping Edge download test on Windows since the installation requires admin privileges");
+    } else if OS.eq("linux") && ARCH.eq("aarch64") && !browser.eq("firefox") {
+        println!("Skipping non-Firefox download test on Linux arm64 since no other browsers are supported yet");
     } else {
         let mut cmd = get_selenium_manager();
         cmd.args([

rust/tests/browser_tests.rs

@@ -19,6 +19,7 @@ use crate::common::{assert_output, get_selenium_manager, get_stdout};
 
 use exitcode::DATAERR;
 use rstest::rstest;
+use std::env::consts::ARCH;
 use std::env::consts::OS;
 use std::path::Path;
 
@@ -40,6 +41,10 @@ fn browser_version_test(
     #[case] browser_version: String,
     #[case] driver_version: String,
 ) {
+    if OS.eq("linux") && ARCH.eq("aarch64") && !browser.eq("firefox") {
+      return
+    }
+
     let mut cmd = get_selenium_manager();
     cmd.args([
         "--browser",
@@ -78,6 +83,10 @@ fn wrong_parameters_test(
     #[case] driver_version: String,
     #[case] error_code: i32,
 ) {
+    if OS.eq("linux") && ARCH.eq("aarch64") && !browser.eq("firefox") {
+      return
+    }
+
     let mut cmd = get_selenium_manager();
     let result = cmd
         .args([

rust/tests/cache_tests.rs

@@ -23,6 +23,7 @@ use std::path::Path;
 
 mod common;
 
+#[cfg(not(all(target_os = "linux", target_arch = "aarch64")))]
 #[rstest]
 #[case("../tmp")]
 #[case("../áèîö")]

rust/tests/common.rs

@@ -23,6 +23,7 @@ use selenium_manager::logger::JsonOutput;
 use selenium_manager::shell;
 use selenium_manager::shell::run_shell_command_by_os;
 use std::borrow::BorrowMut;
+use std::env::consts::ARCH;
 use std::env::consts::OS;
 use std::path::{Path, PathBuf};
 
@@ -119,3 +120,8 @@ pub fn assert_output(
             .contains(&error_code.to_string()));
     }
 }
+
+#[allow(dead_code)]
+pub fn is_linux_arm64() -> bool {
+  OS == "linux" && ARCH == "aarch64"
+}

rust/tests/config_tests.rs

@@ -19,6 +19,8 @@ use crate::common::{assert_browser, assert_driver, get_selenium_manager, get_std
 
 use rstest::rstest;
 
+use std::env::consts::ARCH;
+use std::env::consts::OS;
 use std::fs::File;
 use std::io::{BufWriter, Write};
 use tempfile::Builder;
@@ -30,6 +32,9 @@ mod common;
 #[case("firefox")]
 #[case("edge")]
 fn config_test(#[case] browser_name: String) {
+    if OS.eq("linux") && ARCH.eq("aarch64") && !browser_name.eq("firefox") {
+      return
+    }
     let tmp_dir = Builder::new().prefix("sm-config-test").tempdir().unwrap();
     let config_path = tmp_dir.path().join("se-config.toml");
     let config_file = File::create(config_path.as_path()).unwrap();

rust/tests/exec_driver_tests.rs

@@ -18,6 +18,7 @@
 use crate::common::{assert_browser, assert_driver, exec_driver, get_selenium_manager};
 
 use rstest::rstest;
+use std::env::consts::ARCH;
 use std::env::consts::OS;
 
 mod common;
@@ -28,6 +29,10 @@ mod common;
 #[case("firefox", "geckodriver")]
 #[case("iexplorer", "IEDriverServer")]
 fn exec_driver_test(#[case] browser_name: String, #[case] driver_name: String) {
+    if OS.eq("linux") && ARCH.eq("aarch64") && !browser_name.eq("firefox") {
+      return
+    }
+
     let mut cmd = get_selenium_manager();
     cmd.args(["--browser", &browser_name, "--output", "json"])
         .assert()

rust/tests/mirror_tests.rs

@@ -15,7 +15,7 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use crate::common::{assert_driver, get_selenium_manager};
+use crate::common::{assert_driver, is_linux_arm64, get_selenium_manager};
 
 mod common;
 
@@ -24,7 +24,7 @@ fn mirror_test() {
     let mut cmd = get_selenium_manager();
     cmd.args([
         "--browser",
-        "chrome",
+        if is_linux_arm64() { "firefox" } else { "chrome" },
         "--driver-mirror-url",
         "https://registry.npmmirror.com/-/binary/chromedriver/",
         "--browser-version",

rust/tests/offline_tests.rs

@@ -15,17 +15,22 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use crate::common::{get_selenium_manager, get_stdout};
+use crate::common::{get_selenium_manager, get_stdout, is_linux_arm64};
 
 mod common;
 
 #[test]
 fn offline_test() {
     let mut cmd = get_selenium_manager();
-    cmd.args(["--debug", "--browser", "chrome", "--offline"])
-        .assert()
-        .success()
-        .code(0);
+    cmd.args([
+        "--debug",
+        "--browser",
+        if is_linux_arm64() { "firefox" } else { "chrome" },
+        "--offline"
+    ])
+    .assert()
+    .success()
+    .code(0);
 
     let stdout = get_stdout(&mut cmd);