package scenarios import ( "encoding/json" "fmt" "io/ioutil" "net/http" "testing" "github.com/xtls/xray-core/app/metrics" "github.com/xtls/xray-core/app/proxyman" "github.com/xtls/xray-core/app/router" "github.com/xtls/xray-core/common" "github.com/xtls/xray-core/common/net" "github.com/xtls/xray-core/common/serial" "github.com/xtls/xray-core/core" "github.com/xtls/xray-core/proxy/dokodemo" "github.com/xtls/xray-core/proxy/freedom" "github.com/xtls/xray-core/testing/servers/tcp" ) const expectedMessage = "goroutine profile: total" func TestMetrics(t *testing.T) { tcpServer := tcp.Server{ MsgProcessor: xor, } dest, err := tcpServer.Start() common.Must(err) defer tcpServer.Close() metricsPort := tcp.PickPort() clientConfig := &core.Config{ App: []*serial.TypedMessage{ serial.ToTypedMessage(&metrics.Config{ Tag: "metrics_out", }), serial.ToTypedMessage(&router.Config{ Rule: []*router.RoutingRule{ { InboundTag: []string{"metrics_in"}, TargetTag: &router.RoutingRule_Tag{ Tag: "metrics_out", }, }, }, }), }, Inbound: []*core.InboundHandlerConfig{ { Tag: "metrics_in", ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{ PortList: &net.PortList{Range: []*net.PortRange{net.SinglePortRange(metricsPort)}}, Listen: net.NewIPOrDomain(net.LocalHostIP), }), ProxySettings: serial.ToTypedMessage(&dokodemo.Config{ Address: net.NewIPOrDomain(dest.Address), Port: uint32(dest.Port), Networks: []net.Network{net.Network_TCP}, }), }, }, Outbound: []*core.OutboundHandlerConfig{ { Tag: "default-outbound", ProxySettings: serial.ToTypedMessage(&freedom.Config{}), }, }, } servers, err := InitializeServerConfigs(clientConfig) common.Must(err) defer CloseAllServers(servers) resp, err := http.Get(fmt.Sprintf("http://127.0.0.1:%d/debug/pprof/goroutine?debug=1", metricsPort)) common.Must(err) if resp == nil { t.Error("unexpected pprof nil response") } if resp.StatusCode != http.StatusOK { t.Error("unexpected pprof status code") } body, err := ioutil.ReadAll(resp.Body) if err != nil { t.Fatal(err) } if string(body)[0:len(expectedMessage)] != expectedMessage { t.Error("unexpected response body from pprof handler") } resp2, err2 := http.Get(fmt.Sprintf("http://127.0.0.1:%d/debug/vars", metricsPort)) common.Must(err2) if resp2 == nil { t.Error("unexpected expvars nil response") } if resp2.StatusCode != http.StatusOK { t.Error("unexpected expvars status code") } body2, err2 := ioutil.ReadAll(resp2.Body) if err2 != nil { t.Fatal(err2) } var json2 map[string]interface{} if json.Unmarshal(body2, &json2) != nil { t.Error("unexpected response body from expvars handler") } }