source_ss 0.5.4

Media 
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

use source_ss::define::*;
use source_ss::mng;
use source_ss::relay::SourceRelayBuilder;

#[tokio::test]
async fn test_get_relays() {
    let (event_tx, _) = tokio::sync::mpsc::unbounded_channel();

    let config = SSConfig::default();
    let mng = mng::run(event_tx, &config).unwrap();
    let ctrl = mng.ctrl();

    // Initially should be empty
    let relays = ctrl.get_relays(()).await.unwrap();
    assert!(relays.is_empty(), "relays should be empty initially");

    // Add a relay source
    let relay = SourceRelayBuilder::default()
        .source_id("source_1")
        .relay_id("relay_1")
        .scheme(SSScheme::Es)
        .relay_ip("127.0.0.1")
        .build()
        .unwrap();
    ctrl.add_relay_source(relay).await.unwrap();

    // Now should have one relay
    let relays = ctrl.get_relays(()).await.unwrap();
    assert_eq!(relays.len(), 1);
    assert!(relays.contains_key("relay_1"));

    // Add another relay source
    let relay = SourceRelayBuilder::default()
        .source_id("source_2")
        .relay_id("relay_2")
        .scheme(SSScheme::Es)
        .relay_ip("127.0.0.1")
        .build()
        .unwrap();
    ctrl.add_relay_source(relay).await.unwrap();

    let relays = ctrl.get_relays(()).await.unwrap();
    assert_eq!(relays.len(), 2);
    assert!(relays.contains_key("relay_1"));
    assert!(relays.contains_key("relay_2"));

    // Subscribe with consum=true, relay should be removed
    let sub_args = SubscribeArgsBuilder::default()
        .relay_id("relay_1")
        .consum(true)
        .build()
        .unwrap();
    let _subject = ctrl.subscribe(sub_args).await.unwrap();

    // After consum, relay_1 should be gone
    let relays = ctrl.get_relays(()).await.unwrap();
    assert_eq!(relays.len(), 1);
    assert!(!relays.contains_key("relay_1"));
    assert!(relays.contains_key("relay_2"));

    // Consume the remaining relay
    let sub_args = SubscribeArgsBuilder::default()
        .relay_id("relay_2")
        .consum(true)
        .build()
        .unwrap();
    let _subject = ctrl.subscribe(sub_args).await.unwrap();

    // All relays consumed, should be empty
    let relays = ctrl.get_relays(()).await.unwrap();
    assert!(relays.is_empty(), "relays should be empty after all consumed");
}