test: more granular tests

Author: theg1239

apps/desktop/src-tauri/Cargo.lock

@@ -35,6 +35,9 @@ percent-encoding = "2"
 tauri-plugin-os = "2.3.1"
 tauri-plugin-store = "2.4.0"
 
+[dev-dependencies]
+tempfile = "3.13.0"
+
 [target.'cfg(target_os = "windows")'.dependencies]
 winreg = "0.52"
 

apps/desktop/src-tauri/Cargo.toml

@@ -0,0 +1,93 @@
+#![cfg(target_os = "windows")]
+
+use desktop_lib::browser_details::{get_chrome_profiles, Browsers};
+use std::fs;
+use std::io::Write;
+use std::path::PathBuf;
+use tempfile::TempDir;
+
+fn write_local_state(temp: &TempDir, relative: &str, contents: &str) -> PathBuf {
+    let mut path = temp.path().to_path_buf();
+    path.push(relative);
+    if let Some(parent) = path.parent() {
+        fs::create_dir_all(parent).expect("failed to create Local State parent");
+    }
+    let mut file = fs::File::create(&path).expect("failed to create Local State file");
+    file.write_all(contents.as_bytes())
+        .expect("failed to write Local State");
+    path
+}
+
+fn install_local_state(temp: &TempDir, browser: Browsers, json: &str) {
+    let relative = match browser {
+        Browsers::Chrome => "Google/Chrome/User Data/Local State",
+        Browsers::Edge => "Microsoft/Edge/User Data/Local State",
+        Browsers::Brave => "BraveSoftware/Brave-Browser/User Data/Local State",
+        _ => panic!("unexpected browser for chrome profile test"),
+    };
+    write_local_state(temp, relative, json);
+}
+
+fn with_localappdata<F: FnOnce()>(temp: &TempDir, f: F) {
+    use std::env;
+    let original = env::var_os("LOCALAPPDATA");
+    env::set_var("LOCALAPPDATA", temp.path());
+    f();
+    match original {
+        Some(val) => env::set_var("LOCALAPPDATA", val),
+        None => env::remove_var("LOCALAPPDATA"),
+    }
+}
+
+#[test]
+fn reads_profiles_and_adds_default_when_missing() {
+    let temp = TempDir::new().expect("temp dir");
+    let local_state = r#"{
+        "profile": {
+            "info_cache": {
+                "Profile 1": {
+                    "profile_dir": "Profile 1",
+                    "name": "Personal"
+                },
+                "Profile 2": {
+                    "gaia_name": "Work"
+                }
+            }
+        }
+    }"#;
+    install_local_state(&temp, Browsers::Chrome, local_state);
+
+    with_localappdata(&temp, || {
+        let profiles = get_chrome_profiles(Browsers::Chrome).expect("profiles");
+        let labels: Vec<_> = profiles.iter().map(|p| &p.display_name).collect();
+        assert!(labels.contains(&"Personal".to_string()));
+        assert!(labels.contains(&"Work".to_string()));
+        assert!(labels.contains(&"Default".to_string()));
+    });
+}
+
+#[test]
+fn deduplicates_profile_directories() {
+    let temp = TempDir::new().expect("temp dir");
+    let local_state = r#"{
+        "profile": {
+            "info_cache": {
+                "Profile 1": {
+                    "profile_dir": "Profile 1",
+                    "name": "Personal"
+                },
+                "Duplicate": {
+                    "profile_dir": "Profile 1",
+                    "name": "Duplicate"
+                }
+            }
+        }
+    }"#;
+    install_local_state(&temp, Browsers::Chrome, local_state);
+
+    with_localappdata(&temp, || {
+        let profiles = get_chrome_profiles(Browsers::Chrome).expect("profiles");
+        let dirs: Vec<_> = profiles.iter().map(|p| &p.directory).collect();
+        assert_eq!(dirs.iter().filter(|d| d.as_str() == "Profile 1").count(), 1);
+    });
+}

apps/desktop/src-tauri/tests/browser/chrome_profiles.rs

@@ -0,0 +1,35 @@
+#![cfg(target_os = "windows")]
+
+use desktop_lib::browser_details::get_firefox_profiles;
+use std::fs;
+use tempfile::TempDir;
+
+fn with_localappdata<F: FnOnce()>(temp: &TempDir, f: F) {
+    use std::env;
+    let original = env::var_os("LOCALAPPDATA");
+    env::set_var("LOCALAPPDATA", temp.path());
+    f();
+    match original {
+        Some(val) => env::set_var("LOCALAPPDATA", val),
+        None => env::remove_var("LOCALAPPDATA"),
+    }
+}
+
+#[test]
+fn discovers_profile_directories() {
+    let temp = TempDir::new().expect("temp dir");
+    let base = temp.path().join("Mozilla/Firefox/Profiles");
+    fs::create_dir_all(base.join("abcd.default-release")).expect("create profile dir");
+    fs::create_dir_all(base.join("custom.work"   )).expect("create profile dir");
+    // include a file to ensure non-dirs skipped
+    fs::write(base.join("not_a_dir"), b"noop").expect("create dummy file");
+
+    with_localappdata(&temp, || {
+        let mut profiles = get_firefox_profiles().expect("profiles");
+        profiles.sort_by(|a, b| a.display_name.cmp(&b.display_name));
+        let names: Vec<_> = profiles.iter().map(|p| &p.display_name).collect();
+        assert!(names.contains(&"abcd.default-release".to_string()));
+        assert!(names.contains(&"custom.work".to_string()));
+        assert_eq!(profiles.len(), 2);
+    });
+}