relay-knowledge 1.1.17

Graph-database-based knowledge graph project.
Documentation 
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

use crate::api::AgentProtocolKind;

use super::*;

#[test]
fn session_records_normalize_client_identity() {
    let registry = AcpSessionRegistry::default();
    registry.insert_session(
        "session-1".to_owned(),
        AcpSessionRecord::new(
            Some("  editor  ".to_owned()),
            Some("   ".to_owned()),
            Some(" actor-1 ".to_owned()),
        ),
    );

    let record = registry.session("session-1").expect("session should exist");
    let identity = record.identity("session-1", Some("request-1".to_owned()));
    assert_eq!(identity.protocol, AgentProtocolKind::Acp);
    assert_eq!(identity.client_name.as_deref(), Some("editor"));
    assert_eq!(identity.client_version, None);
    assert_eq!(identity.actor_id.as_deref(), Some("actor-1"));
    assert_eq!(identity.session_id.as_deref(), Some("session-1"));
    assert_eq!(identity.tool_call_id.as_deref(), Some("request-1"));
}

#[test]
fn active_request_drop_removes_cancellation_registration() {
    let registry = AcpSessionRegistry::default();
    let (_receiver, registration) = registry.register_request("session-1", "request-1".to_owned());

    drop(registration);

    assert!(!registry.cancel_request("session-1", "request-1"));
}

#[test]
fn cancellation_notifies_active_request_until_release() {
    let registry = AcpSessionRegistry::default();
    let (receiver, registration) = registry.register_request("session-1", "request-1".to_owned());

    assert!(registry.cancel_request("session-1", "request-1"));
    assert!(*receiver.borrow());
    registration.release();
    assert!(!registry.cancel_request("session-1", "request-1"));
}