阶段性提交

internal/rpc/services/service_admin_test.go

@@ -0,0 +1,80 @@
+package services
+
+import (
+	"context"
+	"encoding/base64"
+	teaconst "github.com/TeaOSLab/EdgeAPI/internal/const"
+	"github.com/TeaOSLab/EdgeAPI/internal/encrypt"
+	"github.com/TeaOSLab/EdgeAPI/internal/rpc/pb"
+	"github.com/iwind/TeaGo/assert"
+	"github.com/iwind/TeaGo/maps"
+	stringutil "github.com/iwind/TeaGo/utils/string"
+	"google.golang.org/grpc/metadata"
+	"testing"
+	"time"
+)
+
+func TestAdminService_Login(t *testing.T) {
+	a := assert.NewAssertion(t)
+	service := &AdminService{
+		debug: true,
+	}
+	resp, err := service.Login(testCtx(t), &pb.AdminLoginRequest{
+		Username: "admin",
+		Password: stringutil.Md5("123456"),
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	a.LogJSON(resp)
+}
+
+func TestAdminService_CreateLog(t *testing.T) {
+	service := &AdminService{debug: true}
+
+	resp, err := service.CreateLog(testCtx(t), &pb.AdminCreateLogRequest{
+		Level:       "info",
+		Description: "这是一个测试日志",
+		Action:      "/login",
+		Ip:          "127.0.0.1",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(resp)
+}
+
+func TestAdminService_FindAllEnabledClusters(t *testing.T) {
+	service := &AdminService{
+		debug: true,
+	}
+	resp, err := service.FindAllEnabledClusters(testCtx(t), &pb.AdminFindAllEnabledClustersRequest{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(resp)
+}
+
+func testCtx(t *testing.T) context.Context {
+	ctx := context.Background()
+	nodeId := "H6sjDf779jimnVPnBFSgZxvr6Ca0wQ0z"
+
+	token := maps.Map{
+		"timestamp": time.Now().Unix(),
+		"adminId":   1,
+	}
+	data := token.AsJSON()
+
+	method, err := encrypt.NewMethodInstance(teaconst.EncryptMethod, "hMHjmEng0SIcT3yiA3HIoUjogwAC9cur", nodeId)
+	if err != nil {
+		t.Fatal(err)
+	}
+	data, err = method.Encrypt(data)
+	if err != nil {
+		t.Fatal(err)
+	}
+	tokenString := base64.StdEncoding.EncodeToString(data)
+
+	ctx = metadata.AppendToOutgoingContext(ctx, "nodeId", nodeId, "token", tokenString)
+	return ctx
+}