spikard 0.6.2 → 0.7.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/README.md +90 -508
- data/ext/spikard_rb/Cargo.lock +3287 -0
- data/ext/spikard_rb/Cargo.toml +1 -1
- data/ext/spikard_rb/extconf.rb +3 -3
- data/lib/spikard/app.rb +72 -49
- data/lib/spikard/background.rb +38 -7
- data/lib/spikard/testing.rb +42 -4
- data/lib/spikard/version.rb +1 -1
- data/sig/spikard.rbs +4 -0
- data/vendor/crates/spikard-bindings-shared/Cargo.toml +1 -1
- data/vendor/crates/spikard-bindings-shared/tests/config_extractor_behavior.rs +191 -0
- data/vendor/crates/spikard-core/Cargo.toml +1 -1
- data/vendor/crates/spikard-core/src/http.rs +1 -0
- data/vendor/crates/spikard-core/src/lifecycle.rs +63 -0
- data/vendor/crates/spikard-core/tests/bindings_response_tests.rs +136 -0
- data/vendor/crates/spikard-core/tests/di_dependency_defaults.rs +37 -0
- data/vendor/crates/spikard-core/tests/error_mapper.rs +761 -0
- data/vendor/crates/spikard-core/tests/parameters_edge_cases.rs +106 -0
- data/vendor/crates/spikard-core/tests/parameters_full.rs +701 -0
- data/vendor/crates/spikard-core/tests/parameters_schema_and_formats.rs +301 -0
- data/vendor/crates/spikard-core/tests/request_data_roundtrip.rs +67 -0
- data/vendor/crates/spikard-core/tests/validation_coverage.rs +250 -0
- data/vendor/crates/spikard-core/tests/validation_error_paths.rs +45 -0
- data/vendor/crates/spikard-http/Cargo.toml +1 -1
- data/vendor/crates/spikard-http/src/jsonrpc/http_handler.rs +502 -0
- data/vendor/crates/spikard-http/src/jsonrpc/method_registry.rs +648 -0
- data/vendor/crates/spikard-http/src/jsonrpc/mod.rs +58 -0
- data/vendor/crates/spikard-http/src/jsonrpc/protocol.rs +1207 -0
- data/vendor/crates/spikard-http/src/jsonrpc/router.rs +2262 -0
- data/vendor/crates/spikard-http/src/testing/test_client.rs +155 -2
- data/vendor/crates/spikard-http/src/testing.rs +171 -0
- data/vendor/crates/spikard-http/src/websocket.rs +79 -6
- data/vendor/crates/spikard-http/tests/auth_integration.rs +647 -0
- data/vendor/crates/spikard-http/tests/common/test_builders.rs +633 -0
- data/vendor/crates/spikard-http/tests/di_handler_error_responses.rs +162 -0
- data/vendor/crates/spikard-http/tests/middleware_stack_integration.rs +389 -0
- data/vendor/crates/spikard-http/tests/request_extraction_full.rs +513 -0
- data/vendor/crates/spikard-http/tests/server_auth_middleware_behavior.rs +244 -0
- data/vendor/crates/spikard-http/tests/server_configured_router_behavior.rs +200 -0
- data/vendor/crates/spikard-http/tests/server_cors_preflight.rs +82 -0
- data/vendor/crates/spikard-http/tests/server_handler_wrappers.rs +464 -0
- data/vendor/crates/spikard-http/tests/server_method_router_additional_behavior.rs +286 -0
- data/vendor/crates/spikard-http/tests/server_method_router_coverage.rs +118 -0
- data/vendor/crates/spikard-http/tests/server_middleware_behavior.rs +99 -0
- data/vendor/crates/spikard-http/tests/server_middleware_branches.rs +206 -0
- data/vendor/crates/spikard-http/tests/server_openapi_jsonrpc_static.rs +281 -0
- data/vendor/crates/spikard-http/tests/server_router_behavior.rs +121 -0
- data/vendor/crates/spikard-http/tests/sse_full_behavior.rs +584 -0
- data/vendor/crates/spikard-http/tests/sse_handler_behavior.rs +130 -0
- data/vendor/crates/spikard-http/tests/test_client_requests.rs +167 -0
- data/vendor/crates/spikard-http/tests/testing_helpers.rs +87 -0
- data/vendor/crates/spikard-http/tests/testing_module_coverage.rs +156 -0
- data/vendor/crates/spikard-http/tests/urlencoded_content_type.rs +82 -0
- data/vendor/crates/spikard-http/tests/websocket_full_behavior.rs +440 -0
- data/vendor/crates/spikard-http/tests/websocket_integration.rs +152 -0
- data/vendor/crates/spikard-rb/Cargo.toml +1 -1
- data/vendor/crates/spikard-rb/src/gvl.rs +80 -0
- data/vendor/crates/spikard-rb/src/handler.rs +12 -9
- data/vendor/crates/spikard-rb/src/lib.rs +137 -124
- data/vendor/crates/spikard-rb/src/request.rs +342 -0
- data/vendor/crates/spikard-rb/src/runtime/server_runner.rs +1 -8
- data/vendor/crates/spikard-rb/src/server.rs +1 -8
- data/vendor/crates/spikard-rb/src/testing/client.rs +168 -9
- data/vendor/crates/spikard-rb/src/websocket.rs +119 -30
- data/vendor/crates/spikard-rb-macros/Cargo.toml +14 -0
- data/vendor/crates/spikard-rb-macros/src/lib.rs +52 -0
- metadata +44 -1
|
@@ -0,0 +1,2262 @@
|
|
|
1
|
+
//! JSON-RPC request router for handling single and batch requests
|
|
2
|
+
//!
|
|
3
|
+
//! This module provides the core routing logic for JSON-RPC 2.0 requests, including
|
|
4
|
+
//! support for batch processing with configurable size limits. The router matches
|
|
5
|
+
//! incoming requests to registered method handlers and returns appropriately formatted
|
|
6
|
+
//! responses according to the JSON-RPC 2.0 specification.
|
|
7
|
+
//!
|
|
8
|
+
//! # Features
|
|
9
|
+
//!
|
|
10
|
+
//! - Single request routing to registered handlers
|
|
11
|
+
//! - Batch request processing with size validation
|
|
12
|
+
//! - Notification handling (requests without IDs)
|
|
13
|
+
//! - Comprehensive error handling for all JSON-RPC error codes
|
|
14
|
+
//! - Thread-safe access via Arc<JsonRpcMethodRegistry>
|
|
15
|
+
//! - Method name validation at routing time (defense in depth)
|
|
16
|
+
//!
|
|
17
|
+
//! # Validation
|
|
18
|
+
//!
|
|
19
|
+
//! Method names are validated at two points:
|
|
20
|
+
//! 1. During request parsing in `parse_request()` - validates method names when requests are deserialized
|
|
21
|
+
//! 2. During routing in `route_single()` - provides defense-in-depth validation even for programmatically created requests
|
|
22
|
+
//!
|
|
23
|
+
//! Invalid method names return a JSON-RPC error with code -32600 (Invalid Request) as per the JSON-RPC 2.0 spec.
|
|
24
|
+
//!
|
|
25
|
+
//! # Example
|
|
26
|
+
//!
|
|
27
|
+
//! ```ignore
|
|
28
|
+
//! use spikard_http::jsonrpc::{JsonRpcRouter, JsonRpcMethodRegistry, JsonRpcRequest};
|
|
29
|
+
//! use std::sync::Arc;
|
|
30
|
+
//! use serde_json::json;
|
|
31
|
+
//!
|
|
32
|
+
//! let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
33
|
+
//! let router = JsonRpcRouter::new(registry, true, 50);
|
|
34
|
+
//!
|
|
35
|
+
//! // Route a single request
|
|
36
|
+
//! let request = JsonRpcRequest::new("user.getById", Some(json!({"id": "123"})), Some(json!(1)));
|
|
37
|
+
//! let response = router.route_single(request).await;
|
|
38
|
+
//! ```
|
|
39
|
+
|
|
40
|
+
use super::method_registry::JsonRpcMethodRegistry;
|
|
41
|
+
use super::protocol::*;
|
|
42
|
+
use crate::handler_trait::RequestData;
|
|
43
|
+
use axum::body::Body;
|
|
44
|
+
use axum::http::Request;
|
|
45
|
+
use serde_json::Value;
|
|
46
|
+
use std::sync::Arc;
|
|
47
|
+
|
|
48
|
+
/// JSON-RPC router for handling single and batch requests
|
|
49
|
+
///
|
|
50
|
+
/// Manages request routing to registered method handlers with support for
|
|
51
|
+
/// batch processing, notifications, and comprehensive error handling.
|
|
52
|
+
pub struct JsonRpcRouter {
|
|
53
|
+
/// Registry of available methods and their handlers
|
|
54
|
+
registry: Arc<JsonRpcMethodRegistry>,
|
|
55
|
+
/// Whether batch requests are enabled
|
|
56
|
+
enable_batch: bool,
|
|
57
|
+
/// Maximum number of requests allowed in a single batch
|
|
58
|
+
max_batch_size: usize,
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
impl JsonRpcRouter {
|
|
62
|
+
/// Creates a new JSON-RPC router
|
|
63
|
+
///
|
|
64
|
+
/// # Arguments
|
|
65
|
+
///
|
|
66
|
+
/// * `registry` - The method registry containing registered handlers
|
|
67
|
+
/// * `enable_batch` - Whether to allow batch requests
|
|
68
|
+
/// * `max_batch_size` - Maximum number of requests per batch
|
|
69
|
+
///
|
|
70
|
+
/// # Example
|
|
71
|
+
///
|
|
72
|
+
/// ```ignore
|
|
73
|
+
/// use spikard_http::jsonrpc::{JsonRpcRouter, JsonRpcMethodRegistry};
|
|
74
|
+
/// use std::sync::Arc;
|
|
75
|
+
///
|
|
76
|
+
/// let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
77
|
+
/// let router = JsonRpcRouter::new(registry, true, 100);
|
|
78
|
+
/// ```
|
|
79
|
+
pub fn new(registry: Arc<JsonRpcMethodRegistry>, enable_batch: bool, max_batch_size: usize) -> Self {
|
|
80
|
+
Self {
|
|
81
|
+
registry,
|
|
82
|
+
enable_batch,
|
|
83
|
+
max_batch_size,
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
|
|
87
|
+
/// Routes a single JSON-RPC request to its handler
|
|
88
|
+
///
|
|
89
|
+
/// Processes a single request by:
|
|
90
|
+
/// 1. Checking if the method exists in the registry
|
|
91
|
+
/// 2. Handling notifications (requests without IDs)
|
|
92
|
+
/// 3. Invoking the handler with the HTTP request context
|
|
93
|
+
/// 4. Converting handler responses to JSON-RPC format
|
|
94
|
+
/// 5. Returning appropriately formatted responses
|
|
95
|
+
///
|
|
96
|
+
/// For notifications, the server MUST NOT send a response.
|
|
97
|
+
/// The response is still generated but marked as not-to-be-sent by the caller.
|
|
98
|
+
///
|
|
99
|
+
/// # Arguments
|
|
100
|
+
///
|
|
101
|
+
/// * `request` - The JSON-RPC request to route
|
|
102
|
+
/// * `http_request` - The HTTP request context (headers, method, etc.)
|
|
103
|
+
/// * `request_data` - Reference to extracted request data (params, body, etc.)
|
|
104
|
+
///
|
|
105
|
+
/// # Returns
|
|
106
|
+
///
|
|
107
|
+
/// A `JsonRpcResponseType` containing either a success response with the
|
|
108
|
+
/// handler's result or an error response if the method is not found or
|
|
109
|
+
/// the handler fails
|
|
110
|
+
pub async fn route_single(
|
|
111
|
+
&self,
|
|
112
|
+
request: JsonRpcRequest,
|
|
113
|
+
http_request: Request<Body>,
|
|
114
|
+
request_data: &RequestData,
|
|
115
|
+
) -> JsonRpcResponseType {
|
|
116
|
+
if let Err(validation_error) = super::protocol::validate_method_name(&request.method) {
|
|
117
|
+
let id = request.id.unwrap_or(Value::Null);
|
|
118
|
+
return JsonRpcResponseType::Error(JsonRpcErrorResponse::error(
|
|
119
|
+
error_codes::INVALID_REQUEST,
|
|
120
|
+
format!("Invalid method name: {}", validation_error),
|
|
121
|
+
id,
|
|
122
|
+
));
|
|
123
|
+
}
|
|
124
|
+
|
|
125
|
+
let handler = match self.registry.get(&request.method) {
|
|
126
|
+
Ok(Some(h)) => h,
|
|
127
|
+
Ok(None) => {
|
|
128
|
+
let id = request.id.unwrap_or(Value::Null);
|
|
129
|
+
return JsonRpcResponseType::Error(JsonRpcErrorResponse::error(
|
|
130
|
+
error_codes::METHOD_NOT_FOUND,
|
|
131
|
+
"Method not found",
|
|
132
|
+
id,
|
|
133
|
+
));
|
|
134
|
+
}
|
|
135
|
+
Err(e) => {
|
|
136
|
+
let id = request.id.unwrap_or(Value::Null);
|
|
137
|
+
return JsonRpcResponseType::Error(JsonRpcErrorResponse::error(
|
|
138
|
+
error_codes::INTERNAL_ERROR,
|
|
139
|
+
format!("Internal error: {}", e),
|
|
140
|
+
id,
|
|
141
|
+
));
|
|
142
|
+
}
|
|
143
|
+
};
|
|
144
|
+
|
|
145
|
+
let _is_notification = request.is_notification();
|
|
146
|
+
|
|
147
|
+
let handler_result = handler.call(http_request, request_data.clone()).await;
|
|
148
|
+
|
|
149
|
+
match handler_result {
|
|
150
|
+
Ok(response) => {
|
|
151
|
+
let body_bytes = axum::body::to_bytes(response.into_body(), usize::MAX)
|
|
152
|
+
.await
|
|
153
|
+
.unwrap_or_default();
|
|
154
|
+
|
|
155
|
+
let result = if body_bytes.is_empty() {
|
|
156
|
+
Value::Null
|
|
157
|
+
} else {
|
|
158
|
+
match serde_json::from_slice::<Value>(&body_bytes) {
|
|
159
|
+
Ok(json_val) => json_val,
|
|
160
|
+
Err(_) => Value::String(
|
|
161
|
+
String::from_utf8(body_bytes.to_vec()).unwrap_or_else(|_| "[binary data]".to_string()),
|
|
162
|
+
),
|
|
163
|
+
}
|
|
164
|
+
};
|
|
165
|
+
|
|
166
|
+
let id = request.id.unwrap_or(Value::Null);
|
|
167
|
+
JsonRpcResponseType::Success(JsonRpcResponse::success(result, id))
|
|
168
|
+
}
|
|
169
|
+
Err((_status, error_msg)) => {
|
|
170
|
+
let id = request.id.unwrap_or(Value::Null);
|
|
171
|
+
let error_data = serde_json::json!({
|
|
172
|
+
"details": error_msg
|
|
173
|
+
});
|
|
174
|
+
JsonRpcResponseType::Error(JsonRpcErrorResponse::error_with_data(
|
|
175
|
+
error_codes::INTERNAL_ERROR,
|
|
176
|
+
"Internal error from handler",
|
|
177
|
+
error_data,
|
|
178
|
+
id,
|
|
179
|
+
))
|
|
180
|
+
}
|
|
181
|
+
}
|
|
182
|
+
}
|
|
183
|
+
|
|
184
|
+
/// Routes a batch of JSON-RPC requests
|
|
185
|
+
///
|
|
186
|
+
/// Processes a batch of requests by:
|
|
187
|
+
/// 1. Checking if batch processing is enabled
|
|
188
|
+
/// 2. Validating batch size doesn't exceed the limit
|
|
189
|
+
/// 3. Ensuring batch is not empty
|
|
190
|
+
/// 4. Routing each request in sequence
|
|
191
|
+
/// 5. Filtering out notification responses
|
|
192
|
+
///
|
|
193
|
+
/// According to JSON-RPC 2.0 spec, the server SHOULD process all requests
|
|
194
|
+
/// in the batch and return a JSON array of responses. Responses for
|
|
195
|
+
/// notifications (requests without IDs) are not included in the result.
|
|
196
|
+
///
|
|
197
|
+
/// # Arguments
|
|
198
|
+
///
|
|
199
|
+
/// * `batch` - A vector of JSON-RPC requests
|
|
200
|
+
/// * `http_request` - The HTTP request context (shared for all batch requests)
|
|
201
|
+
/// * `request_data` - Extracted request data (shared for all batch requests)
|
|
202
|
+
///
|
|
203
|
+
/// # Returns
|
|
204
|
+
///
|
|
205
|
+
/// * `Ok(Vec<JsonRpcResponseType>)` - Array of responses for all non-notification requests
|
|
206
|
+
/// * `Err(JsonRpcErrorResponse)` - Single error if batch validation fails
|
|
207
|
+
///
|
|
208
|
+
/// # Errors
|
|
209
|
+
///
|
|
210
|
+
/// Returns an error if:
|
|
211
|
+
/// - Batch requests are not enabled
|
|
212
|
+
/// - Batch size exceeds the configured maximum
|
|
213
|
+
/// - Batch is empty
|
|
214
|
+
pub async fn route_batch(
|
|
215
|
+
&self,
|
|
216
|
+
batch: Vec<JsonRpcRequest>,
|
|
217
|
+
http_request: Request<Body>,
|
|
218
|
+
request_data: &RequestData,
|
|
219
|
+
) -> Result<Vec<JsonRpcResponseType>, JsonRpcErrorResponse> {
|
|
220
|
+
if !self.enable_batch {
|
|
221
|
+
return Err(JsonRpcErrorResponse::error(
|
|
222
|
+
error_codes::INVALID_REQUEST,
|
|
223
|
+
"Batch requests not enabled",
|
|
224
|
+
Value::Null,
|
|
225
|
+
));
|
|
226
|
+
}
|
|
227
|
+
|
|
228
|
+
if batch.len() > self.max_batch_size {
|
|
229
|
+
return Err(JsonRpcErrorResponse::error(
|
|
230
|
+
error_codes::INVALID_REQUEST,
|
|
231
|
+
format!("Batch size {} exceeds maximum {}", batch.len(), self.max_batch_size),
|
|
232
|
+
Value::Null,
|
|
233
|
+
));
|
|
234
|
+
}
|
|
235
|
+
|
|
236
|
+
if batch.is_empty() {
|
|
237
|
+
return Err(JsonRpcErrorResponse::error(
|
|
238
|
+
error_codes::INVALID_REQUEST,
|
|
239
|
+
"Batch request cannot be empty",
|
|
240
|
+
Value::Null,
|
|
241
|
+
));
|
|
242
|
+
}
|
|
243
|
+
|
|
244
|
+
let (base_parts, _body) = http_request.into_parts();
|
|
245
|
+
|
|
246
|
+
let mut responses = Vec::with_capacity(batch.len());
|
|
247
|
+
for request in batch {
|
|
248
|
+
let is_notification = request.is_notification();
|
|
249
|
+
|
|
250
|
+
let req_for_handler = Request::from_parts(base_parts.clone(), Body::empty());
|
|
251
|
+
|
|
252
|
+
let response = self.route_single(request, req_for_handler, request_data).await;
|
|
253
|
+
|
|
254
|
+
if !is_notification {
|
|
255
|
+
responses.push(response);
|
|
256
|
+
}
|
|
257
|
+
}
|
|
258
|
+
|
|
259
|
+
Ok(responses)
|
|
260
|
+
}
|
|
261
|
+
|
|
262
|
+
/// Parses a JSON string into either a single request or a batch
|
|
263
|
+
///
|
|
264
|
+
/// Attempts to deserialize the input as a single JSON-RPC request first,
|
|
265
|
+
/// then tries batch parsing if that fails. Returns a parse error if both
|
|
266
|
+
/// attempts fail.
|
|
267
|
+
///
|
|
268
|
+
/// # Arguments
|
|
269
|
+
///
|
|
270
|
+
/// * `body` - The raw JSON request body as a string
|
|
271
|
+
///
|
|
272
|
+
/// # Returns
|
|
273
|
+
///
|
|
274
|
+
/// * `Ok(JsonRpcRequestOrBatch::Single(req))` - Parsed single request
|
|
275
|
+
/// * `Ok(JsonRpcRequestOrBatch::Batch(requests))` - Parsed batch request
|
|
276
|
+
/// * `Err(JsonRpcErrorResponse)` - Parse error
|
|
277
|
+
///
|
|
278
|
+
/// # Example
|
|
279
|
+
///
|
|
280
|
+
/// ```ignore
|
|
281
|
+
/// let single_json = r#"{"jsonrpc":"2.0","method":"test","id":1}"#;
|
|
282
|
+
/// let parsed = JsonRpcRouter::parse_request(single_json);
|
|
283
|
+
/// assert!(parsed.is_ok());
|
|
284
|
+
///
|
|
285
|
+
/// let batch_json = r#"[{"jsonrpc":"2.0","method":"test","id":1}]"#;
|
|
286
|
+
/// let parsed = JsonRpcRouter::parse_request(batch_json);
|
|
287
|
+
/// assert!(parsed.is_ok());
|
|
288
|
+
/// ```
|
|
289
|
+
pub fn parse_request(body: &str) -> Result<JsonRpcRequestOrBatch, Box<JsonRpcErrorResponse>> {
|
|
290
|
+
if let Ok(request) = serde_json::from_str::<JsonRpcRequest>(body) {
|
|
291
|
+
if let Err(validation_error) = super::protocol::validate_method_name(&request.method) {
|
|
292
|
+
let id = request.id.unwrap_or(Value::Null);
|
|
293
|
+
return Err(Box::new(JsonRpcErrorResponse::error(
|
|
294
|
+
error_codes::INVALID_REQUEST,
|
|
295
|
+
format!("Invalid method name: {}", validation_error),
|
|
296
|
+
id,
|
|
297
|
+
)));
|
|
298
|
+
}
|
|
299
|
+
return Ok(JsonRpcRequestOrBatch::Single(request));
|
|
300
|
+
}
|
|
301
|
+
|
|
302
|
+
if let Ok(batch) = serde_json::from_str::<Vec<JsonRpcRequest>>(body) {
|
|
303
|
+
for request in &batch {
|
|
304
|
+
if let Err(validation_error) = super::protocol::validate_method_name(&request.method) {
|
|
305
|
+
let id = request.id.clone().unwrap_or(Value::Null);
|
|
306
|
+
return Err(Box::new(JsonRpcErrorResponse::error(
|
|
307
|
+
error_codes::INVALID_REQUEST,
|
|
308
|
+
format!("Invalid method name: {}", validation_error),
|
|
309
|
+
id,
|
|
310
|
+
)));
|
|
311
|
+
}
|
|
312
|
+
}
|
|
313
|
+
return Ok(JsonRpcRequestOrBatch::Batch(batch));
|
|
314
|
+
}
|
|
315
|
+
|
|
316
|
+
Err(Box::new(JsonRpcErrorResponse::error(
|
|
317
|
+
error_codes::PARSE_ERROR,
|
|
318
|
+
"Parse error",
|
|
319
|
+
Value::Null,
|
|
320
|
+
)))
|
|
321
|
+
}
|
|
322
|
+
}
|
|
323
|
+
|
|
324
|
+
/// Represents either a single JSON-RPC request or a batch of requests
|
|
325
|
+
///
|
|
326
|
+
/// Used to distinguish between single and batch requests after parsing,
|
|
327
|
+
/// allowing different routing logic for each case.
|
|
328
|
+
#[derive(Debug)]
|
|
329
|
+
pub enum JsonRpcRequestOrBatch {
|
|
330
|
+
/// A single JSON-RPC request
|
|
331
|
+
Single(JsonRpcRequest),
|
|
332
|
+
/// A batch (array) of JSON-RPC requests
|
|
333
|
+
Batch(Vec<JsonRpcRequest>),
|
|
334
|
+
}
|
|
335
|
+
|
|
336
|
+
#[cfg(test)]
|
|
337
|
+
mod tests {
|
|
338
|
+
use super::*;
|
|
339
|
+
use crate::handler_trait::{Handler, HandlerResult, RequestData};
|
|
340
|
+
use crate::jsonrpc::MethodMetadata;
|
|
341
|
+
use axum::body::Body;
|
|
342
|
+
use axum::http::Request;
|
|
343
|
+
use serde_json::json;
|
|
344
|
+
use std::collections::HashMap;
|
|
345
|
+
use std::sync::Arc;
|
|
346
|
+
|
|
347
|
+
/// Helper function to create minimal RequestData for tests
|
|
348
|
+
fn create_test_request_data() -> RequestData {
|
|
349
|
+
RequestData {
|
|
350
|
+
path_params: Arc::new(HashMap::new()),
|
|
351
|
+
query_params: Value::Object(serde_json::Map::new()),
|
|
352
|
+
validated_params: None,
|
|
353
|
+
raw_query_params: Arc::new(HashMap::new()),
|
|
354
|
+
body: Value::Null,
|
|
355
|
+
raw_body: None,
|
|
356
|
+
headers: Arc::new(HashMap::new()),
|
|
357
|
+
cookies: Arc::new(HashMap::new()),
|
|
358
|
+
method: "POST".to_string(),
|
|
359
|
+
path: "/rpc".to_string(),
|
|
360
|
+
#[cfg(feature = "di")]
|
|
361
|
+
dependencies: None,
|
|
362
|
+
}
|
|
363
|
+
}
|
|
364
|
+
|
|
365
|
+
/// Helper function to create a test HTTP request
|
|
366
|
+
fn create_test_http_request() -> Request<Body> {
|
|
367
|
+
Request::builder()
|
|
368
|
+
.method("POST")
|
|
369
|
+
.uri("/rpc")
|
|
370
|
+
.body(Body::empty())
|
|
371
|
+
.unwrap()
|
|
372
|
+
}
|
|
373
|
+
|
|
374
|
+
/// Mock handler that returns success with JSON
|
|
375
|
+
struct MockSuccessHandler;
|
|
376
|
+
|
|
377
|
+
impl Handler for MockSuccessHandler {
|
|
378
|
+
fn call(
|
|
379
|
+
&self,
|
|
380
|
+
_request: Request<Body>,
|
|
381
|
+
_request_data: RequestData,
|
|
382
|
+
) -> std::pin::Pin<Box<dyn std::future::Future<Output = HandlerResult> + Send + '_>> {
|
|
383
|
+
Box::pin(async {
|
|
384
|
+
use axum::response::Response;
|
|
385
|
+
let response = Response::builder()
|
|
386
|
+
.status(200)
|
|
387
|
+
.header("content-type", "application/json")
|
|
388
|
+
.body(Body::from(r#"{"result":"success"}"#))
|
|
389
|
+
.unwrap();
|
|
390
|
+
Ok(response)
|
|
391
|
+
})
|
|
392
|
+
}
|
|
393
|
+
}
|
|
394
|
+
|
|
395
|
+
/// Mock handler that returns an error
|
|
396
|
+
struct MockErrorHandler;
|
|
397
|
+
|
|
398
|
+
impl Handler for MockErrorHandler {
|
|
399
|
+
fn call(
|
|
400
|
+
&self,
|
|
401
|
+
_request: Request<Body>,
|
|
402
|
+
_request_data: RequestData,
|
|
403
|
+
) -> std::pin::Pin<Box<dyn std::future::Future<Output = HandlerResult> + Send + '_>> {
|
|
404
|
+
Box::pin(async {
|
|
405
|
+
Err((
|
|
406
|
+
axum::http::StatusCode::INTERNAL_SERVER_ERROR,
|
|
407
|
+
"Handler error".to_string(),
|
|
408
|
+
))
|
|
409
|
+
})
|
|
410
|
+
}
|
|
411
|
+
}
|
|
412
|
+
|
|
413
|
+
/// Mock handler that returns success with non-JSON UTF-8 text
|
|
414
|
+
struct MockTextHandler;
|
|
415
|
+
|
|
416
|
+
impl Handler for MockTextHandler {
|
|
417
|
+
fn call(
|
|
418
|
+
&self,
|
|
419
|
+
_request: Request<Body>,
|
|
420
|
+
_request_data: RequestData,
|
|
421
|
+
) -> std::pin::Pin<Box<dyn std::future::Future<Output = HandlerResult> + Send + '_>> {
|
|
422
|
+
Box::pin(async {
|
|
423
|
+
use axum::response::Response;
|
|
424
|
+
let response = Response::builder()
|
|
425
|
+
.status(200)
|
|
426
|
+
.header("content-type", "text/plain")
|
|
427
|
+
.body(Body::from("hello"))
|
|
428
|
+
.unwrap();
|
|
429
|
+
Ok(response)
|
|
430
|
+
})
|
|
431
|
+
}
|
|
432
|
+
}
|
|
433
|
+
|
|
434
|
+
/// Mock handler that returns success with non-UTF-8 bytes
|
|
435
|
+
struct MockBinaryHandler;
|
|
436
|
+
|
|
437
|
+
impl Handler for MockBinaryHandler {
|
|
438
|
+
fn call(
|
|
439
|
+
&self,
|
|
440
|
+
_request: Request<Body>,
|
|
441
|
+
_request_data: RequestData,
|
|
442
|
+
) -> std::pin::Pin<Box<dyn std::future::Future<Output = HandlerResult> + Send + '_>> {
|
|
443
|
+
Box::pin(async {
|
|
444
|
+
use axum::response::Response;
|
|
445
|
+
let response = Response::builder()
|
|
446
|
+
.status(200)
|
|
447
|
+
.header("content-type", "application/octet-stream")
|
|
448
|
+
.body(Body::from(vec![0xff, 0xfe, 0xfd]))
|
|
449
|
+
.unwrap();
|
|
450
|
+
Ok(response)
|
|
451
|
+
})
|
|
452
|
+
}
|
|
453
|
+
}
|
|
454
|
+
|
|
455
|
+
#[tokio::test]
|
|
456
|
+
async fn test_route_single_invalid_method_name_empty() {
|
|
457
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
458
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
459
|
+
|
|
460
|
+
let request = JsonRpcRequest::new("", None, Some(json!(1)));
|
|
461
|
+
let http_request = create_test_http_request();
|
|
462
|
+
let request_data = create_test_request_data();
|
|
463
|
+
|
|
464
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
465
|
+
|
|
466
|
+
match response {
|
|
467
|
+
JsonRpcResponseType::Error(err) => {
|
|
468
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
469
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
470
|
+
assert_eq!(err.id, json!(1));
|
|
471
|
+
}
|
|
472
|
+
_ => panic!("Expected error response for invalid method name"),
|
|
473
|
+
}
|
|
474
|
+
}
|
|
475
|
+
|
|
476
|
+
#[tokio::test]
|
|
477
|
+
async fn test_route_single_invalid_method_name_with_space() {
|
|
478
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
479
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
480
|
+
|
|
481
|
+
let request = JsonRpcRequest::new("method name", None, Some(json!(1)));
|
|
482
|
+
let http_request = create_test_http_request();
|
|
483
|
+
let request_data = create_test_request_data();
|
|
484
|
+
|
|
485
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
486
|
+
|
|
487
|
+
match response {
|
|
488
|
+
JsonRpcResponseType::Error(err) => {
|
|
489
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
490
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
491
|
+
assert_eq!(err.id, json!(1));
|
|
492
|
+
}
|
|
493
|
+
_ => panic!("Expected error response for invalid method name"),
|
|
494
|
+
}
|
|
495
|
+
}
|
|
496
|
+
|
|
497
|
+
#[tokio::test]
|
|
498
|
+
async fn test_route_single_success_non_json_utf8_body_returns_string() {
|
|
499
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
500
|
+
registry
|
|
501
|
+
.register("echo", Arc::new(MockTextHandler), MethodMetadata::new("echo"))
|
|
502
|
+
.unwrap();
|
|
503
|
+
|
|
504
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
505
|
+
let request = JsonRpcRequest::new("echo", None, Some(json!(1)));
|
|
506
|
+
let http_request = create_test_http_request();
|
|
507
|
+
let request_data = create_test_request_data();
|
|
508
|
+
|
|
509
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
510
|
+
match response {
|
|
511
|
+
JsonRpcResponseType::Success(ok) => {
|
|
512
|
+
assert_eq!(ok.result, Value::String("hello".to_string()));
|
|
513
|
+
}
|
|
514
|
+
other => panic!("Expected success response, got: {:?}", other),
|
|
515
|
+
}
|
|
516
|
+
}
|
|
517
|
+
|
|
518
|
+
#[tokio::test]
|
|
519
|
+
async fn test_route_single_success_non_utf8_body_returns_placeholder_string() {
|
|
520
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
521
|
+
registry
|
|
522
|
+
.register("bin", Arc::new(MockBinaryHandler), MethodMetadata::new("bin"))
|
|
523
|
+
.unwrap();
|
|
524
|
+
|
|
525
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
526
|
+
let request = JsonRpcRequest::new("bin", None, Some(json!(1)));
|
|
527
|
+
let http_request = create_test_http_request();
|
|
528
|
+
let request_data = create_test_request_data();
|
|
529
|
+
|
|
530
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
531
|
+
match response {
|
|
532
|
+
JsonRpcResponseType::Success(ok) => {
|
|
533
|
+
assert_eq!(ok.result, Value::String("[binary data]".to_string()));
|
|
534
|
+
}
|
|
535
|
+
other => panic!("Expected success response, got: {:?}", other),
|
|
536
|
+
}
|
|
537
|
+
}
|
|
538
|
+
|
|
539
|
+
#[tokio::test]
|
|
540
|
+
async fn test_route_single_invalid_method_name_control_char() {
|
|
541
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
542
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
543
|
+
|
|
544
|
+
let request = JsonRpcRequest::new("method\nname", None, Some(json!(1)));
|
|
545
|
+
let http_request = create_test_http_request();
|
|
546
|
+
let request_data = create_test_request_data();
|
|
547
|
+
|
|
548
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
549
|
+
|
|
550
|
+
match response {
|
|
551
|
+
JsonRpcResponseType::Error(err) => {
|
|
552
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
553
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
554
|
+
assert_eq!(err.id, json!(1));
|
|
555
|
+
}
|
|
556
|
+
_ => panic!("Expected error response for invalid method name"),
|
|
557
|
+
}
|
|
558
|
+
}
|
|
559
|
+
|
|
560
|
+
#[tokio::test]
|
|
561
|
+
async fn test_route_single_invalid_method_name_special_char() {
|
|
562
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
563
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
564
|
+
|
|
565
|
+
let request = JsonRpcRequest::new("method@name", None, Some(json!(1)));
|
|
566
|
+
let http_request = create_test_http_request();
|
|
567
|
+
let request_data = create_test_request_data();
|
|
568
|
+
|
|
569
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
570
|
+
|
|
571
|
+
match response {
|
|
572
|
+
JsonRpcResponseType::Error(err) => {
|
|
573
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
574
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
575
|
+
assert_eq!(err.id, json!(1));
|
|
576
|
+
}
|
|
577
|
+
_ => panic!("Expected error response for invalid method name"),
|
|
578
|
+
}
|
|
579
|
+
}
|
|
580
|
+
|
|
581
|
+
#[tokio::test]
|
|
582
|
+
async fn test_route_single_invalid_method_name_too_long() {
|
|
583
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
584
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
585
|
+
|
|
586
|
+
let long_method = "a".repeat(256);
|
|
587
|
+
let request = JsonRpcRequest::new(long_method, None, Some(json!(1)));
|
|
588
|
+
let http_request = create_test_http_request();
|
|
589
|
+
let request_data = create_test_request_data();
|
|
590
|
+
|
|
591
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
592
|
+
|
|
593
|
+
match response {
|
|
594
|
+
JsonRpcResponseType::Error(err) => {
|
|
595
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
596
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
597
|
+
assert_eq!(err.id, json!(1));
|
|
598
|
+
}
|
|
599
|
+
_ => panic!("Expected error response for invalid method name"),
|
|
600
|
+
}
|
|
601
|
+
}
|
|
602
|
+
|
|
603
|
+
#[tokio::test]
|
|
604
|
+
async fn test_route_single_method_not_found() {
|
|
605
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
606
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
607
|
+
|
|
608
|
+
let request = JsonRpcRequest::new("unknown_method", None, Some(json!(1)));
|
|
609
|
+
let http_request = create_test_http_request();
|
|
610
|
+
let request_data = create_test_request_data();
|
|
611
|
+
|
|
612
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
613
|
+
|
|
614
|
+
match response {
|
|
615
|
+
JsonRpcResponseType::Error(err) => {
|
|
616
|
+
assert_eq!(err.error.code, error_codes::METHOD_NOT_FOUND);
|
|
617
|
+
assert_eq!(err.id, json!(1));
|
|
618
|
+
}
|
|
619
|
+
_ => panic!("Expected error response"),
|
|
620
|
+
}
|
|
621
|
+
}
|
|
622
|
+
|
|
623
|
+
#[tokio::test]
|
|
624
|
+
async fn test_route_single_notification() {
|
|
625
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
626
|
+
|
|
627
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
628
|
+
let metadata = super::super::method_registry::MethodMetadata::new("notify_method");
|
|
629
|
+
registry.register("notify_method", handler, metadata).unwrap();
|
|
630
|
+
|
|
631
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
632
|
+
|
|
633
|
+
let request = JsonRpcRequest::new("notify_method", None, None);
|
|
634
|
+
assert!(request.is_notification());
|
|
635
|
+
|
|
636
|
+
let http_request = create_test_http_request();
|
|
637
|
+
let request_data = create_test_request_data();
|
|
638
|
+
|
|
639
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
640
|
+
|
|
641
|
+
match response {
|
|
642
|
+
JsonRpcResponseType::Success(resp) => {
|
|
643
|
+
assert_eq!(resp.id, Value::Null);
|
|
644
|
+
}
|
|
645
|
+
_ => panic!("Expected success response for notification"),
|
|
646
|
+
}
|
|
647
|
+
}
|
|
648
|
+
|
|
649
|
+
#[tokio::test]
|
|
650
|
+
async fn test_route_single_with_handler_success() {
|
|
651
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
652
|
+
|
|
653
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
654
|
+
let metadata = super::super::method_registry::MethodMetadata::new("test_method");
|
|
655
|
+
registry.register("test_method", handler, metadata).unwrap();
|
|
656
|
+
|
|
657
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
658
|
+
|
|
659
|
+
let request = JsonRpcRequest::new("test_method", None, Some(json!(1)));
|
|
660
|
+
let http_request = create_test_http_request();
|
|
661
|
+
let request_data = create_test_request_data();
|
|
662
|
+
|
|
663
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
664
|
+
|
|
665
|
+
match response {
|
|
666
|
+
JsonRpcResponseType::Success(resp) => {
|
|
667
|
+
assert_eq!(resp.result, json!({"result":"success"}));
|
|
668
|
+
assert_eq!(resp.id, json!(1));
|
|
669
|
+
}
|
|
670
|
+
_ => panic!("Expected success response"),
|
|
671
|
+
}
|
|
672
|
+
}
|
|
673
|
+
|
|
674
|
+
#[tokio::test]
|
|
675
|
+
async fn test_route_single_with_handler_error() {
|
|
676
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
677
|
+
|
|
678
|
+
let handler = Arc::new(MockErrorHandler);
|
|
679
|
+
let metadata = super::super::method_registry::MethodMetadata::new("error_method");
|
|
680
|
+
registry.register("error_method", handler, metadata).unwrap();
|
|
681
|
+
|
|
682
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
683
|
+
|
|
684
|
+
let request = JsonRpcRequest::new("error_method", None, Some(json!(1)));
|
|
685
|
+
let http_request = create_test_http_request();
|
|
686
|
+
let request_data = create_test_request_data();
|
|
687
|
+
|
|
688
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
689
|
+
|
|
690
|
+
match response {
|
|
691
|
+
JsonRpcResponseType::Error(err) => {
|
|
692
|
+
assert_eq!(err.error.code, error_codes::INTERNAL_ERROR);
|
|
693
|
+
assert_eq!(err.id, json!(1));
|
|
694
|
+
assert!(err.error.data.is_some());
|
|
695
|
+
}
|
|
696
|
+
_ => panic!("Expected error response"),
|
|
697
|
+
}
|
|
698
|
+
}
|
|
699
|
+
|
|
700
|
+
#[tokio::test]
|
|
701
|
+
async fn test_route_batch_disabled() {
|
|
702
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
703
|
+
let router = JsonRpcRouter::new(registry, false, 100);
|
|
704
|
+
|
|
705
|
+
let batch = vec![JsonRpcRequest::new("method", None, Some(json!(1)))];
|
|
706
|
+
let http_request = create_test_http_request();
|
|
707
|
+
let request_data = create_test_request_data();
|
|
708
|
+
|
|
709
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
710
|
+
|
|
711
|
+
assert!(result.is_err());
|
|
712
|
+
let err = result.unwrap_err();
|
|
713
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
714
|
+
}
|
|
715
|
+
|
|
716
|
+
#[tokio::test]
|
|
717
|
+
async fn test_route_batch_empty() {
|
|
718
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
719
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
720
|
+
|
|
721
|
+
let batch = vec![];
|
|
722
|
+
let http_request = create_test_http_request();
|
|
723
|
+
let request_data = create_test_request_data();
|
|
724
|
+
|
|
725
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
726
|
+
|
|
727
|
+
assert!(result.is_err());
|
|
728
|
+
let err = result.unwrap_err();
|
|
729
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
730
|
+
}
|
|
731
|
+
|
|
732
|
+
#[tokio::test]
|
|
733
|
+
async fn test_route_batch_size_exceeded() {
|
|
734
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
735
|
+
let router = JsonRpcRouter::new(registry, true, 5);
|
|
736
|
+
|
|
737
|
+
let batch = (1..=10)
|
|
738
|
+
.map(|i| JsonRpcRequest::new("method", None, Some(json!(i))))
|
|
739
|
+
.collect();
|
|
740
|
+
let http_request = create_test_http_request();
|
|
741
|
+
let request_data = create_test_request_data();
|
|
742
|
+
|
|
743
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
744
|
+
|
|
745
|
+
assert!(result.is_err());
|
|
746
|
+
let err = result.unwrap_err();
|
|
747
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
748
|
+
}
|
|
749
|
+
|
|
750
|
+
#[tokio::test]
|
|
751
|
+
async fn test_route_batch_filters_notifications() {
|
|
752
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
753
|
+
|
|
754
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
755
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
756
|
+
registry.register("method", handler, metadata).unwrap();
|
|
757
|
+
|
|
758
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
759
|
+
|
|
760
|
+
let batch = vec![
|
|
761
|
+
JsonRpcRequest::new("method", None, Some(json!(1))),
|
|
762
|
+
JsonRpcRequest::new("method", None, None),
|
|
763
|
+
JsonRpcRequest::new("method", None, Some(json!(2))),
|
|
764
|
+
];
|
|
765
|
+
|
|
766
|
+
let http_request = create_test_http_request();
|
|
767
|
+
let request_data = create_test_request_data();
|
|
768
|
+
|
|
769
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
770
|
+
assert!(result.is_ok());
|
|
771
|
+
|
|
772
|
+
let responses = result.unwrap();
|
|
773
|
+
assert_eq!(responses.len(), 2);
|
|
774
|
+
}
|
|
775
|
+
|
|
776
|
+
#[test]
|
|
777
|
+
fn test_parse_request_single() {
|
|
778
|
+
let json = r#"{"jsonrpc":"2.0","method":"test","id":1}"#;
|
|
779
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
780
|
+
|
|
781
|
+
assert!(parsed.is_ok());
|
|
782
|
+
match parsed.unwrap() {
|
|
783
|
+
JsonRpcRequestOrBatch::Single(req) => {
|
|
784
|
+
assert_eq!(req.method, "test");
|
|
785
|
+
assert_eq!(req.id, Some(json!(1)));
|
|
786
|
+
}
|
|
787
|
+
_ => panic!("Expected single request"),
|
|
788
|
+
}
|
|
789
|
+
}
|
|
790
|
+
|
|
791
|
+
#[test]
|
|
792
|
+
fn test_parse_request_batch() {
|
|
793
|
+
let json = r#"[{"jsonrpc":"2.0","method":"test","id":1},{"jsonrpc":"2.0","method":"test2","id":2}]"#;
|
|
794
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
795
|
+
|
|
796
|
+
assert!(parsed.is_ok());
|
|
797
|
+
match parsed.unwrap() {
|
|
798
|
+
JsonRpcRequestOrBatch::Batch(batch) => {
|
|
799
|
+
assert_eq!(batch.len(), 2);
|
|
800
|
+
assert_eq!(batch[0].method, "test");
|
|
801
|
+
assert_eq!(batch[1].method, "test2");
|
|
802
|
+
}
|
|
803
|
+
_ => panic!("Expected batch request"),
|
|
804
|
+
}
|
|
805
|
+
}
|
|
806
|
+
|
|
807
|
+
#[test]
|
|
808
|
+
fn test_parse_request_invalid() {
|
|
809
|
+
let json = r#"{"invalid":"json"}"#;
|
|
810
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
811
|
+
|
|
812
|
+
assert!(parsed.is_err());
|
|
813
|
+
let err = parsed.unwrap_err();
|
|
814
|
+
assert_eq!(err.error.code, error_codes::PARSE_ERROR);
|
|
815
|
+
}
|
|
816
|
+
|
|
817
|
+
#[test]
|
|
818
|
+
fn test_parse_request_notification() {
|
|
819
|
+
let json = r#"{"jsonrpc":"2.0","method":"notify"}"#;
|
|
820
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
821
|
+
|
|
822
|
+
assert!(parsed.is_ok());
|
|
823
|
+
match parsed.unwrap() {
|
|
824
|
+
JsonRpcRequestOrBatch::Single(req) => {
|
|
825
|
+
assert!(req.is_notification());
|
|
826
|
+
assert_eq!(req.method, "notify");
|
|
827
|
+
}
|
|
828
|
+
_ => panic!("Expected single request"),
|
|
829
|
+
}
|
|
830
|
+
}
|
|
831
|
+
|
|
832
|
+
#[test]
|
|
833
|
+
fn test_parse_request_with_params() {
|
|
834
|
+
let json = r#"{"jsonrpc":"2.0","method":"subtract","params":{"subtrahend":23,"minuend":42},"id":3}"#;
|
|
835
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
836
|
+
|
|
837
|
+
assert!(parsed.is_ok());
|
|
838
|
+
match parsed.unwrap() {
|
|
839
|
+
JsonRpcRequestOrBatch::Single(req) => {
|
|
840
|
+
assert_eq!(req.method, "subtract");
|
|
841
|
+
assert!(req.params.is_some());
|
|
842
|
+
let params = req.params.unwrap();
|
|
843
|
+
assert_eq!(params["subtrahend"], 23);
|
|
844
|
+
assert_eq!(params["minuend"], 42);
|
|
845
|
+
}
|
|
846
|
+
_ => panic!("Expected single request"),
|
|
847
|
+
}
|
|
848
|
+
}
|
|
849
|
+
|
|
850
|
+
#[test]
|
|
851
|
+
fn test_parse_request_invalid_method_name_empty() {
|
|
852
|
+
let json = r#"{"jsonrpc":"2.0","method":"","id":1}"#;
|
|
853
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
854
|
+
|
|
855
|
+
assert!(parsed.is_err());
|
|
856
|
+
let err = parsed.unwrap_err();
|
|
857
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
858
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
859
|
+
}
|
|
860
|
+
|
|
861
|
+
#[test]
|
|
862
|
+
fn test_parse_request_invalid_method_name_leading_space() {
|
|
863
|
+
let json = r#"{"jsonrpc":"2.0","method":" method","id":1}"#;
|
|
864
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
865
|
+
|
|
866
|
+
assert!(parsed.is_err());
|
|
867
|
+
let err = parsed.unwrap_err();
|
|
868
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
869
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
870
|
+
}
|
|
871
|
+
|
|
872
|
+
#[test]
|
|
873
|
+
fn test_parse_request_invalid_method_name_trailing_space() {
|
|
874
|
+
let json = r#"{"jsonrpc":"2.0","method":"method ","id":1}"#;
|
|
875
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
876
|
+
|
|
877
|
+
assert!(parsed.is_err());
|
|
878
|
+
let err = parsed.unwrap_err();
|
|
879
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
880
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
881
|
+
}
|
|
882
|
+
|
|
883
|
+
#[test]
|
|
884
|
+
fn test_parse_request_invalid_method_name_with_space() {
|
|
885
|
+
let json = r#"{"jsonrpc":"2.0","method":"method name","id":1}"#;
|
|
886
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
887
|
+
|
|
888
|
+
assert!(parsed.is_err());
|
|
889
|
+
let err = parsed.unwrap_err();
|
|
890
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
891
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
892
|
+
}
|
|
893
|
+
|
|
894
|
+
#[test]
|
|
895
|
+
fn test_parse_request_invalid_method_name_special_char() {
|
|
896
|
+
let json = r#"{"jsonrpc":"2.0","method":"method@name","id":1}"#;
|
|
897
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
898
|
+
|
|
899
|
+
assert!(parsed.is_err());
|
|
900
|
+
let err = parsed.unwrap_err();
|
|
901
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
902
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
903
|
+
}
|
|
904
|
+
|
|
905
|
+
#[test]
|
|
906
|
+
fn test_parse_request_invalid_method_name_too_long() {
|
|
907
|
+
let long_method = "a".repeat(256);
|
|
908
|
+
let json = format!(r#"{{"jsonrpc":"2.0","method":"{}","id":1}}"#, long_method);
|
|
909
|
+
let parsed = JsonRpcRouter::parse_request(&json);
|
|
910
|
+
|
|
911
|
+
assert!(parsed.is_err());
|
|
912
|
+
let err = parsed.unwrap_err();
|
|
913
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
914
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
915
|
+
}
|
|
916
|
+
|
|
917
|
+
#[test]
|
|
918
|
+
fn test_parse_request_batch_valid_method_names() {
|
|
919
|
+
let json = r#"[
|
|
920
|
+
{"jsonrpc":"2.0","method":"test.method","id":1},
|
|
921
|
+
{"jsonrpc":"2.0","method":"another_method","id":2}
|
|
922
|
+
]"#;
|
|
923
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
924
|
+
|
|
925
|
+
assert!(parsed.is_ok());
|
|
926
|
+
match parsed.unwrap() {
|
|
927
|
+
JsonRpcRequestOrBatch::Batch(batch) => {
|
|
928
|
+
assert_eq!(batch.len(), 2);
|
|
929
|
+
assert_eq!(batch[0].method, "test.method");
|
|
930
|
+
assert_eq!(batch[1].method, "another_method");
|
|
931
|
+
}
|
|
932
|
+
_ => panic!("Expected batch request"),
|
|
933
|
+
}
|
|
934
|
+
}
|
|
935
|
+
|
|
936
|
+
#[test]
|
|
937
|
+
fn test_parse_request_batch_invalid_first_method_name() {
|
|
938
|
+
let json = r#"[
|
|
939
|
+
{"jsonrpc":"2.0","method":" invalid","id":1},
|
|
940
|
+
{"jsonrpc":"2.0","method":"valid","id":2}
|
|
941
|
+
]"#;
|
|
942
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
943
|
+
|
|
944
|
+
assert!(parsed.is_err());
|
|
945
|
+
let err = parsed.unwrap_err();
|
|
946
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
947
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
948
|
+
}
|
|
949
|
+
|
|
950
|
+
#[test]
|
|
951
|
+
fn test_parse_request_batch_invalid_second_method_name() {
|
|
952
|
+
let json = r#"[
|
|
953
|
+
{"jsonrpc":"2.0","method":"valid","id":1},
|
|
954
|
+
{"jsonrpc":"2.0","method":"invalid#method","id":2}
|
|
955
|
+
]"#;
|
|
956
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
957
|
+
|
|
958
|
+
assert!(parsed.is_err());
|
|
959
|
+
let err = parsed.unwrap_err();
|
|
960
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
961
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
962
|
+
}
|
|
963
|
+
|
|
964
|
+
#[test]
|
|
965
|
+
fn test_parse_request_batch_notification_invalid_method_name() {
|
|
966
|
+
let json = r#"[
|
|
967
|
+
{"jsonrpc":"2.0","method":"valid","id":1},
|
|
968
|
+
{"jsonrpc":"2.0","method":"invalid method"}
|
|
969
|
+
]"#;
|
|
970
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
971
|
+
|
|
972
|
+
assert!(parsed.is_err());
|
|
973
|
+
let err = parsed.unwrap_err();
|
|
974
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
975
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
976
|
+
}
|
|
977
|
+
|
|
978
|
+
#[test]
|
|
979
|
+
fn test_parse_request_method_name_dos_prevention() {
|
|
980
|
+
let json = format!(r#"{{"jsonrpc":"2.0","method":"{}","id":1}}"#, "a".repeat(10000));
|
|
981
|
+
let parsed = JsonRpcRouter::parse_request(&json);
|
|
982
|
+
|
|
983
|
+
assert!(parsed.is_err());
|
|
984
|
+
let err = parsed.unwrap_err();
|
|
985
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
986
|
+
assert!(err.error.message.contains("Invalid method name"));
|
|
987
|
+
}
|
|
988
|
+
|
|
989
|
+
#[test]
|
|
990
|
+
fn test_parse_request_valid_method_names_complex() {
|
|
991
|
+
let json = r#"{"jsonrpc":"2.0","method":"user.getById_v1-2","id":1}"#;
|
|
992
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
993
|
+
|
|
994
|
+
assert!(parsed.is_ok());
|
|
995
|
+
match parsed.unwrap() {
|
|
996
|
+
JsonRpcRequestOrBatch::Single(req) => {
|
|
997
|
+
assert_eq!(req.method, "user.getById_v1-2");
|
|
998
|
+
}
|
|
999
|
+
_ => panic!("Expected single request"),
|
|
1000
|
+
}
|
|
1001
|
+
}
|
|
1002
|
+
|
|
1003
|
+
#[tokio::test]
|
|
1004
|
+
async fn test_error_code_parse_error_invalid_json() {
|
|
1005
|
+
let json = r#"{"invalid json"#;
|
|
1006
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
1007
|
+
|
|
1008
|
+
assert!(parsed.is_err());
|
|
1009
|
+
let err = parsed.unwrap_err();
|
|
1010
|
+
assert_eq!(err.error.code, error_codes::PARSE_ERROR);
|
|
1011
|
+
assert_eq!(err.error.code, -32700);
|
|
1012
|
+
}
|
|
1013
|
+
|
|
1014
|
+
#[tokio::test]
|
|
1015
|
+
async fn test_error_code_method_not_found() {
|
|
1016
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1017
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
1018
|
+
|
|
1019
|
+
let request = JsonRpcRequest::new("nonexistent_method", None, Some(json!(1)));
|
|
1020
|
+
let http_request = create_test_http_request();
|
|
1021
|
+
let request_data = create_test_request_data();
|
|
1022
|
+
|
|
1023
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
1024
|
+
|
|
1025
|
+
match response {
|
|
1026
|
+
JsonRpcResponseType::Error(err) => {
|
|
1027
|
+
assert_eq!(err.error.code, error_codes::METHOD_NOT_FOUND);
|
|
1028
|
+
assert_eq!(err.error.code, -32601);
|
|
1029
|
+
assert_eq!(err.id, json!(1));
|
|
1030
|
+
}
|
|
1031
|
+
_ => panic!("Expected error response"),
|
|
1032
|
+
}
|
|
1033
|
+
}
|
|
1034
|
+
|
|
1035
|
+
#[tokio::test]
|
|
1036
|
+
async fn test_error_code_invalid_request_empty_method() {
|
|
1037
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1038
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
1039
|
+
|
|
1040
|
+
let request = JsonRpcRequest::new("", None, Some(json!(1)));
|
|
1041
|
+
let http_request = create_test_http_request();
|
|
1042
|
+
let request_data = create_test_request_data();
|
|
1043
|
+
|
|
1044
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
1045
|
+
|
|
1046
|
+
match response {
|
|
1047
|
+
JsonRpcResponseType::Error(err) => {
|
|
1048
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
1049
|
+
assert_eq!(err.error.code, -32600);
|
|
1050
|
+
assert_eq!(err.id, json!(1));
|
|
1051
|
+
}
|
|
1052
|
+
_ => panic!("Expected error response"),
|
|
1053
|
+
}
|
|
1054
|
+
}
|
|
1055
|
+
|
|
1056
|
+
#[tokio::test]
|
|
1057
|
+
async fn test_error_code_internal_error_from_handler() {
|
|
1058
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1059
|
+
|
|
1060
|
+
let handler = Arc::new(MockErrorHandler);
|
|
1061
|
+
let metadata = super::super::method_registry::MethodMetadata::new("error_method");
|
|
1062
|
+
registry.register("error_method", handler, metadata).unwrap();
|
|
1063
|
+
|
|
1064
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1065
|
+
|
|
1066
|
+
let request = JsonRpcRequest::new("error_method", None, Some(json!(1)));
|
|
1067
|
+
let http_request = create_test_http_request();
|
|
1068
|
+
let request_data = create_test_request_data();
|
|
1069
|
+
|
|
1070
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
1071
|
+
|
|
1072
|
+
match response {
|
|
1073
|
+
JsonRpcResponseType::Error(err) => {
|
|
1074
|
+
assert_eq!(err.error.code, error_codes::INTERNAL_ERROR);
|
|
1075
|
+
assert_eq!(err.error.code, -32603);
|
|
1076
|
+
assert_eq!(err.id, json!(1));
|
|
1077
|
+
}
|
|
1078
|
+
_ => panic!("Expected error response"),
|
|
1079
|
+
}
|
|
1080
|
+
}
|
|
1081
|
+
|
|
1082
|
+
#[tokio::test]
|
|
1083
|
+
async fn test_error_code_server_error_custom_range() {
|
|
1084
|
+
assert!(error_codes::is_server_error(-32000));
|
|
1085
|
+
assert!(error_codes::is_server_error(-32050));
|
|
1086
|
+
assert!(error_codes::is_server_error(-32099));
|
|
1087
|
+
assert!(!error_codes::is_server_error(-31999));
|
|
1088
|
+
assert!(!error_codes::is_server_error(-32100));
|
|
1089
|
+
}
|
|
1090
|
+
|
|
1091
|
+
#[tokio::test]
|
|
1092
|
+
async fn test_error_response_always_has_code() {
|
|
1093
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1094
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
1095
|
+
|
|
1096
|
+
let request = JsonRpcRequest::new("nonexistent", None, Some(json!(1)));
|
|
1097
|
+
let http_request = create_test_http_request();
|
|
1098
|
+
let request_data = create_test_request_data();
|
|
1099
|
+
|
|
1100
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
1101
|
+
|
|
1102
|
+
match response {
|
|
1103
|
+
JsonRpcResponseType::Error(err) => {
|
|
1104
|
+
assert!(err.error.code != 0);
|
|
1105
|
+
assert!(err.error.code < 0);
|
|
1106
|
+
}
|
|
1107
|
+
_ => panic!("Expected error response"),
|
|
1108
|
+
}
|
|
1109
|
+
}
|
|
1110
|
+
|
|
1111
|
+
#[tokio::test]
|
|
1112
|
+
async fn test_error_response_always_has_message() {
|
|
1113
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1114
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
1115
|
+
|
|
1116
|
+
let request = JsonRpcRequest::new("nonexistent", None, Some(json!(1)));
|
|
1117
|
+
let http_request = create_test_http_request();
|
|
1118
|
+
let request_data = create_test_request_data();
|
|
1119
|
+
|
|
1120
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
1121
|
+
|
|
1122
|
+
match response {
|
|
1123
|
+
JsonRpcResponseType::Error(err) => {
|
|
1124
|
+
assert!(!err.error.message.is_empty());
|
|
1125
|
+
assert!(err.jsonrpc == "2.0");
|
|
1126
|
+
}
|
|
1127
|
+
_ => panic!("Expected error response"),
|
|
1128
|
+
}
|
|
1129
|
+
}
|
|
1130
|
+
|
|
1131
|
+
#[tokio::test]
|
|
1132
|
+
async fn test_error_response_data_optional() {
|
|
1133
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1134
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
1135
|
+
|
|
1136
|
+
let request = JsonRpcRequest::new("nonexistent", None, Some(json!(1)));
|
|
1137
|
+
let http_request = create_test_http_request();
|
|
1138
|
+
let request_data = create_test_request_data();
|
|
1139
|
+
|
|
1140
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
1141
|
+
|
|
1142
|
+
match response {
|
|
1143
|
+
JsonRpcResponseType::Error(err) => {
|
|
1144
|
+
assert_eq!(err.error.code, error_codes::METHOD_NOT_FOUND);
|
|
1145
|
+
}
|
|
1146
|
+
_ => panic!("Expected error response"),
|
|
1147
|
+
}
|
|
1148
|
+
}
|
|
1149
|
+
|
|
1150
|
+
#[tokio::test]
|
|
1151
|
+
async fn test_error_response_id_matches_request() {
|
|
1152
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1153
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
1154
|
+
|
|
1155
|
+
let test_id = json!(42);
|
|
1156
|
+
let request = JsonRpcRequest::new("nonexistent", None, Some(test_id.clone()));
|
|
1157
|
+
let http_request = create_test_http_request();
|
|
1158
|
+
let request_data = create_test_request_data();
|
|
1159
|
+
|
|
1160
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
1161
|
+
|
|
1162
|
+
match response {
|
|
1163
|
+
JsonRpcResponseType::Error(err) => {
|
|
1164
|
+
assert_eq!(err.id, test_id);
|
|
1165
|
+
}
|
|
1166
|
+
_ => panic!("Expected error response"),
|
|
1167
|
+
}
|
|
1168
|
+
}
|
|
1169
|
+
|
|
1170
|
+
#[tokio::test]
|
|
1171
|
+
async fn test_error_response_no_result_field() {
|
|
1172
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1173
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
1174
|
+
|
|
1175
|
+
let request = JsonRpcRequest::new("nonexistent", None, Some(json!(1)));
|
|
1176
|
+
let http_request = create_test_http_request();
|
|
1177
|
+
let request_data = create_test_request_data();
|
|
1178
|
+
|
|
1179
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
1180
|
+
|
|
1181
|
+
match response {
|
|
1182
|
+
JsonRpcResponseType::Error(_) => {}
|
|
1183
|
+
_ => panic!("Expected error response"),
|
|
1184
|
+
}
|
|
1185
|
+
}
|
|
1186
|
+
|
|
1187
|
+
#[tokio::test]
|
|
1188
|
+
async fn test_error_data_includes_details() {
|
|
1189
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1190
|
+
|
|
1191
|
+
let handler = Arc::new(MockErrorHandler);
|
|
1192
|
+
let metadata = super::super::method_registry::MethodMetadata::new("error_method");
|
|
1193
|
+
registry.register("error_method", handler, metadata).unwrap();
|
|
1194
|
+
|
|
1195
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1196
|
+
|
|
1197
|
+
let request = JsonRpcRequest::new("error_method", None, Some(json!(1)));
|
|
1198
|
+
let http_request = create_test_http_request();
|
|
1199
|
+
let request_data = create_test_request_data();
|
|
1200
|
+
|
|
1201
|
+
let response = router.route_single(request, http_request, &request_data).await;
|
|
1202
|
+
|
|
1203
|
+
match response {
|
|
1204
|
+
JsonRpcResponseType::Error(err) => {
|
|
1205
|
+
assert!(err.error.data.is_some());
|
|
1206
|
+
let data = err.error.data.unwrap();
|
|
1207
|
+
assert!(data.get("details").is_some());
|
|
1208
|
+
}
|
|
1209
|
+
_ => panic!("Expected error response"),
|
|
1210
|
+
}
|
|
1211
|
+
}
|
|
1212
|
+
|
|
1213
|
+
#[tokio::test]
|
|
1214
|
+
async fn test_batch_single_request() {
|
|
1215
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1216
|
+
|
|
1217
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1218
|
+
let metadata = super::super::method_registry::MethodMetadata::new("test");
|
|
1219
|
+
registry.register("test", handler, metadata).unwrap();
|
|
1220
|
+
|
|
1221
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1222
|
+
|
|
1223
|
+
let batch = vec![JsonRpcRequest::new("test", None, Some(json!(1)))];
|
|
1224
|
+
let http_request = create_test_http_request();
|
|
1225
|
+
let request_data = create_test_request_data();
|
|
1226
|
+
|
|
1227
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1228
|
+
|
|
1229
|
+
assert!(result.is_ok());
|
|
1230
|
+
let responses = result.unwrap();
|
|
1231
|
+
assert_eq!(responses.len(), 1);
|
|
1232
|
+
}
|
|
1233
|
+
|
|
1234
|
+
#[tokio::test]
|
|
1235
|
+
async fn test_batch_three_requests_all_succeed() {
|
|
1236
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1237
|
+
|
|
1238
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1239
|
+
let metadata = super::super::method_registry::MethodMetadata::new("test");
|
|
1240
|
+
registry.register("test", handler, metadata).unwrap();
|
|
1241
|
+
|
|
1242
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1243
|
+
|
|
1244
|
+
let batch = vec![
|
|
1245
|
+
JsonRpcRequest::new("test", None, Some(json!(1))),
|
|
1246
|
+
JsonRpcRequest::new("test", None, Some(json!(2))),
|
|
1247
|
+
JsonRpcRequest::new("test", None, Some(json!(3))),
|
|
1248
|
+
];
|
|
1249
|
+
let http_request = create_test_http_request();
|
|
1250
|
+
let request_data = create_test_request_data();
|
|
1251
|
+
|
|
1252
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1253
|
+
|
|
1254
|
+
assert!(result.is_ok());
|
|
1255
|
+
let responses = result.unwrap();
|
|
1256
|
+
assert_eq!(responses.len(), 3);
|
|
1257
|
+
for resp in &responses {
|
|
1258
|
+
match resp {
|
|
1259
|
+
JsonRpcResponseType::Success(_) => {}
|
|
1260
|
+
_ => panic!("Expected all success responses"),
|
|
1261
|
+
}
|
|
1262
|
+
}
|
|
1263
|
+
}
|
|
1264
|
+
|
|
1265
|
+
#[tokio::test]
|
|
1266
|
+
async fn test_batch_mixed_success_and_errors() {
|
|
1267
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1268
|
+
|
|
1269
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1270
|
+
let metadata = super::super::method_registry::MethodMetadata::new("valid");
|
|
1271
|
+
registry.register("valid", handler, metadata).unwrap();
|
|
1272
|
+
|
|
1273
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1274
|
+
|
|
1275
|
+
let batch = vec![
|
|
1276
|
+
JsonRpcRequest::new("valid", None, Some(json!(1))),
|
|
1277
|
+
JsonRpcRequest::new("invalid", None, Some(json!(2))),
|
|
1278
|
+
JsonRpcRequest::new("valid", None, Some(json!(3))),
|
|
1279
|
+
];
|
|
1280
|
+
let http_request = create_test_http_request();
|
|
1281
|
+
let request_data = create_test_request_data();
|
|
1282
|
+
|
|
1283
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1284
|
+
|
|
1285
|
+
assert!(result.is_ok());
|
|
1286
|
+
let responses = result.unwrap();
|
|
1287
|
+
assert_eq!(responses.len(), 3);
|
|
1288
|
+
|
|
1289
|
+
match &responses[0] {
|
|
1290
|
+
JsonRpcResponseType::Success(_) => {}
|
|
1291
|
+
_ => panic!("Expected success at index 0"),
|
|
1292
|
+
}
|
|
1293
|
+
match &responses[1] {
|
|
1294
|
+
JsonRpcResponseType::Error(_) => {}
|
|
1295
|
+
_ => panic!("Expected error at index 1"),
|
|
1296
|
+
}
|
|
1297
|
+
match &responses[2] {
|
|
1298
|
+
JsonRpcResponseType::Success(_) => {}
|
|
1299
|
+
_ => panic!("Expected success at index 2"),
|
|
1300
|
+
}
|
|
1301
|
+
}
|
|
1302
|
+
|
|
1303
|
+
#[tokio::test]
|
|
1304
|
+
async fn test_batch_all_errors() {
|
|
1305
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1306
|
+
|
|
1307
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1308
|
+
|
|
1309
|
+
let batch = vec![
|
|
1310
|
+
JsonRpcRequest::new("nonexistent1", None, Some(json!(1))),
|
|
1311
|
+
JsonRpcRequest::new("nonexistent2", None, Some(json!(2))),
|
|
1312
|
+
JsonRpcRequest::new("nonexistent3", None, Some(json!(3))),
|
|
1313
|
+
];
|
|
1314
|
+
let http_request = create_test_http_request();
|
|
1315
|
+
let request_data = create_test_request_data();
|
|
1316
|
+
|
|
1317
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1318
|
+
|
|
1319
|
+
assert!(result.is_ok());
|
|
1320
|
+
let responses = result.unwrap();
|
|
1321
|
+
assert_eq!(responses.len(), 3);
|
|
1322
|
+
for resp in &responses {
|
|
1323
|
+
match resp {
|
|
1324
|
+
JsonRpcResponseType::Error(_) => {}
|
|
1325
|
+
_ => panic!("Expected all error responses"),
|
|
1326
|
+
}
|
|
1327
|
+
}
|
|
1328
|
+
}
|
|
1329
|
+
|
|
1330
|
+
#[tokio::test]
|
|
1331
|
+
async fn test_batch_exceeds_max_size() {
|
|
1332
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1333
|
+
let router = JsonRpcRouter::new(registry, true, 5);
|
|
1334
|
+
|
|
1335
|
+
let batch = (1..=10)
|
|
1336
|
+
.map(|i| JsonRpcRequest::new("method", None, Some(json!(i))))
|
|
1337
|
+
.collect();
|
|
1338
|
+
let http_request = create_test_http_request();
|
|
1339
|
+
let request_data = create_test_request_data();
|
|
1340
|
+
|
|
1341
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1342
|
+
|
|
1343
|
+
assert!(result.is_err());
|
|
1344
|
+
let err = result.unwrap_err();
|
|
1345
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
1346
|
+
}
|
|
1347
|
+
|
|
1348
|
+
#[tokio::test]
|
|
1349
|
+
async fn test_batch_empty_array() {
|
|
1350
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1351
|
+
let router = JsonRpcRouter::new(registry, true, 100);
|
|
1352
|
+
|
|
1353
|
+
let batch = vec![];
|
|
1354
|
+
let http_request = create_test_http_request();
|
|
1355
|
+
let request_data = create_test_request_data();
|
|
1356
|
+
|
|
1357
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1358
|
+
|
|
1359
|
+
assert!(result.is_err());
|
|
1360
|
+
let err = result.unwrap_err();
|
|
1361
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
1362
|
+
}
|
|
1363
|
+
|
|
1364
|
+
#[tokio::test]
|
|
1365
|
+
async fn test_batch_disabled_error() {
|
|
1366
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1367
|
+
let router = JsonRpcRouter::new(registry, false, 100);
|
|
1368
|
+
|
|
1369
|
+
let batch = vec![JsonRpcRequest::new("method", None, Some(json!(1)))];
|
|
1370
|
+
let http_request = create_test_http_request();
|
|
1371
|
+
let request_data = create_test_request_data();
|
|
1372
|
+
|
|
1373
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1374
|
+
|
|
1375
|
+
assert!(result.is_err());
|
|
1376
|
+
let err = result.unwrap_err();
|
|
1377
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
1378
|
+
}
|
|
1379
|
+
|
|
1380
|
+
#[tokio::test]
|
|
1381
|
+
async fn test_notification_no_response_generated() {
|
|
1382
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1383
|
+
|
|
1384
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1385
|
+
let metadata = super::super::method_registry::MethodMetadata::new("notify");
|
|
1386
|
+
registry.register("notify", handler, metadata).unwrap();
|
|
1387
|
+
|
|
1388
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1389
|
+
|
|
1390
|
+
let batch = vec![
|
|
1391
|
+
JsonRpcRequest::new("notify", None, Some(json!(1))),
|
|
1392
|
+
JsonRpcRequest::new("notify", None, None),
|
|
1393
|
+
];
|
|
1394
|
+
let http_request = create_test_http_request();
|
|
1395
|
+
let request_data = create_test_request_data();
|
|
1396
|
+
|
|
1397
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1398
|
+
|
|
1399
|
+
assert!(result.is_ok());
|
|
1400
|
+
let responses = result.unwrap();
|
|
1401
|
+
assert_eq!(responses.len(), 1);
|
|
1402
|
+
}
|
|
1403
|
+
|
|
1404
|
+
#[tokio::test]
|
|
1405
|
+
async fn test_batch_mixed_notifications_responses() {
|
|
1406
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1407
|
+
|
|
1408
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1409
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
1410
|
+
registry.register("method", handler, metadata).unwrap();
|
|
1411
|
+
|
|
1412
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1413
|
+
|
|
1414
|
+
let batch = vec![
|
|
1415
|
+
JsonRpcRequest::new("method", None, Some(json!(1))),
|
|
1416
|
+
JsonRpcRequest::new("method", None, None),
|
|
1417
|
+
JsonRpcRequest::new("method", None, Some(json!(2))),
|
|
1418
|
+
JsonRpcRequest::new("method", None, None),
|
|
1419
|
+
];
|
|
1420
|
+
let http_request = create_test_http_request();
|
|
1421
|
+
let request_data = create_test_request_data();
|
|
1422
|
+
|
|
1423
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1424
|
+
|
|
1425
|
+
assert!(result.is_ok());
|
|
1426
|
+
let responses = result.unwrap();
|
|
1427
|
+
assert_eq!(responses.len(), 2);
|
|
1428
|
+
}
|
|
1429
|
+
|
|
1430
|
+
#[tokio::test]
|
|
1431
|
+
async fn test_batch_all_notifications_empty_response() {
|
|
1432
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1433
|
+
|
|
1434
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1435
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
1436
|
+
registry.register("method", handler, metadata).unwrap();
|
|
1437
|
+
|
|
1438
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1439
|
+
|
|
1440
|
+
let batch = vec![
|
|
1441
|
+
JsonRpcRequest::new("method", None, None),
|
|
1442
|
+
JsonRpcRequest::new("method", None, None),
|
|
1443
|
+
];
|
|
1444
|
+
let http_request = create_test_http_request();
|
|
1445
|
+
let request_data = create_test_request_data();
|
|
1446
|
+
|
|
1447
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1448
|
+
|
|
1449
|
+
assert!(result.is_ok());
|
|
1450
|
+
let responses = result.unwrap();
|
|
1451
|
+
assert_eq!(responses.len(), 0);
|
|
1452
|
+
}
|
|
1453
|
+
|
|
1454
|
+
#[tokio::test]
|
|
1455
|
+
async fn test_notification_error_still_not_responded() {
|
|
1456
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1457
|
+
|
|
1458
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1459
|
+
|
|
1460
|
+
let batch = vec![
|
|
1461
|
+
JsonRpcRequest::new("nonexistent", None, Some(json!(1))),
|
|
1462
|
+
JsonRpcRequest::new("nonexistent", None, None),
|
|
1463
|
+
];
|
|
1464
|
+
let http_request = create_test_http_request();
|
|
1465
|
+
let request_data = create_test_request_data();
|
|
1466
|
+
|
|
1467
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1468
|
+
|
|
1469
|
+
assert!(result.is_ok());
|
|
1470
|
+
let responses = result.unwrap();
|
|
1471
|
+
assert_eq!(responses.len(), 1);
|
|
1472
|
+
match &responses[0] {
|
|
1473
|
+
JsonRpcResponseType::Error(_) => {}
|
|
1474
|
+
_ => panic!("Expected error response"),
|
|
1475
|
+
}
|
|
1476
|
+
}
|
|
1477
|
+
|
|
1478
|
+
#[tokio::test]
|
|
1479
|
+
async fn test_batch_requests_independent_state() {
|
|
1480
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1481
|
+
|
|
1482
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1483
|
+
let metadata = super::super::method_registry::MethodMetadata::new("test");
|
|
1484
|
+
registry.register("test", handler, metadata).unwrap();
|
|
1485
|
+
|
|
1486
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1487
|
+
|
|
1488
|
+
let batch = vec![
|
|
1489
|
+
JsonRpcRequest::new("test", None, Some(json!(1))),
|
|
1490
|
+
JsonRpcRequest::new("test", None, Some(json!(2))),
|
|
1491
|
+
JsonRpcRequest::new("test", None, Some(json!(3))),
|
|
1492
|
+
];
|
|
1493
|
+
let http_request = create_test_http_request();
|
|
1494
|
+
let request_data = create_test_request_data();
|
|
1495
|
+
|
|
1496
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1497
|
+
|
|
1498
|
+
assert!(result.is_ok());
|
|
1499
|
+
let responses = result.unwrap();
|
|
1500
|
+
assert_eq!(responses.len(), 3);
|
|
1501
|
+
for (i, resp) in responses.iter().enumerate() {
|
|
1502
|
+
match resp {
|
|
1503
|
+
JsonRpcResponseType::Success(s) => {
|
|
1504
|
+
assert_eq!(s.id, json!(i + 1));
|
|
1505
|
+
}
|
|
1506
|
+
_ => panic!("Expected success response"),
|
|
1507
|
+
}
|
|
1508
|
+
}
|
|
1509
|
+
}
|
|
1510
|
+
|
|
1511
|
+
#[tokio::test]
|
|
1512
|
+
async fn test_batch_error_preserves_request_order() {
|
|
1513
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1514
|
+
|
|
1515
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1516
|
+
let metadata = super::super::method_registry::MethodMetadata::new("valid");
|
|
1517
|
+
registry.register("valid", handler, metadata).unwrap();
|
|
1518
|
+
|
|
1519
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1520
|
+
|
|
1521
|
+
let batch = vec![
|
|
1522
|
+
JsonRpcRequest::new("valid", None, Some(json!("first"))),
|
|
1523
|
+
JsonRpcRequest::new("invalid", None, Some(json!("second"))),
|
|
1524
|
+
JsonRpcRequest::new("valid", None, Some(json!("third"))),
|
|
1525
|
+
];
|
|
1526
|
+
let http_request = create_test_http_request();
|
|
1527
|
+
let request_data = create_test_request_data();
|
|
1528
|
+
|
|
1529
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1530
|
+
|
|
1531
|
+
assert!(result.is_ok());
|
|
1532
|
+
let responses = result.unwrap();
|
|
1533
|
+
assert_eq!(responses.len(), 3);
|
|
1534
|
+
|
|
1535
|
+
match &responses[0] {
|
|
1536
|
+
JsonRpcResponseType::Success(s) => {
|
|
1537
|
+
assert_eq!(s.id, json!("first"));
|
|
1538
|
+
assert_eq!(s.jsonrpc, "2.0");
|
|
1539
|
+
}
|
|
1540
|
+
_ => panic!("Expected success at index 0"),
|
|
1541
|
+
}
|
|
1542
|
+
match &responses[1] {
|
|
1543
|
+
JsonRpcResponseType::Error(e) => {
|
|
1544
|
+
assert_eq!(e.id, json!("second"));
|
|
1545
|
+
assert_eq!(e.error.code, error_codes::METHOD_NOT_FOUND);
|
|
1546
|
+
}
|
|
1547
|
+
_ => panic!("Expected error at index 1"),
|
|
1548
|
+
}
|
|
1549
|
+
match &responses[2] {
|
|
1550
|
+
JsonRpcResponseType::Success(s) => {
|
|
1551
|
+
assert_eq!(s.id, json!("third"));
|
|
1552
|
+
assert_eq!(s.jsonrpc, "2.0");
|
|
1553
|
+
}
|
|
1554
|
+
_ => panic!("Expected success at index 2"),
|
|
1555
|
+
}
|
|
1556
|
+
}
|
|
1557
|
+
|
|
1558
|
+
#[tokio::test]
|
|
1559
|
+
async fn test_batch_handler_execution_order() {
|
|
1560
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1561
|
+
|
|
1562
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1563
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method1");
|
|
1564
|
+
registry.register("method1", handler.clone(), metadata).unwrap();
|
|
1565
|
+
|
|
1566
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method2");
|
|
1567
|
+
registry.register("method2", handler.clone(), metadata).unwrap();
|
|
1568
|
+
|
|
1569
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method3");
|
|
1570
|
+
registry.register("method3", handler.clone(), metadata).unwrap();
|
|
1571
|
+
|
|
1572
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1573
|
+
|
|
1574
|
+
let batch = vec![
|
|
1575
|
+
JsonRpcRequest::new("method1", None, Some(json!(1))),
|
|
1576
|
+
JsonRpcRequest::new("method2", None, Some(json!(2))),
|
|
1577
|
+
JsonRpcRequest::new("method3", None, Some(json!(3))),
|
|
1578
|
+
];
|
|
1579
|
+
let http_request = create_test_http_request();
|
|
1580
|
+
let request_data = create_test_request_data();
|
|
1581
|
+
|
|
1582
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1583
|
+
|
|
1584
|
+
assert!(result.is_ok());
|
|
1585
|
+
let responses = result.unwrap();
|
|
1586
|
+
assert_eq!(responses.len(), 3);
|
|
1587
|
+
for resp in &responses {
|
|
1588
|
+
match resp {
|
|
1589
|
+
JsonRpcResponseType::Success(_) => {}
|
|
1590
|
+
_ => panic!("Expected all success responses"),
|
|
1591
|
+
}
|
|
1592
|
+
}
|
|
1593
|
+
}
|
|
1594
|
+
|
|
1595
|
+
#[test]
|
|
1596
|
+
fn test_parse_batch_single_request_in_array() {
|
|
1597
|
+
let json = r#"[{"jsonrpc":"2.0","method":"test","id":1}]"#;
|
|
1598
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
1599
|
+
|
|
1600
|
+
assert!(parsed.is_ok());
|
|
1601
|
+
match parsed.unwrap() {
|
|
1602
|
+
JsonRpcRequestOrBatch::Batch(batch) => {
|
|
1603
|
+
assert_eq!(batch.len(), 1);
|
|
1604
|
+
assert_eq!(batch[0].method, "test");
|
|
1605
|
+
}
|
|
1606
|
+
_ => panic!("Expected batch request"),
|
|
1607
|
+
}
|
|
1608
|
+
}
|
|
1609
|
+
|
|
1610
|
+
#[test]
|
|
1611
|
+
fn test_parse_batch_three_requests() {
|
|
1612
|
+
let json = r#"[
|
|
1613
|
+
{"jsonrpc":"2.0","method":"test1","id":1},
|
|
1614
|
+
{"jsonrpc":"2.0","method":"test2","id":2},
|
|
1615
|
+
{"jsonrpc":"2.0","method":"test3","id":3}
|
|
1616
|
+
]"#;
|
|
1617
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
1618
|
+
|
|
1619
|
+
assert!(parsed.is_ok());
|
|
1620
|
+
match parsed.unwrap() {
|
|
1621
|
+
JsonRpcRequestOrBatch::Batch(batch) => {
|
|
1622
|
+
assert_eq!(batch.len(), 3);
|
|
1623
|
+
assert_eq!(batch[0].method, "test1");
|
|
1624
|
+
assert_eq!(batch[1].method, "test2");
|
|
1625
|
+
assert_eq!(batch[2].method, "test3");
|
|
1626
|
+
}
|
|
1627
|
+
_ => panic!("Expected batch request"),
|
|
1628
|
+
}
|
|
1629
|
+
}
|
|
1630
|
+
|
|
1631
|
+
#[test]
|
|
1632
|
+
fn test_parse_batch_mixed_notifications() {
|
|
1633
|
+
let json = r#"[
|
|
1634
|
+
{"jsonrpc":"2.0","method":"method","id":1},
|
|
1635
|
+
{"jsonrpc":"2.0","method":"method"},
|
|
1636
|
+
{"jsonrpc":"2.0","method":"method","id":2}
|
|
1637
|
+
]"#;
|
|
1638
|
+
let parsed = JsonRpcRouter::parse_request(json);
|
|
1639
|
+
|
|
1640
|
+
assert!(parsed.is_ok());
|
|
1641
|
+
match parsed.unwrap() {
|
|
1642
|
+
JsonRpcRequestOrBatch::Batch(batch) => {
|
|
1643
|
+
assert_eq!(batch.len(), 3);
|
|
1644
|
+
assert!(!batch[0].is_notification());
|
|
1645
|
+
assert!(batch[1].is_notification());
|
|
1646
|
+
assert!(!batch[2].is_notification());
|
|
1647
|
+
}
|
|
1648
|
+
_ => panic!("Expected batch request"),
|
|
1649
|
+
}
|
|
1650
|
+
}
|
|
1651
|
+
|
|
1652
|
+
#[tokio::test]
|
|
1653
|
+
async fn test_batch_max_size_boundary_exactly_at_limit() {
|
|
1654
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1655
|
+
let max_size = 10;
|
|
1656
|
+
let router = JsonRpcRouter::new(registry, true, max_size);
|
|
1657
|
+
|
|
1658
|
+
let batch: Vec<JsonRpcRequest> = (1..=max_size)
|
|
1659
|
+
.map(|i| JsonRpcRequest::new("method", None, Some(json!(i))))
|
|
1660
|
+
.collect();
|
|
1661
|
+
|
|
1662
|
+
let http_request = create_test_http_request();
|
|
1663
|
+
let request_data = create_test_request_data();
|
|
1664
|
+
|
|
1665
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1666
|
+
|
|
1667
|
+
assert!(result.is_ok());
|
|
1668
|
+
let responses = result.unwrap();
|
|
1669
|
+
assert_eq!(responses.len(), max_size);
|
|
1670
|
+
}
|
|
1671
|
+
|
|
1672
|
+
#[tokio::test]
|
|
1673
|
+
async fn test_batch_max_size_boundary_one_over_limit() {
|
|
1674
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1675
|
+
let max_size = 10;
|
|
1676
|
+
let router = JsonRpcRouter::new(registry, true, max_size);
|
|
1677
|
+
|
|
1678
|
+
let batch: Vec<JsonRpcRequest> = (1..=max_size + 1)
|
|
1679
|
+
.map(|i| JsonRpcRequest::new("method", None, Some(json!(i))))
|
|
1680
|
+
.collect();
|
|
1681
|
+
|
|
1682
|
+
let http_request = create_test_http_request();
|
|
1683
|
+
let request_data = create_test_request_data();
|
|
1684
|
+
|
|
1685
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1686
|
+
|
|
1687
|
+
assert!(result.is_err());
|
|
1688
|
+
let err = result.unwrap_err();
|
|
1689
|
+
assert_eq!(err.error.code, error_codes::INVALID_REQUEST);
|
|
1690
|
+
assert!(err.error.message.contains("exceeds maximum"));
|
|
1691
|
+
}
|
|
1692
|
+
|
|
1693
|
+
#[tokio::test]
|
|
1694
|
+
async fn test_batch_notification_with_numeric_id_zero() {
|
|
1695
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1696
|
+
|
|
1697
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1698
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
1699
|
+
registry.register("method", handler, metadata).unwrap();
|
|
1700
|
+
|
|
1701
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1702
|
+
|
|
1703
|
+
let batch = vec![
|
|
1704
|
+
JsonRpcRequest::new("method", None, Some(json!(0))),
|
|
1705
|
+
JsonRpcRequest::new("method", None, Some(json!(1))),
|
|
1706
|
+
];
|
|
1707
|
+
let http_request = create_test_http_request();
|
|
1708
|
+
let request_data = create_test_request_data();
|
|
1709
|
+
|
|
1710
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1711
|
+
|
|
1712
|
+
assert!(result.is_ok());
|
|
1713
|
+
let responses = result.unwrap();
|
|
1714
|
+
assert_eq!(responses.len(), 2);
|
|
1715
|
+
match &responses[0] {
|
|
1716
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!(0)),
|
|
1717
|
+
_ => panic!("Expected success response"),
|
|
1718
|
+
}
|
|
1719
|
+
}
|
|
1720
|
+
|
|
1721
|
+
#[tokio::test]
|
|
1722
|
+
async fn test_batch_mixed_id_types_string_and_number() {
|
|
1723
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1724
|
+
|
|
1725
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1726
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
1727
|
+
registry.register("method", handler, metadata).unwrap();
|
|
1728
|
+
|
|
1729
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1730
|
+
|
|
1731
|
+
let batch = vec![
|
|
1732
|
+
JsonRpcRequest::new("method", None, Some(json!("string_id"))),
|
|
1733
|
+
JsonRpcRequest::new("method", None, Some(json!(42))),
|
|
1734
|
+
JsonRpcRequest::new("method", None, Some(json!("another_string"))),
|
|
1735
|
+
];
|
|
1736
|
+
let http_request = create_test_http_request();
|
|
1737
|
+
let request_data = create_test_request_data();
|
|
1738
|
+
|
|
1739
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1740
|
+
|
|
1741
|
+
assert!(result.is_ok());
|
|
1742
|
+
let responses = result.unwrap();
|
|
1743
|
+
assert_eq!(responses.len(), 3);
|
|
1744
|
+
|
|
1745
|
+
match &responses[0] {
|
|
1746
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!("string_id")),
|
|
1747
|
+
_ => panic!("Expected success response"),
|
|
1748
|
+
}
|
|
1749
|
+
match &responses[1] {
|
|
1750
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!(42)),
|
|
1751
|
+
_ => panic!("Expected success response"),
|
|
1752
|
+
}
|
|
1753
|
+
match &responses[2] {
|
|
1754
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!("another_string")),
|
|
1755
|
+
_ => panic!("Expected success response"),
|
|
1756
|
+
}
|
|
1757
|
+
}
|
|
1758
|
+
|
|
1759
|
+
#[tokio::test]
|
|
1760
|
+
async fn test_batch_all_notifications_truly_empty_response() {
|
|
1761
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1762
|
+
|
|
1763
|
+
for i in 1..=5 {
|
|
1764
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1765
|
+
let method_name = format!("method{}", i);
|
|
1766
|
+
let metadata = super::super::method_registry::MethodMetadata::new(&method_name);
|
|
1767
|
+
registry.register(&method_name, handler, metadata).unwrap();
|
|
1768
|
+
}
|
|
1769
|
+
|
|
1770
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1771
|
+
|
|
1772
|
+
let batch = vec![
|
|
1773
|
+
JsonRpcRequest::new("method1", None, None),
|
|
1774
|
+
JsonRpcRequest::new("method2", None, None),
|
|
1775
|
+
JsonRpcRequest::new("method3", None, None),
|
|
1776
|
+
JsonRpcRequest::new("method4", None, None),
|
|
1777
|
+
JsonRpcRequest::new("method5", None, None),
|
|
1778
|
+
];
|
|
1779
|
+
let http_request = create_test_http_request();
|
|
1780
|
+
let request_data = create_test_request_data();
|
|
1781
|
+
|
|
1782
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1783
|
+
|
|
1784
|
+
assert!(result.is_ok());
|
|
1785
|
+
let responses = result.unwrap();
|
|
1786
|
+
assert_eq!(responses.len(), 0);
|
|
1787
|
+
assert!(responses.is_empty());
|
|
1788
|
+
}
|
|
1789
|
+
|
|
1790
|
+
#[tokio::test]
|
|
1791
|
+
async fn test_batch_single_notification_among_requests() {
|
|
1792
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1793
|
+
|
|
1794
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1795
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
1796
|
+
registry.register("method", handler, metadata).unwrap();
|
|
1797
|
+
|
|
1798
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1799
|
+
|
|
1800
|
+
let batch = vec![
|
|
1801
|
+
JsonRpcRequest::new("method", None, Some(json!(1))),
|
|
1802
|
+
JsonRpcRequest::new("method", None, None),
|
|
1803
|
+
JsonRpcRequest::new("method", None, Some(json!(2))),
|
|
1804
|
+
];
|
|
1805
|
+
let http_request = create_test_http_request();
|
|
1806
|
+
let request_data = create_test_request_data();
|
|
1807
|
+
|
|
1808
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1809
|
+
|
|
1810
|
+
assert!(result.is_ok());
|
|
1811
|
+
let responses = result.unwrap();
|
|
1812
|
+
assert_eq!(responses.len(), 2);
|
|
1813
|
+
|
|
1814
|
+
match &responses[0] {
|
|
1815
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!(1)),
|
|
1816
|
+
_ => panic!("Expected success at index 0"),
|
|
1817
|
+
}
|
|
1818
|
+
match &responses[1] {
|
|
1819
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!(2)),
|
|
1820
|
+
_ => panic!("Expected success at index 1"),
|
|
1821
|
+
}
|
|
1822
|
+
}
|
|
1823
|
+
|
|
1824
|
+
#[tokio::test]
|
|
1825
|
+
async fn test_batch_error_one_request_fails_others_succeed() {
|
|
1826
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1827
|
+
|
|
1828
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1829
|
+
let metadata = super::super::method_registry::MethodMetadata::new("success");
|
|
1830
|
+
registry.register("success", handler, metadata).unwrap();
|
|
1831
|
+
|
|
1832
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1833
|
+
|
|
1834
|
+
let batch = vec![
|
|
1835
|
+
JsonRpcRequest::new("success", None, Some(json!(1))),
|
|
1836
|
+
JsonRpcRequest::new("failing_method", None, Some(json!(2))),
|
|
1837
|
+
JsonRpcRequest::new("success", None, Some(json!(3))),
|
|
1838
|
+
JsonRpcRequest::new("failing_method", None, Some(json!(4))),
|
|
1839
|
+
JsonRpcRequest::new("success", None, Some(json!(5))),
|
|
1840
|
+
];
|
|
1841
|
+
let http_request = create_test_http_request();
|
|
1842
|
+
let request_data = create_test_request_data();
|
|
1843
|
+
|
|
1844
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1845
|
+
|
|
1846
|
+
assert!(result.is_ok());
|
|
1847
|
+
let responses = result.unwrap();
|
|
1848
|
+
assert_eq!(responses.len(), 5);
|
|
1849
|
+
|
|
1850
|
+
match &responses[0] {
|
|
1851
|
+
JsonRpcResponseType::Success(s) => {
|
|
1852
|
+
assert_eq!(s.id, json!(1));
|
|
1853
|
+
}
|
|
1854
|
+
_ => panic!("Expected success at index 0"),
|
|
1855
|
+
}
|
|
1856
|
+
match &responses[1] {
|
|
1857
|
+
JsonRpcResponseType::Error(e) => {
|
|
1858
|
+
assert_eq!(e.id, json!(2));
|
|
1859
|
+
assert_eq!(e.error.code, error_codes::METHOD_NOT_FOUND);
|
|
1860
|
+
}
|
|
1861
|
+
_ => panic!("Expected error at index 1"),
|
|
1862
|
+
}
|
|
1863
|
+
match &responses[2] {
|
|
1864
|
+
JsonRpcResponseType::Success(s) => {
|
|
1865
|
+
assert_eq!(s.id, json!(3));
|
|
1866
|
+
}
|
|
1867
|
+
_ => panic!("Expected success at index 2"),
|
|
1868
|
+
}
|
|
1869
|
+
match &responses[3] {
|
|
1870
|
+
JsonRpcResponseType::Error(e) => {
|
|
1871
|
+
assert_eq!(e.id, json!(4));
|
|
1872
|
+
assert_eq!(e.error.code, error_codes::METHOD_NOT_FOUND);
|
|
1873
|
+
}
|
|
1874
|
+
_ => panic!("Expected error at index 3"),
|
|
1875
|
+
}
|
|
1876
|
+
match &responses[4] {
|
|
1877
|
+
JsonRpcResponseType::Success(s) => {
|
|
1878
|
+
assert_eq!(s.id, json!(5));
|
|
1879
|
+
}
|
|
1880
|
+
_ => panic!("Expected success at index 4"),
|
|
1881
|
+
}
|
|
1882
|
+
}
|
|
1883
|
+
|
|
1884
|
+
#[tokio::test]
|
|
1885
|
+
async fn test_batch_response_order_with_complex_id_types() {
|
|
1886
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1887
|
+
|
|
1888
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1889
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
1890
|
+
registry.register("method", handler, metadata).unwrap();
|
|
1891
|
+
|
|
1892
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1893
|
+
|
|
1894
|
+
let batch = vec![
|
|
1895
|
+
JsonRpcRequest::new("method", None, Some(json!("alpha"))),
|
|
1896
|
+
JsonRpcRequest::new("method", None, Some(json!(999))),
|
|
1897
|
+
JsonRpcRequest::new("method", None, Some(json!(null))),
|
|
1898
|
+
JsonRpcRequest::new("method", None, Some(json!("beta"))),
|
|
1899
|
+
JsonRpcRequest::new("method", None, Some(json!(0))),
|
|
1900
|
+
];
|
|
1901
|
+
let http_request = create_test_http_request();
|
|
1902
|
+
let request_data = create_test_request_data();
|
|
1903
|
+
|
|
1904
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1905
|
+
|
|
1906
|
+
assert!(result.is_ok());
|
|
1907
|
+
let responses = result.unwrap();
|
|
1908
|
+
assert_eq!(responses.len(), 5);
|
|
1909
|
+
|
|
1910
|
+
match &responses[0] {
|
|
1911
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!("alpha")),
|
|
1912
|
+
_ => panic!("Expected success at index 0"),
|
|
1913
|
+
}
|
|
1914
|
+
match &responses[1] {
|
|
1915
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!(999)),
|
|
1916
|
+
_ => panic!("Expected success at index 1"),
|
|
1917
|
+
}
|
|
1918
|
+
match &responses[2] {
|
|
1919
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!(null)),
|
|
1920
|
+
_ => panic!("Expected success at index 2"),
|
|
1921
|
+
}
|
|
1922
|
+
match &responses[3] {
|
|
1923
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!("beta")),
|
|
1924
|
+
_ => panic!("Expected success at index 3"),
|
|
1925
|
+
}
|
|
1926
|
+
match &responses[4] {
|
|
1927
|
+
JsonRpcResponseType::Success(s) => assert_eq!(s.id, json!(0)),
|
|
1928
|
+
_ => panic!("Expected success at index 4"),
|
|
1929
|
+
}
|
|
1930
|
+
}
|
|
1931
|
+
|
|
1932
|
+
#[tokio::test]
|
|
1933
|
+
async fn test_batch_handler_error_does_not_stop_batch() {
|
|
1934
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1935
|
+
|
|
1936
|
+
let error_handler = Arc::new(MockErrorHandler);
|
|
1937
|
+
let metadata = super::super::method_registry::MethodMetadata::new("failing");
|
|
1938
|
+
registry.register("failing", error_handler, metadata).unwrap();
|
|
1939
|
+
|
|
1940
|
+
let success_handler = Arc::new(MockSuccessHandler);
|
|
1941
|
+
let metadata = super::super::method_registry::MethodMetadata::new("success");
|
|
1942
|
+
registry.register("success", success_handler, metadata).unwrap();
|
|
1943
|
+
|
|
1944
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1945
|
+
|
|
1946
|
+
let batch = vec![
|
|
1947
|
+
JsonRpcRequest::new("success", None, Some(json!(1))),
|
|
1948
|
+
JsonRpcRequest::new("failing", None, Some(json!(2))),
|
|
1949
|
+
JsonRpcRequest::new("success", None, Some(json!(3))),
|
|
1950
|
+
JsonRpcRequest::new("failing", None, Some(json!(4))),
|
|
1951
|
+
JsonRpcRequest::new("success", None, Some(json!(5))),
|
|
1952
|
+
];
|
|
1953
|
+
let http_request = create_test_http_request();
|
|
1954
|
+
let request_data = create_test_request_data();
|
|
1955
|
+
|
|
1956
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1957
|
+
|
|
1958
|
+
assert!(result.is_ok());
|
|
1959
|
+
let responses = result.unwrap();
|
|
1960
|
+
assert_eq!(responses.len(), 5);
|
|
1961
|
+
|
|
1962
|
+
for resp in &responses {
|
|
1963
|
+
match resp {
|
|
1964
|
+
JsonRpcResponseType::Success(s) => {
|
|
1965
|
+
assert_eq!(s.jsonrpc, "2.0");
|
|
1966
|
+
assert!(s.id != Value::Null);
|
|
1967
|
+
}
|
|
1968
|
+
JsonRpcResponseType::Error(e) => {
|
|
1969
|
+
assert_eq!(e.error.code, error_codes::INTERNAL_ERROR);
|
|
1970
|
+
assert!(e.id != Value::Null);
|
|
1971
|
+
}
|
|
1972
|
+
}
|
|
1973
|
+
}
|
|
1974
|
+
}
|
|
1975
|
+
|
|
1976
|
+
#[tokio::test]
|
|
1977
|
+
async fn test_batch_large_batch_within_limits() {
|
|
1978
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
1979
|
+
|
|
1980
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
1981
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
1982
|
+
registry.register("method", handler, metadata).unwrap();
|
|
1983
|
+
|
|
1984
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
1985
|
+
|
|
1986
|
+
let batch: Vec<JsonRpcRequest> = (1..=50)
|
|
1987
|
+
.map(|i| JsonRpcRequest::new("method", None, Some(json!(i))))
|
|
1988
|
+
.collect();
|
|
1989
|
+
|
|
1990
|
+
let http_request = create_test_http_request();
|
|
1991
|
+
let request_data = create_test_request_data();
|
|
1992
|
+
|
|
1993
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
1994
|
+
|
|
1995
|
+
assert!(result.is_ok());
|
|
1996
|
+
let responses = result.unwrap();
|
|
1997
|
+
assert_eq!(responses.len(), 50);
|
|
1998
|
+
for resp in &responses {
|
|
1999
|
+
match resp {
|
|
2000
|
+
JsonRpcResponseType::Success(_) => {}
|
|
2001
|
+
_ => panic!("Expected all success responses"),
|
|
2002
|
+
}
|
|
2003
|
+
}
|
|
2004
|
+
}
|
|
2005
|
+
|
|
2006
|
+
#[tokio::test]
|
|
2007
|
+
async fn test_batch_id_preservation_with_duplicates() {
|
|
2008
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
2009
|
+
|
|
2010
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
2011
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
2012
|
+
registry.register("method", handler, metadata).unwrap();
|
|
2013
|
+
|
|
2014
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
2015
|
+
|
|
2016
|
+
let batch = vec![
|
|
2017
|
+
JsonRpcRequest::new("method", None, Some(json!(1))),
|
|
2018
|
+
JsonRpcRequest::new("method", None, Some(json!(1))),
|
|
2019
|
+
JsonRpcRequest::new("method", None, Some(json!(2))),
|
|
2020
|
+
JsonRpcRequest::new("method", None, Some(json!(1))),
|
|
2021
|
+
];
|
|
2022
|
+
let http_request = create_test_http_request();
|
|
2023
|
+
let request_data = create_test_request_data();
|
|
2024
|
+
|
|
2025
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
2026
|
+
|
|
2027
|
+
assert!(result.is_ok());
|
|
2028
|
+
let responses = result.unwrap();
|
|
2029
|
+
assert_eq!(responses.len(), 4);
|
|
2030
|
+
|
|
2031
|
+
assert_eq!(responses[0].id(), json!(1));
|
|
2032
|
+
assert_eq!(responses[1].id(), json!(1));
|
|
2033
|
+
assert_eq!(responses[2].id(), json!(2));
|
|
2034
|
+
assert_eq!(responses[3].id(), json!(1));
|
|
2035
|
+
}
|
|
2036
|
+
|
|
2037
|
+
#[tokio::test]
|
|
2038
|
+
async fn test_batch_different_methods_in_sequence() {
|
|
2039
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
2040
|
+
|
|
2041
|
+
for method_name in &["user.getById", "post.create", "comment.delete", "admin.stats"] {
|
|
2042
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
2043
|
+
let metadata = super::super::method_registry::MethodMetadata::new(method_name.to_string());
|
|
2044
|
+
registry.register(method_name.to_string(), handler, metadata).unwrap();
|
|
2045
|
+
}
|
|
2046
|
+
|
|
2047
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
2048
|
+
|
|
2049
|
+
let batch = vec![
|
|
2050
|
+
JsonRpcRequest::new("user.getById", None, Some(json!(1))),
|
|
2051
|
+
JsonRpcRequest::new("post.create", None, Some(json!(2))),
|
|
2052
|
+
JsonRpcRequest::new("comment.delete", None, Some(json!(3))),
|
|
2053
|
+
JsonRpcRequest::new("admin.stats", None, Some(json!(4))),
|
|
2054
|
+
];
|
|
2055
|
+
let http_request = create_test_http_request();
|
|
2056
|
+
let request_data = create_test_request_data();
|
|
2057
|
+
|
|
2058
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
2059
|
+
|
|
2060
|
+
assert!(result.is_ok());
|
|
2061
|
+
let responses = result.unwrap();
|
|
2062
|
+
assert_eq!(responses.len(), 4);
|
|
2063
|
+
for resp in &responses {
|
|
2064
|
+
match resp {
|
|
2065
|
+
JsonRpcResponseType::Success(_) => {}
|
|
2066
|
+
_ => panic!("Expected all success responses"),
|
|
2067
|
+
}
|
|
2068
|
+
}
|
|
2069
|
+
}
|
|
2070
|
+
|
|
2071
|
+
#[tokio::test]
|
|
2072
|
+
async fn test_batch_mixed_valid_and_method_not_found_errors() {
|
|
2073
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
2074
|
+
|
|
2075
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
2076
|
+
let metadata = super::super::method_registry::MethodMetadata::new("exists");
|
|
2077
|
+
registry.register("exists", handler, metadata).unwrap();
|
|
2078
|
+
|
|
2079
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
2080
|
+
|
|
2081
|
+
let batch = vec![
|
|
2082
|
+
JsonRpcRequest::new("exists", None, Some(json!(1))),
|
|
2083
|
+
JsonRpcRequest::new("not_exists", None, Some(json!(2))),
|
|
2084
|
+
JsonRpcRequest::new("exists", None, Some(json!(3))),
|
|
2085
|
+
JsonRpcRequest::new("not_exists", None, Some(json!(4))),
|
|
2086
|
+
JsonRpcRequest::new("not_exists", None, Some(json!(5))),
|
|
2087
|
+
JsonRpcRequest::new("exists", None, Some(json!(6))),
|
|
2088
|
+
];
|
|
2089
|
+
let http_request = create_test_http_request();
|
|
2090
|
+
let request_data = create_test_request_data();
|
|
2091
|
+
|
|
2092
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
2093
|
+
|
|
2094
|
+
assert!(result.is_ok());
|
|
2095
|
+
let responses = result.unwrap();
|
|
2096
|
+
assert_eq!(responses.len(), 6);
|
|
2097
|
+
|
|
2098
|
+
match &responses[0] {
|
|
2099
|
+
JsonRpcResponseType::Success(_) => {}
|
|
2100
|
+
_ => panic!("Expected success at index 0"),
|
|
2101
|
+
}
|
|
2102
|
+
for i in [1, 3, 4] {
|
|
2103
|
+
match &responses[i] {
|
|
2104
|
+
JsonRpcResponseType::Error(e) => {
|
|
2105
|
+
assert_eq!(e.error.code, error_codes::METHOD_NOT_FOUND);
|
|
2106
|
+
}
|
|
2107
|
+
_ => panic!("Expected error at index {}", i),
|
|
2108
|
+
}
|
|
2109
|
+
}
|
|
2110
|
+
match &responses[2] {
|
|
2111
|
+
JsonRpcResponseType::Success(_) => {}
|
|
2112
|
+
_ => panic!("Expected success at index 2"),
|
|
2113
|
+
}
|
|
2114
|
+
match &responses[5] {
|
|
2115
|
+
JsonRpcResponseType::Success(_) => {}
|
|
2116
|
+
_ => panic!("Expected success at index 5"),
|
|
2117
|
+
}
|
|
2118
|
+
}
|
|
2119
|
+
|
|
2120
|
+
#[tokio::test]
|
|
2121
|
+
async fn test_batch_request_data_shared_correctly() {
|
|
2122
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
2123
|
+
|
|
2124
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
2125
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
2126
|
+
registry.register("method", handler, metadata).unwrap();
|
|
2127
|
+
|
|
2128
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
2129
|
+
|
|
2130
|
+
let batch = vec![
|
|
2131
|
+
JsonRpcRequest::new("method", None, Some(json!(1))),
|
|
2132
|
+
JsonRpcRequest::new("method", None, Some(json!(2))),
|
|
2133
|
+
JsonRpcRequest::new("method", None, Some(json!(3))),
|
|
2134
|
+
];
|
|
2135
|
+
let http_request = create_test_http_request();
|
|
2136
|
+
let request_data = create_test_request_data();
|
|
2137
|
+
|
|
2138
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
2139
|
+
|
|
2140
|
+
assert!(result.is_ok());
|
|
2141
|
+
let responses = result.unwrap();
|
|
2142
|
+
assert_eq!(responses.len(), 3);
|
|
2143
|
+
|
|
2144
|
+
for resp in &responses {
|
|
2145
|
+
match resp {
|
|
2146
|
+
JsonRpcResponseType::Success(s) => {
|
|
2147
|
+
assert_eq!(s.jsonrpc, "2.0");
|
|
2148
|
+
assert!(s.result != Value::Null);
|
|
2149
|
+
}
|
|
2150
|
+
_ => panic!("Expected success response"),
|
|
2151
|
+
}
|
|
2152
|
+
}
|
|
2153
|
+
}
|
|
2154
|
+
|
|
2155
|
+
#[tokio::test]
|
|
2156
|
+
async fn test_batch_response_contains_required_fields() {
|
|
2157
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
2158
|
+
|
|
2159
|
+
let handler = Arc::new(MockSuccessHandler);
|
|
2160
|
+
let metadata = super::super::method_registry::MethodMetadata::new("method");
|
|
2161
|
+
registry.register("method", handler, metadata).unwrap();
|
|
2162
|
+
|
|
2163
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
2164
|
+
|
|
2165
|
+
let batch = vec![
|
|
2166
|
+
JsonRpcRequest::new("method", None, Some(json!(1))),
|
|
2167
|
+
JsonRpcRequest::new("method", None, Some(json!(2))),
|
|
2168
|
+
];
|
|
2169
|
+
let http_request = create_test_http_request();
|
|
2170
|
+
let request_data = create_test_request_data();
|
|
2171
|
+
|
|
2172
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
2173
|
+
|
|
2174
|
+
assert!(result.is_ok());
|
|
2175
|
+
let responses = result.unwrap();
|
|
2176
|
+
|
|
2177
|
+
for resp in &responses {
|
|
2178
|
+
match resp {
|
|
2179
|
+
JsonRpcResponseType::Success(s) => {
|
|
2180
|
+
assert_eq!(s.jsonrpc, "2.0");
|
|
2181
|
+
assert!(!s.result.is_null());
|
|
2182
|
+
assert!(!s.id.is_null());
|
|
2183
|
+
}
|
|
2184
|
+
_ => panic!("Expected success response"),
|
|
2185
|
+
}
|
|
2186
|
+
}
|
|
2187
|
+
}
|
|
2188
|
+
|
|
2189
|
+
#[tokio::test]
|
|
2190
|
+
async fn test_batch_error_response_required_fields() {
|
|
2191
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
2192
|
+
|
|
2193
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
2194
|
+
|
|
2195
|
+
let batch = vec![
|
|
2196
|
+
JsonRpcRequest::new("not_found", None, Some(json!(1))),
|
|
2197
|
+
JsonRpcRequest::new("also_not_found", None, Some(json!(2))),
|
|
2198
|
+
];
|
|
2199
|
+
let http_request = create_test_http_request();
|
|
2200
|
+
let request_data = create_test_request_data();
|
|
2201
|
+
|
|
2202
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
2203
|
+
|
|
2204
|
+
assert!(result.is_ok());
|
|
2205
|
+
let responses = result.unwrap();
|
|
2206
|
+
|
|
2207
|
+
for resp in &responses {
|
|
2208
|
+
match resp {
|
|
2209
|
+
JsonRpcResponseType::Error(e) => {
|
|
2210
|
+
assert_eq!(e.jsonrpc, "2.0");
|
|
2211
|
+
assert!(e.error.code < 0);
|
|
2212
|
+
assert!(!e.error.message.is_empty());
|
|
2213
|
+
assert!(!e.id.is_null());
|
|
2214
|
+
}
|
|
2215
|
+
_ => panic!("Expected error response"),
|
|
2216
|
+
}
|
|
2217
|
+
}
|
|
2218
|
+
}
|
|
2219
|
+
|
|
2220
|
+
#[tokio::test]
|
|
2221
|
+
async fn test_batch_notification_errors_still_not_responded() {
|
|
2222
|
+
let registry = Arc::new(JsonRpcMethodRegistry::new());
|
|
2223
|
+
|
|
2224
|
+
let router = JsonRpcRouter::new(registry.clone(), true, 100);
|
|
2225
|
+
|
|
2226
|
+
let batch = vec![
|
|
2227
|
+
JsonRpcRequest::new("not_found", None, Some(json!(1))),
|
|
2228
|
+
JsonRpcRequest::new("not_found", None, None),
|
|
2229
|
+
JsonRpcRequest::new("not_found", None, Some(json!(2))),
|
|
2230
|
+
];
|
|
2231
|
+
let http_request = create_test_http_request();
|
|
2232
|
+
let request_data = create_test_request_data();
|
|
2233
|
+
|
|
2234
|
+
let result = router.route_batch(batch, http_request, &request_data).await;
|
|
2235
|
+
|
|
2236
|
+
assert!(result.is_ok());
|
|
2237
|
+
let responses = result.unwrap();
|
|
2238
|
+
assert_eq!(responses.len(), 2);
|
|
2239
|
+
|
|
2240
|
+
match &responses[0] {
|
|
2241
|
+
JsonRpcResponseType::Error(e) => {
|
|
2242
|
+
assert_eq!(e.id, json!(1));
|
|
2243
|
+
}
|
|
2244
|
+
_ => panic!("Expected error at index 0"),
|
|
2245
|
+
}
|
|
2246
|
+
match &responses[1] {
|
|
2247
|
+
JsonRpcResponseType::Error(e) => {
|
|
2248
|
+
assert_eq!(e.id, json!(2));
|
|
2249
|
+
}
|
|
2250
|
+
_ => panic!("Expected error at index 1"),
|
|
2251
|
+
}
|
|
2252
|
+
}
|
|
2253
|
+
|
|
2254
|
+
impl JsonRpcResponseType {
|
|
2255
|
+
fn id(&self) -> Value {
|
|
2256
|
+
match self {
|
|
2257
|
+
JsonRpcResponseType::Success(s) => s.id.clone(),
|
|
2258
|
+
JsonRpcResponseType::Error(e) => e.id.clone(),
|
|
2259
|
+
}
|
|
2260
|
+
}
|
|
2261
|
+
}
|
|
2262
|
+
}
|