mini-starter/packages/user-module-mt/tests/integration/role.integration.test.ts

import { describe, it, expect, beforeAll, beforeEach, afterEach } from 'vitest';
import { DataSource } from 'typeorm';
import { RoleServiceMt } from '../../src/services/role.service.mt';
import { RoleMt } from '../../src/entities/role.entity';
import { AppDataSource, initializeDataSource } from '@d8d/shared-utils';

// 确保测试环境变量被设置
process.env.NODE_ENV = 'test';

describe('角色集成测试', () => {
  let dataSource: DataSource;
  let roleService: RoleServiceMt;

  beforeAll(() => {
    // 使用预先配置的数据源
    initializeDataSource([RoleMt])
    dataSource = AppDataSource;
  })

  beforeEach(async () => {
    if (!dataSource.isInitialized) {
      await dataSource.initialize();
    }
    roleService = new RoleServiceMt(dataSource);
  });

  afterEach(async () => {
    if (dataSource.isInitialized) {
      await dataSource.destroy();
    }
  });

  describe('角色CRUD操作', () => {
    it('应该创建并检索角色', async () => {
      // Create role
      const roleData = {
        name: 'admin',
        description: 'Administrator role',
        permissions: ['user:create', 'user:delete', 'role:manage'],
        tenantId: 1
      };

      const createdRole = await roleService.create(roleData);

      expect(createdRole.id).toBeDefined();
      expect(createdRole.name).toBe(roleData.name);
      expect(createdRole.description).toBe(roleData.description);
      expect(createdRole.permissions).toEqual(roleData.permissions);

      // Retrieve role
      const retrievedRole = await roleService.getById(createdRole.id);
      expect(retrievedRole).toBeDefined();
      expect(retrievedRole?.name).toBe(roleData.name);
    });

    it('应该更新角色信息', async () => {
      // Create role first
      const roleData = {
        name: 'updaterole',
        description: 'Role to be updated',
        permissions: ['user:read'],
        tenantId: 1
      };

      const createdRole = await roleService.create(roleData);

      // Update role
      const updateData = {
        description: 'Updated role description',
        permissions: ['user:read', 'user:update']
      };

      const updatedRole = await roleService.update(createdRole.id, updateData);

      expect(updatedRole).toBeDefined();
      expect(updatedRole?.description).toBe(updateData.description);
      expect(updatedRole?.permissions).toEqual(updateData.permissions);
    });

    it('应该删除角色', async () => {
      // Create role first
      const roleData = {
        name: 'deleterole',
        description: 'Role to be deleted',
        permissions: ['user:read'],
        tenantId: 1
      };

      const createdRole = await roleService.create(roleData);

      // Delete role
      const deleteResult = await roleService.delete(createdRole.id);
      expect(deleteResult).toBe(true);

      // Verify role is deleted
      const retrievedRole = await roleService.getById(createdRole.id);
      expect(retrievedRole).toBeNull();
    });

    it('应该根据角色名获取角色', async () => {
      const roleData = {
        name: 'specificrole',
        description: 'Specific role for testing',
        permissions: ['user:read'],
        tenantId: 1
      };

      await roleService.create(roleData);

      const foundRole = await roleService.getRoleByName('specificrole');
      expect(foundRole).toBeDefined();
      expect(foundRole?.name).toBe('specificrole');
    });
  });

  describe('角色列表操作', () => {
    it('应该获取分页角色列表', async () => {
      // Create multiple roles
      const rolesData = [
        { name: 'role1', description: 'Role 1', permissions: ['user:read'], tenantId: 1 },
        { name: 'role2', description: 'Role 2', permissions: ['user:read'], tenantId: 1 },
        { name: 'role3', description: 'Role 3', permissions: ['user:read'], tenantId: 1 }
      ];

      for (const roleData of rolesData) {
        await roleService.create(roleData);
      }

      const [roles, total] = await roleService.getList(1, 10);

      expect(total).toBe(3);
      expect(roles).toHaveLength(3);
      expect(roles.map(r => r.name)).toEqual(
        expect.arrayContaining(['role1', 'role2', 'role3'])
      );
    });

    it('应该根据角色名搜索角色', async () => {
      // Create roles with different names
      await roleService.create({ name: 'admin', description: 'Admin role', permissions: ['user:create'], tenantId: 1 });
      await roleService.create({ name: 'user', description: 'User role', permissions: ['user:read'], tenantId: 1 });
      await roleService.create({ name: 'moderator', description: 'Moderator role', permissions: ['user:update'], tenantId: 1 });

      const [adminRoles] = await roleService.getList(1, 10, 'admin', ['name']);
      expect(adminRoles).toHaveLength(1);
      expect(adminRoles[0].name).toBe('admin');

      const [userRoles] = await roleService.getList(1, 10, 'user', ['name']);
      expect(userRoles).toHaveLength(1);
      expect(userRoles[0].name).toBe('user');
    });
  });

  describe('权限检查', () => {
    it('应该检查角色是否拥有权限', async () => {
      const roleData = {
        name: 'permissionrole',
        description: 'Role with specific permissions',
        permissions: ['user:create', 'user:read', 'user:update'],
        tenantId: 1
      };

      const role = await roleService.create(roleData);

      // Check existing permissions
      expect(await roleService.hasPermission(role.id, 'user:create')).toBe(true);
      expect(await roleService.hasPermission(role.id, 'user:read')).toBe(true);
      expect(await roleService.hasPermission(role.id, 'user:update')).toBe(true);

      // Check non-existing permission
      expect(await roleService.hasPermission(role.id, 'user:delete')).toBe(false);
    });

    it('应该对不存在的角色返回false', async () => {
      const hasPermission = await roleService.hasPermission(999, 'user:create');
      expect(hasPermission).toBe(false);
    });
  });

  describe('角色验证', () => {
    it('应该允许不同租户使用相同的角色名', async () => {
      const roleData1 = {
        name: 'same_name',
        description: 'Role in tenant 1',
        permissions: ['user:read'],
        tenantId: 1
      };

      const roleData2 = {
        name: 'same_name',
        description: 'Role in tenant 2',
        permissions: ['user:read'],
        tenantId: 2
      };

      // Create role in tenant 1
      const role1 = await roleService.create(roleData1);
      expect(role1).toBeDefined();
      expect(role1.tenantId).toBe(1);

      // Create role with same name in tenant 2 (should work in multi-tenant)
      const role2 = await roleService.create(roleData2);
      expect(role2).toBeDefined();
      expect(role2.tenantId).toBe(2);
    });

    it('应该要求至少一个权限', async () => {
      const roleData = {
        name: 'nopermission',
        description: 'Role without permissions',
        permissions: [],
        tenantId: 1
      };

      // This should work as TypeORM handles empty arrays
      const role = await roleService.create(roleData);
      expect(role.permissions).toEqual([]);
    });
  });
});