dk-protocol 0.3.0

dkod gRPC protocol definitions and server/client implementations
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use dk_protocol::session::{SessionManager, SessionSnapshot};
use std::time::Duration;

#[test]
fn test_create_and_get_session() {
    let mgr = SessionManager::new(Duration::from_secs(60));
    let sid = mgr.create_session(
        "claude-v3".into(),
        "org/repo".into(),
        "Refactor auth".into(),
        "abc123".into(),
    );

    let session = mgr.get_session(&sid).unwrap();
    assert_eq!(session.agent_id, "claude-v3");
    assert_eq!(session.codebase, "org/repo");
    assert_eq!(session.intent, "Refactor auth");
    assert_eq!(session.codebase_version, "abc123");
}

#[test]
fn test_session_not_found() {
    let mgr = SessionManager::new(Duration::from_secs(60));
    let fake_id = uuid::Uuid::new_v4();
    assert!(mgr.get_session(&fake_id).is_none());
}

#[test]
fn test_touch_session() {
    let mgr = SessionManager::new(Duration::from_secs(60));
    let sid = mgr.create_session("agent".into(), "repo".into(), "test".into(), "v1".into());
    assert!(mgr.touch_session(&sid));
    assert!(!mgr.touch_session(&uuid::Uuid::new_v4()));
}

#[test]
fn test_remove_session() {
    let mgr = SessionManager::new(Duration::from_secs(60));
    let sid = mgr.create_session("agent".into(), "repo".into(), "test".into(), "v1".into());
    assert!(mgr.remove_session(&sid));
    assert!(mgr.get_session(&sid).is_none());
}

#[test]
fn test_expired_session() {
    let mgr = SessionManager::new(Duration::from_millis(1));
    let sid = mgr.create_session("agent".into(), "repo".into(), "test".into(), "v1".into());
    std::thread::sleep(Duration::from_millis(10));
    assert!(mgr.get_session(&sid).is_none());
}

#[test]
fn test_save_and_take_snapshot() {
    let mgr = SessionManager::new(Duration::from_secs(60));
    let sid = mgr.create_session("agent".into(), "repo".into(), "test".into(), "v1".into());

    mgr.save_snapshot(&sid, SessionSnapshot {
        agent_id: "agent".into(),
        codebase: "repo".into(),
        intent: "test".into(),
        codebase_version: "v1".into(),
    });

    let snap = mgr.take_snapshot(&sid).unwrap();
    assert_eq!(snap.agent_id, "agent");
    assert_eq!(snap.codebase, "repo");

    // Second take returns None (consumed)
    assert!(mgr.take_snapshot(&sid).is_none());
}

#[test]
fn test_cleanup_expired_saves_snapshots() {
    let mgr = SessionManager::new(Duration::from_millis(1));
    let sid = mgr.create_session("agent".into(), "repo".into(), "test".into(), "v1".into());
    std::thread::sleep(Duration::from_millis(10));

    mgr.cleanup_expired();

    // Session should be gone
    assert!(mgr.get_session(&sid).is_none());

    // But a snapshot should have been saved
    let snap = mgr.take_snapshot(&sid).unwrap();
    assert_eq!(snap.agent_id, "agent");
    assert_eq!(snap.codebase, "repo");
    assert_eq!(snap.intent, "test");
    assert_eq!(snap.codebase_version, "v1");
}