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,
                   {