add HEAD method

limhyungseok · mrtazz · commit f47acbbedd97 · 2016-11-27T20:28:25.000-05:00
this closes mrtazz#63
diff --git a/README.md b/README.md
@@ -27,6 +27,7 @@ RestClient::Response r = RestClient::get("http://url.com")
 RestClient::Response r = RestClient::post("http://url.com/post", "text/json", "{\"foo\": \"bla\"}")
 RestClient::Response r = RestClient::put("http://url.com/put", "text/json", "{\"foo\": \"bla\"}")
 RestClient::Response r = RestClient::del("http://url.com/delete")
+RestClient::Response r = RestClient::head("http://url.com")
 ```
 
 The response is of type [RestClient::Response][restclient_response] and has
@@ -77,6 +78,7 @@ conn->AppendHeader("X-MY-HEADER", "foo")
 conn->SetCAInfoFilePath("/etc/custom-ca.crt")
 
 RestClient::Response r = conn->get("/get")
+RestClient::Response r = conn->head("/get")
 RestClient::Response r = conn->del("/delete")
 
 // set different content header for POST and PUT
diff --git a/include/restclient-cpp/connection.h b/include/restclient-cpp/connection.h
@@ -148,6 +148,7 @@ class Connection {
     RestClient::Response put(const std::string& uri,
                              const std::string& data);
     RestClient::Response del(const std::string& uri);
+    RestClient::Response head(const std::string& uri);
 
  private:
     CURL* curlHandle;
diff --git a/include/restclient-cpp/restclient.h b/include/restclient-cpp/restclient.h
@@ -58,6 +58,7 @@ Response put(const std::string& url,
               const std::string& content_type,
               const std::string& data);
 Response del(const std::string& url);
+Response head(const std::string& url);
 
 }  // namespace RestClient
 
diff --git a/source/connection.cc b/source/connection.cc
@@ -358,3 +358,21 @@ RestClient::Connection::del(const std::string& url) {
   return this->performCurlRequest(url);
 }
 
+/**
+ * @brief HTTP HEAD method
+ *
+ * @param url to query
+ *
+ * @return response struct
+ */
+RestClient::Response
+RestClient::Connection::head(const std::string& url) {
+    /** we want HTTP HEAD */
+    const char* http_head = "HEAD";
+
+    /** set HTTP HEAD METHOD */
+    curl_easy_setopt(this->curlHandle, CURLOPT_CUSTOMREQUEST, http_head);
+    curl_easy_setopt(this->curlHandle, CURLOPT_NOBODY, 1L);
+
+    return this->performCurlRequest(url);
+}
diff --git a/source/restclient.cc b/source/restclient.cc
@@ -108,3 +108,17 @@ RestClient::Response RestClient::del(const std::string& url) {
   return ret;
 }
 
+/**
+ * @brief HTTP HEAD method
+ *
+ * @param url to query
+ *
+ * @return response struct
+ */
+RestClient::Response RestClient::head(const std::string& url) {
+  RestClient::Response ret;
+  RestClient::Connection *conn = new RestClient::Connection("");
+  ret = conn->head(url);
+  delete conn;
+  return ret;
+}
diff --git a/test/test_connection.cc b/test/test_connection.cc
@@ -182,3 +182,18 @@ TEST_F(ConnectionTest, TestGetInfoFromRedirect)
   EXPECT_NE(0, info.lastRequest.redirectTime);
   EXPECT_NE(0, info.lastRequest.redirectCount);
 }
+
+TEST_F(ConnectionTest, TestHeadHeaders)
+{
+  RestClient::HeaderFields headers;
+  headers["Foo"] = "bar";
+  headers["Bla"] = "lol";
+  conn->SetHeaders(headers);
+  RestClient::Response res = conn->head("/headers");
+  EXPECT_EQ(200, res.code);
+  EXPECT_EQ("", res.body);
+
+  RestClient::HeaderFields headers_returned = conn->GetHeaders();
+  EXPECT_EQ("bar", headers_returned["Foo"]);
+  EXPECT_EQ("lol", headers_returned["Bla"]);
+}
diff --git a/test/test_restclient.cc b/test/test_restclient.cc
@@ -164,3 +164,9 @@ TEST_F(RestClientTest, TestRestClientAuth)
   res = RestClient::get("http://httpbin.org/basic-auth/foo/bar");
   EXPECT_EQ(401, res.code);
 }
+TEST_F(RestClientTest, TestRestClientHeadCode)
+{
+  RestClient::Response res = RestClient::head("http://httpbin.org/get");
+  EXPECT_EQ(200, res.code);
+  EXPECT_EQ("", res.body);
+}
