add user data test

ChangeLog.md

@@ -1,5 +1,8 @@
 Change Log 更新说明
 ------------------------------
+## 2021-07-05
+1. 完善测试用例，加入获取userdata测试用例
+
 ## 2020-11-30 v1.4.1
 1. Group Middleware加入登出放行逻辑；
 

example/sample/main.go

@@ -23,6 +23,7 @@ func main() {
 }
 
 var gfToken *gtoken.GfToken
+var gfAdminToken *gtoken.GfToken
 
 /*
 统一路由注册
@@ -41,7 +42,7 @@ func initRouter(s *ghttp.Server) {
 	// 认证接口
 	loginFunc := Login
 	// 启动gtoken
-	gfToken := &gtoken.GfToken{
+	gfToken = &gtoken.GfToken{
 		ServerName:       TestServerName,
 		LoginPath:        "/login",
 		LoginBeforeFunc:  loginFunc,
@@ -68,7 +69,7 @@ func initRouter(s *ghttp.Server) {
 	})
 
 	// 启动gtoken
-	gfAdminToken := &gtoken.GfToken{
+	gfAdminToken = &gtoken.GfToken{
 		ServerName: TestServerName,
 		//Timeout:         10 * 1000,
 		LoginPath:        "/login",

example/sample1/main.go

@@ -35,6 +35,10 @@ func initRouter(s *ghttp.Server) {
 		group.ALL("/hello", func(r *ghttp.Request) {
 			r.Response.WriteJson(gtoken.Succ("hello"))
 		})
+		// 获取登录扩展属性
+		group.ALL("/system/data", func(r *ghttp.Request) {
+			r.Response.WriteJson(gfToken.GetTokenData(r).Data)
+		})
 		group.ALL("/system/user", func(r *ghttp.Request) {
 			r.Response.WriteJson(gtoken.Succ("system user"))
 		})
@@ -48,7 +52,7 @@ func initRouter(s *ghttp.Server) {
 
 	loginFunc := Login
 	// 启动gtoken
-	gfToken := &gtoken.GfToken{
+	gfToken = &gtoken.GfToken{
 		ServerName: TestServerName,
 		//Timeout:         10 * 1000,
 		CacheMode:        g.Cfg().GetInt8("gToken.CacheMode"),

example/sample1/test/api_test.go

@@ -38,6 +38,31 @@ func TestHello(t *testing.T) {
 	}
 }
 
+func TestUserData(t *testing.T) {
+	// 登录，访问用户信息
+	t.Log("2. execute login and visit user")
+	data := Post(t, "/system/data", "username="+Username)
+	if data.Success() {
+		if data.DataString() == "1" {
+			t.Log("get user data success", data.Json())
+		} else {
+			t.Error("user data not eq 1 ", data.Json())
+		}
+	} else {
+		t.Error("error:", data.Json())
+	}
+
+	// 登出
+	t.Log("3. execute logout")
+	data = Post(t, "/user/logout", "username="+Username)
+	if data.Success() {
+		t.Log(data.Json())
+	} else {
+		t.Error("error:", data.Json())
+	}
+
+}
+
 func TestSystemUser(t *testing.T) {
 	// 未登录
 	t.Log("1. not login and visit user")