add test for basic-lambda (#551) (#612)

Author: peterborkuti

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"

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());
+    }
+}