add test

keyserve/keyserve_test.go

@@ -0,0 +1,86 @@
+package keyserve
+
+import (
+	"context"
+	"crypto/ecdsa"
+	"crypto/elliptic"
+	"crypto/rand"
+	"encoding/json"
+	"log"
+	"net/http"
+	"os"
+	"testing"
+	"time"
+
+	keypairs "github.com/big-squid/go-keypairs"
+)
+
+func TestServeKeys(t *testing.T) {
+	key, _ := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
+	pubs := []keypairs.PublicKey{
+		keypairs.NewPublicKey(key.Public()),
+	}
+
+	addr := ":62017"
+	done := make(chan bool, 1)
+
+	h := &http.Server{Addr: addr, Handler: &server{Middleware: Middleware{Keys: pubs}}}
+
+	logger := log.New(os.Stdout, "", 0)
+
+	go func() {
+		logger.Printf("Listening on http://0.0.0.0%s\n", addr)
+
+		if err := h.ListenAndServe(); err != nil {
+			// TODO check for the non-error server closed error
+			//logger.Fatal(err)
+		}
+
+		done <- true
+	}()
+
+	go func() {
+		time.Sleep(15 * time.Second)
+		ctx, _ := context.WithTimeout(context.Background(), 15*time.Second)
+		h.Shutdown(ctx)
+	}()
+
+	m := map[string]string{}
+	resp, err := http.Get("http://localhost" + addr + "/.well-known/openid-configuration")
+	if nil != err {
+		panic(err)
+	}
+	dec := json.NewDecoder(resp.Body)
+	err = dec.Decode(&m)
+	if nil != err {
+		panic(err)
+	}
+
+	n := struct {
+		Keys []map[string]interface{} `json:"keys"`
+	}{
+		Keys: []map[string]interface{}{},
+	}
+	resp, err = http.Get(m["jwks_uri"])
+	if nil != err {
+		panic(err)
+	}
+	dec = json.NewDecoder(resp.Body)
+	err = dec.Decode(&n)
+	if nil != err {
+		panic(err)
+	}
+	h.Shutdown(context.Background())
+
+	<-done
+}
+
+type server struct {
+	Middleware Middleware
+}
+
+func (s *server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	if !s.Middleware.Handler(w, r) {
+		w.Write([]byte("Try .well-known/openid-configuration or .well-known/jwks.json"))
+	}
+}