atomic_websocket 0.9.5

High level Websocket util library from tokio-tungstenite
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

//! Feature-specific integration tests.
//!
//! Tests that verify functionality depending on specific feature flags.

mod common;

use atomic_websocket::types::save_key;

// ============================================================================
// Save Key Constants
// ============================================================================

#[test]
fn test_save_key_constants() {
    assert_eq!(save_key::SERVER_CONNECT_INFO, "ServerConnectInfo");
    assert_eq!(save_key::VALID_CLIENT, "ValidClient");
    assert_eq!(save_key::CLIENT_ID, "ClientId");
}

// ============================================================================
// Bebop Feature Tests
// ============================================================================

#[cfg(feature = "bebop")]
mod bebop_tests {
    use atomic_websocket::schema::{Category, SaveKey};
    use atomic_websocket::types::save_key;
    use atomic_websocket::types::SaveKeyExt;

    #[test]
    fn test_save_key_ext_trait() {
        assert_eq!(
            SaveKey::ServerConnectInfo.as_str(),
            save_key::SERVER_CONNECT_INFO
        );
        assert_eq!(SaveKey::ValidClient.as_str(), save_key::VALID_CLIENT);
        assert_eq!(SaveKey::ClientId.as_str(), save_key::CLIENT_ID);
    }

    #[test]
    fn test_category_values() {
        assert_eq!(Category::Ping as u16, 10000);
        assert_eq!(Category::Pong as u16, 10001);
        assert_eq!(Category::Expired as u16, 10002);
        assert_eq!(Category::Disconnect as u16, 10003);
    }
}

// ============================================================================
// Native-DB Feature Tests
// ============================================================================

#[cfg(feature = "native-db")]
mod native_db_tests {
    use atomic_websocket::Settings;

    #[test]
    fn test_settings_struct() {
        let settings = Settings {
            key: "test_key".to_string(),
            value: vec![1, 2, 3],
        };

        assert_eq!(settings.key, "test_key");
        assert_eq!(settings.value, vec![1, 2, 3]);
    }
}

// ============================================================================
// In-Memory Storage Tests (when native-db is disabled)
// ============================================================================

#[cfg(not(feature = "native-db"))]
mod in_memory_tests {
    use atomic_websocket::types::InMemoryStorage;

    #[test]
    fn test_in_memory_storage_crud() {
        let mut storage = InMemoryStorage::new();

        // Insert
        storage.insert("key1".to_string(), vec![1, 2, 3]);
        assert_eq!(storage.get("key1"), Some(&vec![1, 2, 3]));

        // Update
        storage.insert("key1".to_string(), vec![4, 5, 6]);
        assert_eq!(storage.get("key1"), Some(&vec![4, 5, 6]));

        // Remove
        let removed = storage.remove("key1");
        assert_eq!(removed, Some(vec![4, 5, 6]));
        assert!(storage.get("key1").is_none());
    }

    #[test]
    fn test_in_memory_storage_multiple_keys() {
        let mut storage = InMemoryStorage::new();

        storage.insert("a".to_string(), vec![1]);
        storage.insert("b".to_string(), vec![2]);
        storage.insert("c".to_string(), vec![3]);

        assert_eq!(storage.get("a"), Some(&vec![1]));
        assert_eq!(storage.get("b"), Some(&vec![2]));
        assert_eq!(storage.get("c"), Some(&vec![3]));
    }
}

// ============================================================================
// Rustls Feature Tests
// ============================================================================
//
// There used to be two tests here for `ClientOptions::use_tls`. Both set the
// field and then asserted it held what they had put in it — neither ever
// checked that it changed a connection. It changed nothing: the dialer read the
// scheme off the URL and never looked at the field. The tests passed for years
// while the option did nothing, which is how it survived.
//
// The scheme rule that actually decides TLS is pinned in
// `get_outer_websocket::outer_ws_url_tests`, against the function that does it.