From 7c6e863c8d2e3f200f047cd5cc446059d731f8af Mon Sep 17 00:00:00 2001
From: Junyi <mytharcher@users.noreply.github.com>
Date: Thu, 5 Dec 2024 16:24:25 +0800
Subject: [PATCH] fix(test): add roleName to login in mock server (#5794)

---
 packages/core/test/src/server/mock-server.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/packages/core/test/src/server/mock-server.ts b/packages/core/test/src/server/mock-server.ts
index 9f58cd5b45..3d1c030bd0 100644
--- a/packages/core/test/src/server/mock-server.ts
+++ b/packages/core/test/src/server/mock-server.ts
@@ -68,8 +68,8 @@ interface Resource {
 }
 
 interface ExtendedAgent extends SuperAgentTest {
-  login: (user: any) => ExtendedAgent;
-  loginUsingId: (userId: number) => ExtendedAgent;
+  login: (user: any, roleName?: string) => ExtendedAgent;
+  loginUsingId: (userId: number, roleName?: string) => ExtendedAgent;
   resource: (name: string, resourceOf?: any) => Resource;
 }
 
@@ -117,13 +117,14 @@ export class MockServer extends Application {
     const proxy = new Proxy(agent, {
       get(target, method: string, receiver) {
         if (['login', 'loginUsingId'].includes(method)) {
-          return (userOrId: any) => {
+          return (userOrId: any, roleName?: string) => {
             return proxy
               .auth(
                 jwt.sign(
                   {
                     userId: typeof userOrId === 'number' ? userOrId : userOrId?.id,
                     temp: true,
+                    roleName,
                   },
                   process.env.APP_KEY,
                   {