nautilus_bitmex/websocket/
error.rs

1// -------------------------------------------------------------------------------------------------
2//  Copyright (C) 2015-2025 Nautech Systems Pty Ltd. All rights reserved.
3//  https://nautechsystems.io
4//
5//  Licensed under the GNU Lesser General Public License Version 3.0 (the "License");
6//  You may not use this file except in compliance with the License.
7//  You may obtain a copy of the License at https://www.gnu.org/licenses/lgpl-3.0.en.html
8//
9//  Unless required by applicable law or agreed to in writing, software
10//  distributed under the License is distributed on an "AS IS" BASIS,
11//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12//  See the License for the specific language governing permissions and
13//  limitations under the License.
14// -------------------------------------------------------------------------------------------------
15
16//! Error definitions for the BitMEX WebSocket client.
17
18use thiserror::Error;
19use tokio_tungstenite::tungstenite;
20
21/// A typed error enumeration for the BitMEX WebSocket client.
22#[derive(Debug, Error)]
23pub enum BitmexWsError {
24    /// Parsing error.
25    #[error("Parsing error: {0}")]
26    ParsingError(String),
27    /// Errors returned directly by BitMEX (non-zero code).
28    #[error("BitMEX error {error_name}: {message}")]
29    BitmexError { error_name: String, message: String },
30    /// Failure during JSON serialization/deserialization.
31    #[error("JSON error: {0}")]
32    JsonError(String),
33    /// Client error.
34    #[error("Client error: {0}")]
35    ClientError(String),
36    /// Authentication error.
37    #[error("Authentication error: {0}")]
38    AuthenticationError(String),
39    /// Subscription error.
40    #[error("Subscription error: {0}")]
41    SubscriptionError(String),
42    /// WebSocket transport error.
43    #[error("Tungstenite error: {0}")]
44    TungsteniteError(#[from] tungstenite::Error),
45    /// Missing credentials for authenticated operation.
46    #[error("Missing credentials: API authentication required for this operation")]
47    MissingCredentials,
48}
49
50impl From<serde_json::Error> for BitmexWsError {
51    fn from(error: serde_json::Error) -> Self {
52        Self::JsonError(error.to_string())
53    }
54}
55
56#[cfg(test)]
57mod tests {
58    use rstest::rstest;
59
60    use super::*;
61
62    #[rstest]
63    fn test_bitmex_ws_error_display() {
64        let error = BitmexWsError::ParsingError("Invalid message format".to_string());
65        assert_eq!(error.to_string(), "Parsing error: Invalid message format");
66
67        let error = BitmexWsError::BitmexError {
68            error_name: "InvalidTopic".to_string(),
69            message: "Unknown subscription topic".to_string(),
70        };
71        assert_eq!(
72            error.to_string(),
73            "BitMEX error InvalidTopic: Unknown subscription topic"
74        );
75
76        let error = BitmexWsError::ClientError("Connection lost".to_string());
77        assert_eq!(error.to_string(), "Client error: Connection lost");
78
79        let error = BitmexWsError::AuthenticationError("Invalid API key".to_string());
80        assert_eq!(error.to_string(), "Authentication error: Invalid API key");
81
82        let error = BitmexWsError::SubscriptionError("Topic not available".to_string());
83        assert_eq!(error.to_string(), "Subscription error: Topic not available");
84
85        let error = BitmexWsError::MissingCredentials;
86        assert_eq!(
87            error.to_string(),
88            "Missing credentials: API authentication required for this operation"
89        );
90    }
91
92    #[rstest]
93    fn test_bitmex_ws_error_from_json_error() {
94        let json_err = serde_json::from_str::<serde_json::Value>("invalid json").unwrap_err();
95        let ws_error: BitmexWsError = json_err.into();
96        assert!(ws_error.to_string().contains("JSON error"));
97    }
98
99    #[rstest]
100    fn test_bitmex_ws_error_from_tungstenite() {
101        use tokio_tungstenite::tungstenite::Error as WsError;
102
103        let tungstenite_err = WsError::ConnectionClosed;
104        let ws_error: BitmexWsError = tungstenite_err.into();
105        assert!(ws_error.to_string().contains("Tungstenite error"));
106    }
107}