78 lines
1.7 KiB
Go
78 lines
1.7 KiB
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)
|
|
}
|