Fix bug with ":" symbol in http headers (#29)

Add support for `:` in http headers.

Closes: #28

.gitignore

@@ -1,3 +1,6 @@
+# IDE
+.vscode
+
 # Ignore vendor to reduce repo size
 vendor
 

README.md

@@ -1,6 +1,6 @@
 # Minigun
 
-[![Minigun Version](https://img.shields.io/badge/Minigun-0.4.1-7f187f.svg)](https://github.com/wayfair-incubator/minigun/blob/main/CHANGELOG.md)
+[![Minigun Version](https://img.shields.io/badge/Minigun-0.4.2-7f187f.svg)](https://github.com/wayfair-incubator/minigun/blob/main/CHANGELOG.md)
 [![Contributor Covenant](https://img.shields.io/badge/Contributor%20Covenant-2.0-4baaaa.svg)](CODE_OF_CONDUCT.md)
 
 ## About The Project

main.go

@@ -35,7 +35,7 @@ import (
 )
 
 // Constants and vars
-const version = "0.4.1"
+const version = "0.4.2"
 const workersCannelSize = 1024
 const errorBadHTTPCode = "Bad HTTP status code"
 
@@ -120,7 +120,7 @@ func (headers *httpHeaders) Set(value string) error {
 	var tmpHeaders httpHeaders
 
 	s := strings.Split(value, ":")
-	if len(s) != 2 {
+	if len(s) < 2 {
 		return fmt.Errorf("Wrong header argument")
 	}
 
@@ -130,7 +130,10 @@ func (headers *httpHeaders) Set(value string) error {
 		tmpHeaders = *headers
 	}
 
-	tmpHeaders[s[0]] = s[1]
+	tmpHeader := strings.TrimSpace(s[0])
+	tmpValue := strings.TrimSpace(strings.Join(s[1:], ":"))
+
+	tmpHeaders[tmpHeader] = tmpValue
 	*headers = tmpHeaders
 
 	return nil

main_test.go

@@ -17,3 +17,29 @@ func TestRandomBytes(t *testing.T) {
 			len(result))
 	}
 }
+
+func TestHeadersParsing(t *testing.T) {
+	var sendHTTPHeaders httpHeaders
+
+	sendHTTPHeaders.Set("Host: hostname.local")
+	sendHTTPHeaders.Set("H1  : V1,V2  ")
+	sendHTTPHeaders.Set("Authorization: id:hash")
+	sendHTTPHeaders.Set("Keep-Alive: timeout=2, max=100")
+	sendHTTPHeaders.Set("Content-Type: application/xml")
+	sendHTTPHeaders.Set("Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8   ")
+
+	expected := httpHeaders{
+		"Authorization": "id:hash",
+		"H1":            "V1,V2",
+		"Host":          "hostname.local",
+		"Keep-Alive":    "timeout=2, max=100",
+		"Content-Type":  "application/xml",
+		"Accept":        "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8",
+	}
+
+	for k, v := range expected {
+		if sendHTTPHeaders[k] != v {
+			t.Errorf("Expected header: %q => %q, got header:  %q => %q", k, v, k, sendHTTPHeaders[k])
+		}
+	}
+}