feat: add test for 404 fallback mechanism

wucke13 · lovasoa · commit d5cbee8f67ff · 2024-08-22T20:48:03.000+02:00
diff --git a/tests/404.sql b/tests/404.sql
@@ -0,0 +1,3 @@
+SELECT 'alert' AS component,
+    'We almost got an oopsie' AS title,
+    'But the `404.sql` file saved the day!' AS description_md; 
diff --git a/tests/index.rs b/tests/index.rs
@@ -51,6 +51,28 @@ async fn test_404() {
     }
 }
 
+#[actix_web::test]
+async fn test_404_fallback() {
+    for f in [
+        "/tests/does_not_exist.sql",
+        "/tests/does_not_exist.html",
+        "/tests/does_not_exist/",
+    ] {
+        let resp_result = req_path(f).await;
+        let resp = resp_result.unwrap();
+        assert_eq!(resp.status(), http::StatusCode::OK, "{f} isnt 200");
+
+        let body = test::read_body(resp).await;
+        assert!(body.starts_with(b"<!DOCTYPE html>"));
+        // the body should contain our happy string, but not the string "error"
+        let body = String::from_utf8(body.to_vec()).unwrap();
+        assert!(body.contains("But the "));
+        assert!(body.contains("404.sql"));
+        assert!(body.contains("file saved the day!"));
+        assert!(!body.contains("error"));
+    }
+}
+
 #[actix_web::test]
 async fn test_concurrent_requests() {
     // send 32 requests (less than the default postgres pool size)

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+SELECT 'alert' AS component,`
	`2`	`+ 'We almost got an oopsie' AS title,`
	`3`	+ 'But the `404.sql` file saved the day!' AS description_md;