test concurrent requests

lovasoa · lovasoa · commit d38ca15b7410 · 2024-01-30T14:35:22.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,44 @@ 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 app_data = make_app_data().await;
+    let reqs = (0..64)
+        .map(|i| {
+            let component = components[i % components.len()];
+            req_path_with_app_data(
+                format!("/tests/any_component.sql?component={}", component),
+                app_data.clone(),
+            )
+        })
+        .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 +97,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") {
@@ -141,28 +179,46 @@ async fn test_csv_upload() -> actix_web::Result<()> {
 }
 
 async fn get_request_to(path: &str) -> actix_web::Result<TestRequest> {
+    let data = make_app_data().await;
+    Ok(test::TestRequest::get().uri(path).app_data(data))
+}
+
+async fn make_app_data() -> actix_web::web::Data<AppState> {
     init_log();
     let config = test_config();
     let state = AppState::init(&config).await.unwrap();
     let data = actix_web::web::Data::new(state);
-    Ok(test::TestRequest::get().uri(path).app_data(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();
     main_handler(req).await
 }
 
+async fn req_path_with_app_data(
+    path: impl AsRef<str>,
+    app_data: actix_web::web::Data<AppState>,
+) -> Result<actix_web::dev::ServiceResponse, actix_web::Error> {
+    let req = test::TestRequest::get()
+        .uri(path.as_ref())
+        .app_data(app_data)
+        .to_srv_request();
+    main_handler(req).await
+}
+
 pub fn test_config() -> AppConfig {
     let db_url = std::env::var("DATABASE_URL").unwrap_or_else(|_| "sqlite::memory:".to_string());
     serde_json::from_str::<AppConfig>(&format!(
         r#"{{
         "database_url": "{}",
         "database_connection_retries": 2,
-        "database_connection_acquire_timeout_seconds": 1,
+        "database_connection_acquire_timeout_seconds": 10,
         "allow_exec": true,
         "listen_on": "111.111.111.111:1"
     }}"#,
