tea-session 0.1.0

Append-only session log, replay, and storage contracts for tea-rs
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151

use crate::common;

use common::{SESSION_ID, creation, envelope};
use serde_json::json;
use std::str::FromStr;
use tea_protocol::{ModelId, SessionId, SessionSequence};
use tea_session::{
    AppendTransaction, InMemorySessionStore, SessionReducer, SessionStore, SessionStoreErrorCode,
    StoredSession, apply_transaction_in_place,
};

#[tokio::test(flavor = "current_thread")]
async fn create_append_load_and_replay_are_equivalent() {
    let store = InMemorySessionStore::new();
    let session_id = SessionId::from_str(SESSION_ID).unwrap();
    let created = store
        .append(AppendTransaction::new(session_id, None, vec![creation()]))
        .await
        .unwrap();
    assert_eq!(created.previous_sequence(), None);
    assert_eq!(created.current_sequence(), SessionSequence::new(0));

    let changed = envelope(
        1,
        "configuration_changed",
        json!({"model":{"providerId":"openai","modelId":"openai/gpt-5"}}),
    );
    let outcome = store
        .append(AppendTransaction::new(
            session_id,
            Some(SessionSequence::new(0)),
            vec![changed],
        ))
        .await
        .unwrap();
    assert_eq!(outcome.current_sequence(), SessionSequence::new(1));

    let loaded = store.load(session_id).await.unwrap();
    assert_eq!(loaded.records().len(), 2);
    assert_eq!(
        loaded.state(),
        &SessionReducer::replay(loaded.records().to_vec()).unwrap()
    );
    assert_eq!(
        loaded.state().configuration().model_id(),
        Some(&ModelId::from_str("openai/gpt-5").unwrap())
    );
}

#[tokio::test(flavor = "current_thread")]
async fn expected_sequence_and_creation_conflicts_are_stable() {
    let store = InMemorySessionStore::new();
    let session_id = SessionId::from_str(SESSION_ID).unwrap();
    store
        .append(AppendTransaction::new(session_id, None, vec![creation()]))
        .await
        .unwrap();

    let duplicate = store
        .append(AppendTransaction::new(session_id, None, vec![creation()]))
        .await
        .unwrap_err();
    assert_eq!(
        duplicate.code(),
        SessionStoreErrorCode::SessionAlreadyExists
    );

    let stale = store
        .append(AppendTransaction::new(
            session_id,
            Some(SessionSequence::new(9)),
            vec![envelope(
                1,
                "configuration_changed",
                json!({"model":{"providerId":"openai","modelId":"openai/gpt-5"}}),
            )],
        ))
        .await
        .unwrap_err();
    assert_eq!(stale.code(), SessionStoreErrorCode::SequenceConflict);

    let missing_id = SessionId::from_str("0195a0b1-5e3a-7d72-a902-c4e85d828bf2").unwrap();
    let mut missing_creation = serde_json::to_value(creation()).unwrap();
    missing_creation["sessionId"] = json!(missing_id.to_string());
    let missing_creation = serde_json::from_value(missing_creation).unwrap();
    let missing = store
        .append(AppendTransaction::new(
            missing_id,
            Some(SessionSequence::new(0)),
            vec![missing_creation],
        ))
        .await
        .unwrap_err();
    assert_eq!(missing.code(), SessionStoreErrorCode::SessionNotFound);
}

#[tokio::test(flavor = "current_thread")]
async fn invalid_batch_rolls_back_all_records() {
    let store = InMemorySessionStore::new();
    let session_id = SessionId::from_str(SESSION_ID).unwrap();
    store
        .append(AppendTransaction::new(session_id, None, vec![creation()]))
        .await
        .unwrap();
    let first = envelope(
        1,
        "configuration_changed",
        json!({"model":{"providerId":"openai","modelId":"openai/gpt-5"}}),
    );
    let gap = envelope(3, "configuration_changed", json!({"profileId":"coding"}));
    assert!(
        store
            .append(AppendTransaction::new(
                session_id,
                Some(SessionSequence::new(0)),
                vec![first, gap],
            ))
            .await
            .is_err()
    );
    let loaded = store.load(session_id).await.unwrap();
    assert_eq!(loaded.records().len(), 1);
    assert_eq!(loaded.state().tail_sequence(), SessionSequence::new(0));
}

#[test]
fn incremental_apply_fails_closed_for_an_inconsistent_cache() {
    let session_id = SessionId::from_str(SESSION_ID).unwrap();
    let mut stored = StoredSession::from_durable_facts(vec![creation()], vec![], vec![]).unwrap();
    stored.reducer = SessionReducer::new();
    let transaction = AppendTransaction::new(
        session_id,
        Some(SessionSequence::new(0)),
        vec![envelope(
            1,
            "configuration_changed",
            json!({"profileId":"coding"}),
        )],
    );

    let error = apply_transaction_in_place(&transaction, &mut stored, true, |_| false).unwrap_err();

    assert_eq!(error.code(), SessionStoreErrorCode::CorruptionDetected);
}

#[test]
fn store_is_object_safe_send_and_sync() {
    fn assert_store<T: SessionStore + Send + Sync>() {}
    assert_store::<InMemorySessionStore>();
    let _object: Box<dyn SessionStore> = Box::new(InMemorySessionStore::new());
}