deribit-websocket 0.3.4

WebSocket client for Deribit trading platform real-time data
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

//! Subscribe / unsubscribe round-trip against the mock server.
//!
//! Verifies that [`DeribitWebSocketClient::subscribe`] reconciles the
//! local [`SubscriptionManager`] from the server-confirmed channel list
//! and that [`DeribitWebSocketClient::unsubscribe`] removes them again.

use std::time::Duration;

use futures_util::{SinkExt, StreamExt};
use serde_json::Value;
use tokio::time::timeout;
use tokio_tungstenite::tungstenite::Message;

use super::helpers::client::mock_client;
use super::helpers::frames::{subscribe_success, unsubscribe_success};
use super::helpers::mock_server::spawn_mock_server;

const CHANNEL: &str = "ticker.BTC-PERPETUAL.raw";

#[tokio::test]
async fn subscribe_and_unsubscribe_update_manager() {
    let server = spawn_mock_server(|mut sink, mut stream| async move {
        // Accept the subscribe first, assert its method is literally
        // `public/subscribe`, then echo the channel list.
        let subscribe_text = match stream.next().await {
            Some(Ok(Message::Text(t))) => t,
            _ => return,
        };
        let subscribe_request: Value =
            serde_json::from_str(&subscribe_text).expect("subscribe request parses as JSON");
        let subscribe_id = subscribe_request.get("id").cloned().unwrap_or(Value::Null);
        assert_eq!(
            subscribe_request["method"].as_str().unwrap_or(""),
            "public/subscribe",
            "first request must be public/subscribe, got {}",
            subscribe_request["method"]
        );
        let _ = sink
            .send(Message::Text(
                subscribe_success(&subscribe_id, &[CHANNEL]).into(),
            ))
            .await;

        // Then the unsubscribe, same assertion pattern.
        let unsubscribe_text = match stream.next().await {
            Some(Ok(Message::Text(t))) => t,
            _ => return,
        };
        let unsubscribe_request: Value =
            serde_json::from_str(&unsubscribe_text).expect("unsubscribe request parses as JSON");
        let unsubscribe_id = unsubscribe_request
            .get("id")
            .cloned()
            .unwrap_or(Value::Null);
        assert_eq!(
            unsubscribe_request["method"].as_str().unwrap_or(""),
            "public/unsubscribe",
            "second request must be public/unsubscribe, got {}",
            unsubscribe_request["method"]
        );
        let _ = sink
            .send(Message::Text(
                unsubscribe_success(&unsubscribe_id, &[CHANNEL]).into(),
            ))
            .await;
    })
    .await;

    let client = mock_client(&server);

    timeout(Duration::from_secs(5), async {
        client.connect().await.expect("connect");

        client
            .subscribe(vec![CHANNEL.to_string()])
            .await
            .expect("subscribe");

        let active_after_sub = client
            .subscription_manager()
            .lock()
            .await
            .get_active_channels();
        assert!(
            active_after_sub.iter().any(|c| c == CHANNEL),
            "{} must be active after subscribe, got {:?}",
            CHANNEL,
            active_after_sub
        );

        client
            .unsubscribe(vec![CHANNEL.to_string()])
            .await
            .expect("unsubscribe");

        let active_after_unsub = client
            .subscription_manager()
            .lock()
            .await
            .get_active_channels();
        assert!(
            !active_after_unsub.iter().any(|c| c == CHANNEL),
            "{} must be gone after unsubscribe, got {:?}",
            CHANNEL,
            active_after_unsub
        );

        client.disconnect().await.expect("disconnect");
    })
    .await
    .expect("subscribe flow finishes within 5s");

    server.finish().await;
}