gamegroup/src/modules/assets/assets.service.spec.ts

import { Test, TestingModule } from '@nestjs/testing';
import { getRepositoryToken } from '@nestjs/typeorm';
import { Repository, DataSource } from 'typeorm';
import { AssetsService } from './assets.service';
import { Asset } from '../../entities/asset.entity';
import { AssetLog } from '../../entities/asset-log.entity';
import { Group } from '../../entities/group.entity';
import { GroupMember } from '../../entities/group-member.entity';
import { AssetType, AssetStatus, GroupMemberRole, AssetLogAction } from '../../common/enums';
import { NotFoundException, ForbiddenException, BadRequestException } from '@nestjs/common';

describe('AssetsService', () => {
  let service: AssetsService;
  let assetRepository: Repository<Asset>;
  let assetLogRepository: Repository<AssetLog>;
  let groupRepository: Repository<Group>;
  let groupMemberRepository: Repository<GroupMember>;

  const mockAsset = {
    id: 'asset-1',
    groupId: 'group-1',
    type: AssetType.ACCOUNT,
    name: '测试账号',
    description: '测试描述',
    accountCredentials: 'encrypted-data',
    quantity: 1,
    status: AssetStatus.AVAILABLE,
    currentBorrowerId: null,
    createdAt: new Date(),
    updatedAt: new Date(),
  };

  const mockGroup = {
    id: 'group-1',
    name: '测试小组',
  };

  const mockGroupMember = {
    id: 'member-1',
    userId: 'user-1',
    groupId: 'group-1',
    role: GroupMemberRole.ADMIN,
  };

  const mockDataSource = {
    createQueryRunner: jest.fn().mockReturnValue({
      connect: jest.fn(),
      startTransaction: jest.fn(),
      commitTransaction: jest.fn(),
      rollbackTransaction: jest.fn(),
      release: jest.fn(),
      manager: {
        findOne: jest.fn(),
        find: jest.fn(),
        create: jest.fn(),
        save: jest.fn(),
      },
    }),
  };

  beforeEach(async () => {
    const module: TestingModule = await Test.createTestingModule({
      providers: [
        AssetsService,
        {
          provide: getRepositoryToken(Asset),
          useValue: {
            create: jest.fn(),
            save: jest.fn(),
            find: jest.fn(),
            findOne: jest.fn(),
            remove: jest.fn(),
          },
        },
        {
          provide: getRepositoryToken(AssetLog),
          useValue: {
            create: jest.fn(),
            save: jest.fn(),
            find: jest.fn(),
          },
        },
        {
          provide: getRepositoryToken(Group),
          useValue: {
            findOne: jest.fn(),
          },
        },
        {
          provide: getRepositoryToken(GroupMember),
          useValue: {
            findOne: jest.fn(),
          },
        },
        {
          provide: DataSource,
          useValue: mockDataSource,
        },
      ],
    }).compile();

    service = module.get<AssetsService>(AssetsService);
    assetRepository = module.get<Repository<Asset>>(getRepositoryToken(Asset));
    assetLogRepository = module.get<Repository<AssetLog>>(getRepositoryToken(AssetLog));
    groupRepository = module.get<Repository<Group>>(getRepositoryToken(Group));
    groupMemberRepository = module.get<Repository<GroupMember>>(getRepositoryToken(GroupMember));
  });

  it('should be defined', () => {
    expect(service).toBeDefined();
  });

  describe('create', () => {
    it('应该成功创建资产', async () => {
      const createDto = {
        groupId: 'group-1',
        type: AssetType.ACCOUNT,
        name: '测试账号',
        description: '测试描述',
      };

      jest.spyOn(groupRepository, 'findOne').mockResolvedValue(mockGroup as any);
      jest.spyOn(groupMemberRepository, 'findOne').mockResolvedValue(mockGroupMember as any);
      jest.spyOn(assetRepository, 'create').mockReturnValue(mockAsset as any);
      jest.spyOn(assetRepository, 'save').mockResolvedValue(mockAsset as any);
      jest.spyOn(assetRepository, 'findOne').mockResolvedValue(mockAsset as any);

      const result = await service.create('user-1', createDto);

      expect(result).toBeDefined();
      expect(groupRepository.findOne).toHaveBeenCalledWith({ where: { id: 'group-1' } });
      expect(groupMemberRepository.findOne).toHaveBeenCalled();
    });

    it('小组不存在时应该抛出异常', async () => {
      const createDto = {
        groupId: 'group-1',
        type: AssetType.ACCOUNT,
        name: '测试账号',
      };

      jest.spyOn(groupRepository, 'findOne').mockResolvedValue(null);

      await expect(service.create('user-1', createDto)).rejects.toThrow(NotFoundException);
    });

    it('无权限时应该抛出异常', async () => {
      const createDto = {
        groupId: 'group-1',
        type: AssetType.ACCOUNT,
        name: '测试账号',
      };

      jest.spyOn(groupRepository, 'findOne').mockResolvedValue(mockGroup as any);
      jest.spyOn(groupMemberRepository, 'findOne').mockResolvedValue({
        ...mockGroupMember,
        role: GroupMemberRole.MEMBER,
      } as any);

      await expect(service.create('user-1', createDto)).rejects.toThrow(ForbiddenException);
    });
  });

  describe('findAll', () => {
    it('应该返回资产列表', async () => {
      jest.spyOn(assetRepository, 'find').mockResolvedValue([mockAsset] as any);

      const result = await service.findAll('group-1');

      expect(result).toHaveLength(1);
      expect(result[0].accountCredentials).toBeUndefined();
    });
  });

  describe('borrow', () => {
    it('应该成功借用资产', async () => {
      const borrowDto = { reason: '需要使用' };

      jest.spyOn(assetRepository, 'findOne').mockResolvedValue(mockAsset as any);
      jest.spyOn(groupMemberRepository, 'findOne').mockResolvedValue(mockGroupMember as any);
      jest.spyOn(assetRepository, 'save').mockResolvedValue({ ...mockAsset, status: AssetStatus.IN_USE } as any);
      jest.spyOn(assetLogRepository, 'create').mockReturnValue({} as any);
      jest.spyOn(assetLogRepository, 'save').mockResolvedValue({} as any);

      const result = await service.borrow('user-1', 'asset-1', borrowDto);

      expect(result.message).toBe('借用成功');
      expect(assetRepository.save).toHaveBeenCalled();
      expect(assetLogRepository.save).toHaveBeenCalled();
    });

    it('资产不可用时应该抛出异常', async () => {
      const borrowDto = { reason: '需要使用' };

      jest.spyOn(assetRepository, 'findOne').mockResolvedValue({
        ...mockAsset,
        status: AssetStatus.IN_USE,
      } as any);

      await expect(service.borrow('user-1', 'asset-1', borrowDto)).rejects.toThrow(BadRequestException);
    });
  });

  describe('return', () => {
    it('应该成功归还资产', async () => {
      jest.spyOn(assetRepository, 'findOne').mockResolvedValue({
        ...mockAsset,
        currentBorrowerId: 'user-1',
      } as any);
      jest.spyOn(assetRepository, 'save').mockResolvedValue(mockAsset as any);
      jest.spyOn(assetLogRepository, 'create').mockReturnValue({} as any);
      jest.spyOn(assetLogRepository, 'save').mockResolvedValue({} as any);

      const result = await service.return('user-1', 'asset-1', '已归还');

      expect(result.message).toBe('归还成功');
      expect(assetRepository.save).toHaveBeenCalled();
    });

    it('非借用人归还时应该抛出异常', async () => {
      jest.spyOn(assetRepository, 'findOne').mockResolvedValue({
        ...mockAsset,
        currentBorrowerId: 'user-2',
      } as any);

      await expect(service.return('user-1', 'asset-1')).rejects.toThrow(ForbiddenException);
    });
  });

  describe('remove', () => {
    it('应该成功删除资产', async () => {
      jest.spyOn(assetRepository, 'findOne').mockResolvedValue(mockAsset as any);
      jest.spyOn(groupMemberRepository, 'findOne').mockResolvedValue(mockGroupMember as any);
      jest.spyOn(assetRepository, 'remove').mockResolvedValue(mockAsset as any);

      const result = await service.remove('user-1', 'asset-1');

      expect(result.message).toBe('删除成功');
      expect(assetRepository.remove).toHaveBeenCalled();
    });
  });
});