test concurrent requests

lovasoa · lovasoa · commit b1c671600891 · 2024-01-30T12:17:47.000+01:00
diff --git a/tests/any_component.sql b/tests/any_component.sql
@@ -0,0 +1,6 @@
+select $component as component;
+select 
+    'It works !' as title,
+    'It works !'  as description;
+
+select 'divider' as component, 'the end' as contents;
diff --git a/tests/index.rs b/tests/index.rs
@@ -11,7 +11,7 @@ async fn test_index_ok() {
     assert_eq!(resp.status(), http::StatusCode::OK);
     let body = test::read_body(resp).await;
     assert!(body.starts_with(b"<!DOCTYPE html>"));
-    // the body should contain the strint "It works!" and should not contain the string "error"
+    // the body should contain the string "It works!" and should not contain the string "error"
     let body = String::from_utf8(body.to_vec()).unwrap();
     assert!(body.contains("It works !"));
     assert!(!body.contains("error"));
@@ -43,6 +43,40 @@ async fn test_404() {
     }
 }
 
+#[actix_web::test]
+async fn test_concurrent_requests() {
+    // send 32 requests (less than the default postgres pool size)
+    // at the same time to /tests/multiple_components.sql
+    let components = [
+        "table", "form", "card", "datagrid", "hero", "list", "timeline",
+    ];
+    let reqs = (0..32)
+        .map(|i| {
+            let component = components[i % components.len()];
+            req_path(format!("/tests/any_component.sql?component={}", component))
+        })
+        .collect::<Vec<_>>();
+    // send all requests at the same time
+    let results = futures_util::future::join_all(reqs).await;
+    // check that all requests succeeded
+    for result in results.into_iter() {
+        let resp = result.unwrap();
+        assert_eq!(resp.status(), http::StatusCode::OK);
+        let body = test::read_body(resp).await;
+        assert!(
+            body.starts_with(b"<!DOCTYPE html>"),
+            "Expected html doctype"
+        );
+        // the body should contain the string "It works!" and should not contain the string "error"
+        let body = String::from_utf8(body.to_vec()).unwrap();
+        assert!(
+            body.contains("It works !"),
+            "Expected to contain: It works !, but got: {body}"
+        );
+        assert!(!body.contains("error"));
+    }
+}
+
 #[actix_web::test]
 async fn test_files() {
     // Iterate over all the sql test files in the tests/ directory
@@ -59,7 +93,7 @@ async fn test_files() {
         let resp = req_path(&req_str).await.unwrap();
         let body = test::read_body(resp).await;
         assert!(body.starts_with(b"<!DOCTYPE html>"));
-        // the body should contain the strint "It works!" and should not contain the string "error"
+        // the body should contain the string "It works!" and should not contain the string "error"
         let body = String::from_utf8(body.to_vec()).unwrap();
         let lowercase_body = body.to_lowercase();
         if stem.starts_with("it_works") {
@@ -148,8 +182,10 @@ async fn get_request_to(path: &str) -> actix_web::Result<TestRequest> {
     Ok(test::TestRequest::get().uri(path).app_data(data))
 }
 
-async fn req_path(path: &str) -> Result<actix_web::dev::ServiceResponse, actix_web::Error> {
-    let req = get_request_to(path)
+async fn req_path(
+    path: impl AsRef<str>,
+) -> Result<actix_web::dev::ServiceResponse, actix_web::Error> {
+    let req = get_request_to(path.as_ref())
         .await?
         .insert_header(ContentType::plaintext())
         .to_srv_request();
