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.
Files changed (68) hide show
  1. checksums.yaml +4 -4
  2. data/README.md +90 -508
  3. data/ext/spikard_rb/Cargo.lock +3287 -0
  4. data/ext/spikard_rb/Cargo.toml +1 -1
  5. data/ext/spikard_rb/extconf.rb +3 -3
  6. data/lib/spikard/app.rb +72 -49
  7. data/lib/spikard/background.rb +38 -7
  8. data/lib/spikard/testing.rb +42 -4
  9. data/lib/spikard/version.rb +1 -1
  10. data/sig/spikard.rbs +4 -0
  11. data/vendor/crates/spikard-bindings-shared/Cargo.toml +1 -1
  12. data/vendor/crates/spikard-bindings-shared/tests/config_extractor_behavior.rs +191 -0
  13. data/vendor/crates/spikard-core/Cargo.toml +1 -1
  14. data/vendor/crates/spikard-core/src/http.rs +1 -0
  15. data/vendor/crates/spikard-core/src/lifecycle.rs +63 -0
  16. data/vendor/crates/spikard-core/tests/bindings_response_tests.rs +136 -0
  17. data/vendor/crates/spikard-core/tests/di_dependency_defaults.rs +37 -0
  18. data/vendor/crates/spikard-core/tests/error_mapper.rs +761 -0
  19. data/vendor/crates/spikard-core/tests/parameters_edge_cases.rs +106 -0
  20. data/vendor/crates/spikard-core/tests/parameters_full.rs +701 -0
  21. data/vendor/crates/spikard-core/tests/parameters_schema_and_formats.rs +301 -0
  22. data/vendor/crates/spikard-core/tests/request_data_roundtrip.rs +67 -0
  23. data/vendor/crates/spikard-core/tests/validation_coverage.rs +250 -0
  24. data/vendor/crates/spikard-core/tests/validation_error_paths.rs +45 -0
  25. data/vendor/crates/spikard-http/Cargo.toml +1 -1
  26. data/vendor/crates/spikard-http/src/jsonrpc/http_handler.rs +502 -0
  27. data/vendor/crates/spikard-http/src/jsonrpc/method_registry.rs +648 -0
  28. data/vendor/crates/spikard-http/src/jsonrpc/mod.rs +58 -0
  29. data/vendor/crates/spikard-http/src/jsonrpc/protocol.rs +1207 -0
  30. data/vendor/crates/spikard-http/src/jsonrpc/router.rs +2262 -0
  31. data/vendor/crates/spikard-http/src/testing/test_client.rs +155 -2
  32. data/vendor/crates/spikard-http/src/testing.rs +171 -0
  33. data/vendor/crates/spikard-http/src/websocket.rs +79 -6
  34. data/vendor/crates/spikard-http/tests/auth_integration.rs +647 -0
  35. data/vendor/crates/spikard-http/tests/common/test_builders.rs +633 -0
  36. data/vendor/crates/spikard-http/tests/di_handler_error_responses.rs +162 -0
  37. data/vendor/crates/spikard-http/tests/middleware_stack_integration.rs +389 -0
  38. data/vendor/crates/spikard-http/tests/request_extraction_full.rs +513 -0
  39. data/vendor/crates/spikard-http/tests/server_auth_middleware_behavior.rs +244 -0
  40. data/vendor/crates/spikard-http/tests/server_configured_router_behavior.rs +200 -0
  41. data/vendor/crates/spikard-http/tests/server_cors_preflight.rs +82 -0
  42. data/vendor/crates/spikard-http/tests/server_handler_wrappers.rs +464 -0
  43. data/vendor/crates/spikard-http/tests/server_method_router_additional_behavior.rs +286 -0
  44. data/vendor/crates/spikard-http/tests/server_method_router_coverage.rs +118 -0
  45. data/vendor/crates/spikard-http/tests/server_middleware_behavior.rs +99 -0
  46. data/vendor/crates/spikard-http/tests/server_middleware_branches.rs +206 -0
  47. data/vendor/crates/spikard-http/tests/server_openapi_jsonrpc_static.rs +281 -0
  48. data/vendor/crates/spikard-http/tests/server_router_behavior.rs +121 -0
  49. data/vendor/crates/spikard-http/tests/sse_full_behavior.rs +584 -0
  50. data/vendor/crates/spikard-http/tests/sse_handler_behavior.rs +130 -0
  51. data/vendor/crates/spikard-http/tests/test_client_requests.rs +167 -0
  52. data/vendor/crates/spikard-http/tests/testing_helpers.rs +87 -0
  53. data/vendor/crates/spikard-http/tests/testing_module_coverage.rs +156 -0
  54. data/vendor/crates/spikard-http/tests/urlencoded_content_type.rs +82 -0
  55. data/vendor/crates/spikard-http/tests/websocket_full_behavior.rs +440 -0
  56. data/vendor/crates/spikard-http/tests/websocket_integration.rs +152 -0
  57. data/vendor/crates/spikard-rb/Cargo.toml +1 -1
  58. data/vendor/crates/spikard-rb/src/gvl.rs +80 -0
  59. data/vendor/crates/spikard-rb/src/handler.rs +12 -9
  60. data/vendor/crates/spikard-rb/src/lib.rs +137 -124
  61. data/vendor/crates/spikard-rb/src/request.rs +342 -0
  62. data/vendor/crates/spikard-rb/src/runtime/server_runner.rs +1 -8
  63. data/vendor/crates/spikard-rb/src/server.rs +1 -8
  64. data/vendor/crates/spikard-rb/src/testing/client.rs +168 -9
  65. data/vendor/crates/spikard-rb/src/websocket.rs +119 -30
  66. data/vendor/crates/spikard-rb-macros/Cargo.toml +14 -0
  67. data/vendor/crates/spikard-rb-macros/src/lib.rs +52 -0
  68. 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
+ }