rho-coding-agent 2.19.0

A fast Rust agent harness with a small footprint and opinionated defaults
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use super::authenticate;
use crate::acp_runtime::{
    scripted_agent::{Script, ScriptedAgent},
    test_support::TEST_BUDGET,
};
use agent_client_protocol::Channel;
use pretty_assertions::assert_eq;
use serde_json::{json, Value};

// Covers: login sends the caller's method after initialize, and the agent's
// own failure text (for example a sign-in timeout) reaches the user.
// Owner: ACP login. Delegated runs never authenticate (driver tests).
#[tokio::test]
async fn authenticate_sends_the_method_and_keeps_agent_errors() {
    let cases = [
        (None, Ok(())),
        (
            Some("Onboarding failed: Timed out"),
            Err("Onboarding failed: Timed out"),
        ),
    ];
    for (agent_error, expected) in cases {
        let (client, agent) = Channel::duplex();
        let (fake, record, _signals) = ScriptedAgent::new(Script {
            authenticate_error: agent_error.map(str::to_owned),
            ..Script::default()
        });
        let (outcome, _) = tokio::time::timeout(TEST_BUDGET, async {
            tokio::join!(
                authenticate(client, "oauth-personal", TEST_BUDGET),
                fake.run(agent)
            )
        })
        .await
        .expect("login settles within the test budget");
        match (outcome, expected) {
            (Ok(()), Ok(())) => {}
            (Err(message), Err(detail)) => assert!(
                message.contains("authenticate") && message.contains(detail),
                "agent error kept with its step: {message}"
            ),
            (actual, expected) => panic!("{actual:?} != {expected:?}"),
        }
        let requests = record.lock().unwrap().requests.clone();
        assert_eq!(
            requests
                .iter()
                .map(|request| (
                    request["method"].clone(),
                    request["params"]["methodId"].clone()
                ))
                .collect::<Vec<_>>(),
            vec![
                (json!("initialize"), Value::Null),
                (json!("authenticate"), json!("oauth-personal")),
            ]
        );
    }
}