108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package scenarios
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"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 := io.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 := io.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")
|
|
}
|
|
}
|