1. TODO: guest_id 2. jwt的封装

server/home-user-auth/test/basic.go

@@ -0,0 +1,58 @@
+package logic_test
+
+import (
+	"flag"
+	"fmt"
+	"fusenapi/server/home-user-auth/internal/config"
+	"fusenapi/server/home-user-auth/internal/handler"
+	"fusenapi/server/home-user-auth/internal/svc"
+	"testing"
+
+	"github.com/474420502/requests"
+	"github.com/zeromicro/go-zero/core/conf"
+	"github.com/zeromicro/go-zero/rest"
+)
+
+var testConfigFile = flag.String("f", "../etc/home-user-auth.yaml", "the config file")
+var cnf config.Config
+var gserver *rest.Server
+
+func init() {
+	gserver = GetTestServer()
+}
+
+func GetTestServer() *rest.Server {
+	flag.Parse()
+
+	conf.MustLoad(*testConfigFile, &cnf)
+
+	server := rest.MustNewServer(cnf.RestConf)
+	defer server.Stop()
+
+	ctx := svc.NewServiceContext(cnf)
+	handler.RegisterHandlers(server, ctx)
+
+	fmt.Printf("Starting server at %s:%d...\n", cnf.Host, cnf.Port)
+	return server
+}
+
+func GetSesssionWithJwtToken(t *testing.T, server requests.ITestServer) *requests.Session {
+	ses := requests.NewSession()
+	tp := ses.Post(fmt.Sprintf("http://%s:%d/user/login", cnf.Host, cnf.Port))
+	tp.SetBodyJson(map[string]interface{}{
+		"name": "devenv@sina.cn",
+		"pwd":  "$2y$13$6UFDMZQMEfqFYiNLpiUCi.B3fpvGEamPAjIgzUqv/u7jT05nB3pOC",
+	})
+	resp, err := tp.TestInServer(server)
+	if err != nil {
+		t.Error(err)
+	}
+	result := resp.Json()
+	token := result.Get("data.token")
+	if !token.Exists() {
+		t.Error("data.token is not exists")
+	}
+	ses.Header.Add("Authorization", token.String())
+
+	return ses
+}

server/home-user-auth/test/useraddresslistlogic_test.go

@@ -1,57 +1,21 @@
-package server_test
+package logic_test
 
 import (
-	"flag"
 	"fmt"
-	"fusenapi/server/home-user-auth/internal/config"
-	"fusenapi/server/home-user-auth/internal/handler"
-	"fusenapi/server/home-user-auth/internal/svc"
 	"testing"
 
 	"github.com/474420502/requests"
-	"github.com/zeromicro/go-zero/core/conf"
-	"github.com/zeromicro/go-zero/rest"
+	"github.com/tidwall/gjson"
 )
 
-var testConfigFile = flag.String("f", "../etc/home-user-auth.yaml", "the config file")
-var cnf config.Config
-
-func GetTestServer() *rest.Server {
-	flag.Parse()
-
-	conf.MustLoad(*testConfigFile, &cnf)
-
-	server := rest.MustNewServer(cnf.RestConf)
-	defer server.Stop()
-
-	ctx := svc.NewServiceContext(cnf)
-	handler.RegisterHandlers(server, ctx)
-
-	fmt.Printf("Starting server at %s:%d...\n", cnf.Host, cnf.Port)
-	return server
-}
-
 func TestCaseAddressList(t *testing.T) {
-	server := GetTestServer()
+	var err error
+	var resp *requests.Response
+	var result gjson.Result
 
-	ses := requests.NewSession()
-	tp := ses.Post(fmt.Sprintf("http://%s:%d/user/login", cnf.Host, cnf.Port))
-	tp.SetBodyJson(map[string]interface{}{
-		"name": "devenv@sina.cn",
-		"pwd":  "$2y$13$6UFDMZQMEfqFYiNLpiUCi.B3fpvGEamPAjIgzUqv/u7jT05nB3pOC",
-	})
-	resp, err := tp.TestInServer(server)
-	if err != nil {
-		t.Error(err)
-	}
-	result := resp.Json()
-	token := result.Get("data.token")
-	if !token.Exists() {
-		t.Error("data.token is not exists")
-	}
-	ses.Header.Add("Authorization", token.String())
+	ses := GetSesssionWithJwtToken(t, gserver)
 
-	resp, err = ses.Get(fmt.Sprintf("http://%s:%d/user/address-list", cnf.Host, cnf.Port)).TestInServer(server)
+	resp, err = ses.Get(fmt.Sprintf("http://%s:%d/user/address-list", cnf.Host, cnf.Port)).TestInServer(gserver)
 	if err != nil {
 		t.Error(err)
 	}

server/home-user-auth/test/userbasicinfologic_test.go

@@ -0,0 +1 @@
+package logic_test