MiFaZhan
89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
package grpcapi
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net"
|
|
"testing"
|
|
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
"google.golang.org/grpc/status"
|
|
"google.golang.org/grpc/test/bufconn"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-sms/api/pb"
|
|
"git.5iot.com/iotlcm-public/iotlcm-sms/internal/sms"
|
|
)
|
|
|
|
type fakeSender struct {
|
|
config sms.Config
|
|
message sms.Message
|
|
result sms.Result
|
|
}
|
|
|
|
func (f *fakeSender) Send(_ context.Context, config sms.Config, message sms.Message) (sms.Result, error) {
|
|
f.config = config
|
|
f.message = message
|
|
return f.result, nil
|
|
}
|
|
|
|
func TestSend(t *testing.T) {
|
|
fake := &fakeSender{result: sms.Result{Code: 0, Message: "发送成功"}}
|
|
client, closeServer := newTestClient(t, fake)
|
|
defer closeServer()
|
|
|
|
config, _ := json.Marshal(sms.Config{ServiceType: sms.ServiceAliyun})
|
|
params, _ := json.Marshal(sms.Message{PhoneNumber: "18600001858", MsgContent: `{"code":"123456"}`})
|
|
response, err := client.Send(context.Background(), &pb.SendReq{
|
|
Head: &pb.HeadReq{RouterId: 42, TraceId: "trace-1"},
|
|
Config: config,
|
|
Params: params,
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if response.Head.Code != 0 || response.Head.RouterId != 42 || response.Head.TraceId != "trace-1" {
|
|
t.Fatalf("unexpected response: %+v", response.Head)
|
|
}
|
|
if fake.config.ServiceType != sms.ServiceAliyun || fake.message.PhoneNumber != "18600001858" {
|
|
t.Fatalf("unexpected request: config=%+v message=%+v", fake.config, fake.message)
|
|
}
|
|
}
|
|
|
|
func TestSendRejectsInvalidJSON(t *testing.T) {
|
|
client, closeServer := newTestClient(t, &fakeSender{})
|
|
defer closeServer()
|
|
|
|
_, err := client.Send(context.Background(), &pb.SendReq{Config: []byte(`{`), Params: []byte(`{}`)})
|
|
if status.Code(err) != codes.InvalidArgument {
|
|
t.Fatalf("code = %v, want %v; err=%v", status.Code(err), codes.InvalidArgument, err)
|
|
}
|
|
}
|
|
|
|
func newTestClient(t *testing.T, sender Sender) (pb.NoticeClient, func()) {
|
|
t.Helper()
|
|
listener := bufconn.Listen(1024 * 1024)
|
|
server := grpc.NewServer()
|
|
pb.RegisterNoticeServer(server, NewServer(sender))
|
|
go func() {
|
|
_ = server.Serve(listener)
|
|
}()
|
|
|
|
conn, err := grpc.NewClient(
|
|
"passthrough:///bufnet",
|
|
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
|
grpc.WithContextDialer(func(context.Context, string) (net.Conn, error) {
|
|
return listener.Dial()
|
|
}),
|
|
)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return pb.NewNoticeClient(conn), func() {
|
|
_ = conn.Close()
|
|
server.Stop()
|
|
_ = listener.Close()
|
|
}
|
|
}
|