spikard 0.12.0 → 0.15.2

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 (206) hide show
  1. checksums.yaml +4 -4
  2. data/Steepfile +6 -0
  3. data/ext/spikard_rb/extconf.rb +1 -2
  4. data/ext/spikard_rb/{Cargo.lock → native/Cargo.lock} +897 -451
  5. data/ext/spikard_rb/native/Cargo.toml +24 -0
  6. data/ext/spikard_rb/src/lib.rs +5366 -3
  7. data/lib/spikard/native.rb +86 -0
  8. data/lib/spikard/version.rb +6 -1
  9. data/lib/spikard.rb +8 -45
  10. data/lib/spikard_rb.so +0 -0
  11. data/sig/types.rbs +427 -0
  12. metadata +14 -242
  13. data/LICENSE +0 -1
  14. data/README.md +0 -267
  15. data/ext/spikard_rb/Cargo.toml +0 -17
  16. data/lib/spikard/app.rb +0 -428
  17. data/lib/spikard/background.rb +0 -58
  18. data/lib/spikard/config.rb +0 -506
  19. data/lib/spikard/converters.rb +0 -13
  20. data/lib/spikard/grpc.rb +0 -182
  21. data/lib/spikard/handler_wrapper.rb +0 -113
  22. data/lib/spikard/provide.rb +0 -214
  23. data/lib/spikard/response.rb +0 -173
  24. data/lib/spikard/schema.rb +0 -243
  25. data/lib/spikard/sse.rb +0 -111
  26. data/lib/spikard/streaming_response.rb +0 -44
  27. data/lib/spikard/testing.rb +0 -432
  28. data/lib/spikard/upload_file.rb +0 -131
  29. data/lib/spikard/websocket.rb +0 -59
  30. data/sig/spikard.rbs +0 -719
  31. data/vendor/crates/spikard-bindings-shared/Cargo.toml +0 -80
  32. data/vendor/crates/spikard-bindings-shared/examples/config_extraction.rs +0 -132
  33. data/vendor/crates/spikard-bindings-shared/src/config_extractor.rs +0 -905
  34. data/vendor/crates/spikard-bindings-shared/src/conversion_traits.rs +0 -210
  35. data/vendor/crates/spikard-bindings-shared/src/di_traits.rs +0 -252
  36. data/vendor/crates/spikard-bindings-shared/src/error_response.rs +0 -404
  37. data/vendor/crates/spikard-bindings-shared/src/grpc_metadata.rs +0 -199
  38. data/vendor/crates/spikard-bindings-shared/src/handler_base.rs +0 -252
  39. data/vendor/crates/spikard-bindings-shared/src/json_conversion.rs +0 -829
  40. data/vendor/crates/spikard-bindings-shared/src/lazy_cache.rs +0 -587
  41. data/vendor/crates/spikard-bindings-shared/src/lib.rs +0 -33
  42. data/vendor/crates/spikard-bindings-shared/src/lifecycle_base.rs +0 -298
  43. data/vendor/crates/spikard-bindings-shared/src/lifecycle_executor.rs +0 -594
  44. data/vendor/crates/spikard-bindings-shared/src/response_builder.rs +0 -743
  45. data/vendor/crates/spikard-bindings-shared/src/response_interpreter.rs +0 -944
  46. data/vendor/crates/spikard-bindings-shared/src/test_client_base.rs +0 -260
  47. data/vendor/crates/spikard-bindings-shared/src/validation_helpers.rs +0 -369
  48. data/vendor/crates/spikard-bindings-shared/tests/config_extractor_behavior.rs +0 -192
  49. data/vendor/crates/spikard-bindings-shared/tests/error_response_edge_cases.rs +0 -383
  50. data/vendor/crates/spikard-bindings-shared/tests/full_coverage.rs +0 -459
  51. data/vendor/crates/spikard-bindings-shared/tests/handler_base_integration.rs +0 -280
  52. data/vendor/crates/spikard-bindings-shared/tests/integration_tests.rs +0 -669
  53. data/vendor/crates/spikard-core/Cargo.toml +0 -60
  54. data/vendor/crates/spikard-core/src/bindings/mod.rs +0 -3
  55. data/vendor/crates/spikard-core/src/bindings/response.rs +0 -130
  56. data/vendor/crates/spikard-core/src/debug.rs +0 -127
  57. data/vendor/crates/spikard-core/src/di/container.rs +0 -702
  58. data/vendor/crates/spikard-core/src/di/dependency.rs +0 -273
  59. data/vendor/crates/spikard-core/src/di/error.rs +0 -118
  60. data/vendor/crates/spikard-core/src/di/factory.rs +0 -538
  61. data/vendor/crates/spikard-core/src/di/graph.rs +0 -507
  62. data/vendor/crates/spikard-core/src/di/mod.rs +0 -192
  63. data/vendor/crates/spikard-core/src/di/resolved.rs +0 -428
  64. data/vendor/crates/spikard-core/src/di/value.rs +0 -282
  65. data/vendor/crates/spikard-core/src/errors.rs +0 -72
  66. data/vendor/crates/spikard-core/src/http.rs +0 -492
  67. data/vendor/crates/spikard-core/src/lib.rs +0 -29
  68. data/vendor/crates/spikard-core/src/lifecycle.rs +0 -1273
  69. data/vendor/crates/spikard-core/src/metadata.rs +0 -378
  70. data/vendor/crates/spikard-core/src/parameters.rs +0 -2546
  71. data/vendor/crates/spikard-core/src/problem.rs +0 -358
  72. data/vendor/crates/spikard-core/src/request_data.rs +0 -1146
  73. data/vendor/crates/spikard-core/src/router.rs +0 -530
  74. data/vendor/crates/spikard-core/src/schema_registry.rs +0 -197
  75. data/vendor/crates/spikard-core/src/type_hints.rs +0 -311
  76. data/vendor/crates/spikard-core/src/validation/error_mapper.rs +0 -710
  77. data/vendor/crates/spikard-core/src/validation/mod.rs +0 -470
  78. data/vendor/crates/spikard-core/tests/bindings_response_tests.rs +0 -136
  79. data/vendor/crates/spikard-core/tests/di_dependency_defaults.rs +0 -37
  80. data/vendor/crates/spikard-core/tests/error_mapper.rs +0 -761
  81. data/vendor/crates/spikard-core/tests/parameters_edge_cases.rs +0 -106
  82. data/vendor/crates/spikard-core/tests/parameters_full.rs +0 -701
  83. data/vendor/crates/spikard-core/tests/parameters_schema_and_formats.rs +0 -301
  84. data/vendor/crates/spikard-core/tests/request_data_roundtrip.rs +0 -67
  85. data/vendor/crates/spikard-core/tests/validation_coverage.rs +0 -250
  86. data/vendor/crates/spikard-core/tests/validation_error_paths.rs +0 -45
  87. data/vendor/crates/spikard-http/Cargo.toml +0 -87
  88. data/vendor/crates/spikard-http/examples/sse-notifications.rs +0 -148
  89. data/vendor/crates/spikard-http/examples/websocket-chat.rs +0 -92
  90. data/vendor/crates/spikard-http/src/auth.rs +0 -301
  91. data/vendor/crates/spikard-http/src/background.rs +0 -1860
  92. data/vendor/crates/spikard-http/src/bindings/mod.rs +0 -3
  93. data/vendor/crates/spikard-http/src/bindings/response.rs +0 -1
  94. data/vendor/crates/spikard-http/src/body_metadata.rs +0 -8
  95. data/vendor/crates/spikard-http/src/cors.rs +0 -1026
  96. data/vendor/crates/spikard-http/src/debug.rs +0 -128
  97. data/vendor/crates/spikard-http/src/di_handler.rs +0 -1672
  98. data/vendor/crates/spikard-http/src/grpc/framing.rs +0 -469
  99. data/vendor/crates/spikard-http/src/grpc/handler.rs +0 -1122
  100. data/vendor/crates/spikard-http/src/grpc/mod.rs +0 -434
  101. data/vendor/crates/spikard-http/src/grpc/service.rs +0 -622
  102. data/vendor/crates/spikard-http/src/grpc/streaming.rs +0 -319
  103. data/vendor/crates/spikard-http/src/handler_response.rs +0 -901
  104. data/vendor/crates/spikard-http/src/handler_trait.rs +0 -1015
  105. data/vendor/crates/spikard-http/src/handler_trait_tests.rs +0 -290
  106. data/vendor/crates/spikard-http/src/jsonrpc/http_handler.rs +0 -502
  107. data/vendor/crates/spikard-http/src/jsonrpc/method_registry.rs +0 -648
  108. data/vendor/crates/spikard-http/src/jsonrpc/mod.rs +0 -58
  109. data/vendor/crates/spikard-http/src/jsonrpc/protocol.rs +0 -1207
  110. data/vendor/crates/spikard-http/src/jsonrpc/router.rs +0 -2262
  111. data/vendor/crates/spikard-http/src/lib.rs +0 -548
  112. data/vendor/crates/spikard-http/src/lifecycle/adapter.rs +0 -230
  113. data/vendor/crates/spikard-http/src/lifecycle.rs +0 -1193
  114. data/vendor/crates/spikard-http/src/middleware/mod.rs +0 -560
  115. data/vendor/crates/spikard-http/src/middleware/multipart.rs +0 -912
  116. data/vendor/crates/spikard-http/src/middleware/urlencoded.rs +0 -513
  117. data/vendor/crates/spikard-http/src/middleware/validation.rs +0 -768
  118. data/vendor/crates/spikard-http/src/openapi/mod.rs +0 -309
  119. data/vendor/crates/spikard-http/src/openapi/parameter_extraction.rs +0 -535
  120. data/vendor/crates/spikard-http/src/openapi/schema_conversion.rs +0 -1363
  121. data/vendor/crates/spikard-http/src/openapi/spec_generation.rs +0 -667
  122. data/vendor/crates/spikard-http/src/query_parser.rs +0 -793
  123. data/vendor/crates/spikard-http/src/response.rs +0 -720
  124. data/vendor/crates/spikard-http/src/server/fast_router.rs +0 -186
  125. data/vendor/crates/spikard-http/src/server/grpc_routing.rs +0 -858
  126. data/vendor/crates/spikard-http/src/server/handler.rs +0 -1661
  127. data/vendor/crates/spikard-http/src/server/lifecycle_execution.rs +0 -253
  128. data/vendor/crates/spikard-http/src/server/mod.rs +0 -1649
  129. data/vendor/crates/spikard-http/src/server/request_extraction.rs +0 -871
  130. data/vendor/crates/spikard-http/src/server/routing_factory.rs +0 -618
  131. data/vendor/crates/spikard-http/src/sse.rs +0 -1409
  132. data/vendor/crates/spikard-http/src/testing/form.rs +0 -52
  133. data/vendor/crates/spikard-http/src/testing/multipart.rs +0 -64
  134. data/vendor/crates/spikard-http/src/testing/test_client.rs +0 -787
  135. data/vendor/crates/spikard-http/src/testing.rs +0 -617
  136. data/vendor/crates/spikard-http/src/websocket.rs +0 -1477
  137. data/vendor/crates/spikard-http/tests/auth_integration.rs +0 -645
  138. data/vendor/crates/spikard-http/tests/background_behavior.rs +0 -832
  139. data/vendor/crates/spikard-http/tests/common/grpc_helpers.rs +0 -1012
  140. data/vendor/crates/spikard-http/tests/common/handlers.rs +0 -309
  141. data/vendor/crates/spikard-http/tests/common/mod.rs +0 -33
  142. data/vendor/crates/spikard-http/tests/common/test_builders.rs +0 -628
  143. data/vendor/crates/spikard-http/tests/di_handler_error_responses.rs +0 -162
  144. data/vendor/crates/spikard-http/tests/di_integration.rs +0 -192
  145. data/vendor/crates/spikard-http/tests/doc_snippets.rs +0 -5
  146. data/vendor/crates/spikard-http/tests/grpc_bidirectional_streaming.rs +0 -430
  147. data/vendor/crates/spikard-http/tests/grpc_client_streaming.rs +0 -738
  148. data/vendor/crates/spikard-http/tests/grpc_error_handling_test.rs +0 -652
  149. data/vendor/crates/spikard-http/tests/grpc_integration_test.rs +0 -334
  150. data/vendor/crates/spikard-http/tests/grpc_metadata_test.rs +0 -532
  151. data/vendor/crates/spikard-http/tests/grpc_server_integration.rs +0 -495
  152. data/vendor/crates/spikard-http/tests/grpc_server_streaming.rs +0 -974
  153. data/vendor/crates/spikard-http/tests/lifecycle_execution.rs +0 -1093
  154. data/vendor/crates/spikard-http/tests/middleware_stack_integration.rs +0 -389
  155. data/vendor/crates/spikard-http/tests/multipart_behavior.rs +0 -656
  156. data/vendor/crates/spikard-http/tests/request_extraction_full.rs +0 -513
  157. data/vendor/crates/spikard-http/tests/server_auth_middleware_behavior.rs +0 -328
  158. data/vendor/crates/spikard-http/tests/server_config_builder.rs +0 -314
  159. data/vendor/crates/spikard-http/tests/server_configured_router_behavior.rs +0 -200
  160. data/vendor/crates/spikard-http/tests/server_cors_preflight.rs +0 -83
  161. data/vendor/crates/spikard-http/tests/server_handler_wrappers.rs +0 -464
  162. data/vendor/crates/spikard-http/tests/server_method_router_additional_behavior.rs +0 -286
  163. data/vendor/crates/spikard-http/tests/server_method_router_coverage.rs +0 -118
  164. data/vendor/crates/spikard-http/tests/server_middleware_behavior.rs +0 -99
  165. data/vendor/crates/spikard-http/tests/server_middleware_branches.rs +0 -204
  166. data/vendor/crates/spikard-http/tests/server_openapi_jsonrpc_static.rs +0 -421
  167. data/vendor/crates/spikard-http/tests/server_router_behavior.rs +0 -121
  168. data/vendor/crates/spikard-http/tests/sse_behavior.rs +0 -620
  169. data/vendor/crates/spikard-http/tests/sse_full_behavior.rs +0 -584
  170. data/vendor/crates/spikard-http/tests/sse_handler_behavior.rs +0 -130
  171. data/vendor/crates/spikard-http/tests/test_client_requests.rs +0 -167
  172. data/vendor/crates/spikard-http/tests/testing_helpers.rs +0 -87
  173. data/vendor/crates/spikard-http/tests/testing_module_coverage.rs +0 -155
  174. data/vendor/crates/spikard-http/tests/urlencoded_content_type.rs +0 -82
  175. data/vendor/crates/spikard-http/tests/websocket_behavior.rs +0 -663
  176. data/vendor/crates/spikard-http/tests/websocket_full_behavior.rs +0 -440
  177. data/vendor/crates/spikard-http/tests/websocket_integration.rs +0 -150
  178. data/vendor/crates/spikard-rb/Cargo.toml +0 -68
  179. data/vendor/crates/spikard-rb/build.rs +0 -200
  180. data/vendor/crates/spikard-rb/src/background.rs +0 -63
  181. data/vendor/crates/spikard-rb/src/config/mod.rs +0 -5
  182. data/vendor/crates/spikard-rb/src/config/server_config.rs +0 -401
  183. data/vendor/crates/spikard-rb/src/conversion.rs +0 -688
  184. data/vendor/crates/spikard-rb/src/di/builder.rs +0 -100
  185. data/vendor/crates/spikard-rb/src/di/mod.rs +0 -375
  186. data/vendor/crates/spikard-rb/src/grpc/handler.rs +0 -834
  187. data/vendor/crates/spikard-rb/src/grpc/mod.rs +0 -13
  188. data/vendor/crates/spikard-rb/src/gvl.rs +0 -80
  189. data/vendor/crates/spikard-rb/src/handler.rs +0 -699
  190. data/vendor/crates/spikard-rb/src/integration/mod.rs +0 -3
  191. data/vendor/crates/spikard-rb/src/lib.rs +0 -2264
  192. data/vendor/crates/spikard-rb/src/lifecycle.rs +0 -303
  193. data/vendor/crates/spikard-rb/src/metadata/mod.rs +0 -5
  194. data/vendor/crates/spikard-rb/src/metadata/route_extraction.rs +0 -507
  195. data/vendor/crates/spikard-rb/src/request.rs +0 -439
  196. data/vendor/crates/spikard-rb/src/runtime/mod.rs +0 -5
  197. data/vendor/crates/spikard-rb/src/runtime/server_runner.rs +0 -344
  198. data/vendor/crates/spikard-rb/src/server.rs +0 -307
  199. data/vendor/crates/spikard-rb/src/sse.rs +0 -231
  200. data/vendor/crates/spikard-rb/src/testing/client.rs +0 -698
  201. data/vendor/crates/spikard-rb/src/testing/mod.rs +0 -7
  202. data/vendor/crates/spikard-rb/src/testing/sse.rs +0 -108
  203. data/vendor/crates/spikard-rb/src/testing/websocket.rs +0 -573
  204. data/vendor/crates/spikard-rb/src/websocket.rs +0 -475
  205. data/vendor/crates/spikard-rb-macros/Cargo.toml +0 -25
  206. data/vendor/crates/spikard-rb-macros/src/lib.rs +0 -51
@@ -1,2262 +0,0 @@
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: Arc::new(Value::Object(serde_json::Map::new())),
352
- validated_params: None,
353
- raw_query_params: Arc::new(HashMap::new()),
354
- body: Arc::new(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
- }