test: add test cases for serve

Signed-off-by: Frank Yang <[email protected]>

Author: FrankYang0529

Makefile

@@ -0,0 +1,11 @@
+.PHONY: test
+test: lint test-unit
+
+.PHONY: lint 
+lint:
+	cargo clippy --all-targets --all-features -- -D warnings
+	cargo fmt --all -- --check
+
+.PHONY: test-unit
+test-unit:
+	RUST_LOG=$(LOG_LEVEL) cargo test --all --no-fail-fast --target=$$(rustc -vV | sed -n 's|host: ||p') -- --nocapture --include-ignored

readme.md

@@ -58,3 +58,17 @@ Compiling the component:
 ```shell
 $ cargo build --release
 ```
+
+### Testing
+
+Prerequisites:
+
+- [Rust](https://www.rust-lang.org/) at
+  [1.56+](https://www.rust-lang.org/tools/install) with the `wasm32-wasi` target
+  configured
+
+Running test cases:
+
+```shell
+$ make test
+```

src/lib.rs

@@ -27,7 +27,7 @@ const BROTLI_ENCODING: &str = "br";
 const PATH_INFO_HEADER: &str = "spin-path-info";
 
 /// Common Content Encodings
-#[derive(PartialEq)]
+#[derive(PartialEq, Debug)]
 pub enum ContentEncoding {
     Brotli,
     //Deflate, // Could use flate2 for this
@@ -156,3 +156,115 @@ impl FileServer {
         state.finish().to_string()
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use http::header::{ACCEPT_ENCODING, IF_NONE_MATCH};
+
+    use super::*;
+
+    #[test]
+    fn test_best_encoding_none() {
+        let req = http::Request::builder()
+            .uri("http://thisistest.com")
+            .body(Some(bytes::Bytes::default()))
+            .unwrap();
+        let enc = ContentEncoding::best_encoding(&req).unwrap();
+        assert_eq!(enc, ContentEncoding::None);
+    }
+
+    #[test]
+    fn test_best_encoding_not_br() {
+        let req = http::Request::builder()
+            .uri("http://thisistest.com")
+            .header(ACCEPT_ENCODING, "gzip")
+            .body(Some(bytes::Bytes::default()))
+            .unwrap();
+        let enc = ContentEncoding::best_encoding(&req).unwrap();
+        assert_eq!(enc, ContentEncoding::None);
+    }
+
+    #[test]
+    fn test_best_encoding_with_br() {
+        let req = http::Request::builder()
+            .uri("http://thisistest.com")
+            .header(ACCEPT_ENCODING, "gzip,br")
+            .body(Some(bytes::Bytes::default()))
+            .unwrap();
+        let enc = ContentEncoding::best_encoding(&req).unwrap();
+        assert_eq!(enc, ContentEncoding::Brotli);
+    }
+
+    #[test]
+    fn test_serve_file_found() {
+        let req = spin_http::Request {
+            method: spin_http::Method::Get,
+            uri: "http://thisistest.com".to_string(),
+            headers: vec![(
+                PATH_INFO_HEADER.to_string(),
+                "./content/hello-test.txt".to_string(),
+            )],
+            params: vec![],
+            body: None,
+        };
+        let rsp = <super::SpinHttp as spin_http::SpinHttp>::handle_http_request(req);
+        assert_eq!(rsp.status, 200);
+    }
+
+    #[test]
+    fn test_serve_with_etag() {
+        let req = spin_http::Request {
+            method: spin_http::Method::Get,
+            uri: "http://thisistest.com".to_string(),
+            headers: vec![
+                (
+                    PATH_INFO_HEADER.to_string(),
+                    "./content/hello-test.txt".to_string(),
+                ),
+                (
+                    IF_NONE_MATCH.to_string(),
+                    "13946318585003701156".to_string(),
+                ),
+            ],
+            params: vec![],
+            body: None,
+        };
+        let rsp = <super::SpinHttp as spin_http::SpinHttp>::handle_http_request(req);
+        assert_eq!(rsp.status, 304);
+    }
+
+    #[test]
+    fn test_serve_with_not_matched_etag() {
+        let req = spin_http::Request {
+            method: spin_http::Method::Get,
+            uri: "http://thisistest.com".to_string(),
+            headers: vec![
+                (
+                    PATH_INFO_HEADER.to_string(),
+                    "./content/hello-test.txt".to_string(),
+                ),
+                (IF_NONE_MATCH.to_string(), "".to_string()),
+            ],
+            params: vec![],
+            body: None,
+        };
+        let rsp = <super::SpinHttp as spin_http::SpinHttp>::handle_http_request(req);
+        assert_eq!(rsp.status, 200);
+    }
+
+    #[test]
+    fn test_serve_file_not_found() {
+        let req = spin_http::Request {
+            method: spin_http::Method::Get,
+            uri: "http://thisistest.com".to_string(),
+            headers: vec![(
+                PATH_INFO_HEADER.to_string(),
+                "not-existent-file".to_string(),
+            )],
+            params: vec![],
+            body: None,
+        };
+        let rsp = <super::SpinHttp as spin_http::SpinHttp>::handle_http_request(req);
+        assert_eq!(rsp.status, 404);
+    }
+}