chore: 并入 iotlcm-sms 历史到 sms/（原提交 1d38bc4）

2026-11-02 13:28:02 +08:00
parent b10b8cfc51 1d38bc4226
commit 691b792391
34 changed files with 2776 additions and 0 deletions
@@ -0,0 +1,88 @@
+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()
+	}
+}