botrs 0.12.1

A Rust QQ Bot framework based on QQ Guild Bot API
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

use std::sync::Arc;
use std::time::Duration;

use tokio::sync::mpsc;

use super::*;
use crate::error::{CodeConnCloseCantIdentify, CodeConnCloseCantResume, CodeNeedReConnect, New};
use crate::intents::Intents;
use crate::models::api::{GatewayResponse, SessionStartLimit};
use crate::token::Token;

fn ap_info(shards: u32, remaining: u32, max_concurrency: u32) -> GatewayResponse {
    GatewayResponse {
        url: "wss://example.com".to_string(),
        shards,
        session_start_limit: SessionStartLimit {
            total: 10,
            remaining,
            reset_after: 1000,
            max_concurrency,
        },
    }
}

#[test]
fn calc_interval_matches_expected() {
    assert_eq!(CalcInterval(0), Duration::from_secs(2));
    assert_eq!(CalcInterval(1), Duration::from_secs(2));
    assert_eq!(CalcInterval(2), Duration::from_secs(1));
    assert_eq!(CalcInterval(3), Duration::from_secs(1));
    assert_eq!(CalcInterval(100), Duration::from_secs(1));
}

#[test]
fn check_session_limit_matches_expected() {
    assert!(CheckSessionLimit(&ap_info(2, 2, 1)).is_ok());

    let err = CheckSessionLimit(&ap_info(3, 2, 1)).unwrap_err();
    assert!(CanNotIdentify(&err));
}

#[test]
fn resume_and_identify_error_sets_match_expected() {
    let resume = New(CodeConnCloseCantResume, "invalid session");
    let identify = New(CodeConnCloseCantIdentify, "bot banned");
    let reconnect = New(CodeNeedReConnect, "need reconnect");

    assert!(CanNotResume(&resume));
    assert!(!CanNotResume(&identify));
    assert!(CanNotIdentify(&identify));
    assert!(!CanNotIdentify(&resume));
    assert!(!CanNotIdentify(&reconnect));
}

#[test]
fn sessions_are_generated_per_shard() {
    let token = Token::new("app_id", "secret");
    let sessions = ChanManager::sessions(&ap_info(3, 3, 1), token, Intents::default());

    let shards = sessions
        .into_iter()
        .map(|session| session.shard())
        .collect::<Vec<_>>();
    assert_eq!(shards, vec![[0, 3], [1, 3], [2, 3]]);
}

#[test]
fn app_id_session_matches_webhook_shape() {
    let session = Session::from_app_id("app-id-1");

    assert_eq!(session.app_id.as_deref(), Some("app-id-1"));
    assert!(session.id.is_empty());
    assert_eq!(session.last_seq, 0);
    assert_eq!(session.shard(), [0, 0]);
}

#[tokio::test]
async fn non_resumable_error_clears_session_before_requeue() {
    let (session_tx, mut session_rx) = mpsc::unbounded_channel();
    let (event_tx, _event_rx) = mpsc::unbounded_channel();
    let connect_fn: Arc<SessionConnectFn> = Arc::new(|session, _event_sender| {
        Box::pin(async move {
            let mut next = session;
            next.id = "stale-session".to_string();
            next.last_seq = 42;
            (
                next,
                Err(New(CodeConnCloseCantResume, "invalid session").into()),
            )
        })
    });

    ChanManager::new_connect(
        session_tx,
        connect_fn,
        event_tx,
        Session::new(
            "wss://example.com",
            Token::new("app_id", "secret"),
            Intents::default(),
            0,
            1,
        ),
    )
    .await;

    let session = session_rx.recv().await.unwrap();
    assert!(session.id.is_empty());
    assert_eq!(session.last_seq, 0);
}