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.
102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
package zrpc
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/tal-tech/go-zero/core/logx"
|
|
"github.com/tal-tech/go-zero/zrpc/internal/mock"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
"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{})
|
|
|
|
go func() {
|
|
if err := server.Serve(listener); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
return func(context.Context, string) (net.Conn, error) {
|
|
return listener.Dial()
|
|
}
|
|
}
|
|
|
|
func TestDepositServer_Deposit(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
amount float32
|
|
res *mock.DepositResponse
|
|
errCode codes.Code
|
|
errMsg string
|
|
}{
|
|
{
|
|
"invalid request with negative amount",
|
|
-1.11,
|
|
nil,
|
|
codes.InvalidArgument,
|
|
fmt.Sprintf("cannot deposit %v", -1.11),
|
|
},
|
|
{
|
|
"valid request with non negative amount",
|
|
0.00,
|
|
&mock.DepositResponse{Ok: true},
|
|
codes.OK,
|
|
"",
|
|
},
|
|
}
|
|
|
|
directClient := MustNewClient(RpcClientConf{
|
|
Endpoints: []string{"foo"},
|
|
App: "foo",
|
|
Token: "bar",
|
|
Timeout: 1000,
|
|
}, WithDialOption(grpc.WithInsecure()), WithDialOption(grpc.WithContextDialer(dialer())))
|
|
targetClient, err := NewClientWithTarget("foo", WithDialOption(grpc.WithInsecure()),
|
|
WithDialOption(grpc.WithContextDialer(dialer())))
|
|
assert.Nil(t, err)
|
|
clients := []Client{
|
|
directClient,
|
|
targetClient,
|
|
}
|
|
for _, tt := range tests {
|
|
for _, client := range clients {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
cli := mock.NewDepositServiceClient(client.Conn())
|
|
request := &mock.DepositRequest{Amount: tt.amount}
|
|
response, err := cli.Deposit(context.Background(), request)
|
|
if response != nil {
|
|
assert.True(t, len(response.String()) > 0)
|
|
if response.GetOk() != tt.res.GetOk() {
|
|
t.Error("response: expected", tt.res.GetOk(), "received", response.GetOk())
|
|
}
|
|
}
|
|
if err != nil {
|
|
if e, ok := status.FromError(err); ok {
|
|
if e.Code() != tt.errCode {
|
|
t.Error("error code: expected", codes.InvalidArgument, "received", e.Code())
|
|
}
|
|
if e.Message() != tt.errMsg {
|
|
t.Error("error message: expected", tt.errMsg, "received", e.Message())
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
}
|