Skip to content

Commit 38e08d7

Browse files
cursoragentlovasoa
andcommitted
Refactor: Pass ServerTiming to extract_request_info
Co-authored-by: contact <[email protected]>
1 parent 5c41598 commit 38e08d7

File tree

4 files changed

+21
-16
lines changed

4 files changed

+21
-16
lines changed

src/webserver/http_request_info.rs

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,8 @@ mod test {
288288
serde_json::from_str::<AppConfig>(r#"{"listen_on": "localhost:1234"}"#).unwrap();
289289
let mut service_request = TestRequest::default().to_srv_request();
290290
let app_data = Arc::new(AppState::init(&config).await.unwrap());
291-
let request_info = extract_request_info(&mut service_request, app_data)
291+
let server_timing = super::server_timing::ServerTiming::new(false);
292+
let request_info = extract_request_info(&mut service_request, app_data, server_timing)
292293
.await
293294
.unwrap();
294295
assert_eq!(request_info.post_variables.len(), 0);
@@ -306,7 +307,8 @@ mod test {
306307
.set_payload("my_array[]=3&my_array[]=Hello%20World&repeated=1&repeated=2")
307308
.to_srv_request();
308309
let app_data = Arc::new(AppState::init(&config).await.unwrap());
309-
let request_info = extract_request_info(&mut service_request, app_data)
310+
let server_timing = super::server_timing::ServerTiming::new(false);
311+
let request_info = extract_request_info(&mut service_request, app_data, server_timing)
310312
.await
311313
.unwrap();
312314
assert_eq!(
@@ -355,7 +357,8 @@ mod test {
355357
)
356358
.to_srv_request();
357359
let app_data = Arc::new(AppState::init(&config).await.unwrap());
358-
let request_info = extract_request_info(&mut service_request, app_data)
360+
let server_timing = super::server_timing::ServerTiming::new(false);
361+
let request_info = extract_request_info(&mut service_request, app_data, server_timing)
359362
.await
360363
.unwrap();
361364
assert_eq!(

src/webserver/server_timing.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
use std::fmt::Write;
12
use std::time::Instant;
23

34
#[derive(Debug, Clone)]
@@ -28,8 +29,7 @@ impl ServerTiming {
2829
if !self.header.is_empty() {
2930
self.header.push_str(", ");
3031
}
31-
use std::fmt::Write;
32-
write!(&mut self.header, "{};dur={:.2}", name, duration_ms).unwrap();
32+
write!(&mut self.header, "{name};dur={duration_ms:.2}").unwrap();
3333
}
3434

3535
#[must_use]

tests/server_timing/mod.rs

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,12 @@ async fn test_server_timing_disabled_in_production() -> actix_web::Result<()> {
99
config.environment = sqlpage::app_config::DevOrProd::Production;
1010
let app_data = make_app_data_from_config(config).await;
1111

12-
let req =
13-
crate::common::get_request_to_with_data("/tests/server_timing/simple_query.sql", app_data)
14-
.await?
15-
.to_srv_request();
12+
let req = crate::common::get_request_to_with_data(
13+
"/tests/sql_test_files/it_works_simple.sql",
14+
app_data,
15+
)
16+
.await?
17+
.to_srv_request();
1618
let resp = main_handler(req).await?;
1719

1820
assert_eq!(resp.status(), StatusCode::OK);
@@ -29,10 +31,12 @@ async fn test_server_timing_enabled_in_development() -> actix_web::Result<()> {
2931
config.environment = sqlpage::app_config::DevOrProd::Development;
3032
let app_data = make_app_data_from_config(config).await;
3133

32-
let req =
33-
crate::common::get_request_to_with_data("/tests/server_timing/simple_query.sql", app_data)
34-
.await?
35-
.to_srv_request();
34+
let req = crate::common::get_request_to_with_data(
35+
"/tests/sql_test_files/it_works_simple.sql",
36+
app_data,
37+
)
38+
.await?
39+
.to_srv_request();
3640
let resp = main_handler(req).await?;
3741

3842
assert_eq!(resp.status(), StatusCode::OK);
@@ -68,7 +72,7 @@ async fn test_server_timing_enabled_in_development() -> actix_web::Result<()> {
6872

6973
#[actix_web::test]
7074
async fn test_server_timing_format() -> actix_web::Result<()> {
71-
let req = get_request_to("/tests/server_timing/simple_query.sql")
75+
let req = get_request_to("/tests/sql_test_files/it_works_simple.sql")
7276
.await?
7377
.to_srv_request();
7478
let resp = main_handler(req).await?;

tests/server_timing/simple_query.sql

Lines changed: 0 additions & 2 deletions
This file was deleted.

0 commit comments

Comments
 (0)