Only test received headers when it is required, improve syntax

server_test.go

@@ -60,20 +60,21 @@ func TestHttpReverseProxy(t *testing.T) {
 				response := httptest.NewRecorder()
 				reverseProxy.Handler.ServeHTTP(response, request)
 
-				got := header
+				gotHeader := header
 				header = nil
 
 				assertStatus(t, response.Code, test.wantResponseCode)
-				assertHeader(t, got, "X-Forwarded-Host", test.wantXForwardedHostHeader)
+				if len(test.wantXForwardedHostHeader) > 0 {
+					assertHeader(t, gotHeader, "X-Forwarded-Host", test.wantXForwardedHostHeader)
+				}
 			})
 	}
 }
 
 func assertHeader(t *testing.T, gotHeader map[string][]string, headerName, want string) {
 	t.Helper()
-	got := ""
-	lookup, ok := gotHeader[headerName]
-	if ok {
+	var got string
+	if lookup, ok := gotHeader[headerName]; ok {
 		got = lookup[0]
 	}
 	if got != want {