ELY-Browser/crates/ely_browser_core/tests/profiles.rs

use std::error::Error;

use ely_browser_core::{BrowserCore, CoreError, InitialBrowserConfig};
use ely_domain::{CommandIntent, ProfileKind, ProfileSyncPolicy, UrlText};

#[test]
fn new_private_profile_command_creates_private_profile() -> Result<(), Box<dyn Error>> {
    let mut core = BrowserCore::new(InitialBrowserConfig::ely_defaults()?)?;

    core.set_command_query(">new-private-profile Private");
    let intent = core.submit_command()?;

    assert_eq!(intent, Some(CommandIntent::Command("new-private-profile Private".to_string())),);
    let snapshot = core.snapshot()?;
    let Some(profile) = snapshot.profiles.iter().find(|profile| profile.name() == "Private") else {
        return Err("missing private profile".into());
    };

    assert_eq!(profile.kind(), &ProfileKind::Private);
    assert_eq!(profile.sync_policy(), ProfileSyncPolicy::Paused);
    assert_eq!(&snapshot.active_profile_id, profile.id());
    assert_eq!(snapshot.active_profile_name, "Private");
    assert_eq!(snapshot.command_query, "");
    Ok(())
}

#[test]
fn private_window_config_starts_with_private_profile() -> Result<(), Box<dyn Error>> {
    let mut core = BrowserCore::new(InitialBrowserConfig::private_window()?)?;
    let snapshot = core.snapshot()?;
    let Some(private_profile) =
        snapshot.profiles.iter().find(|profile| profile.id() == &snapshot.active_profile_id)
    else {
        return Err("missing active private profile".into());
    };

    assert_eq!(snapshot.active_space_name, "Private");
    assert_eq!(snapshot.active_profile_name, "Private");
    assert_eq!(private_profile.kind(), &ProfileKind::Private);
    assert_eq!(private_profile.sync_policy(), ProfileSyncPolicy::Paused);

    core.open_tab(UrlText::parse("https://example.com/private-window")?);
    assert!(core.snapshot()?.history_entries.is_empty());
    Ok(())
}

#[test]
fn private_profiles_do_not_record_history() -> Result<(), Box<dyn Error>> {
    let mut core = BrowserCore::new(InitialBrowserConfig::ely_defaults()?)?;
    let default_profile_id = core.snapshot()?.active_profile_id;
    let private_profile_id = core.create_profile("Private", 0x807d72, ProfileKind::Private)?;

    core.open_tab(UrlText::parse("https://example.com/private")?);
    let private_snapshot = core.snapshot()?;
    assert_eq!(private_snapshot.active_profile_id, private_profile_id);
    assert!(private_snapshot.history_entries.is_empty());

    core.select_profile(&default_profile_id)?;
    core.open_tab(UrlText::parse("https://example.com/standard")?);
    let default_snapshot = core.snapshot()?;
    assert_eq!(default_snapshot.active_profile_id, default_profile_id);
    assert_eq!(default_snapshot.history_entries.len(), 1);

    core.select_profile(&private_profile_id)?;
    let private_snapshot = core.snapshot()?;
    assert!(private_snapshot.history_entries.is_empty());
    Ok(())
}

#[test]
fn private_profile_download_index_clears_when_leaving_profile() -> Result<(), Box<dyn Error>> {
    let mut core = BrowserCore::new(InitialBrowserConfig::ely_defaults()?)?;
    let default_profile_id = core.snapshot()?.active_profile_id;
    let private_profile_id = core.create_profile("Private", 0x807d72, ProfileKind::Private)?;

    core.record_download_started(
        UrlText::parse("https://example.com/private.dmg")?,
        "private.dmg",
        Some(4096),
    )?;
    assert_eq!(core.snapshot()?.download_entries.len(), 1);

    core.select_profile(&default_profile_id)?;
    assert_eq!(core.snapshot()?.active_profile_id, default_profile_id);
    assert!(core.snapshot()?.download_entries.is_empty());

    core.select_profile(&private_profile_id)?;
    assert_eq!(core.snapshot()?.active_profile_id, private_profile_id);
    assert!(core.snapshot()?.download_entries.is_empty());
    Ok(())
}

#[test]
fn selecting_standard_tab_clears_private_profile_download_index() -> Result<(), Box<dyn Error>> {
    let mut core = BrowserCore::new(InitialBrowserConfig::ely_defaults()?)?;
    let default_tab_id = core.snapshot()?.active_tab_id;
    let private_profile_id = core.create_profile("Private", 0x807d72, ProfileKind::Private)?;

    core.record_download_started(
        UrlText::parse("https://example.com/private.zip")?,
        "private.zip",
        Some(1024),
    )?;
    assert_eq!(core.snapshot()?.download_entries.len(), 1);

    core.select_tab(&default_tab_id)?;
    assert!(core.snapshot()?.download_entries.is_empty());

    core.select_profile(&private_profile_id)?;
    assert!(core.snapshot()?.download_entries.is_empty());
    Ok(())
}

#[test]
fn private_profile_starts_with_sync_paused() -> Result<(), Box<dyn Error>> {
    let mut core = BrowserCore::new(InitialBrowserConfig::ely_defaults()?)?;
    let private_profile_id = core.create_profile("Private", 0x807d72, ProfileKind::Private)?;
    let snapshot = core.snapshot()?;
    let Some(private_profile) =
        snapshot.profiles.iter().find(|profile| profile.id() == &private_profile_id)
    else {
        return Err("missing private profile".into());
    };

    assert_eq!(private_profile.sync_policy(), ProfileSyncPolicy::Paused);
    Ok(())
}

#[test]
fn profile_sync_policy_can_pause_one_profile() -> Result<(), Box<dyn Error>> {
    let mut core = BrowserCore::new(InitialBrowserConfig::ely_defaults()?)?;
    let default_profile_id = core.snapshot()?.active_profile_id;
    let research_profile_id = core.create_profile("Research", 0x9fc9a2, ProfileKind::Standard)?;

    core.set_profile_sync_policy(&research_profile_id, ProfileSyncPolicy::Paused)?;
    let snapshot = core.snapshot()?;
    let Some(default_profile) =
        snapshot.profiles.iter().find(|profile| profile.id() == &default_profile_id)
    else {
        return Err("missing default profile".into());
    };
    let Some(research_profile) =
        snapshot.profiles.iter().find(|profile| profile.id() == &research_profile_id)
    else {
        return Err("missing research profile".into());
    };

    assert_eq!(default_profile.sync_policy(), ProfileSyncPolicy::Enabled);
    assert_eq!(research_profile.sync_policy(), ProfileSyncPolicy::Paused);
    Ok(())
}

#[test]
fn private_profile_sync_policy_stays_paused() -> Result<(), Box<dyn Error>> {
    let mut core = BrowserCore::new(InitialBrowserConfig::ely_defaults()?)?;
    let private_profile_id = core.create_profile("Private", 0x807d72, ProfileKind::Private)?;

    let result = core.set_profile_sync_policy(&private_profile_id, ProfileSyncPolicy::Enabled);
    let error = match result {
        Ok(()) => return Err("private profile sync enabled".into()),
        Err(error) => error,
    };

    assert_eq!(error, CoreError::PrivateProfileSyncLocked { id: private_profile_id.clone() });
    let snapshot = core.snapshot()?;
    let Some(private_profile) =
        snapshot.profiles.iter().find(|profile| profile.id() == &private_profile_id)
    else {
        return Err("missing private profile".into());
    };
    assert_eq!(private_profile.sync_policy(), ProfileSyncPolicy::Paused);
    Ok(())
}