From 22f7a7c65d2fce58b6cb1e32865a800473c4b196 Mon Sep 17 00:00:00 2001
From: Paul-Christian Volkmer <code@pcvolkmer.de>
Date: Tue, 26 Mar 2024 18:11:54 +0100
Subject: [PATCH] test: add tests for response generation

---
 src/main.rs | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index 5735d49..7a96447 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -154,3 +154,26 @@ fn error_response() -> Response {
         .body(Body::empty())
         .expect("response built")
 }
+
+#[cfg(test)]
+mod tests {
+    use axum::http::StatusCode;
+    use uuid::Uuid;
+
+    use crate::{error_response, success_response};
+
+    #[test]
+    fn should_return_success_response() {
+        let response = success_response(&Uuid::new_v4().to_string());
+        assert_eq!(response.status(), StatusCode::ACCEPTED);
+        assert!(response.headers().contains_key("x-request-id"));
+    }
+
+    #[test]
+    fn should_return_error_response() {
+        let response = error_response();
+        assert_eq!(response.status(), StatusCode::INTERNAL_SERVER_ERROR);
+        assert_eq!(response.headers().contains_key("x-request-id"), false);
+    }
+
+}
\ No newline at end of file