feat: add test for 404 fallback mechanism

sqlpage · Aug 22, 2024 · d5cbee8 · d5cbee8
1 parent 3ac80fb
commit d5cbee8
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
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)