From 9bf309f3f81c2ab43eee355b73d1b8ffe2846033 Mon Sep 17 00:00:00 2001
From: Doug Davis <dug@us.ibm.com>
Date: Fri, 28 Aug 2015 11:54:44 -0700
Subject: [PATCH] More httputil tests for the docker header

follow-on to #15911

Signed-off-by: Doug Davis <dug@us.ibm.com>
---
 httputils/httputils.go      |  9 ++---
 httputils/httputils_test.go | 72 ++++++++++++++++++++++---------------
 2 files changed, 49 insertions(+), 32 deletions(-)

diff --git a/httputils/httputils.go b/httputils/httputils.go
index 0fd90ea..d7dc438 100644
--- a/httputils/httputils.go
+++ b/httputils/httputils.go
@@ -5,12 +5,13 @@ import (
 	"fmt"
 	"net/http"
 	"regexp"
+	"strings"
 
 	"github.com/docker/docker/pkg/jsonmessage"
 )
 
 var (
-	headerRegexp     = regexp.MustCompile(`^(?:(.+)/(.+))\s\((.+)\).*$`)
+	headerRegexp     = regexp.MustCompile(`^(?:(.+)/(.+?))\((.+)\).*$`)
 	errInvalidHeader = errors.New("Bad header, should be in format `docker/version (platform)`")
 )
 
@@ -48,8 +49,8 @@ func ParseServerHeader(hdr string) (*ServerHeader, error) {
 		return nil, errInvalidHeader
 	}
 	return &ServerHeader{
-		App: matches[1],
-		Ver: matches[2],
-		OS:  matches[3],
+		App: strings.TrimSpace(matches[1]),
+		Ver: strings.TrimSpace(matches[2]),
+		OS:  strings.TrimSpace(matches[3]),
 	}, nil
 }
diff --git a/httputils/httputils_test.go b/httputils/httputils_test.go
index 61e98ba..41281ec 100644
--- a/httputils/httputils_test.go
+++ b/httputils/httputils_test.go
@@ -24,36 +24,52 @@ func TestNewHTTPRequestError(t *testing.T) {
 }
 
 func TestParseServerHeader(t *testing.T) {
-	if _, err := ParseServerHeader("bad header"); err != errInvalidHeader {
-		t.Fatalf("Should fail when header can not be parsed")
+	inputs := map[string][]string{
+		"bad header":           {"error"},
+		"(bad header)":         {"error"},
+		"(without/spaces)":     {"error"},
+		"(header/with spaces)": {"error"},
+		"foo/bar (baz)":        {"foo", "bar", "baz"},
+		"foo/bar":              {"error"},
+		"foo":                  {"error"},
+		"foo/bar (baz space)":           {"foo", "bar", "baz space"},
+		"  f  f  /  b  b  (  b  s  )  ": {"f  f", "b  b", "b  s"},
+		"foo/bar (baz) ignore":          {"foo", "bar", "baz"},
+		"foo/bar ()":                    {"error"},
+		"foo/bar()":                     {"error"},
+		"foo/bar(baz)":                  {"foo", "bar", "baz"},
+		"foo/bar/zzz(baz)":              {"foo/bar", "zzz", "baz"},
+		"foo/bar(baz/abc)":              {"foo", "bar", "baz/abc"},
+		"foo/bar(baz (abc))":            {"foo", "bar", "baz (abc)"},
 	}
 
-	if _, err := ParseServerHeader("(bad header)"); err != errInvalidHeader {
-		t.Fatalf("Should fail when header can not be parsed")
+	for header, values := range inputs {
+		serverHeader, err := ParseServerHeader(header)
+		if err != nil {
+			if err != errInvalidHeader {
+				t.Fatalf("Failed to parse %q, and got some unexpected error: %q", header, err)
+			}
+			if values[0] == "error" {
+				continue
+			}
+			t.Fatalf("Header %q failed to parse when it shouldn't have", header)
+		}
+		if values[0] == "error" {
+			t.Fatalf("Header %q parsed ok when it should have failed(%q).", header, serverHeader)
+		}
+
+		if serverHeader.App != values[0] {
+			t.Fatalf("Expected serverHeader.App for %q to equal %q, got %q", header, values[0], serverHeader.App)
+		}
+
+		if serverHeader.Ver != values[1] {
+			t.Fatalf("Expected serverHeader.Ver for %q to equal %q, got %q", header, values[1], serverHeader.Ver)
+		}
+
+		if serverHeader.OS != values[2] {
+			t.Fatalf("Expected serverHeader.OS for %q to equal %q, got %q", header, values[2], serverHeader.OS)
+		}
+
 	}
 
-	if _, err := ParseServerHeader("(without/spaces)"); err != errInvalidHeader {
-		t.Fatalf("Should fail when header can not be parsed")
-	}
-
-	if _, err := ParseServerHeader("(header/with space)"); err != errInvalidHeader {
-		t.Fatalf("Expected err to not exist when ParseServerHeader(\"(header/with space)\")")
-	}
-
-	serverHeader, err := ParseServerHeader("foo/bar (baz)")
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if serverHeader.App != "foo" {
-		t.Fatalf("Expected serverHeader.App to equal \"foo\", got %s", serverHeader.App)
-	}
-
-	if serverHeader.Ver != "bar" {
-		t.Fatalf("Expected serverHeader.Ver to equal \"bar\", got %s", serverHeader.Ver)
-	}
-
-	if serverHeader.OS != "baz" {
-		t.Fatalf("Expected serverHeader.OS to equal \"baz\", got %s", serverHeader.OS)
-	}
 }