spikard 0.5.0 → 0.6.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/LICENSE +1 -1
- data/README.md +674 -674
- data/ext/spikard_rb/Cargo.toml +17 -17
- data/ext/spikard_rb/extconf.rb +13 -10
- data/ext/spikard_rb/src/lib.rs +6 -6
- data/lib/spikard/app.rb +405 -405
- data/lib/spikard/background.rb +27 -27
- data/lib/spikard/config.rb +396 -396
- data/lib/spikard/converters.rb +13 -13
- data/lib/spikard/handler_wrapper.rb +113 -113
- data/lib/spikard/provide.rb +214 -214
- data/lib/spikard/response.rb +173 -173
- data/lib/spikard/schema.rb +243 -243
- data/lib/spikard/sse.rb +111 -111
- data/lib/spikard/streaming_response.rb +44 -44
- data/lib/spikard/testing.rb +256 -256
- data/lib/spikard/upload_file.rb +131 -131
- data/lib/spikard/version.rb +5 -5
- data/lib/spikard/websocket.rb +59 -59
- data/lib/spikard.rb +43 -43
- data/sig/spikard.rbs +366 -366
- data/vendor/crates/spikard-bindings-shared/Cargo.toml +63 -63
- data/vendor/crates/spikard-bindings-shared/examples/config_extraction.rs +132 -132
- data/vendor/crates/spikard-bindings-shared/src/config_extractor.rs +752 -752
- data/vendor/crates/spikard-bindings-shared/src/conversion_traits.rs +194 -194
- data/vendor/crates/spikard-bindings-shared/src/di_traits.rs +246 -246
- data/vendor/crates/spikard-bindings-shared/src/error_response.rs +401 -401
- data/vendor/crates/spikard-bindings-shared/src/handler_base.rs +238 -238
- data/vendor/crates/spikard-bindings-shared/src/lib.rs +24 -24
- data/vendor/crates/spikard-bindings-shared/src/lifecycle_base.rs +292 -292
- data/vendor/crates/spikard-bindings-shared/src/lifecycle_executor.rs +616 -616
- data/vendor/crates/spikard-bindings-shared/src/response_builder.rs +305 -305
- data/vendor/crates/spikard-bindings-shared/src/test_client_base.rs +248 -248
- data/vendor/crates/spikard-bindings-shared/src/validation_helpers.rs +351 -351
- data/vendor/crates/spikard-bindings-shared/tests/comprehensive_coverage.rs +454 -454
- data/vendor/crates/spikard-bindings-shared/tests/error_response_edge_cases.rs +383 -383
- data/vendor/crates/spikard-bindings-shared/tests/handler_base_integration.rs +280 -280
- data/vendor/crates/spikard-core/Cargo.toml +40 -40
- data/vendor/crates/spikard-core/src/bindings/mod.rs +3 -3
- data/vendor/crates/spikard-core/src/bindings/response.rs +133 -133
- data/vendor/crates/spikard-core/src/debug.rs +127 -127
- data/vendor/crates/spikard-core/src/di/container.rs +702 -702
- data/vendor/crates/spikard-core/src/di/dependency.rs +273 -273
- data/vendor/crates/spikard-core/src/di/error.rs +118 -118
- data/vendor/crates/spikard-core/src/di/factory.rs +534 -534
- data/vendor/crates/spikard-core/src/di/graph.rs +506 -506
- data/vendor/crates/spikard-core/src/di/mod.rs +192 -192
- data/vendor/crates/spikard-core/src/di/resolved.rs +405 -405
- data/vendor/crates/spikard-core/src/di/value.rs +281 -281
- data/vendor/crates/spikard-core/src/errors.rs +69 -69
- data/vendor/crates/spikard-core/src/http.rs +415 -415
- data/vendor/crates/spikard-core/src/lib.rs +29 -29
- data/vendor/crates/spikard-core/src/lifecycle.rs +1186 -1186
- data/vendor/crates/spikard-core/src/metadata.rs +389 -389
- data/vendor/crates/spikard-core/src/parameters.rs +2525 -2525
- data/vendor/crates/spikard-core/src/problem.rs +344 -344
- data/vendor/crates/spikard-core/src/request_data.rs +1154 -1154
- data/vendor/crates/spikard-core/src/router.rs +510 -510
- data/vendor/crates/spikard-core/src/schema_registry.rs +183 -183
- data/vendor/crates/spikard-core/src/type_hints.rs +304 -304
- data/vendor/crates/spikard-core/src/validation/error_mapper.rs +696 -688
- data/vendor/crates/spikard-core/src/validation/mod.rs +457 -457
- data/vendor/crates/spikard-http/Cargo.toml +62 -64
- data/vendor/crates/spikard-http/examples/sse-notifications.rs +148 -148
- data/vendor/crates/spikard-http/examples/websocket-chat.rs +92 -92
- data/vendor/crates/spikard-http/src/auth.rs +296 -296
- data/vendor/crates/spikard-http/src/background.rs +1860 -1860
- data/vendor/crates/spikard-http/src/bindings/mod.rs +3 -3
- data/vendor/crates/spikard-http/src/bindings/response.rs +1 -1
- data/vendor/crates/spikard-http/src/body_metadata.rs +8 -8
- data/vendor/crates/spikard-http/src/cors.rs +1005 -1005
- data/vendor/crates/spikard-http/src/debug.rs +128 -128
- data/vendor/crates/spikard-http/src/di_handler.rs +1668 -1668
- data/vendor/crates/spikard-http/src/handler_response.rs +901 -901
- data/vendor/crates/spikard-http/src/handler_trait.rs +838 -830
- data/vendor/crates/spikard-http/src/handler_trait_tests.rs +290 -290
- data/vendor/crates/spikard-http/src/lib.rs +534 -534
- data/vendor/crates/spikard-http/src/lifecycle/adapter.rs +230 -230
- data/vendor/crates/spikard-http/src/lifecycle.rs +1193 -1193
- data/vendor/crates/spikard-http/src/middleware/mod.rs +560 -540
- data/vendor/crates/spikard-http/src/middleware/multipart.rs +912 -912
- data/vendor/crates/spikard-http/src/middleware/urlencoded.rs +513 -513
- data/vendor/crates/spikard-http/src/middleware/validation.rs +768 -735
- data/vendor/crates/spikard-http/src/openapi/mod.rs +309 -309
- data/vendor/crates/spikard-http/src/openapi/parameter_extraction.rs +535 -535
- data/vendor/crates/spikard-http/src/openapi/schema_conversion.rs +1363 -1363
- data/vendor/crates/spikard-http/src/openapi/spec_generation.rs +665 -665
- data/vendor/crates/spikard-http/src/query_parser.rs +793 -793
- data/vendor/crates/spikard-http/src/response.rs +720 -720
- data/vendor/crates/spikard-http/src/server/handler.rs +1650 -1650
- data/vendor/crates/spikard-http/src/server/lifecycle_execution.rs +234 -234
- data/vendor/crates/spikard-http/src/server/mod.rs +1593 -1502
- data/vendor/crates/spikard-http/src/server/request_extraction.rs +789 -770
- data/vendor/crates/spikard-http/src/server/routing_factory.rs +629 -599
- data/vendor/crates/spikard-http/src/sse.rs +1409 -1409
- data/vendor/crates/spikard-http/src/testing/form.rs +52 -52
- data/vendor/crates/spikard-http/src/testing/multipart.rs +64 -60
- data/vendor/crates/spikard-http/src/testing/test_client.rs +311 -283
- data/vendor/crates/spikard-http/src/testing.rs +406 -377
- data/vendor/crates/spikard-http/src/websocket.rs +1404 -1375
- data/vendor/crates/spikard-http/tests/background_behavior.rs +832 -832
- data/vendor/crates/spikard-http/tests/common/handlers.rs +309 -309
- data/vendor/crates/spikard-http/tests/common/mod.rs +26 -26
- data/vendor/crates/spikard-http/tests/di_integration.rs +192 -192
- data/vendor/crates/spikard-http/tests/doc_snippets.rs +5 -5
- data/vendor/crates/spikard-http/tests/lifecycle_execution.rs +1093 -1093
- data/vendor/crates/spikard-http/tests/multipart_behavior.rs +656 -656
- data/vendor/crates/spikard-http/tests/server_config_builder.rs +314 -314
- data/vendor/crates/spikard-http/tests/sse_behavior.rs +620 -620
- data/vendor/crates/spikard-http/tests/websocket_behavior.rs +663 -663
- data/vendor/crates/spikard-rb/Cargo.toml +48 -48
- data/vendor/crates/spikard-rb/build.rs +199 -199
- data/vendor/crates/spikard-rb/src/background.rs +63 -63
- data/vendor/crates/spikard-rb/src/config/mod.rs +5 -5
- data/vendor/crates/spikard-rb/src/config/server_config.rs +285 -285
- data/vendor/crates/spikard-rb/src/conversion.rs +554 -554
- data/vendor/crates/spikard-rb/src/di/builder.rs +100 -100
- data/vendor/crates/spikard-rb/src/di/mod.rs +375 -375
- data/vendor/crates/spikard-rb/src/handler.rs +618 -618
- data/vendor/crates/spikard-rb/src/integration/mod.rs +3 -3
- data/vendor/crates/spikard-rb/src/lib.rs +1806 -1810
- data/vendor/crates/spikard-rb/src/lifecycle.rs +275 -275
- data/vendor/crates/spikard-rb/src/metadata/mod.rs +5 -5
- data/vendor/crates/spikard-rb/src/metadata/route_extraction.rs +442 -447
- data/vendor/crates/spikard-rb/src/runtime/mod.rs +5 -5
- data/vendor/crates/spikard-rb/src/runtime/server_runner.rs +324 -324
- data/vendor/crates/spikard-rb/src/server.rs +305 -308
- data/vendor/crates/spikard-rb/src/sse.rs +231 -231
- data/vendor/crates/spikard-rb/src/testing/client.rs +538 -551
- data/vendor/crates/spikard-rb/src/testing/mod.rs +7 -7
- data/vendor/crates/spikard-rb/src/testing/sse.rs +143 -143
- data/vendor/crates/spikard-rb/src/testing/websocket.rs +608 -635
- data/vendor/crates/spikard-rb/src/websocket.rs +377 -374
- metadata +15 -1
|
@@ -1,901 +1,901 @@
|
|
|
1
|
-
use axum::{
|
|
2
|
-
BoxError,
|
|
3
|
-
body::Body,
|
|
4
|
-
http::{HeaderMap, HeaderName, HeaderValue, StatusCode},
|
|
5
|
-
response::Response as AxumResponse,
|
|
6
|
-
};
|
|
7
|
-
use bytes::Bytes;
|
|
8
|
-
use futures::{Stream, StreamExt};
|
|
9
|
-
use std::pin::Pin;
|
|
10
|
-
|
|
11
|
-
/// Unified response type that can represent either a ready response or a streaming body.
|
|
12
|
-
///
|
|
13
|
-
/// This enum allows handlers to return either:
|
|
14
|
-
/// - A complete response that's ready to send (`Response` variant)
|
|
15
|
-
/// - A streaming response with potentially unbounded data (`Stream` variant)
|
|
16
|
-
///
|
|
17
|
-
/// # Variants
|
|
18
|
-
///
|
|
19
|
-
/// * `Response` - A complete Axum response ready to send to the client. Use this for
|
|
20
|
-
/// responses where you have all the data ready (files, JSON bodies, HTML, etc.)
|
|
21
|
-
///
|
|
22
|
-
/// * `Stream` - A streaming response that produces data chunks over time. Use this for:
|
|
23
|
-
/// - Large files (avoid loading entire file in memory)
|
|
24
|
-
/// - Server-Sent Events (SSE)
|
|
25
|
-
/// - Long-polling responses
|
|
26
|
-
/// - Real-time data feeds
|
|
27
|
-
/// - Any unbounded or very large responses
|
|
28
|
-
///
|
|
29
|
-
/// # Examples
|
|
30
|
-
///
|
|
31
|
-
/// ```ignore
|
|
32
|
-
/// // Regular response
|
|
33
|
-
/// let response = AxumResponse::builder()
|
|
34
|
-
/// .status(StatusCode::OK)
|
|
35
|
-
/// .body(Body::from("Hello"))
|
|
36
|
-
/// .unwrap();
|
|
37
|
-
/// let handler_response = HandlerResponse::from(response);
|
|
38
|
-
///
|
|
39
|
-
/// // Streaming response
|
|
40
|
-
/// let stream = futures::stream::iter(vec![
|
|
41
|
-
/// Ok::<_, Box<dyn std::error::Error>>(Bytes::from("chunk1")),
|
|
42
|
-
/// Ok(Bytes::from("chunk2")),
|
|
43
|
-
/// ]);
|
|
44
|
-
/// let response = HandlerResponse::stream(stream)
|
|
45
|
-
/// .with_status(StatusCode::OK);
|
|
46
|
-
/// ```
|
|
47
|
-
pub enum HandlerResponse {
|
|
48
|
-
/// A complete response ready to send
|
|
49
|
-
Response(AxumResponse<Body>),
|
|
50
|
-
/// A streaming response with custom status and headers
|
|
51
|
-
Stream {
|
|
52
|
-
/// The byte stream that will be sent to the client
|
|
53
|
-
stream: Pin<Box<dyn Stream<Item = Result<Bytes, BoxError>> + Send + 'static>>,
|
|
54
|
-
/// HTTP status code for the response
|
|
55
|
-
status: StatusCode,
|
|
56
|
-
/// Response headers to send
|
|
57
|
-
headers: HeaderMap,
|
|
58
|
-
},
|
|
59
|
-
}
|
|
60
|
-
|
|
61
|
-
impl From<AxumResponse<Body>> for HandlerResponse {
|
|
62
|
-
fn from(response: AxumResponse<Body>) -> Self {
|
|
63
|
-
HandlerResponse::Response(response)
|
|
64
|
-
}
|
|
65
|
-
}
|
|
66
|
-
|
|
67
|
-
impl HandlerResponse {
|
|
68
|
-
/// Convert the handler response into an Axum response.
|
|
69
|
-
///
|
|
70
|
-
/// Consumes the `HandlerResponse` and produces an `AxumResponse<Body>` ready
|
|
71
|
-
/// to be sent to the client. For streaming responses, wraps the stream in an
|
|
72
|
-
/// Axum Body.
|
|
73
|
-
///
|
|
74
|
-
/// # Returns
|
|
75
|
-
/// An `AxumResponse<Body>` ready to be returned from an Axum handler
|
|
76
|
-
pub fn into_response(self) -> AxumResponse<Body> {
|
|
77
|
-
match self {
|
|
78
|
-
HandlerResponse::Response(response) => response,
|
|
79
|
-
HandlerResponse::Stream {
|
|
80
|
-
stream,
|
|
81
|
-
status,
|
|
82
|
-
mut headers,
|
|
83
|
-
} => {
|
|
84
|
-
let body = Body::from_stream(stream);
|
|
85
|
-
let mut response = AxumResponse::new(body);
|
|
86
|
-
*response.status_mut() = status;
|
|
87
|
-
response.headers_mut().extend(headers.drain());
|
|
88
|
-
response
|
|
89
|
-
}
|
|
90
|
-
}
|
|
91
|
-
}
|
|
92
|
-
|
|
93
|
-
/// Create a streaming response from any async stream of byte chunks.
|
|
94
|
-
///
|
|
95
|
-
/// Wraps an async stream of byte chunks into a `HandlerResponse::Stream`.
|
|
96
|
-
/// This is useful for large files, real-time data, or any unbounded response.
|
|
97
|
-
///
|
|
98
|
-
/// # Type Parameters
|
|
99
|
-
/// * `S` - The stream type implementing `Stream<Item = Result<Bytes, E>>`
|
|
100
|
-
/// * `E` - The error type that can be converted to `BoxError`
|
|
101
|
-
///
|
|
102
|
-
/// # Arguments
|
|
103
|
-
/// * `stream` - An async stream that yields byte chunks or errors
|
|
104
|
-
///
|
|
105
|
-
/// # Returns
|
|
106
|
-
/// A `HandlerResponse` with 200 OK status and empty headers (customize with
|
|
107
|
-
/// `with_status()` and `with_header()`)
|
|
108
|
-
///
|
|
109
|
-
/// # Example
|
|
110
|
-
///
|
|
111
|
-
/// ```ignore
|
|
112
|
-
/// use futures::stream;
|
|
113
|
-
/// use spikard_http::HandlerResponse;
|
|
114
|
-
/// use bytes::Bytes;
|
|
115
|
-
///
|
|
116
|
-
/// let stream = stream::iter(vec![
|
|
117
|
-
/// Ok::<_, Box<dyn std::error::Error>>(Bytes::from("Hello ")),
|
|
118
|
-
/// Ok(Bytes::from("World")),
|
|
119
|
-
/// ]);
|
|
120
|
-
/// let response = HandlerResponse::stream(stream)
|
|
121
|
-
/// .with_status(StatusCode::OK);
|
|
122
|
-
/// ```
|
|
123
|
-
pub fn stream<S, E>(stream: S) -> Self
|
|
124
|
-
where
|
|
125
|
-
S: Stream<Item = Result<Bytes, E>> + Send + 'static,
|
|
126
|
-
E: Into<BoxError>,
|
|
127
|
-
{
|
|
128
|
-
let mapped = stream.map(|chunk| chunk.map_err(Into::into));
|
|
129
|
-
HandlerResponse::Stream {
|
|
130
|
-
stream: Box::pin(mapped),
|
|
131
|
-
status: StatusCode::OK,
|
|
132
|
-
headers: HeaderMap::new(),
|
|
133
|
-
}
|
|
134
|
-
}
|
|
135
|
-
|
|
136
|
-
/// Override the HTTP status code for the streaming response.
|
|
137
|
-
///
|
|
138
|
-
/// Sets the HTTP status code to be used in the response. This only affects
|
|
139
|
-
/// `Stream` variants; regular responses already have their status set.
|
|
140
|
-
///
|
|
141
|
-
/// # Arguments
|
|
142
|
-
/// * `status` - The HTTP status code to use (e.g., `StatusCode::OK`)
|
|
143
|
-
///
|
|
144
|
-
/// # Returns
|
|
145
|
-
/// Self for method chaining
|
|
146
|
-
///
|
|
147
|
-
/// # Example
|
|
148
|
-
///
|
|
149
|
-
/// ```ignore
|
|
150
|
-
/// let response = HandlerResponse::stream(my_stream)
|
|
151
|
-
/// .with_status(StatusCode::PARTIAL_CONTENT);
|
|
152
|
-
/// ```
|
|
153
|
-
pub fn with_status(mut self, status: StatusCode) -> Self {
|
|
154
|
-
if let HandlerResponse::Stream { status: s, .. } = &mut self {
|
|
155
|
-
*s = status;
|
|
156
|
-
}
|
|
157
|
-
self
|
|
158
|
-
}
|
|
159
|
-
|
|
160
|
-
/// Insert or replace a header on the streaming response.
|
|
161
|
-
///
|
|
162
|
-
/// Adds an HTTP header to the response. This only affects `Stream` variants;
|
|
163
|
-
/// regular responses already have their headers set. If a header with the same
|
|
164
|
-
/// name already exists, it will be replaced.
|
|
165
|
-
///
|
|
166
|
-
/// # Arguments
|
|
167
|
-
/// * `name` - The header name (e.g., `HeaderName::from_static("content-type")`)
|
|
168
|
-
/// * `value` - The header value
|
|
169
|
-
///
|
|
170
|
-
/// # Returns
|
|
171
|
-
/// Self for method chaining
|
|
172
|
-
///
|
|
173
|
-
/// # Example
|
|
174
|
-
///
|
|
175
|
-
/// ```ignore
|
|
176
|
-
/// use axum::http::{HeaderName, HeaderValue};
|
|
177
|
-
///
|
|
178
|
-
/// let response = HandlerResponse::stream(my_stream)
|
|
179
|
-
/// .with_header(
|
|
180
|
-
/// HeaderName::from_static("content-type"),
|
|
181
|
-
/// HeaderValue::from_static("application/octet-stream")
|
|
182
|
-
/// );
|
|
183
|
-
/// ```
|
|
184
|
-
pub fn with_header(mut self, name: HeaderName, value: HeaderValue) -> Self {
|
|
185
|
-
if let HandlerResponse::Stream { headers, .. } = &mut self {
|
|
186
|
-
headers.insert(name, value);
|
|
187
|
-
}
|
|
188
|
-
self
|
|
189
|
-
}
|
|
190
|
-
}
|
|
191
|
-
|
|
192
|
-
#[cfg(test)]
|
|
193
|
-
mod tests {
|
|
194
|
-
use super::*;
|
|
195
|
-
use axum::http::header;
|
|
196
|
-
use http_body_util::BodyExt;
|
|
197
|
-
|
|
198
|
-
/// Test 1: Convert AxumResponse → HandlerResponse::Response
|
|
199
|
-
#[test]
|
|
200
|
-
fn test_from_axum_response() {
|
|
201
|
-
let axum_response = AxumResponse::new(Body::from("test body"));
|
|
202
|
-
let handler_response = HandlerResponse::from(axum_response);
|
|
203
|
-
|
|
204
|
-
match handler_response {
|
|
205
|
-
HandlerResponse::Response(_) => {}
|
|
206
|
-
HandlerResponse::Stream { .. } => panic!("Expected Response variant"),
|
|
207
|
-
}
|
|
208
|
-
}
|
|
209
|
-
|
|
210
|
-
/// Test 2: Create stream with chunks, verify wrapping
|
|
211
|
-
#[tokio::test]
|
|
212
|
-
async fn test_stream_creation_with_chunks() {
|
|
213
|
-
let chunks = vec![
|
|
214
|
-
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("chunk1")),
|
|
215
|
-
Ok(Bytes::from("chunk2")),
|
|
216
|
-
Ok(Bytes::from("chunk3")),
|
|
217
|
-
];
|
|
218
|
-
let stream = futures::stream::iter(chunks);
|
|
219
|
-
let handler_response = HandlerResponse::stream(stream);
|
|
220
|
-
|
|
221
|
-
match handler_response {
|
|
222
|
-
HandlerResponse::Stream { status, headers, .. } => {
|
|
223
|
-
assert_eq!(status, StatusCode::OK);
|
|
224
|
-
assert!(headers.is_empty());
|
|
225
|
-
}
|
|
226
|
-
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
227
|
-
}
|
|
228
|
-
}
|
|
229
|
-
|
|
230
|
-
/// Test 3: Stream with custom status code (206 Partial Content)
|
|
231
|
-
#[tokio::test]
|
|
232
|
-
async fn test_stream_with_custom_status() {
|
|
233
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
234
|
-
"partial",
|
|
235
|
-
))];
|
|
236
|
-
let stream = futures::stream::iter(chunks);
|
|
237
|
-
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::PARTIAL_CONTENT);
|
|
238
|
-
|
|
239
|
-
match handler_response {
|
|
240
|
-
HandlerResponse::Stream { status, .. } => {
|
|
241
|
-
assert_eq!(status, StatusCode::PARTIAL_CONTENT);
|
|
242
|
-
}
|
|
243
|
-
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
244
|
-
}
|
|
245
|
-
}
|
|
246
|
-
|
|
247
|
-
/// Test 4: Stream with headers via with_header()
|
|
248
|
-
#[tokio::test]
|
|
249
|
-
async fn test_stream_with_headers() {
|
|
250
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("test"))];
|
|
251
|
-
let stream = futures::stream::iter(chunks);
|
|
252
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
253
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/x-ndjson"))
|
|
254
|
-
.with_header(header::CACHE_CONTROL, HeaderValue::from_static("no-cache"));
|
|
255
|
-
|
|
256
|
-
match handler_response {
|
|
257
|
-
HandlerResponse::Stream { headers, .. } => {
|
|
258
|
-
assert_eq!(headers.get(header::CONTENT_TYPE).unwrap(), "application/x-ndjson");
|
|
259
|
-
assert_eq!(headers.get(header::CACHE_CONTROL).unwrap(), "no-cache");
|
|
260
|
-
}
|
|
261
|
-
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
262
|
-
}
|
|
263
|
-
}
|
|
264
|
-
|
|
265
|
-
/// Test 5: Stream body consumption - read all chunks from stream
|
|
266
|
-
#[tokio::test]
|
|
267
|
-
async fn test_stream_body_consumption() {
|
|
268
|
-
let chunks = vec![
|
|
269
|
-
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("hello ")),
|
|
270
|
-
Ok(Bytes::from("world")),
|
|
271
|
-
Ok(Bytes::from("!")),
|
|
272
|
-
];
|
|
273
|
-
let stream = futures::stream::iter(chunks);
|
|
274
|
-
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::OK);
|
|
275
|
-
|
|
276
|
-
let axum_response = handler_response.into_response();
|
|
277
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
278
|
-
let bytes = body.to_bytes();
|
|
279
|
-
|
|
280
|
-
assert_eq!(bytes, "hello world!");
|
|
281
|
-
}
|
|
282
|
-
|
|
283
|
-
/// Test 6: Into response for Response variant - passthrough conversion
|
|
284
|
-
#[tokio::test]
|
|
285
|
-
async fn test_into_response_for_response_variant() {
|
|
286
|
-
let original_body = "test response body";
|
|
287
|
-
let axum_response = AxumResponse::new(Body::from(original_body));
|
|
288
|
-
let handler_response = HandlerResponse::from(axum_response);
|
|
289
|
-
|
|
290
|
-
let result = handler_response.into_response();
|
|
291
|
-
let body = result.into_body().collect().await.unwrap();
|
|
292
|
-
let bytes = body.to_bytes();
|
|
293
|
-
|
|
294
|
-
assert_eq!(bytes, original_body);
|
|
295
|
-
}
|
|
296
|
-
|
|
297
|
-
/// Test 7: Method chaining - with_status() → with_header() → with_header()
|
|
298
|
-
#[tokio::test]
|
|
299
|
-
async fn test_method_chaining() {
|
|
300
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
301
|
-
"chained",
|
|
302
|
-
))];
|
|
303
|
-
let stream = futures::stream::iter(chunks);
|
|
304
|
-
|
|
305
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
306
|
-
.with_status(StatusCode::CREATED)
|
|
307
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("text/plain"))
|
|
308
|
-
.with_header(header::ETAG, HeaderValue::from_static("\"abc123\""));
|
|
309
|
-
|
|
310
|
-
match handler_response {
|
|
311
|
-
HandlerResponse::Stream { status, headers, .. } => {
|
|
312
|
-
assert_eq!(status, StatusCode::CREATED);
|
|
313
|
-
assert_eq!(headers.get(header::CONTENT_TYPE).unwrap(), "text/plain");
|
|
314
|
-
assert_eq!(headers.get(header::ETAG).unwrap(), "\"abc123\"");
|
|
315
|
-
}
|
|
316
|
-
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
317
|
-
}
|
|
318
|
-
}
|
|
319
|
-
|
|
320
|
-
/// Test 8: Empty stream - zero-byte stream handling
|
|
321
|
-
#[tokio::test]
|
|
322
|
-
async fn test_empty_stream() {
|
|
323
|
-
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![];
|
|
324
|
-
let stream = futures::stream::iter(chunks);
|
|
325
|
-
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::NO_CONTENT);
|
|
326
|
-
|
|
327
|
-
let axum_response = handler_response.into_response();
|
|
328
|
-
let status = axum_response.status();
|
|
329
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
330
|
-
let bytes = body.to_bytes();
|
|
331
|
-
|
|
332
|
-
assert!(bytes.is_empty());
|
|
333
|
-
assert_eq!(status, StatusCode::NO_CONTENT);
|
|
334
|
-
}
|
|
335
|
-
|
|
336
|
-
/// Test 9: Large stream - many chunks (100+ items)
|
|
337
|
-
#[tokio::test]
|
|
338
|
-
async fn test_large_stream() {
|
|
339
|
-
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> =
|
|
340
|
-
(0..150).map(|i| Ok(Bytes::from(format!("chunk{} ", i)))).collect();
|
|
341
|
-
|
|
342
|
-
let stream = futures::stream::iter(chunks);
|
|
343
|
-
let handler_response = HandlerResponse::stream(stream);
|
|
344
|
-
|
|
345
|
-
let axum_response = handler_response.into_response();
|
|
346
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
347
|
-
let bytes = body.to_bytes();
|
|
348
|
-
|
|
349
|
-
assert!(bytes.len() > 1000);
|
|
350
|
-
for i in 0..150 {
|
|
351
|
-
let expected = format!("chunk{} ", i);
|
|
352
|
-
assert!(std::str::from_utf8(&bytes).unwrap().contains(&expected));
|
|
353
|
-
}
|
|
354
|
-
}
|
|
355
|
-
|
|
356
|
-
/// Test 10: Error in stream - stream item returns Err, verify error propagation
|
|
357
|
-
#[tokio::test]
|
|
358
|
-
async fn test_stream_error_propagation() {
|
|
359
|
-
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![
|
|
360
|
-
Ok(Bytes::from("good1 ")),
|
|
361
|
-
Err("custom error".into()),
|
|
362
|
-
Ok(Bytes::from("good2")),
|
|
363
|
-
];
|
|
364
|
-
|
|
365
|
-
let stream = futures::stream::iter(chunks);
|
|
366
|
-
let handler_response = HandlerResponse::stream(stream);
|
|
367
|
-
|
|
368
|
-
let axum_response = handler_response.into_response();
|
|
369
|
-
let result = axum_response.into_body().collect().await;
|
|
370
|
-
|
|
371
|
-
assert!(result.is_err());
|
|
372
|
-
}
|
|
373
|
-
|
|
374
|
-
/// Test 11: Response variant ignores with_status()
|
|
375
|
-
#[test]
|
|
376
|
-
fn test_response_variant_ignores_with_status() {
|
|
377
|
-
let axum_response = AxumResponse::builder()
|
|
378
|
-
.status(StatusCode::OK)
|
|
379
|
-
.body(Body::from("test"))
|
|
380
|
-
.unwrap();
|
|
381
|
-
let handler_response = HandlerResponse::from(axum_response);
|
|
382
|
-
|
|
383
|
-
let result = handler_response.with_status(StatusCode::NOT_FOUND);
|
|
384
|
-
|
|
385
|
-
match result {
|
|
386
|
-
HandlerResponse::Response(resp) => {
|
|
387
|
-
assert_eq!(resp.status(), StatusCode::OK);
|
|
388
|
-
}
|
|
389
|
-
HandlerResponse::Stream { .. } => panic!("Expected Response variant"),
|
|
390
|
-
}
|
|
391
|
-
}
|
|
392
|
-
|
|
393
|
-
/// Test 12: Response variant ignores with_header()
|
|
394
|
-
#[test]
|
|
395
|
-
fn test_response_variant_ignores_with_header() {
|
|
396
|
-
let axum_response = AxumResponse::builder()
|
|
397
|
-
.status(StatusCode::OK)
|
|
398
|
-
.header(header::CONTENT_TYPE, "text/plain")
|
|
399
|
-
.body(Body::from("test"))
|
|
400
|
-
.unwrap();
|
|
401
|
-
let handler_response = HandlerResponse::from(axum_response);
|
|
402
|
-
|
|
403
|
-
let result = handler_response.with_header(header::CACHE_CONTROL, HeaderValue::from_static("max-age=3600"));
|
|
404
|
-
|
|
405
|
-
match result {
|
|
406
|
-
HandlerResponse::Response(resp) => {
|
|
407
|
-
assert!(resp.headers().get(header::CACHE_CONTROL).is_none());
|
|
408
|
-
}
|
|
409
|
-
HandlerResponse::Stream { .. } => panic!("Expected Response variant"),
|
|
410
|
-
}
|
|
411
|
-
}
|
|
412
|
-
|
|
413
|
-
/// Test 13: Stream into_response applies status and headers
|
|
414
|
-
#[tokio::test]
|
|
415
|
-
async fn test_stream_into_response_applies_status_and_headers() {
|
|
416
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
417
|
-
"stream data",
|
|
418
|
-
))];
|
|
419
|
-
let stream = futures::stream::iter(chunks);
|
|
420
|
-
|
|
421
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
422
|
-
.with_status(StatusCode::PARTIAL_CONTENT)
|
|
423
|
-
.with_header(header::CONTENT_RANGE, HeaderValue::from_static("bytes 0-10/100"));
|
|
424
|
-
|
|
425
|
-
let axum_response = handler_response.into_response();
|
|
426
|
-
|
|
427
|
-
assert_eq!(axum_response.status(), StatusCode::PARTIAL_CONTENT);
|
|
428
|
-
assert_eq!(
|
|
429
|
-
axum_response.headers().get(header::CONTENT_RANGE).unwrap(),
|
|
430
|
-
"bytes 0-10/100"
|
|
431
|
-
);
|
|
432
|
-
|
|
433
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
434
|
-
assert_eq!(body.to_bytes(), "stream data");
|
|
435
|
-
}
|
|
436
|
-
|
|
437
|
-
/// Test 14: Multiple header replacements via with_header()
|
|
438
|
-
#[tokio::test]
|
|
439
|
-
async fn test_multiple_header_replacements() {
|
|
440
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("data"))];
|
|
441
|
-
let stream = futures::stream::iter(chunks);
|
|
442
|
-
|
|
443
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
444
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/json"))
|
|
445
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/x-ndjson"));
|
|
446
|
-
|
|
447
|
-
match handler_response {
|
|
448
|
-
HandlerResponse::Stream { headers, .. } => {
|
|
449
|
-
assert_eq!(headers.get(header::CONTENT_TYPE).unwrap(), "application/x-ndjson");
|
|
450
|
-
}
|
|
451
|
-
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
452
|
-
}
|
|
453
|
-
}
|
|
454
|
-
|
|
455
|
-
/// Test 15: Stream with various status codes
|
|
456
|
-
#[tokio::test]
|
|
457
|
-
async fn test_stream_with_various_status_codes() {
|
|
458
|
-
let status_codes = vec![
|
|
459
|
-
StatusCode::OK,
|
|
460
|
-
StatusCode::CREATED,
|
|
461
|
-
StatusCode::ACCEPTED,
|
|
462
|
-
StatusCode::PARTIAL_CONTENT,
|
|
463
|
-
StatusCode::MULTI_STATUS,
|
|
464
|
-
];
|
|
465
|
-
|
|
466
|
-
for status in status_codes {
|
|
467
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("test"))];
|
|
468
|
-
let stream = futures::stream::iter(chunks);
|
|
469
|
-
let handler_response = HandlerResponse::stream(stream).with_status(status);
|
|
470
|
-
|
|
471
|
-
match handler_response {
|
|
472
|
-
HandlerResponse::Stream { status: s, .. } => {
|
|
473
|
-
assert_eq!(s, status);
|
|
474
|
-
}
|
|
475
|
-
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
476
|
-
}
|
|
477
|
-
}
|
|
478
|
-
}
|
|
479
|
-
|
|
480
|
-
/// Test 16: Stream with JSON lines content (fixture-based)
|
|
481
|
-
#[tokio::test]
|
|
482
|
-
async fn test_stream_with_json_lines_content() {
|
|
483
|
-
let chunks = vec![
|
|
484
|
-
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(r#"{"index":0,"payload":"alpha"}"#)),
|
|
485
|
-
Ok(Bytes::from("\n")),
|
|
486
|
-
Ok(Bytes::from(r#"{"index":1,"payload":"beta"}"#)),
|
|
487
|
-
Ok(Bytes::from("\n")),
|
|
488
|
-
Ok(Bytes::from(r#"{"index":2,"payload":"gamma"}"#)),
|
|
489
|
-
Ok(Bytes::from("\n")),
|
|
490
|
-
];
|
|
491
|
-
|
|
492
|
-
let stream = futures::stream::iter(chunks);
|
|
493
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
494
|
-
.with_status(StatusCode::OK)
|
|
495
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/x-ndjson"));
|
|
496
|
-
|
|
497
|
-
let axum_response = handler_response.into_response();
|
|
498
|
-
let status = axum_response.status();
|
|
499
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
500
|
-
let bytes = body.to_bytes();
|
|
501
|
-
let body_str = std::str::from_utf8(&bytes).unwrap();
|
|
502
|
-
|
|
503
|
-
assert_eq!(status, StatusCode::OK);
|
|
504
|
-
assert!(body_str.contains("alpha"));
|
|
505
|
-
assert!(body_str.contains("beta"));
|
|
506
|
-
assert!(body_str.contains("gamma"));
|
|
507
|
-
}
|
|
508
|
-
|
|
509
|
-
/// Test 17: Round-trip Response → HandlerResponse → Response
|
|
510
|
-
#[tokio::test]
|
|
511
|
-
async fn test_response_roundtrip() {
|
|
512
|
-
let original = AxumResponse::builder()
|
|
513
|
-
.status(StatusCode::OK)
|
|
514
|
-
.header(header::CONTENT_TYPE, "text/plain")
|
|
515
|
-
.body(Body::from("roundtrip test"))
|
|
516
|
-
.unwrap();
|
|
517
|
-
|
|
518
|
-
let handler_response = HandlerResponse::from(original);
|
|
519
|
-
let result = handler_response.into_response();
|
|
520
|
-
|
|
521
|
-
assert_eq!(result.status(), StatusCode::OK);
|
|
522
|
-
assert_eq!(result.headers().get(header::CONTENT_TYPE).unwrap(), "text/plain");
|
|
523
|
-
|
|
524
|
-
let body = result.into_body().collect().await.unwrap();
|
|
525
|
-
assert_eq!(body.to_bytes(), "roundtrip test");
|
|
526
|
-
}
|
|
527
|
-
|
|
528
|
-
/// Test 18: Single chunk stream (minimal data)
|
|
529
|
-
#[tokio::test]
|
|
530
|
-
async fn test_single_chunk_stream() {
|
|
531
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("only"))];
|
|
532
|
-
let stream = futures::stream::iter(chunks);
|
|
533
|
-
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::OK);
|
|
534
|
-
|
|
535
|
-
let axum_response = handler_response.into_response();
|
|
536
|
-
let status = axum_response.status();
|
|
537
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
538
|
-
let bytes = body.to_bytes();
|
|
539
|
-
|
|
540
|
-
assert_eq!(bytes, "only");
|
|
541
|
-
assert_eq!(status, StatusCode::OK);
|
|
542
|
-
}
|
|
543
|
-
|
|
544
|
-
/// Test 19: Stream with 1000+ chunks (performance edge case)
|
|
545
|
-
#[tokio::test]
|
|
546
|
-
async fn test_very_large_stream_many_chunks() {
|
|
547
|
-
let chunk_count = 1500;
|
|
548
|
-
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> =
|
|
549
|
-
(0..chunk_count).map(|_| Ok(Bytes::from(format!("x")))).collect();
|
|
550
|
-
|
|
551
|
-
let stream = futures::stream::iter(chunks);
|
|
552
|
-
let handler_response = HandlerResponse::stream(stream);
|
|
553
|
-
|
|
554
|
-
let axum_response = handler_response.into_response();
|
|
555
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
556
|
-
let bytes = body.to_bytes();
|
|
557
|
-
|
|
558
|
-
assert_eq!(bytes.len(), chunk_count);
|
|
559
|
-
}
|
|
560
|
-
|
|
561
|
-
/// Test 20: Stream with varying chunk sizes (1 byte to 1MB)
|
|
562
|
-
#[tokio::test]
|
|
563
|
-
async fn test_stream_with_varying_chunk_sizes() {
|
|
564
|
-
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![
|
|
565
|
-
Ok(Bytes::from("x")),
|
|
566
|
-
Ok(Bytes::from("xx".repeat(100))),
|
|
567
|
-
Ok(Bytes::from("x".repeat(10_000))),
|
|
568
|
-
Ok(Bytes::from("x".repeat(100_000))),
|
|
569
|
-
];
|
|
570
|
-
|
|
571
|
-
let stream = futures::stream::iter(chunks);
|
|
572
|
-
let handler_response = HandlerResponse::stream(stream);
|
|
573
|
-
|
|
574
|
-
let axum_response = handler_response.into_response();
|
|
575
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
576
|
-
let bytes = body.to_bytes();
|
|
577
|
-
|
|
578
|
-
assert_eq!(bytes.len(), 110_201);
|
|
579
|
-
}
|
|
580
|
-
|
|
581
|
-
/// Test 21: Stream with error in the middle (chunk 500/1000)
|
|
582
|
-
#[tokio::test]
|
|
583
|
-
async fn test_stream_error_in_middle() {
|
|
584
|
-
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = (0..1000)
|
|
585
|
-
.map(|i| {
|
|
586
|
-
if i == 500 {
|
|
587
|
-
Err("midstream error".into())
|
|
588
|
-
} else {
|
|
589
|
-
Ok(Bytes::from("chunk"))
|
|
590
|
-
}
|
|
591
|
-
})
|
|
592
|
-
.collect();
|
|
593
|
-
|
|
594
|
-
let stream = futures::stream::iter(chunks);
|
|
595
|
-
let handler_response = HandlerResponse::stream(stream);
|
|
596
|
-
|
|
597
|
-
let axum_response = handler_response.into_response();
|
|
598
|
-
let result = axum_response.into_body().collect().await;
|
|
599
|
-
|
|
600
|
-
assert!(result.is_err());
|
|
601
|
-
}
|
|
602
|
-
|
|
603
|
-
/// Test 22: Stream with SSE-like headers
|
|
604
|
-
#[tokio::test]
|
|
605
|
-
async fn test_stream_with_sse_headers() {
|
|
606
|
-
let chunks = vec![
|
|
607
|
-
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("event: message\n")),
|
|
608
|
-
Ok(Bytes::from("data: {\"msg\": \"hello\"}\n\n")),
|
|
609
|
-
];
|
|
610
|
-
let stream = futures::stream::iter(chunks);
|
|
611
|
-
|
|
612
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
613
|
-
.with_status(StatusCode::OK)
|
|
614
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("text/event-stream"))
|
|
615
|
-
.with_header(header::CACHE_CONTROL, HeaderValue::from_static("no-cache"))
|
|
616
|
-
.with_header(header::CONNECTION, HeaderValue::from_static("keep-alive"));
|
|
617
|
-
|
|
618
|
-
let axum_response = handler_response.into_response();
|
|
619
|
-
|
|
620
|
-
assert_eq!(axum_response.status(), StatusCode::OK);
|
|
621
|
-
assert_eq!(
|
|
622
|
-
axum_response.headers().get(header::CONTENT_TYPE).unwrap(),
|
|
623
|
-
"text/event-stream"
|
|
624
|
-
);
|
|
625
|
-
assert_eq!(axum_response.headers().get(header::CACHE_CONTROL).unwrap(), "no-cache");
|
|
626
|
-
|
|
627
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
628
|
-
let body_bytes = body.to_bytes();
|
|
629
|
-
let body_str = std::str::from_utf8(&body_bytes).unwrap();
|
|
630
|
-
assert!(body_str.contains("event: message"));
|
|
631
|
-
}
|
|
632
|
-
|
|
633
|
-
/// Test 23: Stream with WebSocket-like upgrade headers (200 OK with Upgrade)
|
|
634
|
-
#[tokio::test]
|
|
635
|
-
async fn test_stream_with_websocket_headers() {
|
|
636
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
637
|
-
"ws-frame-data",
|
|
638
|
-
))];
|
|
639
|
-
let stream = futures::stream::iter(chunks);
|
|
640
|
-
|
|
641
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
642
|
-
.with_status(StatusCode::OK)
|
|
643
|
-
.with_header(header::UPGRADE, HeaderValue::from_static("websocket"))
|
|
644
|
-
.with_header(
|
|
645
|
-
HeaderName::from_static("sec-websocket-accept"),
|
|
646
|
-
HeaderValue::from_static("s3pPLMBiTxaQ9kYGzzhZRbK+xOo="),
|
|
647
|
-
);
|
|
648
|
-
|
|
649
|
-
let axum_response = handler_response.into_response();
|
|
650
|
-
|
|
651
|
-
assert_eq!(axum_response.status(), StatusCode::OK);
|
|
652
|
-
assert_eq!(axum_response.headers().get(header::UPGRADE).unwrap(), "websocket");
|
|
653
|
-
|
|
654
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
655
|
-
assert_eq!(body.to_bytes(), "ws-frame-data");
|
|
656
|
-
}
|
|
657
|
-
|
|
658
|
-
/// Test 24: Stream status transitions (from 200 OK to 206 Partial Content)
|
|
659
|
-
#[tokio::test]
|
|
660
|
-
async fn test_stream_status_transition() {
|
|
661
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("data"))];
|
|
662
|
-
let stream = futures::stream::iter(chunks);
|
|
663
|
-
|
|
664
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
665
|
-
.with_status(StatusCode::OK)
|
|
666
|
-
.with_status(StatusCode::PARTIAL_CONTENT);
|
|
667
|
-
|
|
668
|
-
match handler_response {
|
|
669
|
-
HandlerResponse::Stream { status, .. } => {
|
|
670
|
-
assert_eq!(status, StatusCode::PARTIAL_CONTENT);
|
|
671
|
-
}
|
|
672
|
-
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
673
|
-
}
|
|
674
|
-
}
|
|
675
|
-
|
|
676
|
-
/// Test 25: Stream with chunked transfer encoding simulation
|
|
677
|
-
#[tokio::test]
|
|
678
|
-
async fn test_stream_chunked_encoding_simulation() {
|
|
679
|
-
let chunks = vec![
|
|
680
|
-
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("5\r\nhello\r\n")),
|
|
681
|
-
Ok(Bytes::from("5\r\nworld\r\n")),
|
|
682
|
-
Ok(Bytes::from("0\r\n\r\n")),
|
|
683
|
-
];
|
|
684
|
-
|
|
685
|
-
let stream = futures::stream::iter(chunks);
|
|
686
|
-
let handler_response =
|
|
687
|
-
HandlerResponse::stream(stream).with_header(header::TRANSFER_ENCODING, HeaderValue::from_static("chunked"));
|
|
688
|
-
|
|
689
|
-
let axum_response = handler_response.into_response();
|
|
690
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
691
|
-
let body_bytes = body.to_bytes();
|
|
692
|
-
|
|
693
|
-
assert!(std::str::from_utf8(&body_bytes).unwrap().contains("hello"));
|
|
694
|
-
}
|
|
695
|
-
|
|
696
|
-
/// Test 26: Stream with binary data (non-UTF8)
|
|
697
|
-
#[tokio::test]
|
|
698
|
-
async fn test_stream_with_binary_data() {
|
|
699
|
-
let chunks = vec![
|
|
700
|
-
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(vec![0xFF, 0xD8, 0xFF])),
|
|
701
|
-
Ok(Bytes::from(vec![0xE0, 0x00, 0x10])),
|
|
702
|
-
Ok(Bytes::from(vec![0x4A, 0x46, 0x49])),
|
|
703
|
-
];
|
|
704
|
-
|
|
705
|
-
let stream = futures::stream::iter(chunks);
|
|
706
|
-
let handler_response = HandlerResponse::stream(stream).with_header(
|
|
707
|
-
header::CONTENT_TYPE,
|
|
708
|
-
HeaderValue::from_static("application/octet-stream"),
|
|
709
|
-
);
|
|
710
|
-
|
|
711
|
-
let axum_response = handler_response.into_response();
|
|
712
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
713
|
-
let bytes = body.to_bytes();
|
|
714
|
-
|
|
715
|
-
assert_eq!(bytes[0], 0xFF);
|
|
716
|
-
assert_eq!(bytes[1], 0xD8);
|
|
717
|
-
assert_eq!(bytes[2], 0xFF);
|
|
718
|
-
assert_eq!(bytes[3], 0xE0);
|
|
719
|
-
assert_eq!(bytes[4], 0x00);
|
|
720
|
-
}
|
|
721
|
-
|
|
722
|
-
/// Test 27: Stream with null bytes in payload
|
|
723
|
-
#[tokio::test]
|
|
724
|
-
async fn test_stream_with_null_bytes() {
|
|
725
|
-
let chunks = vec![
|
|
726
|
-
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(vec![0x00, 0x01, 0x02])),
|
|
727
|
-
Ok(Bytes::from(vec![0x00, 0x00, 0x00])),
|
|
728
|
-
Ok(Bytes::from(vec![0xFF, 0xFE, 0xFD])),
|
|
729
|
-
];
|
|
730
|
-
|
|
731
|
-
let stream = futures::stream::iter(chunks);
|
|
732
|
-
let handler_response = HandlerResponse::stream(stream);
|
|
733
|
-
|
|
734
|
-
let axum_response = handler_response.into_response();
|
|
735
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
736
|
-
let bytes = body.to_bytes();
|
|
737
|
-
|
|
738
|
-
assert_eq!(bytes.len(), 9);
|
|
739
|
-
assert_eq!(bytes[0], 0x00);
|
|
740
|
-
assert_eq!(bytes[4], 0x00);
|
|
741
|
-
assert_eq!(bytes[8], 0xFD);
|
|
742
|
-
}
|
|
743
|
-
|
|
744
|
-
/// Test 28: Stream with maximum header count
|
|
745
|
-
#[tokio::test]
|
|
746
|
-
async fn test_stream_with_many_headers() {
|
|
747
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("data"))];
|
|
748
|
-
let stream = futures::stream::iter(chunks);
|
|
749
|
-
|
|
750
|
-
let mut handler_response = HandlerResponse::stream(stream);
|
|
751
|
-
|
|
752
|
-
for i in 0..50 {
|
|
753
|
-
let header_name = format!("x-custom-{}", i);
|
|
754
|
-
handler_response = handler_response.with_header(
|
|
755
|
-
HeaderName::from_bytes(header_name.as_bytes()).unwrap(),
|
|
756
|
-
HeaderValue::from_static("value"),
|
|
757
|
-
);
|
|
758
|
-
}
|
|
759
|
-
|
|
760
|
-
let axum_response = handler_response.into_response();
|
|
761
|
-
assert_eq!(axum_response.status(), StatusCode::OK);
|
|
762
|
-
assert_eq!(axum_response.headers().len(), 50);
|
|
763
|
-
}
|
|
764
|
-
|
|
765
|
-
/// Test 29: Empty stream with 204 No Content status
|
|
766
|
-
#[tokio::test]
|
|
767
|
-
async fn test_empty_stream_with_204_no_content() {
|
|
768
|
-
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![];
|
|
769
|
-
let stream = futures::stream::iter(chunks);
|
|
770
|
-
|
|
771
|
-
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::NO_CONTENT);
|
|
772
|
-
|
|
773
|
-
let axum_response = handler_response.into_response();
|
|
774
|
-
|
|
775
|
-
assert_eq!(axum_response.status(), StatusCode::NO_CONTENT);
|
|
776
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
777
|
-
assert!(body.to_bytes().is_empty());
|
|
778
|
-
}
|
|
779
|
-
|
|
780
|
-
/// Test 30: Stream with repeated header replacement
|
|
781
|
-
#[tokio::test]
|
|
782
|
-
async fn test_stream_repeated_header_updates() {
|
|
783
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("test"))];
|
|
784
|
-
let stream = futures::stream::iter(chunks);
|
|
785
|
-
|
|
786
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
787
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("text/plain"))
|
|
788
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/json"))
|
|
789
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/xml"));
|
|
790
|
-
|
|
791
|
-
match handler_response {
|
|
792
|
-
HandlerResponse::Stream { headers, .. } => {
|
|
793
|
-
assert_eq!(headers.get(header::CONTENT_TYPE).unwrap(), "application/xml");
|
|
794
|
-
}
|
|
795
|
-
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
796
|
-
}
|
|
797
|
-
}
|
|
798
|
-
|
|
799
|
-
/// Test 31: Stream with extremely long chunk
|
|
800
|
-
#[tokio::test]
|
|
801
|
-
async fn test_stream_with_extremely_long_chunk() {
|
|
802
|
-
let large_chunk = "x".repeat(10_000_000);
|
|
803
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
804
|
-
large_chunk,
|
|
805
|
-
))];
|
|
806
|
-
let stream = futures::stream::iter(chunks);
|
|
807
|
-
|
|
808
|
-
let handler_response = HandlerResponse::stream(stream);
|
|
809
|
-
|
|
810
|
-
let axum_response = handler_response.into_response();
|
|
811
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
812
|
-
let bytes = body.to_bytes();
|
|
813
|
-
|
|
814
|
-
assert_eq!(bytes.len(), 10_000_000);
|
|
815
|
-
}
|
|
816
|
-
|
|
817
|
-
/// Test 32: Stream with zero-length chunks mixed in
|
|
818
|
-
#[tokio::test]
|
|
819
|
-
async fn test_stream_with_zero_length_chunks() {
|
|
820
|
-
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![
|
|
821
|
-
Ok(Bytes::from("hello")),
|
|
822
|
-
Ok(Bytes::new()),
|
|
823
|
-
Ok(Bytes::from("world")),
|
|
824
|
-
Ok(Bytes::new()),
|
|
825
|
-
Ok(Bytes::from("!")),
|
|
826
|
-
];
|
|
827
|
-
|
|
828
|
-
let stream = futures::stream::iter(chunks);
|
|
829
|
-
let handler_response = HandlerResponse::stream(stream);
|
|
830
|
-
|
|
831
|
-
let axum_response = handler_response.into_response();
|
|
832
|
-
let body = axum_response.into_body().collect().await.unwrap();
|
|
833
|
-
let bytes = body.to_bytes();
|
|
834
|
-
|
|
835
|
-
assert_eq!(bytes, "helloworld!");
|
|
836
|
-
}
|
|
837
|
-
|
|
838
|
-
/// Test 33: Handler response variant preserves custom status on failure
|
|
839
|
-
#[test]
|
|
840
|
-
fn test_response_variant_preserves_original_status() {
|
|
841
|
-
let axum_response = AxumResponse::builder()
|
|
842
|
-
.status(StatusCode::BAD_REQUEST)
|
|
843
|
-
.body(Body::from("error"))
|
|
844
|
-
.unwrap();
|
|
845
|
-
|
|
846
|
-
let handler_response = HandlerResponse::from(axum_response);
|
|
847
|
-
|
|
848
|
-
let result = handler_response
|
|
849
|
-
.with_status(StatusCode::OK)
|
|
850
|
-
.with_status(StatusCode::INTERNAL_SERVER_ERROR);
|
|
851
|
-
|
|
852
|
-
match result {
|
|
853
|
-
HandlerResponse::Response(resp) => {
|
|
854
|
-
assert_eq!(resp.status(), StatusCode::BAD_REQUEST);
|
|
855
|
-
}
|
|
856
|
-
HandlerResponse::Stream { .. } => panic!("Expected Response variant"),
|
|
857
|
-
}
|
|
858
|
-
}
|
|
859
|
-
|
|
860
|
-
/// Test 34: Stream response conversion preserves header ordering
|
|
861
|
-
#[tokio::test]
|
|
862
|
-
async fn test_stream_into_response_preserves_headers() {
|
|
863
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("data"))];
|
|
864
|
-
let stream = futures::stream::iter(chunks);
|
|
865
|
-
|
|
866
|
-
let handler_response = HandlerResponse::stream(stream)
|
|
867
|
-
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/json"))
|
|
868
|
-
.with_header(header::CACHE_CONTROL, HeaderValue::from_static("max-age=3600"))
|
|
869
|
-
.with_header(header::ETAG, HeaderValue::from_static("\"abc123\""));
|
|
870
|
-
|
|
871
|
-
let axum_response = handler_response.into_response();
|
|
872
|
-
|
|
873
|
-
assert!(axum_response.headers().get(header::CONTENT_TYPE).is_some());
|
|
874
|
-
assert!(axum_response.headers().get(header::CACHE_CONTROL).is_some());
|
|
875
|
-
assert!(axum_response.headers().get(header::ETAG).is_some());
|
|
876
|
-
assert_eq!(axum_response.headers().len(), 3);
|
|
877
|
-
}
|
|
878
|
-
|
|
879
|
-
/// Test 35: Stream with 5xx status codes
|
|
880
|
-
#[tokio::test]
|
|
881
|
-
async fn test_stream_with_error_status_codes() {
|
|
882
|
-
let error_statuses = vec![
|
|
883
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
884
|
-
StatusCode::SERVICE_UNAVAILABLE,
|
|
885
|
-
StatusCode::GATEWAY_TIMEOUT,
|
|
886
|
-
];
|
|
887
|
-
|
|
888
|
-
for status in error_statuses {
|
|
889
|
-
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("error"))];
|
|
890
|
-
let stream = futures::stream::iter(chunks);
|
|
891
|
-
let handler_response = HandlerResponse::stream(stream).with_status(status);
|
|
892
|
-
|
|
893
|
-
match handler_response {
|
|
894
|
-
HandlerResponse::Stream { status: s, .. } => {
|
|
895
|
-
assert_eq!(s, status);
|
|
896
|
-
}
|
|
897
|
-
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
898
|
-
}
|
|
899
|
-
}
|
|
900
|
-
}
|
|
901
|
-
}
|
|
1
|
+
use axum::{
|
|
2
|
+
BoxError,
|
|
3
|
+
body::Body,
|
|
4
|
+
http::{HeaderMap, HeaderName, HeaderValue, StatusCode},
|
|
5
|
+
response::Response as AxumResponse,
|
|
6
|
+
};
|
|
7
|
+
use bytes::Bytes;
|
|
8
|
+
use futures::{Stream, StreamExt};
|
|
9
|
+
use std::pin::Pin;
|
|
10
|
+
|
|
11
|
+
/// Unified response type that can represent either a ready response or a streaming body.
|
|
12
|
+
///
|
|
13
|
+
/// This enum allows handlers to return either:
|
|
14
|
+
/// - A complete response that's ready to send (`Response` variant)
|
|
15
|
+
/// - A streaming response with potentially unbounded data (`Stream` variant)
|
|
16
|
+
///
|
|
17
|
+
/// # Variants
|
|
18
|
+
///
|
|
19
|
+
/// * `Response` - A complete Axum response ready to send to the client. Use this for
|
|
20
|
+
/// responses where you have all the data ready (files, JSON bodies, HTML, etc.)
|
|
21
|
+
///
|
|
22
|
+
/// * `Stream` - A streaming response that produces data chunks over time. Use this for:
|
|
23
|
+
/// - Large files (avoid loading entire file in memory)
|
|
24
|
+
/// - Server-Sent Events (SSE)
|
|
25
|
+
/// - Long-polling responses
|
|
26
|
+
/// - Real-time data feeds
|
|
27
|
+
/// - Any unbounded or very large responses
|
|
28
|
+
///
|
|
29
|
+
/// # Examples
|
|
30
|
+
///
|
|
31
|
+
/// ```ignore
|
|
32
|
+
/// // Regular response
|
|
33
|
+
/// let response = AxumResponse::builder()
|
|
34
|
+
/// .status(StatusCode::OK)
|
|
35
|
+
/// .body(Body::from("Hello"))
|
|
36
|
+
/// .unwrap();
|
|
37
|
+
/// let handler_response = HandlerResponse::from(response);
|
|
38
|
+
///
|
|
39
|
+
/// // Streaming response
|
|
40
|
+
/// let stream = futures::stream::iter(vec![
|
|
41
|
+
/// Ok::<_, Box<dyn std::error::Error>>(Bytes::from("chunk1")),
|
|
42
|
+
/// Ok(Bytes::from("chunk2")),
|
|
43
|
+
/// ]);
|
|
44
|
+
/// let response = HandlerResponse::stream(stream)
|
|
45
|
+
/// .with_status(StatusCode::OK);
|
|
46
|
+
/// ```
|
|
47
|
+
pub enum HandlerResponse {
|
|
48
|
+
/// A complete response ready to send
|
|
49
|
+
Response(AxumResponse<Body>),
|
|
50
|
+
/// A streaming response with custom status and headers
|
|
51
|
+
Stream {
|
|
52
|
+
/// The byte stream that will be sent to the client
|
|
53
|
+
stream: Pin<Box<dyn Stream<Item = Result<Bytes, BoxError>> + Send + 'static>>,
|
|
54
|
+
/// HTTP status code for the response
|
|
55
|
+
status: StatusCode,
|
|
56
|
+
/// Response headers to send
|
|
57
|
+
headers: HeaderMap,
|
|
58
|
+
},
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
impl From<AxumResponse<Body>> for HandlerResponse {
|
|
62
|
+
fn from(response: AxumResponse<Body>) -> Self {
|
|
63
|
+
HandlerResponse::Response(response)
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
|
|
67
|
+
impl HandlerResponse {
|
|
68
|
+
/// Convert the handler response into an Axum response.
|
|
69
|
+
///
|
|
70
|
+
/// Consumes the `HandlerResponse` and produces an `AxumResponse<Body>` ready
|
|
71
|
+
/// to be sent to the client. For streaming responses, wraps the stream in an
|
|
72
|
+
/// Axum Body.
|
|
73
|
+
///
|
|
74
|
+
/// # Returns
|
|
75
|
+
/// An `AxumResponse<Body>` ready to be returned from an Axum handler
|
|
76
|
+
pub fn into_response(self) -> AxumResponse<Body> {
|
|
77
|
+
match self {
|
|
78
|
+
HandlerResponse::Response(response) => response,
|
|
79
|
+
HandlerResponse::Stream {
|
|
80
|
+
stream,
|
|
81
|
+
status,
|
|
82
|
+
mut headers,
|
|
83
|
+
} => {
|
|
84
|
+
let body = Body::from_stream(stream);
|
|
85
|
+
let mut response = AxumResponse::new(body);
|
|
86
|
+
*response.status_mut() = status;
|
|
87
|
+
response.headers_mut().extend(headers.drain());
|
|
88
|
+
response
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
|
|
93
|
+
/// Create a streaming response from any async stream of byte chunks.
|
|
94
|
+
///
|
|
95
|
+
/// Wraps an async stream of byte chunks into a `HandlerResponse::Stream`.
|
|
96
|
+
/// This is useful for large files, real-time data, or any unbounded response.
|
|
97
|
+
///
|
|
98
|
+
/// # Type Parameters
|
|
99
|
+
/// * `S` - The stream type implementing `Stream<Item = Result<Bytes, E>>`
|
|
100
|
+
/// * `E` - The error type that can be converted to `BoxError`
|
|
101
|
+
///
|
|
102
|
+
/// # Arguments
|
|
103
|
+
/// * `stream` - An async stream that yields byte chunks or errors
|
|
104
|
+
///
|
|
105
|
+
/// # Returns
|
|
106
|
+
/// A `HandlerResponse` with 200 OK status and empty headers (customize with
|
|
107
|
+
/// `with_status()` and `with_header()`)
|
|
108
|
+
///
|
|
109
|
+
/// # Example
|
|
110
|
+
///
|
|
111
|
+
/// ```ignore
|
|
112
|
+
/// use futures::stream;
|
|
113
|
+
/// use spikard_http::HandlerResponse;
|
|
114
|
+
/// use bytes::Bytes;
|
|
115
|
+
///
|
|
116
|
+
/// let stream = stream::iter(vec![
|
|
117
|
+
/// Ok::<_, Box<dyn std::error::Error>>(Bytes::from("Hello ")),
|
|
118
|
+
/// Ok(Bytes::from("World")),
|
|
119
|
+
/// ]);
|
|
120
|
+
/// let response = HandlerResponse::stream(stream)
|
|
121
|
+
/// .with_status(StatusCode::OK);
|
|
122
|
+
/// ```
|
|
123
|
+
pub fn stream<S, E>(stream: S) -> Self
|
|
124
|
+
where
|
|
125
|
+
S: Stream<Item = Result<Bytes, E>> + Send + 'static,
|
|
126
|
+
E: Into<BoxError>,
|
|
127
|
+
{
|
|
128
|
+
let mapped = stream.map(|chunk| chunk.map_err(Into::into));
|
|
129
|
+
HandlerResponse::Stream {
|
|
130
|
+
stream: Box::pin(mapped),
|
|
131
|
+
status: StatusCode::OK,
|
|
132
|
+
headers: HeaderMap::new(),
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
|
|
136
|
+
/// Override the HTTP status code for the streaming response.
|
|
137
|
+
///
|
|
138
|
+
/// Sets the HTTP status code to be used in the response. This only affects
|
|
139
|
+
/// `Stream` variants; regular responses already have their status set.
|
|
140
|
+
///
|
|
141
|
+
/// # Arguments
|
|
142
|
+
/// * `status` - The HTTP status code to use (e.g., `StatusCode::OK`)
|
|
143
|
+
///
|
|
144
|
+
/// # Returns
|
|
145
|
+
/// Self for method chaining
|
|
146
|
+
///
|
|
147
|
+
/// # Example
|
|
148
|
+
///
|
|
149
|
+
/// ```ignore
|
|
150
|
+
/// let response = HandlerResponse::stream(my_stream)
|
|
151
|
+
/// .with_status(StatusCode::PARTIAL_CONTENT);
|
|
152
|
+
/// ```
|
|
153
|
+
pub fn with_status(mut self, status: StatusCode) -> Self {
|
|
154
|
+
if let HandlerResponse::Stream { status: s, .. } = &mut self {
|
|
155
|
+
*s = status;
|
|
156
|
+
}
|
|
157
|
+
self
|
|
158
|
+
}
|
|
159
|
+
|
|
160
|
+
/// Insert or replace a header on the streaming response.
|
|
161
|
+
///
|
|
162
|
+
/// Adds an HTTP header to the response. This only affects `Stream` variants;
|
|
163
|
+
/// regular responses already have their headers set. If a header with the same
|
|
164
|
+
/// name already exists, it will be replaced.
|
|
165
|
+
///
|
|
166
|
+
/// # Arguments
|
|
167
|
+
/// * `name` - The header name (e.g., `HeaderName::from_static("content-type")`)
|
|
168
|
+
/// * `value` - The header value
|
|
169
|
+
///
|
|
170
|
+
/// # Returns
|
|
171
|
+
/// Self for method chaining
|
|
172
|
+
///
|
|
173
|
+
/// # Example
|
|
174
|
+
///
|
|
175
|
+
/// ```ignore
|
|
176
|
+
/// use axum::http::{HeaderName, HeaderValue};
|
|
177
|
+
///
|
|
178
|
+
/// let response = HandlerResponse::stream(my_stream)
|
|
179
|
+
/// .with_header(
|
|
180
|
+
/// HeaderName::from_static("content-type"),
|
|
181
|
+
/// HeaderValue::from_static("application/octet-stream")
|
|
182
|
+
/// );
|
|
183
|
+
/// ```
|
|
184
|
+
pub fn with_header(mut self, name: HeaderName, value: HeaderValue) -> Self {
|
|
185
|
+
if let HandlerResponse::Stream { headers, .. } = &mut self {
|
|
186
|
+
headers.insert(name, value);
|
|
187
|
+
}
|
|
188
|
+
self
|
|
189
|
+
}
|
|
190
|
+
}
|
|
191
|
+
|
|
192
|
+
#[cfg(test)]
|
|
193
|
+
mod tests {
|
|
194
|
+
use super::*;
|
|
195
|
+
use axum::http::header;
|
|
196
|
+
use http_body_util::BodyExt;
|
|
197
|
+
|
|
198
|
+
/// Test 1: Convert AxumResponse → HandlerResponse::Response
|
|
199
|
+
#[test]
|
|
200
|
+
fn test_from_axum_response() {
|
|
201
|
+
let axum_response = AxumResponse::new(Body::from("test body"));
|
|
202
|
+
let handler_response = HandlerResponse::from(axum_response);
|
|
203
|
+
|
|
204
|
+
match handler_response {
|
|
205
|
+
HandlerResponse::Response(_) => {}
|
|
206
|
+
HandlerResponse::Stream { .. } => panic!("Expected Response variant"),
|
|
207
|
+
}
|
|
208
|
+
}
|
|
209
|
+
|
|
210
|
+
/// Test 2: Create stream with chunks, verify wrapping
|
|
211
|
+
#[tokio::test]
|
|
212
|
+
async fn test_stream_creation_with_chunks() {
|
|
213
|
+
let chunks = vec![
|
|
214
|
+
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("chunk1")),
|
|
215
|
+
Ok(Bytes::from("chunk2")),
|
|
216
|
+
Ok(Bytes::from("chunk3")),
|
|
217
|
+
];
|
|
218
|
+
let stream = futures::stream::iter(chunks);
|
|
219
|
+
let handler_response = HandlerResponse::stream(stream);
|
|
220
|
+
|
|
221
|
+
match handler_response {
|
|
222
|
+
HandlerResponse::Stream { status, headers, .. } => {
|
|
223
|
+
assert_eq!(status, StatusCode::OK);
|
|
224
|
+
assert!(headers.is_empty());
|
|
225
|
+
}
|
|
226
|
+
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
227
|
+
}
|
|
228
|
+
}
|
|
229
|
+
|
|
230
|
+
/// Test 3: Stream with custom status code (206 Partial Content)
|
|
231
|
+
#[tokio::test]
|
|
232
|
+
async fn test_stream_with_custom_status() {
|
|
233
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
234
|
+
"partial",
|
|
235
|
+
))];
|
|
236
|
+
let stream = futures::stream::iter(chunks);
|
|
237
|
+
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::PARTIAL_CONTENT);
|
|
238
|
+
|
|
239
|
+
match handler_response {
|
|
240
|
+
HandlerResponse::Stream { status, .. } => {
|
|
241
|
+
assert_eq!(status, StatusCode::PARTIAL_CONTENT);
|
|
242
|
+
}
|
|
243
|
+
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
244
|
+
}
|
|
245
|
+
}
|
|
246
|
+
|
|
247
|
+
/// Test 4: Stream with headers via with_header()
|
|
248
|
+
#[tokio::test]
|
|
249
|
+
async fn test_stream_with_headers() {
|
|
250
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("test"))];
|
|
251
|
+
let stream = futures::stream::iter(chunks);
|
|
252
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
253
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/x-ndjson"))
|
|
254
|
+
.with_header(header::CACHE_CONTROL, HeaderValue::from_static("no-cache"));
|
|
255
|
+
|
|
256
|
+
match handler_response {
|
|
257
|
+
HandlerResponse::Stream { headers, .. } => {
|
|
258
|
+
assert_eq!(headers.get(header::CONTENT_TYPE).unwrap(), "application/x-ndjson");
|
|
259
|
+
assert_eq!(headers.get(header::CACHE_CONTROL).unwrap(), "no-cache");
|
|
260
|
+
}
|
|
261
|
+
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
262
|
+
}
|
|
263
|
+
}
|
|
264
|
+
|
|
265
|
+
/// Test 5: Stream body consumption - read all chunks from stream
|
|
266
|
+
#[tokio::test]
|
|
267
|
+
async fn test_stream_body_consumption() {
|
|
268
|
+
let chunks = vec![
|
|
269
|
+
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("hello ")),
|
|
270
|
+
Ok(Bytes::from("world")),
|
|
271
|
+
Ok(Bytes::from("!")),
|
|
272
|
+
];
|
|
273
|
+
let stream = futures::stream::iter(chunks);
|
|
274
|
+
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::OK);
|
|
275
|
+
|
|
276
|
+
let axum_response = handler_response.into_response();
|
|
277
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
278
|
+
let bytes = body.to_bytes();
|
|
279
|
+
|
|
280
|
+
assert_eq!(bytes, "hello world!");
|
|
281
|
+
}
|
|
282
|
+
|
|
283
|
+
/// Test 6: Into response for Response variant - passthrough conversion
|
|
284
|
+
#[tokio::test]
|
|
285
|
+
async fn test_into_response_for_response_variant() {
|
|
286
|
+
let original_body = "test response body";
|
|
287
|
+
let axum_response = AxumResponse::new(Body::from(original_body));
|
|
288
|
+
let handler_response = HandlerResponse::from(axum_response);
|
|
289
|
+
|
|
290
|
+
let result = handler_response.into_response();
|
|
291
|
+
let body = result.into_body().collect().await.unwrap();
|
|
292
|
+
let bytes = body.to_bytes();
|
|
293
|
+
|
|
294
|
+
assert_eq!(bytes, original_body);
|
|
295
|
+
}
|
|
296
|
+
|
|
297
|
+
/// Test 7: Method chaining - with_status() → with_header() → with_header()
|
|
298
|
+
#[tokio::test]
|
|
299
|
+
async fn test_method_chaining() {
|
|
300
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
301
|
+
"chained",
|
|
302
|
+
))];
|
|
303
|
+
let stream = futures::stream::iter(chunks);
|
|
304
|
+
|
|
305
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
306
|
+
.with_status(StatusCode::CREATED)
|
|
307
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("text/plain"))
|
|
308
|
+
.with_header(header::ETAG, HeaderValue::from_static("\"abc123\""));
|
|
309
|
+
|
|
310
|
+
match handler_response {
|
|
311
|
+
HandlerResponse::Stream { status, headers, .. } => {
|
|
312
|
+
assert_eq!(status, StatusCode::CREATED);
|
|
313
|
+
assert_eq!(headers.get(header::CONTENT_TYPE).unwrap(), "text/plain");
|
|
314
|
+
assert_eq!(headers.get(header::ETAG).unwrap(), "\"abc123\"");
|
|
315
|
+
}
|
|
316
|
+
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
317
|
+
}
|
|
318
|
+
}
|
|
319
|
+
|
|
320
|
+
/// Test 8: Empty stream - zero-byte stream handling
|
|
321
|
+
#[tokio::test]
|
|
322
|
+
async fn test_empty_stream() {
|
|
323
|
+
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![];
|
|
324
|
+
let stream = futures::stream::iter(chunks);
|
|
325
|
+
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::NO_CONTENT);
|
|
326
|
+
|
|
327
|
+
let axum_response = handler_response.into_response();
|
|
328
|
+
let status = axum_response.status();
|
|
329
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
330
|
+
let bytes = body.to_bytes();
|
|
331
|
+
|
|
332
|
+
assert!(bytes.is_empty());
|
|
333
|
+
assert_eq!(status, StatusCode::NO_CONTENT);
|
|
334
|
+
}
|
|
335
|
+
|
|
336
|
+
/// Test 9: Large stream - many chunks (100+ items)
|
|
337
|
+
#[tokio::test]
|
|
338
|
+
async fn test_large_stream() {
|
|
339
|
+
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> =
|
|
340
|
+
(0..150).map(|i| Ok(Bytes::from(format!("chunk{} ", i)))).collect();
|
|
341
|
+
|
|
342
|
+
let stream = futures::stream::iter(chunks);
|
|
343
|
+
let handler_response = HandlerResponse::stream(stream);
|
|
344
|
+
|
|
345
|
+
let axum_response = handler_response.into_response();
|
|
346
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
347
|
+
let bytes = body.to_bytes();
|
|
348
|
+
|
|
349
|
+
assert!(bytes.len() > 1000);
|
|
350
|
+
for i in 0..150 {
|
|
351
|
+
let expected = format!("chunk{} ", i);
|
|
352
|
+
assert!(std::str::from_utf8(&bytes).unwrap().contains(&expected));
|
|
353
|
+
}
|
|
354
|
+
}
|
|
355
|
+
|
|
356
|
+
/// Test 10: Error in stream - stream item returns Err, verify error propagation
|
|
357
|
+
#[tokio::test]
|
|
358
|
+
async fn test_stream_error_propagation() {
|
|
359
|
+
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![
|
|
360
|
+
Ok(Bytes::from("good1 ")),
|
|
361
|
+
Err("custom error".into()),
|
|
362
|
+
Ok(Bytes::from("good2")),
|
|
363
|
+
];
|
|
364
|
+
|
|
365
|
+
let stream = futures::stream::iter(chunks);
|
|
366
|
+
let handler_response = HandlerResponse::stream(stream);
|
|
367
|
+
|
|
368
|
+
let axum_response = handler_response.into_response();
|
|
369
|
+
let result = axum_response.into_body().collect().await;
|
|
370
|
+
|
|
371
|
+
assert!(result.is_err());
|
|
372
|
+
}
|
|
373
|
+
|
|
374
|
+
/// Test 11: Response variant ignores with_status()
|
|
375
|
+
#[test]
|
|
376
|
+
fn test_response_variant_ignores_with_status() {
|
|
377
|
+
let axum_response = AxumResponse::builder()
|
|
378
|
+
.status(StatusCode::OK)
|
|
379
|
+
.body(Body::from("test"))
|
|
380
|
+
.unwrap();
|
|
381
|
+
let handler_response = HandlerResponse::from(axum_response);
|
|
382
|
+
|
|
383
|
+
let result = handler_response.with_status(StatusCode::NOT_FOUND);
|
|
384
|
+
|
|
385
|
+
match result {
|
|
386
|
+
HandlerResponse::Response(resp) => {
|
|
387
|
+
assert_eq!(resp.status(), StatusCode::OK);
|
|
388
|
+
}
|
|
389
|
+
HandlerResponse::Stream { .. } => panic!("Expected Response variant"),
|
|
390
|
+
}
|
|
391
|
+
}
|
|
392
|
+
|
|
393
|
+
/// Test 12: Response variant ignores with_header()
|
|
394
|
+
#[test]
|
|
395
|
+
fn test_response_variant_ignores_with_header() {
|
|
396
|
+
let axum_response = AxumResponse::builder()
|
|
397
|
+
.status(StatusCode::OK)
|
|
398
|
+
.header(header::CONTENT_TYPE, "text/plain")
|
|
399
|
+
.body(Body::from("test"))
|
|
400
|
+
.unwrap();
|
|
401
|
+
let handler_response = HandlerResponse::from(axum_response);
|
|
402
|
+
|
|
403
|
+
let result = handler_response.with_header(header::CACHE_CONTROL, HeaderValue::from_static("max-age=3600"));
|
|
404
|
+
|
|
405
|
+
match result {
|
|
406
|
+
HandlerResponse::Response(resp) => {
|
|
407
|
+
assert!(resp.headers().get(header::CACHE_CONTROL).is_none());
|
|
408
|
+
}
|
|
409
|
+
HandlerResponse::Stream { .. } => panic!("Expected Response variant"),
|
|
410
|
+
}
|
|
411
|
+
}
|
|
412
|
+
|
|
413
|
+
/// Test 13: Stream into_response applies status and headers
|
|
414
|
+
#[tokio::test]
|
|
415
|
+
async fn test_stream_into_response_applies_status_and_headers() {
|
|
416
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
417
|
+
"stream data",
|
|
418
|
+
))];
|
|
419
|
+
let stream = futures::stream::iter(chunks);
|
|
420
|
+
|
|
421
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
422
|
+
.with_status(StatusCode::PARTIAL_CONTENT)
|
|
423
|
+
.with_header(header::CONTENT_RANGE, HeaderValue::from_static("bytes 0-10/100"));
|
|
424
|
+
|
|
425
|
+
let axum_response = handler_response.into_response();
|
|
426
|
+
|
|
427
|
+
assert_eq!(axum_response.status(), StatusCode::PARTIAL_CONTENT);
|
|
428
|
+
assert_eq!(
|
|
429
|
+
axum_response.headers().get(header::CONTENT_RANGE).unwrap(),
|
|
430
|
+
"bytes 0-10/100"
|
|
431
|
+
);
|
|
432
|
+
|
|
433
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
434
|
+
assert_eq!(body.to_bytes(), "stream data");
|
|
435
|
+
}
|
|
436
|
+
|
|
437
|
+
/// Test 14: Multiple header replacements via with_header()
|
|
438
|
+
#[tokio::test]
|
|
439
|
+
async fn test_multiple_header_replacements() {
|
|
440
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("data"))];
|
|
441
|
+
let stream = futures::stream::iter(chunks);
|
|
442
|
+
|
|
443
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
444
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/json"))
|
|
445
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/x-ndjson"));
|
|
446
|
+
|
|
447
|
+
match handler_response {
|
|
448
|
+
HandlerResponse::Stream { headers, .. } => {
|
|
449
|
+
assert_eq!(headers.get(header::CONTENT_TYPE).unwrap(), "application/x-ndjson");
|
|
450
|
+
}
|
|
451
|
+
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
452
|
+
}
|
|
453
|
+
}
|
|
454
|
+
|
|
455
|
+
/// Test 15: Stream with various status codes
|
|
456
|
+
#[tokio::test]
|
|
457
|
+
async fn test_stream_with_various_status_codes() {
|
|
458
|
+
let status_codes = vec![
|
|
459
|
+
StatusCode::OK,
|
|
460
|
+
StatusCode::CREATED,
|
|
461
|
+
StatusCode::ACCEPTED,
|
|
462
|
+
StatusCode::PARTIAL_CONTENT,
|
|
463
|
+
StatusCode::MULTI_STATUS,
|
|
464
|
+
];
|
|
465
|
+
|
|
466
|
+
for status in status_codes {
|
|
467
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("test"))];
|
|
468
|
+
let stream = futures::stream::iter(chunks);
|
|
469
|
+
let handler_response = HandlerResponse::stream(stream).with_status(status);
|
|
470
|
+
|
|
471
|
+
match handler_response {
|
|
472
|
+
HandlerResponse::Stream { status: s, .. } => {
|
|
473
|
+
assert_eq!(s, status);
|
|
474
|
+
}
|
|
475
|
+
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
476
|
+
}
|
|
477
|
+
}
|
|
478
|
+
}
|
|
479
|
+
|
|
480
|
+
/// Test 16: Stream with JSON lines content (fixture-based)
|
|
481
|
+
#[tokio::test]
|
|
482
|
+
async fn test_stream_with_json_lines_content() {
|
|
483
|
+
let chunks = vec![
|
|
484
|
+
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(r#"{"index":0,"payload":"alpha"}"#)),
|
|
485
|
+
Ok(Bytes::from("\n")),
|
|
486
|
+
Ok(Bytes::from(r#"{"index":1,"payload":"beta"}"#)),
|
|
487
|
+
Ok(Bytes::from("\n")),
|
|
488
|
+
Ok(Bytes::from(r#"{"index":2,"payload":"gamma"}"#)),
|
|
489
|
+
Ok(Bytes::from("\n")),
|
|
490
|
+
];
|
|
491
|
+
|
|
492
|
+
let stream = futures::stream::iter(chunks);
|
|
493
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
494
|
+
.with_status(StatusCode::OK)
|
|
495
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/x-ndjson"));
|
|
496
|
+
|
|
497
|
+
let axum_response = handler_response.into_response();
|
|
498
|
+
let status = axum_response.status();
|
|
499
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
500
|
+
let bytes = body.to_bytes();
|
|
501
|
+
let body_str = std::str::from_utf8(&bytes).unwrap();
|
|
502
|
+
|
|
503
|
+
assert_eq!(status, StatusCode::OK);
|
|
504
|
+
assert!(body_str.contains("alpha"));
|
|
505
|
+
assert!(body_str.contains("beta"));
|
|
506
|
+
assert!(body_str.contains("gamma"));
|
|
507
|
+
}
|
|
508
|
+
|
|
509
|
+
/// Test 17: Round-trip Response → HandlerResponse → Response
|
|
510
|
+
#[tokio::test]
|
|
511
|
+
async fn test_response_roundtrip() {
|
|
512
|
+
let original = AxumResponse::builder()
|
|
513
|
+
.status(StatusCode::OK)
|
|
514
|
+
.header(header::CONTENT_TYPE, "text/plain")
|
|
515
|
+
.body(Body::from("roundtrip test"))
|
|
516
|
+
.unwrap();
|
|
517
|
+
|
|
518
|
+
let handler_response = HandlerResponse::from(original);
|
|
519
|
+
let result = handler_response.into_response();
|
|
520
|
+
|
|
521
|
+
assert_eq!(result.status(), StatusCode::OK);
|
|
522
|
+
assert_eq!(result.headers().get(header::CONTENT_TYPE).unwrap(), "text/plain");
|
|
523
|
+
|
|
524
|
+
let body = result.into_body().collect().await.unwrap();
|
|
525
|
+
assert_eq!(body.to_bytes(), "roundtrip test");
|
|
526
|
+
}
|
|
527
|
+
|
|
528
|
+
/// Test 18: Single chunk stream (minimal data)
|
|
529
|
+
#[tokio::test]
|
|
530
|
+
async fn test_single_chunk_stream() {
|
|
531
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("only"))];
|
|
532
|
+
let stream = futures::stream::iter(chunks);
|
|
533
|
+
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::OK);
|
|
534
|
+
|
|
535
|
+
let axum_response = handler_response.into_response();
|
|
536
|
+
let status = axum_response.status();
|
|
537
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
538
|
+
let bytes = body.to_bytes();
|
|
539
|
+
|
|
540
|
+
assert_eq!(bytes, "only");
|
|
541
|
+
assert_eq!(status, StatusCode::OK);
|
|
542
|
+
}
|
|
543
|
+
|
|
544
|
+
/// Test 19: Stream with 1000+ chunks (performance edge case)
|
|
545
|
+
#[tokio::test]
|
|
546
|
+
async fn test_very_large_stream_many_chunks() {
|
|
547
|
+
let chunk_count = 1500;
|
|
548
|
+
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> =
|
|
549
|
+
(0..chunk_count).map(|_| Ok(Bytes::from(format!("x")))).collect();
|
|
550
|
+
|
|
551
|
+
let stream = futures::stream::iter(chunks);
|
|
552
|
+
let handler_response = HandlerResponse::stream(stream);
|
|
553
|
+
|
|
554
|
+
let axum_response = handler_response.into_response();
|
|
555
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
556
|
+
let bytes = body.to_bytes();
|
|
557
|
+
|
|
558
|
+
assert_eq!(bytes.len(), chunk_count);
|
|
559
|
+
}
|
|
560
|
+
|
|
561
|
+
/// Test 20: Stream with varying chunk sizes (1 byte to 1MB)
|
|
562
|
+
#[tokio::test]
|
|
563
|
+
async fn test_stream_with_varying_chunk_sizes() {
|
|
564
|
+
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![
|
|
565
|
+
Ok(Bytes::from("x")),
|
|
566
|
+
Ok(Bytes::from("xx".repeat(100))),
|
|
567
|
+
Ok(Bytes::from("x".repeat(10_000))),
|
|
568
|
+
Ok(Bytes::from("x".repeat(100_000))),
|
|
569
|
+
];
|
|
570
|
+
|
|
571
|
+
let stream = futures::stream::iter(chunks);
|
|
572
|
+
let handler_response = HandlerResponse::stream(stream);
|
|
573
|
+
|
|
574
|
+
let axum_response = handler_response.into_response();
|
|
575
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
576
|
+
let bytes = body.to_bytes();
|
|
577
|
+
|
|
578
|
+
assert_eq!(bytes.len(), 110_201);
|
|
579
|
+
}
|
|
580
|
+
|
|
581
|
+
/// Test 21: Stream with error in the middle (chunk 500/1000)
|
|
582
|
+
#[tokio::test]
|
|
583
|
+
async fn test_stream_error_in_middle() {
|
|
584
|
+
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = (0..1000)
|
|
585
|
+
.map(|i| {
|
|
586
|
+
if i == 500 {
|
|
587
|
+
Err("midstream error".into())
|
|
588
|
+
} else {
|
|
589
|
+
Ok(Bytes::from("chunk"))
|
|
590
|
+
}
|
|
591
|
+
})
|
|
592
|
+
.collect();
|
|
593
|
+
|
|
594
|
+
let stream = futures::stream::iter(chunks);
|
|
595
|
+
let handler_response = HandlerResponse::stream(stream);
|
|
596
|
+
|
|
597
|
+
let axum_response = handler_response.into_response();
|
|
598
|
+
let result = axum_response.into_body().collect().await;
|
|
599
|
+
|
|
600
|
+
assert!(result.is_err());
|
|
601
|
+
}
|
|
602
|
+
|
|
603
|
+
/// Test 22: Stream with SSE-like headers
|
|
604
|
+
#[tokio::test]
|
|
605
|
+
async fn test_stream_with_sse_headers() {
|
|
606
|
+
let chunks = vec![
|
|
607
|
+
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("event: message\n")),
|
|
608
|
+
Ok(Bytes::from("data: {\"msg\": \"hello\"}\n\n")),
|
|
609
|
+
];
|
|
610
|
+
let stream = futures::stream::iter(chunks);
|
|
611
|
+
|
|
612
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
613
|
+
.with_status(StatusCode::OK)
|
|
614
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("text/event-stream"))
|
|
615
|
+
.with_header(header::CACHE_CONTROL, HeaderValue::from_static("no-cache"))
|
|
616
|
+
.with_header(header::CONNECTION, HeaderValue::from_static("keep-alive"));
|
|
617
|
+
|
|
618
|
+
let axum_response = handler_response.into_response();
|
|
619
|
+
|
|
620
|
+
assert_eq!(axum_response.status(), StatusCode::OK);
|
|
621
|
+
assert_eq!(
|
|
622
|
+
axum_response.headers().get(header::CONTENT_TYPE).unwrap(),
|
|
623
|
+
"text/event-stream"
|
|
624
|
+
);
|
|
625
|
+
assert_eq!(axum_response.headers().get(header::CACHE_CONTROL).unwrap(), "no-cache");
|
|
626
|
+
|
|
627
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
628
|
+
let body_bytes = body.to_bytes();
|
|
629
|
+
let body_str = std::str::from_utf8(&body_bytes).unwrap();
|
|
630
|
+
assert!(body_str.contains("event: message"));
|
|
631
|
+
}
|
|
632
|
+
|
|
633
|
+
/// Test 23: Stream with WebSocket-like upgrade headers (200 OK with Upgrade)
|
|
634
|
+
#[tokio::test]
|
|
635
|
+
async fn test_stream_with_websocket_headers() {
|
|
636
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
637
|
+
"ws-frame-data",
|
|
638
|
+
))];
|
|
639
|
+
let stream = futures::stream::iter(chunks);
|
|
640
|
+
|
|
641
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
642
|
+
.with_status(StatusCode::OK)
|
|
643
|
+
.with_header(header::UPGRADE, HeaderValue::from_static("websocket"))
|
|
644
|
+
.with_header(
|
|
645
|
+
HeaderName::from_static("sec-websocket-accept"),
|
|
646
|
+
HeaderValue::from_static("s3pPLMBiTxaQ9kYGzzhZRbK+xOo="),
|
|
647
|
+
);
|
|
648
|
+
|
|
649
|
+
let axum_response = handler_response.into_response();
|
|
650
|
+
|
|
651
|
+
assert_eq!(axum_response.status(), StatusCode::OK);
|
|
652
|
+
assert_eq!(axum_response.headers().get(header::UPGRADE).unwrap(), "websocket");
|
|
653
|
+
|
|
654
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
655
|
+
assert_eq!(body.to_bytes(), "ws-frame-data");
|
|
656
|
+
}
|
|
657
|
+
|
|
658
|
+
/// Test 24: Stream status transitions (from 200 OK to 206 Partial Content)
|
|
659
|
+
#[tokio::test]
|
|
660
|
+
async fn test_stream_status_transition() {
|
|
661
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("data"))];
|
|
662
|
+
let stream = futures::stream::iter(chunks);
|
|
663
|
+
|
|
664
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
665
|
+
.with_status(StatusCode::OK)
|
|
666
|
+
.with_status(StatusCode::PARTIAL_CONTENT);
|
|
667
|
+
|
|
668
|
+
match handler_response {
|
|
669
|
+
HandlerResponse::Stream { status, .. } => {
|
|
670
|
+
assert_eq!(status, StatusCode::PARTIAL_CONTENT);
|
|
671
|
+
}
|
|
672
|
+
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
673
|
+
}
|
|
674
|
+
}
|
|
675
|
+
|
|
676
|
+
/// Test 25: Stream with chunked transfer encoding simulation
|
|
677
|
+
#[tokio::test]
|
|
678
|
+
async fn test_stream_chunked_encoding_simulation() {
|
|
679
|
+
let chunks = vec![
|
|
680
|
+
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("5\r\nhello\r\n")),
|
|
681
|
+
Ok(Bytes::from("5\r\nworld\r\n")),
|
|
682
|
+
Ok(Bytes::from("0\r\n\r\n")),
|
|
683
|
+
];
|
|
684
|
+
|
|
685
|
+
let stream = futures::stream::iter(chunks);
|
|
686
|
+
let handler_response =
|
|
687
|
+
HandlerResponse::stream(stream).with_header(header::TRANSFER_ENCODING, HeaderValue::from_static("chunked"));
|
|
688
|
+
|
|
689
|
+
let axum_response = handler_response.into_response();
|
|
690
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
691
|
+
let body_bytes = body.to_bytes();
|
|
692
|
+
|
|
693
|
+
assert!(std::str::from_utf8(&body_bytes).unwrap().contains("hello"));
|
|
694
|
+
}
|
|
695
|
+
|
|
696
|
+
/// Test 26: Stream with binary data (non-UTF8)
|
|
697
|
+
#[tokio::test]
|
|
698
|
+
async fn test_stream_with_binary_data() {
|
|
699
|
+
let chunks = vec![
|
|
700
|
+
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(vec![0xFF, 0xD8, 0xFF])),
|
|
701
|
+
Ok(Bytes::from(vec![0xE0, 0x00, 0x10])),
|
|
702
|
+
Ok(Bytes::from(vec![0x4A, 0x46, 0x49])),
|
|
703
|
+
];
|
|
704
|
+
|
|
705
|
+
let stream = futures::stream::iter(chunks);
|
|
706
|
+
let handler_response = HandlerResponse::stream(stream).with_header(
|
|
707
|
+
header::CONTENT_TYPE,
|
|
708
|
+
HeaderValue::from_static("application/octet-stream"),
|
|
709
|
+
);
|
|
710
|
+
|
|
711
|
+
let axum_response = handler_response.into_response();
|
|
712
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
713
|
+
let bytes = body.to_bytes();
|
|
714
|
+
|
|
715
|
+
assert_eq!(bytes[0], 0xFF);
|
|
716
|
+
assert_eq!(bytes[1], 0xD8);
|
|
717
|
+
assert_eq!(bytes[2], 0xFF);
|
|
718
|
+
assert_eq!(bytes[3], 0xE0);
|
|
719
|
+
assert_eq!(bytes[4], 0x00);
|
|
720
|
+
}
|
|
721
|
+
|
|
722
|
+
/// Test 27: Stream with null bytes in payload
|
|
723
|
+
#[tokio::test]
|
|
724
|
+
async fn test_stream_with_null_bytes() {
|
|
725
|
+
let chunks = vec![
|
|
726
|
+
Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(vec![0x00, 0x01, 0x02])),
|
|
727
|
+
Ok(Bytes::from(vec![0x00, 0x00, 0x00])),
|
|
728
|
+
Ok(Bytes::from(vec![0xFF, 0xFE, 0xFD])),
|
|
729
|
+
];
|
|
730
|
+
|
|
731
|
+
let stream = futures::stream::iter(chunks);
|
|
732
|
+
let handler_response = HandlerResponse::stream(stream);
|
|
733
|
+
|
|
734
|
+
let axum_response = handler_response.into_response();
|
|
735
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
736
|
+
let bytes = body.to_bytes();
|
|
737
|
+
|
|
738
|
+
assert_eq!(bytes.len(), 9);
|
|
739
|
+
assert_eq!(bytes[0], 0x00);
|
|
740
|
+
assert_eq!(bytes[4], 0x00);
|
|
741
|
+
assert_eq!(bytes[8], 0xFD);
|
|
742
|
+
}
|
|
743
|
+
|
|
744
|
+
/// Test 28: Stream with maximum header count
|
|
745
|
+
#[tokio::test]
|
|
746
|
+
async fn test_stream_with_many_headers() {
|
|
747
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("data"))];
|
|
748
|
+
let stream = futures::stream::iter(chunks);
|
|
749
|
+
|
|
750
|
+
let mut handler_response = HandlerResponse::stream(stream);
|
|
751
|
+
|
|
752
|
+
for i in 0..50 {
|
|
753
|
+
let header_name = format!("x-custom-{}", i);
|
|
754
|
+
handler_response = handler_response.with_header(
|
|
755
|
+
HeaderName::from_bytes(header_name.as_bytes()).unwrap(),
|
|
756
|
+
HeaderValue::from_static("value"),
|
|
757
|
+
);
|
|
758
|
+
}
|
|
759
|
+
|
|
760
|
+
let axum_response = handler_response.into_response();
|
|
761
|
+
assert_eq!(axum_response.status(), StatusCode::OK);
|
|
762
|
+
assert_eq!(axum_response.headers().len(), 50);
|
|
763
|
+
}
|
|
764
|
+
|
|
765
|
+
/// Test 29: Empty stream with 204 No Content status
|
|
766
|
+
#[tokio::test]
|
|
767
|
+
async fn test_empty_stream_with_204_no_content() {
|
|
768
|
+
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![];
|
|
769
|
+
let stream = futures::stream::iter(chunks);
|
|
770
|
+
|
|
771
|
+
let handler_response = HandlerResponse::stream(stream).with_status(StatusCode::NO_CONTENT);
|
|
772
|
+
|
|
773
|
+
let axum_response = handler_response.into_response();
|
|
774
|
+
|
|
775
|
+
assert_eq!(axum_response.status(), StatusCode::NO_CONTENT);
|
|
776
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
777
|
+
assert!(body.to_bytes().is_empty());
|
|
778
|
+
}
|
|
779
|
+
|
|
780
|
+
/// Test 30: Stream with repeated header replacement
|
|
781
|
+
#[tokio::test]
|
|
782
|
+
async fn test_stream_repeated_header_updates() {
|
|
783
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("test"))];
|
|
784
|
+
let stream = futures::stream::iter(chunks);
|
|
785
|
+
|
|
786
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
787
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("text/plain"))
|
|
788
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/json"))
|
|
789
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/xml"));
|
|
790
|
+
|
|
791
|
+
match handler_response {
|
|
792
|
+
HandlerResponse::Stream { headers, .. } => {
|
|
793
|
+
assert_eq!(headers.get(header::CONTENT_TYPE).unwrap(), "application/xml");
|
|
794
|
+
}
|
|
795
|
+
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
796
|
+
}
|
|
797
|
+
}
|
|
798
|
+
|
|
799
|
+
/// Test 31: Stream with extremely long chunk
|
|
800
|
+
#[tokio::test]
|
|
801
|
+
async fn test_stream_with_extremely_long_chunk() {
|
|
802
|
+
let large_chunk = "x".repeat(10_000_000);
|
|
803
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from(
|
|
804
|
+
large_chunk,
|
|
805
|
+
))];
|
|
806
|
+
let stream = futures::stream::iter(chunks);
|
|
807
|
+
|
|
808
|
+
let handler_response = HandlerResponse::stream(stream);
|
|
809
|
+
|
|
810
|
+
let axum_response = handler_response.into_response();
|
|
811
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
812
|
+
let bytes = body.to_bytes();
|
|
813
|
+
|
|
814
|
+
assert_eq!(bytes.len(), 10_000_000);
|
|
815
|
+
}
|
|
816
|
+
|
|
817
|
+
/// Test 32: Stream with zero-length chunks mixed in
|
|
818
|
+
#[tokio::test]
|
|
819
|
+
async fn test_stream_with_zero_length_chunks() {
|
|
820
|
+
let chunks: Vec<Result<Bytes, Box<dyn std::error::Error + Send + Sync>>> = vec![
|
|
821
|
+
Ok(Bytes::from("hello")),
|
|
822
|
+
Ok(Bytes::new()),
|
|
823
|
+
Ok(Bytes::from("world")),
|
|
824
|
+
Ok(Bytes::new()),
|
|
825
|
+
Ok(Bytes::from("!")),
|
|
826
|
+
];
|
|
827
|
+
|
|
828
|
+
let stream = futures::stream::iter(chunks);
|
|
829
|
+
let handler_response = HandlerResponse::stream(stream);
|
|
830
|
+
|
|
831
|
+
let axum_response = handler_response.into_response();
|
|
832
|
+
let body = axum_response.into_body().collect().await.unwrap();
|
|
833
|
+
let bytes = body.to_bytes();
|
|
834
|
+
|
|
835
|
+
assert_eq!(bytes, "helloworld!");
|
|
836
|
+
}
|
|
837
|
+
|
|
838
|
+
/// Test 33: Handler response variant preserves custom status on failure
|
|
839
|
+
#[test]
|
|
840
|
+
fn test_response_variant_preserves_original_status() {
|
|
841
|
+
let axum_response = AxumResponse::builder()
|
|
842
|
+
.status(StatusCode::BAD_REQUEST)
|
|
843
|
+
.body(Body::from("error"))
|
|
844
|
+
.unwrap();
|
|
845
|
+
|
|
846
|
+
let handler_response = HandlerResponse::from(axum_response);
|
|
847
|
+
|
|
848
|
+
let result = handler_response
|
|
849
|
+
.with_status(StatusCode::OK)
|
|
850
|
+
.with_status(StatusCode::INTERNAL_SERVER_ERROR);
|
|
851
|
+
|
|
852
|
+
match result {
|
|
853
|
+
HandlerResponse::Response(resp) => {
|
|
854
|
+
assert_eq!(resp.status(), StatusCode::BAD_REQUEST);
|
|
855
|
+
}
|
|
856
|
+
HandlerResponse::Stream { .. } => panic!("Expected Response variant"),
|
|
857
|
+
}
|
|
858
|
+
}
|
|
859
|
+
|
|
860
|
+
/// Test 34: Stream response conversion preserves header ordering
|
|
861
|
+
#[tokio::test]
|
|
862
|
+
async fn test_stream_into_response_preserves_headers() {
|
|
863
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("data"))];
|
|
864
|
+
let stream = futures::stream::iter(chunks);
|
|
865
|
+
|
|
866
|
+
let handler_response = HandlerResponse::stream(stream)
|
|
867
|
+
.with_header(header::CONTENT_TYPE, HeaderValue::from_static("application/json"))
|
|
868
|
+
.with_header(header::CACHE_CONTROL, HeaderValue::from_static("max-age=3600"))
|
|
869
|
+
.with_header(header::ETAG, HeaderValue::from_static("\"abc123\""));
|
|
870
|
+
|
|
871
|
+
let axum_response = handler_response.into_response();
|
|
872
|
+
|
|
873
|
+
assert!(axum_response.headers().get(header::CONTENT_TYPE).is_some());
|
|
874
|
+
assert!(axum_response.headers().get(header::CACHE_CONTROL).is_some());
|
|
875
|
+
assert!(axum_response.headers().get(header::ETAG).is_some());
|
|
876
|
+
assert_eq!(axum_response.headers().len(), 3);
|
|
877
|
+
}
|
|
878
|
+
|
|
879
|
+
/// Test 35: Stream with 5xx status codes
|
|
880
|
+
#[tokio::test]
|
|
881
|
+
async fn test_stream_with_error_status_codes() {
|
|
882
|
+
let error_statuses = vec![
|
|
883
|
+
StatusCode::INTERNAL_SERVER_ERROR,
|
|
884
|
+
StatusCode::SERVICE_UNAVAILABLE,
|
|
885
|
+
StatusCode::GATEWAY_TIMEOUT,
|
|
886
|
+
];
|
|
887
|
+
|
|
888
|
+
for status in error_statuses {
|
|
889
|
+
let chunks = vec![Ok::<_, Box<dyn std::error::Error + Send + Sync>>(Bytes::from("error"))];
|
|
890
|
+
let stream = futures::stream::iter(chunks);
|
|
891
|
+
let handler_response = HandlerResponse::stream(stream).with_status(status);
|
|
892
|
+
|
|
893
|
+
match handler_response {
|
|
894
|
+
HandlerResponse::Stream { status: s, .. } => {
|
|
895
|
+
assert_eq!(s, status);
|
|
896
|
+
}
|
|
897
|
+
HandlerResponse::Response(_) => panic!("Expected Stream variant"),
|
|
898
|
+
}
|
|
899
|
+
}
|
|
900
|
+
}
|
|
901
|
+
}
|