replace zxq.co/ripple/hanayo

vendor/github.com/valyala/fasthttp/expvarhandler/expvar_test.go

@@ -0,0 +1,67 @@
+package expvarhandler
+
+import (
+	"encoding/json"
+	"expvar"
+	"strings"
+	"testing"
+
+	"github.com/valyala/fasthttp"
+)
+
+func TestExpvarHandlerBasic(t *testing.T) {
+	expvar.Publish("customVar", expvar.Func(func() interface{} {
+		return "foobar"
+	}))
+
+	var ctx fasthttp.RequestCtx
+
+	expvarHandlerCalls.Set(0)
+
+	ExpvarHandler(&ctx)
+
+	body := ctx.Response.Body()
+
+	var m map[string]interface{}
+	if err := json.Unmarshal(body, &m); err != nil {
+		t.Fatalf("unexpected error: %s", err)
+	}
+
+	if _, ok := m["cmdline"]; !ok {
+		t.Fatalf("cannot locate cmdline expvar")
+	}
+	if _, ok := m["memstats"]; !ok {
+		t.Fatalf("cannot locate memstats expvar")
+	}
+
+	v := m["customVar"]
+	sv, ok := v.(string)
+	if !ok {
+		t.Fatalf("unexpected custom var type %T. Expecting string", v)
+	}
+	if sv != "foobar" {
+		t.Fatalf("unexpected custom var value: %q. Expecting %q", v, "foobar")
+	}
+
+	v = m["expvarHandlerCalls"]
+	fv, ok := v.(float64)
+	if !ok {
+		t.Fatalf("unexpected expvarHandlerCalls type %T. Expecting float64", v)
+	}
+	if int(fv) != 1 {
+		t.Fatalf("unexpected value for expvarHandlerCalls: %v. Expecting %v", fv, 1)
+	}
+}
+
+func TestExpvarHandlerRegexp(t *testing.T) {
+	var ctx fasthttp.RequestCtx
+	ctx.QueryArgs().Set("r", "cmd")
+	ExpvarHandler(&ctx)
+	body := string(ctx.Response.Body())
+	if !strings.Contains(body, `"cmdline"`) {
+		t.Fatalf("missing 'cmdline' expvar")
+	}
+	if strings.Contains(body, `"memstats"`) {
+		t.Fatalf("unexpected memstats expvar found")
+	}
+}