From 026318c54c13f8da8551e5155eceef17e1a962f3 Mon Sep 17 00:00:00 2001
From: jlizen <jackliebiz@gmail.com>
Date: Tue, 6 May 2025 02:38:06 +0000
Subject: [PATCH] fix(docs): examples/extension-internal-flush: don't refer to
 a standalone layer arn for internal extension deploy, add testing information

---
 examples/extension-internal-flush/README.md   | 18 +++++++++++++++++-
 examples/extension-internal-flush/src/main.rs |  2 ++
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/examples/extension-internal-flush/README.md b/examples/extension-internal-flush/README.md
index 553f7a3d..c6408676 100644
--- a/examples/extension-internal-flush/README.md
+++ b/examples/extension-internal-flush/README.md
@@ -18,9 +18,10 @@ the Lambda service returns the response to the caller immediately. Extensions ma
 without introducing an observable delay.
 
 ## Build & Deploy
+Two deploy options for internal extensions: 
 
 1. Install [cargo-lambda](https://github.com/cargo-lambda/cargo-lambda#installation)
-2. Build the extension with `cargo lambda build --release`
+2. Build a function with the internal extension embedded with `cargo lambda build --release`
 3. Deploy the function to AWS Lambda with `cargo lambda deploy --iam-role YOUR_ROLE`
 
 The last command will give you an ARN for the extension layer that you can use in your functions.
@@ -28,3 +29,18 @@ The last command will give you an ARN for the extension layer that you can use i
 ## Build for ARM 64
 
 Build the extension with `cargo lambda build --release --arm64`
+
+
+## Test your Function
+
+From your local:
+```
+cargo lambda watch
+# in new terminal window
+cargo lambda invoke --data-ascii '{"Records":[{"messageId":"MessageID_1","receiptHandle":"MessageReceiptHandle","body":"{\"a\":\"Test\",\"b\":123}"}]}'
+```
+
+If you have deployed to AWS using the commands in the 'Build & Deploy' section, you can also invoke your function remotely:
+```
+cargo lambda invoke extension-internal-flush --remote --data-ascii '{"Records":[{"messageId":"MessageID_1","receiptHandle":"MessageReceiptHandle","body":"{\"a\":\"Test\",\"b\":123}"}]}'
+```
\ No newline at end of file
diff --git a/examples/extension-internal-flush/src/main.rs b/examples/extension-internal-flush/src/main.rs
index ff1d10da..c728030b 100644
--- a/examples/extension-internal-flush/src/main.rs
+++ b/examples/extension-internal-flush/src/main.rs
@@ -101,6 +101,8 @@ async fn main() -> Result<(), Error> {
 
     let handler = Arc::new(EventHandler::new(request_done_sender));
 
+    // TODO: add biased! to always poll the handler future first, once supported:
+    // https://github.com/tokio-rs/tokio/issues/7304
     tokio::try_join!(
         lambda_runtime::run(service_fn(|event| {
             let handler = handler.clone();