Merge pull request #107 from fermyon/replace-body

Correctly implement new-request.
fermyon · Jul 11, 2024 · 23ca498 · 23ca498
2 parents 8d44bd6 + 1d31834
commit 23ca498
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/crates/spin-test-virt/src/wasi/http.rs b/crates/spin-test-virt/src/wasi/http.rs
@@ -164,7 +164,7 @@ pub struct OutgoingRequest {
     pub authority: RefCell<Option<String>>,
     pub path_with_query: RefCell<Option<String>>,
     pub headers: Fields,
-    body: Consumable<OutgoingBody>,
+    pub body: Consumable<OutgoingBody>,
 }
 
 impl exports::types::GuestOutgoingRequest for OutgoingRequest {
@@ -259,6 +259,12 @@ impl OutgoingBody {
     }
 }
 
+impl From<IncomingBody> for OutgoingBody {
+    fn from(i: IncomingBody) -> Self {
+        Self(i.0)
+    }
+}
+
 impl exports::types::GuestOutgoingBody for OutgoingBody {
     fn write(&self) -> Result<io::exports::streams::OutputStream, ()> {
         Ok(io::exports::streams::OutputStream::new(
@@ -274,7 +280,7 @@ impl exports::types::GuestOutgoingBody for OutgoingBody {
     }
 }
 
-#[derive(Clone)]
+#[derive(Clone, Debug)]
 pub struct IncomingBody(io::Buffer);
 
 impl IncomingBody {

diff --git a/crates/spin-test-virt/src/wasi/http_helper.rs b/crates/spin-test-virt/src/wasi/http_helper.rs
@@ -34,7 +34,10 @@ impl exports::Guest for Component {
             authority,
             path_with_query,
             headers,
-            body: body.unwrap_or_else(IncomingBody::empty).into(),
+            // Either override the body with `incoming_body`, or use the body from the original request
+            body: body
+                .map(Into::into)
+                .unwrap_or_else(|| request.body.unconsume().map(Into::into)),
         })
     }