| 1720 | |
| 1721 | #[tokio::test] |
| 1722 | async fn p2_cli_serve_echo_env() -> Result<()> { |
| 1723 | let server = WasmtimeServe::new(P2_CLI_SERVE_ECHO_ENV_COMPONENT, |cmd| { |
| 1724 | cmd.arg("--env=FOO=bar"); |
| 1725 | cmd.arg("--env=BAR"); |
| 1726 | cmd.arg("-Scli"); |
| 1727 | cmd.env_remove("BAR"); |
| 1728 | })?; |
| 1729 | |
| 1730 | let foo_env = server |
| 1731 | .send_request( |
| 1732 | hyper::Request::builder() |
| 1733 | .uri("http://localhost/") |
| 1734 | .header("env", "FOO") |
| 1735 | .body(String::new()) |
| 1736 | .context("failed to make request")?, |
| 1737 | ) |
| 1738 | .await?; |
| 1739 | |
| 1740 | assert!(foo_env.status().is_success()); |
| 1741 | assert!(foo_env.body().is_empty()); |
| 1742 | let headers = foo_env.headers(); |
| 1743 | assert_eq!(headers.get("env"), Some(&HeaderValue::from_static("bar"))); |
| 1744 | |
| 1745 | let bar_env = server |
| 1746 | .send_request( |
| 1747 | hyper::Request::builder() |
| 1748 | .uri("http://localhost/") |
| 1749 | .header("env", "BAR") |
| 1750 | .body(String::new()) |
| 1751 | .context("failed to make request")?, |
| 1752 | ) |
| 1753 | .await?; |
| 1754 | |
| 1755 | assert!(bar_env.status().is_success()); |
| 1756 | assert!(bar_env.body().is_empty()); |
| 1757 | let headers = bar_env.headers(); |
| 1758 | assert_eq!(headers.get("env"), None); |
| 1759 | |
| 1760 | server.finish()?; |
| 1761 | Ok(()) |
| 1762 | } |
| 1763 | |
| 1764 | #[tokio::test] |
| 1765 | async fn p2_cli_serve_header_replaces_request_header() -> Result<()> { |