Merge pull request #16 from Till--H/feature/head

Add support for HEAD requests
ducaale · Feb 6, 2021 · c049e25 · c049e25
2 parents 23014db + 2488176
commit c049e25
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 0 deletions.
diff --git a/src/cli.rs b/src/cli.rs
@@ -121,6 +121,7 @@ impl Cli {
 #[derive(Debug, Clone, Copy)]
 pub enum Method {
     GET,
+    HEAD,
     POST,
     PUT,
     PATCH,
@@ -132,6 +133,7 @@ impl FromStr for Method {
     fn from_str(s: &str) -> Result<Method> {
         match s.to_ascii_uppercase().as_str() {
             "GET" => Ok(Method::GET),
+            "HEAD" => Ok(Method::HEAD),
             "POST" => Ok(Method::POST),
             "PUT" => Ok(Method::PUT),
             "PATCH" => Ok(Method::PATCH),
@@ -150,6 +152,7 @@ impl From<Method> for reqwest::Method {
     fn from(method: Method) -> Self {
         match method {
             Method::GET => reqwest::Method::GET,
+            Method::HEAD => reqwest::Method::HEAD,
             Method::POST => reqwest::Method::POST,
             Method::PUT => reqwest::Method::PUT,
             Method::PATCH => reqwest::Method::PATCH,

diff --git a/tests/cli.rs b/tests/cli.rs
@@ -30,3 +30,48 @@ fn basic_post() -> Result<(), Box<dyn std::error::Error>> {
 
     Ok(())
 }
+
+#[test]
+fn basic_get() -> Result<(), Box<dyn std::error::Error>> {
+    let mut cmd = Command::cargo_bin("ht")?;
+    cmd.arg("-v")
+        .arg("--offline")
+        .arg("--ignore-stdin")
+        .arg("--pretty=format")
+        .arg("get")
+        .arg("httpbin.org/get");
+
+    cmd.assert().stdout(indoc! {r#"
+        GET /get HTTP/1.1
+        accept: */*
+        accept-encoding: gzip, deflate
+        connection: keep-alive
+        host: httpbin.org
+
+    "#});
+
+    Ok(())
+}
+
+#[test]
+fn basic_head() -> Result<(), Box<dyn std::error::Error>> {
+    let mut cmd = Command::cargo_bin("ht")?;
+    cmd.arg("-v")
+        .arg("--offline")
+        .arg("--ignore-stdin")
+        .arg("--pretty=format")
+        .arg("head")
+        .arg("httpbin.org/get");
+
+    cmd.assert().stdout(indoc! {r#"
+        HEAD /get HTTP/1.1
+        accept: */*
+        accept-encoding: gzip, deflate
+        connection: keep-alive
+        host: httpbin.org
+
+    "#});
+
+    Ok(())
+}
+