test: fix get user token

openimsdk · Sep 29, 2024 · 67e43eb · 67e43eb
1 parent aa37216
commit 67e43eb
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 6 deletions.
diff --git a/integration_test/internal/manager/manager.go b/integration_test/internal/manager/manager.go
@@ -56,9 +56,19 @@ func (m *MetaManager) GetSecret() string {
 	return m.secret
 }
 
-func (m *MetaManager) GetToken(userID string, platformID int32) (string, error) {
+func (m *MetaManager) GetAdminToken(userID string, platformID int32) (string, error) {
 	req := authPB.UserTokenReq{PlatformID: platformID, UserID: userID, Secret: m.secret}
 	resp := authPB.UserTokenResp{}
+	err := m.PostWithCtx(api.UsersToken.Route(), &req, &resp)
+	if err != nil {
+		return "", err
+	}
+	return resp.Token, nil
+}
+
+func (m *MetaManager) GetUserToken(userID string, platformID int32) (string, error) {
+	req := authPB.GetUserTokenReq{PlatformID: platformID, UserID: userID}
+	resp := authPB.GetUserTokenResp{}
 	err := m.PostWithCtx(api.GetUsersToken.Route(), &req, &resp)
 	if err != nil {
 		return "", err
@@ -67,7 +77,7 @@ func (m *MetaManager) GetToken(userID string, platformID int32) (string, error)
 }
 
 func (m *MetaManager) WithAdminToken() (err error) {
-	token, err := m.GetToken(config.AdminUserID, config.PlatformID)
+	token, err := m.GetAdminToken(config.AdminUserID, config.PlatformID)
 	if err != nil {
 		return err
 	}

diff --git a/integration_test/internal/manager/user_manager.go b/integration_test/internal/manager/user_manager.go
@@ -134,7 +134,7 @@ func (t *TestUserManager) login(ctx context.Context, userIDs ...string) error {
 	for _, userID := range userIDs {
 		userID := userID
 		gr.Go(func() error {
-			token, err := t.GetToken(userID, config.PlatformID)
+			token, err := t.GetAdminToken(userID, config.PlatformID)
 			if err != nil {
 				return err
 			}

diff --git a/msgtest/module/manager.go b/msgtest/module/manager.go
@@ -118,7 +118,7 @@ func (m *MetaManager) buildCtx() context.Context {
 func (m *MetaManager) getToken(userID string, platformID int32) (string, error) {
 	req := authPB.UserTokenReq{PlatformID: platformID, UserID: userID, Secret: m.secret}
 	resp := authPB.UserTokenResp{}
-	err := m.postWithCtx(api.GetUsersToken.Route(), &req, &resp)
+	err := m.postWithCtx(api.UsersToken.Route(), &req, &resp)
 	if err != nil {
 		return "", err
 	}

diff --git a/pkg/api/api.go b/pkg/api/api.go
@@ -95,7 +95,8 @@ var (
 )
 
 var (
-	GetUsersToken = newApi[auth.UserTokenReq, auth.UserTokenResp]("/auth/user_token")
+	UsersToken    = newApi[auth.UserTokenReq, auth.UserTokenResp]("/auth/user_token")
+	GetUsersToken = newApi[auth.UserTokenReq, auth.UserTokenResp]("/auth/get_user_token")
 )
 
 var (

diff --git a/test/init.go b/test/init.go
@@ -83,5 +83,5 @@ func GetUserToken(ctx context.Context, userID string, platformID int32, secret s
 		PlatformID: platformID,
 		Secret:     secret,
 	}
-	return api.ExtractField(ctx, api.GetUsersToken.Invoke, req, (*auth.UserTokenResp).GetToken)
+	return api.ExtractField(ctx, api.UsersToken.Invoke, req, (*auth.UserTokenResp).GetToken)
 }