mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-22 18:17:52 -05:00
d9181ad84a
* Add instance management based testing * Add testing for meek transport * Add testing for grpc, httpupgrade transport
92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
package scenarios
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/v2fly/v2ray-core/v5/common"
|
|
"github.com/v2fly/v2ray-core/v5/testing/servers/tcp"
|
|
|
|
_ "github.com/v2fly/v2ray-core/v5/main/distro/all"
|
|
)
|
|
|
|
func TestGRPCDefault(t *testing.T) {
|
|
tcpServer := tcp.Server{
|
|
MsgProcessor: xor,
|
|
}
|
|
dest, err := tcpServer.Start()
|
|
common.Must(err)
|
|
defer tcpServer.Close()
|
|
|
|
coreInst, InstMgrIfce := NewInstanceManagerCoreInstance()
|
|
defer coreInst.Close()
|
|
|
|
common.Must(InstMgrIfce.AddInstance(
|
|
context.TODO(),
|
|
"grpc_client",
|
|
common.Must2(os.ReadFile("config/grpc_client.json")).([]byte),
|
|
"jsonv5"))
|
|
|
|
common.Must(InstMgrIfce.AddInstance(
|
|
context.TODO(),
|
|
"grpc_server",
|
|
common.Must2(os.ReadFile("config/grpc_server.json")).([]byte),
|
|
"jsonv5"))
|
|
|
|
common.Must(InstMgrIfce.StartInstance(context.TODO(), "grpc_server"))
|
|
common.Must(InstMgrIfce.StartInstance(context.TODO(), "grpc_client"))
|
|
|
|
defer func() {
|
|
common.Must(InstMgrIfce.StopInstance(context.TODO(), "grpc_server"))
|
|
common.Must(InstMgrIfce.StopInstance(context.TODO(), "grpc_client"))
|
|
common.Must(InstMgrIfce.UntrackInstance(context.TODO(), "grpc_server"))
|
|
common.Must(InstMgrIfce.UntrackInstance(context.TODO(), "grpc_client"))
|
|
coreInst.Close()
|
|
}()
|
|
|
|
if err := testTCPConnViaSocks(17784, dest.Port, 1024, time.Second*2)(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestGRPCWithServiceName(t *testing.T) {
|
|
tcpServer := tcp.Server{
|
|
MsgProcessor: xor,
|
|
}
|
|
dest, err := tcpServer.Start()
|
|
common.Must(err)
|
|
defer tcpServer.Close()
|
|
|
|
coreInst, InstMgrIfce := NewInstanceManagerCoreInstance()
|
|
defer coreInst.Close()
|
|
|
|
common.Must(InstMgrIfce.AddInstance(
|
|
context.TODO(),
|
|
"grpc_client",
|
|
common.Must2(os.ReadFile("config/grpc_servicename_client.json")).([]byte),
|
|
"jsonv5"))
|
|
|
|
common.Must(InstMgrIfce.AddInstance(
|
|
context.TODO(),
|
|
"grpc_server",
|
|
common.Must2(os.ReadFile("config/grpc_servicename_server.json")).([]byte),
|
|
"jsonv5"))
|
|
|
|
common.Must(InstMgrIfce.StartInstance(context.TODO(), "grpc_server"))
|
|
common.Must(InstMgrIfce.StartInstance(context.TODO(), "grpc_client"))
|
|
|
|
defer func() {
|
|
common.Must(InstMgrIfce.StopInstance(context.TODO(), "grpc_server"))
|
|
common.Must(InstMgrIfce.StopInstance(context.TODO(), "grpc_client"))
|
|
common.Must(InstMgrIfce.UntrackInstance(context.TODO(), "grpc_server"))
|
|
common.Must(InstMgrIfce.UntrackInstance(context.TODO(), "grpc_client"))
|
|
coreInst.Close()
|
|
}()
|
|
|
|
if err := testTCPConnViaSocks(17794, dest.Port, 1024, time.Second*2)(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|