open-vault/http/testing.go

package http

import (
	"fmt"
	"net"
	"net/http"
	"net/http/cookiejar"
	"testing"
	"time"

	"github.com/hashicorp/vault/vault"
)

func TestListener(t *testing.T) (net.Listener, string) {
	fail := func(format string, args ...interface{}) {
		panic(fmt.Sprintf(format, args...))
	}
	if t != nil {
		fail = t.Fatalf
	}

	ln, err := net.Listen("tcp", "127.0.0.1:0")
	if err != nil {
		fail("err: %s", err)
	}
	addr := "http://" + ln.Addr().String()
	return ln, addr
}

func TestServerWithListener(t *testing.T, ln net.Listener, addr string, core *vault.Core) {
	// Create a muxer to handle our requests so that we can authenticate
	// for tests.
	mux := http.NewServeMux()
	mux.Handle("/_test/auth", http.HandlerFunc(testHandleAuth))
	mux.Handle("/", Handler(core))

	server := &http.Server{
		Addr:    ln.Addr().String(),
		Handler: mux,
	}
	go server.Serve(ln)
}

func TestServer(t *testing.T, core *vault.Core) (net.Listener, string) {
	ln, addr := TestListener(t)
	TestServerWithListener(t, ln, addr, core)
	return ln, addr
}

func TestServerAuth(t *testing.T, addr string, token string) {
	// If no cookie jar is set on the default HTTP client, then setup the jar
	if http.DefaultClient.Jar == nil {
		jar, err := cookiejar.New(&cookiejar.Options{})
		if err != nil {
			t.Fatalf("err: %s", err)
		}

		http.DefaultClient.Jar = jar
	}

	// Get the internal path so that we set the cookie
	if _, err := http.Get(addr + "/_test/auth?token=" + token); err != nil {
		t.Fatalf("error authenticating: %s", err)
	}
}

func testHandleAuth(w http.ResponseWriter, req *http.Request) {
	token := req.URL.Query().Get("token")
	http.SetCookie(w, &http.Cookie{
		Name:    AuthCookieName,
		Value:   token,
		Path:    "/",
		Expires: time.Now().UTC().Add(1 * time.Hour),
	})

	respondOk(w, nil)
}