rho-coding-agent 1.48.0

A lightweight agent harness inspired by Pi
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

use std::num::NonZeroUsize;

use pretty_assertions::assert_eq;
use rho_sdk::tool::tool_progress_channel;
use rmcp::model::{NumberOrString, ProgressNotificationParam, ProgressToken};

use super::McpProgressRouter;

fn token(value: i64) -> ProgressToken {
    ProgressToken(NumberOrString::Number(value))
}

// Covers: progress must reach only the subscribed call, must carry counts only
// when the server supplied a usable total, and must stop when the call ends.
// Owner: MCP progress routing.
#[tokio::test]
async fn progress_reaches_the_subscribed_call_only() {
    let router = McpProgressRouter::new();
    let (sender, mut receiver) = tool_progress_channel(NonZeroUsize::new(8).unwrap());
    let subscription = router.subscribe(token(1), sender);

    // Unsubscribed token: dropped rather than misrouted.
    router
        .dispatch(ProgressNotificationParam::new(token(2), 1.0))
        .await;
    router
        .dispatch(
            ProgressNotificationParam::new(token(1), 3.0)
                .with_total(10.0)
                .with_message("indexing"),
        )
        .await;
    // No total: nothing to render a count against.
    router
        .dispatch(ProgressNotificationParam::new(token(1), 4.0))
        .await;

    let counted = receiver.recv().await.unwrap();
    assert_eq!(
        (
            counted.text(),
            counted.completed_units(),
            counted.total_units()
        ),
        ("indexing", Some(3), Some(10))
    );
    let uncounted = receiver.recv().await.unwrap();
    assert_eq!(
        (uncounted.completed_units(), uncounted.total_units()),
        (None, None)
    );

    drop(subscription);
    router
        .dispatch(ProgressNotificationParam::new(token(1), 5.0))
        .await;
    assert!(receiver.recv().await.is_none());
}