From 9c5edc5c7ae67c5a36ebdf36106d92569830210f Mon Sep 17 00:00:00 2001
From: Peter Borkuti <peterborkuti@gmail.com>
Date: Sat, 4 Mar 2023 21:19:01 +0100
Subject: [PATCH] add test for basic-lambda (#551)

---
 examples/basic-lambda/Cargo.toml  |  1 +
 examples/basic-lambda/src/main.rs | 24 ++++++++++++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/examples/basic-lambda/Cargo.toml b/examples/basic-lambda/Cargo.toml
index c2c8b429..fd6bd5b2 100644
--- a/examples/basic-lambda/Cargo.toml
+++ b/examples/basic-lambda/Cargo.toml
@@ -16,3 +16,4 @@ serde = "1.0.136"
 tokio = { version = "1", features = ["macros"] }
 tracing = { version = "0.1", features = ["log"] }
 tracing-subscriber = { version = "0.3", default-features = false, features = ["ansi", "fmt"] }
+tokio-test = "0.4.2"
\ No newline at end of file
diff --git a/examples/basic-lambda/src/main.rs b/examples/basic-lambda/src/main.rs
index bdf39112..2bb4aeb3 100644
--- a/examples/basic-lambda/src/main.rs
+++ b/examples/basic-lambda/src/main.rs
@@ -54,3 +54,27 @@ pub(crate) async fn my_handler(event: LambdaEvent<Request>) -> Result<Response,
     // return `Response` (it will be serialized to JSON automatically by the runtime)
     Ok(resp)
 }
+
+#[cfg(test)]
+mod tests {
+    use crate::{my_handler, Request};
+    use lambda_runtime::{Context, LambdaEvent};
+
+    #[tokio::test]
+    async fn response_is_good_for_simple_input() {
+        let id = "ID";
+
+        let mut context = Context::default();
+        context.request_id = id.to_string();
+
+        let payload = Request {
+            command: "X".to_string(),
+        };
+        let event = LambdaEvent { payload, context };
+
+        let result = my_handler(event).await.unwrap();
+
+        assert_eq!(result.msg, "Command X executed.");
+        assert_eq!(result.req_id, id.to_string());
+    }
+}