mobius 0.9.21

A small, modular Rust framework for building coding agents
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

//! Agent input-boundary validation tests.

use crate::agent::validate_submission;
use crate::middleware::session_files::session_file_limits;
use crate::protocol::{MessageAuthor, MessageSubmission, Op, SessionFileReference, Submission};

fn attachment(index: usize, size: u64) -> SessionFileReference {
    SessionFileReference {
        id: format!("00000000-0000-0000-0000-{index:012x}"),
        name: format!("{index}.bin"),
        size,
        media_type: "application/octet-stream".into(),
    }
}

fn submission(attachments: Vec<SessionFileReference>) -> Submission {
    Submission {
        id: "submission".into(),
        op: Op::Message {
            message: MessageSubmission {
                author: MessageAuthor::User,
                text: String::new(),
                attachments,
                requested_delivery: None,
                target_turn_id: None,
            },
        },
    }
}

#[test]
fn advertised_attachment_reference_limit_is_enforced() {
    let limits = session_file_limits();
    let attachments = (1..=limits.max_attachment_references + 1)
        .map(|index| attachment(index, 1))
        .collect();

    let error = validate_submission(&submission(attachments)).expect_err("too many references");

    assert!(
        error
            .to_string()
            .contains(&limits.max_attachment_references.to_string())
    );
}

#[test]
fn advertised_file_size_limit_is_enforced() {
    let limits = session_file_limits();

    let error = validate_submission(&submission(vec![attachment(1, limits.max_file_bytes + 1)]))
        .expect_err("oversized attachment");

    assert!(
        error
            .to_string()
            .contains(&limits.max_file_bytes.to_string())
    );
}

#[test]
fn advertised_session_byte_limit_is_enforced() {
    let limits = session_file_limits();
    let attachment_count = limits.max_session_bytes / limits.max_file_bytes + 1;
    let attachments = (1..=attachment_count)
        .map(|index| {
            attachment(
                usize::try_from(index).expect("attachment index"),
                limits.max_file_bytes,
            )
        })
        .collect();

    let error = validate_submission(&submission(attachments)).expect_err("session quota");

    assert!(
        error
            .to_string()
            .contains(&limits.max_session_bytes.to_string())
    );
}