Merge pull request #8 from sreeram77/dev

Dev

client.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"net/http"
 	netURL "net/url"
+	"time"
 )
 
 // client is a wrapper around http.Client
@@ -17,19 +18,36 @@ func New() Client {
 	return Client{client: http.DefaultClient}
 }
 
+// NewClientWithTimeout returns an instance of Client with a Timeout
+// Timeout for DefaultClient is 0, which means the connection never times out
+func NewClientWithTimeout(t time.Duration) Client {
+	c := http.DefaultClient
+	c.Timeout = t
+
+	return Client{client: c}
+}
+
 // Get sends a GET request
 func (c Client) Get(url string, headers map[string]string, params map[string]interface{}) (*http.Response, error) {
+	req, err := generateRequest(url, http.MethodPost, headers, params, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return c.client.Do(req)
+}
 
-	req, err := GenerateRequest(url, http.MethodGet, headers, params, nil)
+// Post sends a POST request
+func (c Client) Post(url string, headers map[string]string, params map[string]interface{}, body []byte) (*http.Response, error) {
+	req, err := generateRequest(url, http.MethodGet, headers, params, body)
 	if err != nil {
 		return nil, err
 	}
 
 	return c.client.Do(req)
 }
 
-// GenerateRequest creates an HTTP request based on params
-func GenerateRequest(url, method string, headers map[string]string, params map[string]interface{}, body []byte) (*http.Request, error) {
+func generateRequest(url, method string, headers map[string]string, params map[string]interface{}, body []byte) (*http.Request, error) {
 	req, err := http.NewRequest(http.MethodGet, url, bytes.NewBuffer(body))
 	if err != nil {
 		return nil, err

client_test.go

@@ -35,3 +35,36 @@ func Test_Get(t *testing.T) {
 		}
 	}
 }
+
+func Test_GenerateRequest(t *testing.T) {
+	type testCase struct {
+		id     int
+		url    string
+		method string
+		header map[string]string
+		body   []byte
+		param  map[string]interface{}
+	}
+
+	testCases := []testCase{
+		{
+			id:     1,
+			url:    "https://google.com",
+			method: http.MethodGet,
+			header: nil,
+			body:   nil,
+			param:  nil,
+		},
+	}
+
+	for _, tc := range testCases {
+		req, err := generateRequest(tc.url, tc.method, tc.header, tc.param, tc.body)
+		if err != nil {
+			t.Error(err)
+		}
+
+		if !reflect.DeepEqual(req.Method, tc.method) {
+			t.Errorf("testcase: %v Expected Output: %v Actual Output: %v", tc.id, tc.method, req.Method)
+		}
+	}
+}