feat: Simple extraction of Content-Type header is now possible throug…

…h helper method

internal/http/http.go

@@ -28,17 +28,6 @@ type Client interface {
 	Get(string) (Response, error)
 }
 
-type Response struct {
-	StatusLine StatusLine
-	Headers    map[string]string
-}
-
-type StatusLine struct {
-	HttpVersion  string
-	StatusCode   uint16
-	ReasonPhrase string
-}
-
 type HttpClient struct {
 	net.Conn
 }

internal/http/response.go

@@ -0,0 +1,27 @@
+package http
+
+import "errors"
+
+type Response struct {
+	StatusLine StatusLine
+	Headers    map[string]string
+}
+
+type StatusLine struct {
+	HttpVersion  string
+	StatusCode   uint16
+	ReasonPhrase string
+}
+
+var ErrNoContentTypefound = errors.New("no content type")
+
+func (r Response) ContentType() (string, error) {
+
+	res, ok := r.Headers["Content-Type"]
+
+	if !ok {
+		return "", ErrNoContentTypefound
+	}
+
+	return res, nil
+}

internal/http/response_test.go

@@ -0,0 +1,34 @@
+package http
+
+import "testing"
+
+func TestContentType(t *testing.T) {
+
+	t.Run("application/json is set", func(t *testing.T) {
+		resp := Response{
+			Headers: map[string]string{
+				"Content-Type": "application/json",
+			},
+		}
+
+		content_type, err := resp.ContentType()
+
+		if err != nil {
+			t.Errorf("unexpected error occured: %s", err)
+		}
+
+		if content_type != "application/json" {
+			t.Errorf("got %s want %s", content_type, "application/json")
+		}
+	})
+
+	t.Run("header is not set", func(t *testing.T) {
+		resp := Response{}
+
+		_, err := resp.ContentType()
+
+		if err != ErrNoContentTypefound {
+			t.Errorf("got %s want %s", err, ErrNoContentTypefound)
+		}
+	})
+}