pitboard-core 0.5.2

//! A switch of a tool that is not Claude Code, through the same engine.
//!
//! Codex disagrees with Claude Code on nearly every fact the boundary carries: its login
//! names its own account, nothing follows a switch until it is restarted, a park may never
//! be a copy, it takes no write lock and caches no identity. These are the cases that
//! prove the switch asks the tool rather than assuming Claude Code's answers.

use super::enroll;
use super::harness::{
    NOW, codex_access, codex_account, codex_id, codex_login, codex_machine, hold,
};
use super::*;
use crate::api::scripted::Trouble;
use crate::provider::codex::api::Fresh;
use crate::provider::{Adoption, Credential};

fn whose(m: &super::harness::Machine) -> Option<String> {
    let live = m.live()?;
    provider::of(m.which)
        .identify(&m.ctx, &Credential::new(m.which, live))
        .ok()
        .map(|found| found.account_id)
}

/// The login moves: `there`'s goes live, `here`'s goes into the vault, and nothing of
/// either is left anywhere else. A copy of `there` left parked beside its live twin is a
/// token `codex logout` would revoke in both places at once.
#[test]
fn a_codex_switch_moves_one_login_in_and_one_out() {
    let m = codex_machine("moves");
    let settled = settle(&m.ctx, None).expect("nothing to recover").0;

    let (outcome, _) = switch(settled, &m.key("there")).expect("switched");

    assert_eq!(whose(&m), Some(codex_id("there")));
    let Outcome::Switched { adoption, .. } = outcome else {
        panic!("expected a switch, got {outcome:?}");
    };
    assert_eq!(
        adoption,
        Adoption::RestartRequired { program: "codex" },
        "a running codex never notices, so it must not be told it will"
    );
    let state = state::load(&m.ctx).expect("state");
    assert_eq!(state.active_for(ProviderId::Codex), Some("there"));
    assert!(state.get(&m.key("there")).unwrap().parked.is_none());
    let parked = state
        .get(&m.key("here"))
        .unwrap()
        .parked
        .clone()
        .expect("here parked");
    assert_eq!(
        m.mem.vault().services(),
        vec![parked.service],
        "one copy in the vault, and it is the outgoing login"
    );
    hold(&m, "after a Codex switch");
}

/// Claude Code's login sits in a keychain item that Codex's chain never reads. A Codex
/// switch that reached it would sign somebody out of the other tool.
#[test]
fn a_codex_switch_leaves_claude_code_alone() {
    let m = codex_machine("claude-untouched");
    let claude = crate::provider::claude::paths::live_service(&m.ctx);
    m.mem
        .live()
        .plant(&claude, "{\"claudeAiOauth\":{\"refreshToken\":\"c\"}}");

    let settled = settle(&m.ctx, None).expect("nothing to recover").0;
    switch(settled, &m.key("there")).expect("switched");

    assert_eq!(
        m.mem.live().peek(&claude).as_deref(),
        Some("{\"claudeAiOauth\":{\"refreshToken\":\"c\"}}")
    );
}

/// Both tools have a `there`. The Codex one is switched to, and the Claude Code one is
/// neither parked, nor made active, nor looked up in its place.
#[test]
fn the_same_label_on_two_tools_is_two_accounts() {
    let m = codex_machine("same-label");
    let mut state = state::load(&m.ctx).expect("state");
    state.upsert(super::harness::account("there", "claude-there", None));
    state::save(&m.ctx, &state).expect("saved");

    let settled = settle(&m.ctx, None).expect("nothing to recover").0;
    switch(settled, &m.key("there")).expect("switched");

    let state = state::load(&m.ctx).expect("state");
    let claude = Key::new(ProviderId::Claude, "there");
    assert_eq!(state.get(&claude).unwrap().account_uuid, "claude-there");
    assert!(state.get(&claude).unwrap().parked.is_none());
    assert_eq!(state.active_for(ProviderId::Claude), None);
    assert_eq!(state.active_for(ProviderId::Codex), Some("there"));
}

/// Codex's login names its account whether or not OpenAI still accepts it, so the park
/// going in is asked about first. Refused as lapsed, it is renewed and the fresh tokens
/// are what goes live.
#[test]
fn a_park_openai_refuses_is_renewed_before_it_goes_live() {
    let m = codex_machine("renewed-first");
    m.api
        .token_trouble(&codex_access("there-refresh"), Trouble::Unauthorized);
    let fresh_access = codex_access("there-renewed");
    m.api.codex_renews(
        "there-refresh",
        Fresh {
            id_token: None,
            access_token: Some(fresh_access.clone()),
            refresh_token: Some("there-renewed".into()),
            at: Some(NOW),
        },
    );

    let settled = settle(&m.ctx, None).expect("nothing to recover").0;
    switch(settled, &m.key("there")).expect("switched");

    let live = m.live().expect("a live login");
    assert_eq!(live["tokens"]["refresh_token"], "there-renewed");
    assert_eq!(live["tokens"]["access_token"], fresh_access.as_str());
    assert_eq!(whose(&m), Some(codex_id("there")));
    hold(&m, "after a renewed Codex switch");
}

/// A park OpenAI will not renew is dropped and nothing moves: `here` is still signed in
/// and still the only copy of itself.
#[test]
fn a_park_openai_will_not_renew_moves_nothing() {
    let m = codex_machine("refused");
    m.api
        .token_trouble(&codex_access("there-refresh"), Trouble::Unauthorized);
    m.api
        .codex_renew_trouble("there-refresh", Trouble::InvalidGrant);
    let before = m.live();

    let settled = settle(&m.ctx, None).expect("nothing to recover").0;
    let refused = switch(settled, &m.key("there")).expect_err("refused");

    assert_eq!(refused.code(), "parked_login_refused");
    assert_eq!(m.live(), before);
    let here = store::fingerprint("here-refresh");
    for service in m.mem.vault().services() {
        let parked: Value =
            serde_json::from_str(&m.mem.vault().peek(&service).expect("listed")).expect("json");
        assert_ne!(
            provider::of(ProviderId::Codex).fingerprint(&parked),
            here,
            "nothing of `here` was parked"
        );
    }
    let state = state::load(&m.ctx).expect("state");
    assert!(
        state.get(&m.key("there")).unwrap().parked.is_none(),
        "the refused park is dropped rather than offered again"
    );
    hold(&m, "after a refused Codex switch");
}

/// A park that could not be kept is found before the live login is replaced. For a tool
/// whose park may never be a copy, the outgoing login stays where it was: it is the only
/// one there is.
#[test]
fn a_park_that_does_not_stick_leaves_the_live_login_in_place() {
    let m = codex_machine("park-lost");
    m.mem
        .vault()
        .fault_all(crate::store::memory::Fault::DeletedAfterWrite);
    let before = m.live();

    let settled = settle(&m.ctx, None).expect("nothing to recover").0;
    assert!(switch(settled, &m.key("there")).is_err());

    assert_eq!(m.live(), before, "`here` must still be signed in");
    m.mem.vault().heal_all();
    hold(&m, "after a park that did not stick");
}

/// A Codex account enrolled alongside keeps its own detail: the plan its login names.
#[test]
fn a_codex_account_is_a_codex_account() {
    let account = codex_account("work", "acc", None);
    assert_eq!(account.provider(), ProviderId::Codex);
    assert!(account.claude().is_none());
    let login = codex_login("acc", "r");
    assert_eq!(
        provider::of(ProviderId::Codex).fingerprint(&login),
        store::fingerprint("r")
    );
}

/// A tool that never follows a switch goes on using the outgoing account in every session
/// already running, and signing out inside one would revoke the login just parked. Both are
/// said, with how many sessions there are.
#[test]
fn running_codex_sessions_are_counted_and_warned_about() {
    let m = codex_machine("sessions");
    m.mem.runs("codex", 2);

    let settled = settle(&m.ctx, Some(ProviderId::Codex))
        .expect("nothing to recover")
        .0;
    let (_, warnings) = switch(settled, &m.key("there")).expect("switched");

    let said = warnings
        .iter()
        .find(|w| w.code() == "sessions_still_running")
        .expect("warned");
    let text = said.to_string();
    assert!(text.contains("2 `codex` sessions"), "{text}");
    assert!(
        text.contains("`codex/here`"),
        "names the account they still use: {text}"
    );
    assert!(
        text.contains("signing out"),
        "and the one thing not to do: {text}"
    );
}

/// Nothing running is nothing to say.
#[test]
fn no_running_session_is_no_warning() {
    let m = codex_machine("no-sessions");
    let settled = settle(&m.ctx, Some(ProviderId::Codex))
        .expect("nothing to recover")
        .0;
    let (_, warnings) = switch(settled, &m.key("there")).expect("switched");
    assert!(
        warnings
            .iter()
            .all(|w| w.code() != "sessions_still_running"),
        "{warnings:?}"
    );
}

/// `CLAUDE_CODE_CUSTOM_OAUTH_URL` moves Claude Code's login and nothing of Codex's, so it
/// refuses changes to Claude Code and lets a Codex switch through.
#[test]
fn a_custom_claude_endpoint_does_not_stop_a_codex_switch() {
    let m = codex_machine("custom-oauth");
    let mut ctx = m.ctx.clone();
    ctx.custom_oauth = true;
    assert!(settle(&ctx, Some(ProviderId::Codex)).is_ok());
    assert_eq!(
        settle(&ctx, Some(ProviderId::Claude))
            .err()
            .map(|e| e.code()),
        Some("custom_oauth_endpoint")
    );
    assert_eq!(
        settle(&ctx, None).err().map(|e| e.code()),
        Some("custom_oauth_endpoint"),
        "a change that could touch every tool's accounts is Claude Code's business too"
    );
}

/// A sign-in run for one tool cannot be enrolled as another's account.
#[test]
fn a_sign_in_is_enrolled_only_under_its_own_tool() {
    let m = codex_machine("wrong-tool");
    let login = super::enroll::planted(
        &m.ctx,
        ProviderId::Codex,
        codex_login("third", "third-refresh"),
    )
    .expect("a sign-in");
    let settled = settle(&m.ctx, None).expect("nothing to recover").0;
    let refused =
        enroll(settled, &Key::new(ProviderId::Claude, "third"), Some(login)).expect_err("refused");
    assert_eq!(refused.code(), "usage");
}

/// A Codex sign-in, planted the way `codex login` leaves one, is parked as a Codex account.
#[test]
fn a_codex_sign_in_is_parked_as_a_codex_account() {
    let m = codex_machine("sign-in");
    let login = super::enroll::planted(
        &m.ctx,
        ProviderId::Codex,
        codex_login("third", "third-refresh"),
    )
    .expect("a sign-in");
    let settled = settle(&m.ctx, None).expect("nothing to recover").0;
    let key = m.key("third");
    enroll(settled, &key, Some(login)).expect("enrolled");

    let state = state::load(&m.ctx).expect("state");
    let third = state.get(&key).expect("enrolled");
    assert_eq!(third.provider(), ProviderId::Codex);
    assert_eq!(third.account_uuid, codex_id("third"));
    let parked = third.parked.clone().expect("parked");
    assert_eq!(
        parked.refresh_fingerprint,
        store::fingerprint("third-refresh")
    );
    hold(&m, "after a Codex sign-in");
}

/// Abandoning an interrupted Codex switch keeps nothing that copies the live login. Killed
/// after parking `here` and before installing `there`, the park is `here`'s own refresh
/// token beside the live one, which a sign-out would revoke in both places.
#[test]
fn abandoning_a_codex_switch_keeps_no_twin_of_the_live_login() {
    let m = codex_machine("abandon-twin");
    let settled = settle(&m.ctx, None).expect("nothing to recover").0;
    let died = crate::fault::killing("switch.park_recorded", || switch(settled, &m.key("there")));
    assert_eq!(died.unwrap_err(), "switch.park_recorded");

    abandon(&m.ctx).expect("abandoned");

    let state = state::load(&m.ctx).expect("state");
    assert!(
        state.get(&m.key("here")).unwrap().parked.is_none(),
        "`here` is signed in, so a park of it is a twin"
    );
    assert!(
        state.get(&m.key("there")).unwrap().parked.is_some(),
        "and `there`'s park, the only copy of it, is kept"
    );
    settle(&m.ctx, None).expect("purged on the next change");
    hold(&m, "after abandoning a Codex switch");
}

/// A codex still running from before the switch refreshes its token while the switch is
/// under way. Codex takes no lock, so nothing kept it out; what matters is that the switch
/// notices before writing over the file that now holds the only live copy of that chain.
/// Nothing is installed, and the park, a copy of the chain the refresh just spent, is
/// dropped rather than kept as though it worked.
#[test]
fn a_codex_refresh_during_a_switch_is_not_written_over() {
    let m = codex_machine("refreshed-meanwhile");
    let refreshed = codex_login("here", "here-refresh-2");
    let writer = m.ctx.clone();
    let settled = settle(&m.ctx, None).expect("nothing to recover").0;

    let refused = crate::fault::meanwhile(
        "switch.park_recorded",
        move || {
            let live = provider::of(ProviderId::Codex).live(&writer).unwrap();
            store::write_raw(&live.chain, &live.service, &refreshed.to_string()).unwrap();
        },
        || switch(settled, &m.key("there")),
    )
    .expect_err("nothing is installed over a login that moved");

    assert_eq!(refused.code(), "signed_in_account_changed");
    assert_eq!(
        m.live().unwrap()["tokens"]["refresh_token"],
        "here-refresh-2",
        "the refreshed login is still there"
    );
    let state = state::load(&m.ctx).expect("state");
    assert!(
        state.get(&m.key("here")).unwrap().parked.is_none(),
        "the park copied a chain the refresh spent, so it is not offered as a login"
    );
    assert!(state.get(&m.key("there")).unwrap().parked.is_some());
    settle(&m.ctx, None).expect("the next change purges");
    hold(&m, "after a refresh during a Codex switch");
}

/// Somebody signs Codex in to another account while the switch is under way. The park of
/// the outgoing account may now be its only login, so it is kept, and nothing is written
/// over the account that is now live.
#[test]
fn another_sign_in_during_a_switch_keeps_the_outgoing_park() {
    let m = codex_machine("signed-in-meanwhile");
    let other = codex_login("other", "other-refresh");
    let writer = m.ctx.clone();
    let settled = settle(&m.ctx, None).expect("nothing to recover").0;

    let refused = crate::fault::meanwhile(
        "switch.park_recorded",
        move || {
            let live = provider::of(ProviderId::Codex).live(&writer).unwrap();
            store::write_raw(&live.chain, &live.service, &other.to_string()).unwrap();
        },
        || switch(settled, &m.key("there")),
    )
    .expect_err("nothing is installed over a login that moved");

    assert_eq!(refused.code(), "signed_in_account_changed");
    assert_eq!(
        m.live().unwrap()["tokens"]["refresh_token"],
        "other-refresh"
    );
    let state = state::load(&m.ctx).expect("state");
    assert!(
        state.get(&m.key("here")).unwrap().parked.is_some(),
        "`here` is no longer live, and its park may be the only login it has"
    );
}

/// An interrupted switch is judged where it ran. Read from another `CODEX_HOME`, the live
/// login there has nothing to do with the switch's two sides, and recovery would guess.
#[test]
fn an_interrupted_switch_is_recovered_only_where_it_ran() {
    let m = codex_machine("slot-bound");
    let settled = settle(&m.ctx, None).expect("nothing to recover").0;
    let died = crate::fault::killing("switch.park_recorded", || switch(settled, &m.key("there")));
    assert_eq!(died.unwrap_err(), "switch.park_recorded");

    let elsewhere = m.ctx.clone().with_codex_home("/somewhere/else".into());
    let refused = settle(&elsewhere, None).err().expect("refused elsewhere");
    assert_eq!(refused.code(), "recovery_elsewhere");
    assert!(refused.to_string().contains("CODEX_HOME"), "{refused}");

    settle(&m.ctx, None).expect("recovered where it ran");
    hold(&m, "after recovering where the switch ran");
}