func TestMoreDefaultsBJSON(t *testing.T) {
seed := time.Now().UnixNano()
popr := math_rand.New(math_rand.NewSource(seed))
p := NewPopulatedMoreDefaultsB(popr, true)
marshaler := github_com_maditya_protobuf_jsonpb.Marshaler{}
jsondata, err := marshaler.MarshalToString(p)
if err != nil {
t.Fatalf("seed = %d, err = %v", seed, err)
}
msg := &MoreDefaultsB{}
err = github_com_maditya_protobuf_jsonpb.UnmarshalString(jsondata, msg)
if err != nil {
t.Fatalf("seed = %d, err = %v", seed, err)
}
if !p.Equal(msg) {
t.Fatalf("seed = %d, %#v !Json Equal %#v", seed, msg, p)
}
}
func TestEncodedSignedEntryUpdateJSON(t *testing.T) {
popr := math_rand.New(math_rand.NewSource(time.Now().UnixNano()))
p := NewPopulatedEncodedSignedEntryUpdate(popr, true)
marshaler := github_com_gogo_protobuf_jsonpb.Marshaler{}
jsondata, err := marshaler.MarshalToString(p)
if err != nil {
t.Fatal(err)
}
msg := &EncodedSignedEntryUpdate{}
err = github_com_gogo_protobuf_jsonpb.UnmarshalString(jsondata, msg)
if err != nil {
t.Fatal(err)
}
if err := p.VerboseEqual(msg); err != nil {
t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)
}
if !p.Equal(msg) {
t.Fatalf("%#v !Json Equal %#v", msg, p)
}
}
func TestWilsonJSON(t *testing.T) {
seed := time.Now().UnixNano()
popr := math_rand.New(math_rand.NewSource(seed))
p := NewPopulatedWilson(popr, true)
marshaler := github_com_maditya_protobuf_jsonpb.Marshaler{}
jsondata, err := marshaler.MarshalToString(p)
if err != nil {
t.Fatalf("seed = %d, err = %v", seed, err)
}
msg := &Wilson{}
err = github_com_maditya_protobuf_jsonpb.UnmarshalString(jsondata, msg)
if err != nil {
t.Fatalf("seed = %d, err = %v", seed, err)
}
if err := p.VerboseEqual(msg); err != nil {
t.Fatalf("seed = %d, %#v !VerboseProto %#v, since %v", seed, msg, p, err)
}
if !p.Equal(msg) {
t.Fatalf("seed = %d, %#v !Json Equal %#v", seed, msg, p)
}
}
func (h *HTTPFront) ServeHTTP(w http.ResponseWriter, r *http.Request) {
path := r.URL.Path
method := r.Method
// service healthcheck
if (method == "HEAD" || method == "GET") && (path == "/status" || path == "/lb") {
if !h.InRotation() {
http.Error(w, `server out of rotation`, http.StatusNotFound)
return
}
if method == "GET" {
w.Write([]byte("OK"))
}
return
}
if method == "GET" && path == "/saml" {
url, err := h.SAMLRequest()
if err != nil {
http.Error(w, err.Error(), http.StatusNotFound)
return
}
http.Redirect(w, r, url, http.StatusFound)
return
}
if method == "GET" && path == "/oidc" {
u, err := url.ParseQuery(r.URL.RawQuery)
if err != nil {
http.Error(w, `error parsing query string`, http.StatusBadRequest)
return
}
d := u.Get("domain")
if d == "" {
http.Error(w, `domain not found`, http.StatusBadRequest)
return
}
url, err := h.OIDCRequest(d, "https://"+r.Host+"/oidcsso")
if err != nil {
http.Error(w, err.Error(), http.StatusNotFound)
return
}
http.Redirect(w, r, url, http.StatusFound)
}
if method != "POST" || (path != "/lookup" && path != "/update") {
http.Error(w, `this server only supports queries of the POST /lookup or POST /update`, http.StatusNotFound)
return
}
pf := &proto.LookupProof{}
var err error
ctx := context.Background()
if path == "/lookup" {
pf, err = h.doLookup(r.Body, ctx)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
} else if path == "/update" {
pf, err = h.doUpdate(r.Body, ctx)
if err != nil {
status := http.StatusBadRequest
if h.IsAuthExpired(err) {
status = http.StatusUnauthorized
}
http.Error(w, err.Error(), status)
return
}
}
// preserve the original field name
marshaler := jsonpb.Marshaler{OrigName: true}
err = marshaler.Marshal(w, pf)
if err != nil {
http.Error(w, `Internal server error`, http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
return
}
