From 522ef21b2866ab453bf62b8750c10dde3151e036 Mon Sep 17 00:00:00 2001
From: amr-crabnebula <amr@crabnebula.dev>
Date: Fri, 17 Nov 2023 03:59:17 +0200
Subject: [PATCH] fix integration test on Linux finally

---
 crates/updater/tests/app/src/main.rs | 38 ++++++++++++++++++++--------
 1 file changed, 28 insertions(+), 10 deletions(-)

diff --git a/crates/updater/tests/app/src/main.rs b/crates/updater/tests/app/src/main.rs
index f13396e9..290c0dbf 100644
--- a/crates/updater/tests/app/src/main.rs
+++ b/crates/updater/tests/app/src/main.rs
@@ -14,17 +14,35 @@ fn main() {
             ..Default::default()
         },
     );
-    if std::env::var("UPDATER_FORMAT").unwrap_or_default() == "nsis" {
-        // /D sets the default installation directory ($INSTDIR),
-        // overriding InstallDir and InstallDirRegKey.
-        // It must be the last parameter used in the command line and must not contain any quotes, even if the path contains spaces.
-        // Only absolute paths are supported.
-        // NOTE: we only need this because this is an integration test and we don't want to install the app in the programs folder
-        builder = builder.installer_args(vec![format!(
-            "/D={}",
-            std::env::current_exe().unwrap().parent().unwrap().display()
-        )]);
+    let format = std::env::var("UPDATER_FORMAT").unwrap_or_default();
+
+    match format.as_str() {
+        "nsis" => {
+            // /D sets the default installation directory ($INSTDIR),
+            // overriding InstallDir and InstallDirRegKey.
+            // It must be the last parameter used in the command line and must not contain any quotes, even if the path contains spaces.
+            // Only absolute paths are supported.
+            // NOTE: we only need this because this is an integration test and we don't want to install the app in the programs folder
+            builder = builder.installer_args(vec![format!(
+                "/D={}",
+                std::env::current_exe().unwrap().parent().unwrap().display()
+            )]);
+        }
+        #[cfg(any(
+            target_os = "linux",
+            target_os = "dragonfly",
+            target_os = "freebsd",
+            target_os = "netbsd",
+            target_os = "openbsd"
+        ))]
+        "appimage" => {
+            if let Some(p) = std::env::var_os("APPIMAGE") {
+                builder = builder.executable_path(p);
+            }
+        }
+        _ => {}
     }
+
     let updater = builder.build().unwrap();
 
     println!("{version}");