You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
2.4 KiB
Go
106 lines
2.4 KiB
Go
package gateway
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/core/conf"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"github.com/zeromicro/go-zero/internal/mock"
|
|
"github.com/zeromicro/go-zero/rest/httpc"
|
|
"github.com/zeromicro/go-zero/zrpc"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/reflection"
|
|
"google.golang.org/grpc/test/bufconn"
|
|
)
|
|
|
|
func init() {
|
|
logx.Disable()
|
|
}
|
|
|
|
func dialer() func(context.Context, string) (net.Conn, error) {
|
|
listener := bufconn.Listen(1024 * 1024)
|
|
server := grpc.NewServer()
|
|
mock.RegisterDepositServiceServer(server, &mock.DepositServer{})
|
|
|
|
reflection.Register(server)
|
|
|
|
go func() {
|
|
if err := server.Serve(listener); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
return func(context.Context, string) (net.Conn, error) {
|
|
return listener.Dial()
|
|
}
|
|
}
|
|
|
|
func TestMustNewServer(t *testing.T) {
|
|
var c GatewayConf
|
|
assert.NoError(t, conf.FillDefault(&c))
|
|
// avoid popup alert on macos for asking permissions
|
|
c.DevServer.Host = "localhost"
|
|
c.Host = "localhost"
|
|
c.Port = 18881
|
|
|
|
s := MustNewServer(c, withDialer(func(conf zrpc.RpcClientConf) zrpc.Client {
|
|
return zrpc.MustNewClient(conf, zrpc.WithDialOption(grpc.WithContextDialer(dialer())))
|
|
}))
|
|
s.upstreams = []Upstream{
|
|
{
|
|
Mappings: []RouteMapping{
|
|
{
|
|
Method: "get",
|
|
Path: "/deposit/:amount",
|
|
RpcPath: "mock.DepositService/Deposit",
|
|
},
|
|
},
|
|
Grpc: zrpc.RpcClientConf{
|
|
Endpoints: []string{"foo"},
|
|
Timeout: 1000,
|
|
Middlewares: zrpc.ClientMiddlewaresConf{
|
|
Trace: true,
|
|
Duration: true,
|
|
Prometheus: true,
|
|
Breaker: true,
|
|
Timeout: true,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
assert.NoError(t, s.build())
|
|
go s.Server.Start()
|
|
|
|
time.Sleep(time.Millisecond * 200)
|
|
|
|
resp, err := httpc.Do(context.Background(), http.MethodGet, "http://localhost:18881/deposit/100", nil)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
|
|
resp, err = httpc.Do(context.Background(), http.MethodGet, "http://localhost:18881/deposit_fail/100", nil)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, http.StatusNotFound, resp.StatusCode)
|
|
}
|
|
|
|
func TestServer_ensureUpstreamNames(t *testing.T) {
|
|
var s = Server{
|
|
upstreams: []Upstream{
|
|
{
|
|
Grpc: zrpc.RpcClientConf{
|
|
Target: "target",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
assert.NoError(t, s.ensureUpstreamNames())
|
|
assert.Equal(t, "target", s.upstreams[0].Name)
|
|
}
|