1 use anyhow::Context;
2 use std::io::{self, Read};
3 use test_programs::p3::wasi::http::types::{Method, Scheme};
4 
5 struct Component;
6 
7 test_programs::p3::export!(Component);
8 
9 impl test_programs::p3::exports::wasi::cli::run::Guest for Component {
10     async fn run() -> Result<(), ()> {
11         // Make sure the final body is larger than 1024*1024, but we cannot allocate
12         // so much memory directly in the wasm program, so we use the `repeat`
13         // method to increase the body size.
14         const LEN: usize = 1024;
15         const REPEAT: usize = 1025;
16         let mut buffer = [0; LEN];
17         let addr = test_programs::p3::wasi::cli::environment::get_environment()
18             .into_iter()
19             .find_map(|(k, v)| k.eq("HTTP_SERVER").then_some(v))
20             .unwrap();
21         io::repeat(0b001).read_exact(&mut buffer).unwrap();
22         let res = test_programs::p3::http::request(
23             Method::Post,
24             Scheme::Http,
25             &addr,
26             "/post",
27             Some(&buffer.repeat(REPEAT)),
28             None,
29             None,
30             None,
31             None,
32         )
33         .await
34         .context("/post large")
35         .unwrap();
36 
37         println!("/post large: {}", res.status);
38         assert_eq!(res.status, 200);
39         let method = res.header("x-wasmtime-test-method").unwrap();
40         assert_eq!(std::str::from_utf8(method).unwrap(), "POST");
41         assert_eq!(res.body.len(), LEN * REPEAT);
42         Ok(())
43     }
44 }
45 
46 fn main() {}
47