weave-scope/vendor/github.com/crossdock/crossdock-go/client.go

// Copyright (c) 2016 Uber Technologies, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

package crossdock

import (
	"encoding/json"
	"log"
	"net/http"
	"net/url"
)

// BehaviorFunc executes a set of tests against T
type BehaviorFunc func(t T)

// Behaviors is a map of BehaviorFuncs to dispatch to
type Behaviors map[string]BehaviorFunc

// BehaviorParam is the url param representing the test to run
const BehaviorParam = "behavior"

// Start begins a blocking Crossdock client
func Start(behaviors Behaviors) {
	http.Handle("/", Handler(behaviors, false))
	log.Fatal(http.ListenAndServe(":8080", nil))
}

// Handler returns an http.Handler that dispatches to functions defined in behaviors map.
// If failOnUnknown is true, a behavior that is not in the behaviors map will cause the
// handler to return an error status. Otherwise, the handler returns skipped status.
func Handler(behaviors Behaviors, failOnUnknown bool) http.Handler {
	return requestHandler{behaviors: behaviors, failOnUnknown: failOnUnknown}
}

type requestHandler struct {
	behaviors     Behaviors
	failOnUnknown bool
}

func (h requestHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	if r.Method == "HEAD" {
		w.Header().Add("Content-Length", "0")
		return
	}
	params := extractParams(r.URL.Query())
	entries := Run(params, func(t T) {
		behavior := h.behaviors[t.Behavior()]
		if behavior == nil {
			if h.failOnUnknown {
				t.Errorf("unknown behavior %q", t.Behavior())
			} else {
				t.Skipf("unknown behavior %q", t.Behavior())
			}
			return
		}
		behavior(t)
	})

	enc := json.NewEncoder(w)
	if err := enc.Encode(entries); err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
	}
}

// extractParams returns a map of params from url values
func extractParams(p url.Values) Params {
	params := Params{}
	for k, l := range p {
		for _, v := range l {
			params[k] = v
		}
	}
	return params
}