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

import { Test, TestingModule } from '@nestjs/testing';
import { getRepositoryToken } from '@nestjs/typeorm';
import {
  NotFoundException,
  BadRequestException,
  ForbiddenException,
} from '@nestjs/common';
import { LedgersService } from './ledgers.service';
import { Ledger } from '../../entities/ledger.entity';
import { Group } from '../../entities/group.entity';
import { GroupMember } from '../../entities/group-member.entity';

enum LedgerType {
  INCOME = 'income',
  EXPENSE = 'expense',
}

describe('LedgersService', () => {
  let service: LedgersService;
  let mockLedgerRepository: any;
  let mockGroupRepository: any;
  let mockGroupMemberRepository: any;

  const mockUser = { id: 'user-1', username: 'testuser' };
  const mockGroup = { 
    id: 'group-1', 
    name: '测试小组', 
    isActive: true,
    parentId: null,
  };
  const mockMembership = {
    id: 'member-1',
    userId: 'user-1',
    groupId: 'group-1',
    role: 'member',
    isActive: true,
  };

  const mockLedger = {
    id: 'ledger-1',
    groupId: 'group-1',
    creatorId: 'user-1',
    type: LedgerType.INCOME,
    amount: 100,
    category: '聚餐费用',
    description: '周末聚餐',
    createdAt: new Date('2024-01-20T10:00:00Z'),
    updatedAt: new Date(),
  };

  beforeEach(async () => {
    mockLedgerRepository = {
      create: jest.fn(),
      save: jest.fn(),
      findOne: jest.fn(),
      remove: jest.fn(),
      createQueryBuilder: jest.fn(),
    };

    mockGroupRepository = {
      findOne: jest.fn(),
      find: jest.fn(),
    };

    mockGroupMemberRepository = {
      findOne: jest.fn(),
    };

    const module: TestingModule = await Test.createTestingModule({
      providers: [
        LedgersService,
        {
          provide: getRepositoryToken(Ledger),
          useValue: mockLedgerRepository,
        },
        {
          provide: getRepositoryToken(Group),
          useValue: mockGroupRepository,
        },
        {
          provide: getRepositoryToken(GroupMember),
          useValue: mockGroupMemberRepository,
        },
      ],
    }).compile();

    service = module.get<LedgersService>(LedgersService);
  });

  describe('create', () => {
    it('应该成功创建账目', async () => {
      mockGroupRepository.findOne.mockResolvedValue(mockGroup);
      mockGroupMemberRepository.findOne.mockResolvedValue(mockMembership);
      mockLedgerRepository.create.mockReturnValue(mockLedger);
      mockLedgerRepository.save.mockResolvedValue(mockLedger);
      mockLedgerRepository.findOne.mockResolvedValue({
        ...mockLedger,
        group: mockGroup,
        creator: mockUser,
      });

      const result = await service.create('user-1', {
        groupId: 'group-1',
        type: LedgerType.INCOME,
        amount: 100,
        category: '聚餐费用',
        description: '周末聚餐',
      });

      expect(result).toHaveProperty('id');
      expect(result.amount).toBe(100);
      expect(mockLedgerRepository.save).toHaveBeenCalled();
    });

    it('应该在小组不存在时抛出异常', async () => {
      mockGroupRepository.findOne.mockResolvedValue(null);

      await expect(
        service.create('user-1', {
          groupId: 'group-1',
          type: LedgerType.INCOME,
          amount: 100,
          category: '聚餐费用',
          description: '测试',
        }),
      ).rejects.toThrow(NotFoundException);
    });

    it('应该在用户不在小组中时抛出异常', async () => {
      mockGroupRepository.findOne.mockResolvedValue(mockGroup);
      mockGroupMemberRepository.findOne.mockResolvedValue(null);

      await expect(
        service.create('user-1', {
          groupId: 'group-1',
          type: LedgerType.INCOME,
          amount: 100,
          category: '聚餐费用',
          description: '测试',
        }),
      ).rejects.toThrow(ForbiddenException);
    });

    it('应该在金额无效时抛出异常', async () => {
      mockGroupRepository.findOne.mockResolvedValue(mockGroup);
      mockGroupMemberRepository.findOne.mockResolvedValue(mockMembership);

      await expect(
        service.create('user-1', {
          groupId: 'group-1',
          type: LedgerType.INCOME,
          amount: -100,
          category: '聚餐费用',
          description: '测试',
        }),
      ).rejects.toThrow(BadRequestException);
    });
  });

  describe('findAll', () => {
    it('应该成功获取账目列表', async () => {
      const mockQueryBuilder = {
        leftJoinAndSelect: jest.fn().mockReturnThis(),
        where: jest.fn().mockReturnThis(),
        andWhere: jest.fn().mockReturnThis(),
        orderBy: jest.fn().mockReturnThis(),
        skip: jest.fn().mockReturnThis(),
        take: jest.fn().mockReturnThis(),
        getManyAndCount: jest.fn().mockResolvedValue([[mockLedger], 1]),
      };

      mockLedgerRepository.createQueryBuilder.mockReturnValue(mockQueryBuilder);
      mockGroupMemberRepository.findOne.mockResolvedValue(mockMembership);

      const result = await service.findAll('user-1', {
        groupId: 'group-1',
        page: 1,
        limit: 10,
      });

      expect(result).toHaveProperty('items');
      expect(result).toHaveProperty('total');
      expect(result.items).toHaveLength(1);
    });

    it('应该支持按类型筛选', async () => {
      const mockQueryBuilder = {
        leftJoinAndSelect: jest.fn().mockReturnThis(),
        where: jest.fn().mockReturnThis(),
        andWhere: jest.fn().mockReturnThis(),
        orderBy: jest.fn().mockReturnThis(),
        skip: jest.fn().mockReturnThis(),
        take: jest.fn().mockReturnThis(),
        getManyAndCount: jest.fn().mockResolvedValue([[mockLedger], 1]),
      };

      mockLedgerRepository.createQueryBuilder.mockReturnValue(mockQueryBuilder);
      mockGroupMemberRepository.findOne.mockResolvedValue(mockMembership);

      const result = await service.findAll('user-1', {
        groupId: 'group-1',
        type: LedgerType.INCOME,
        page: 1,
        limit: 10,
      });

      expect(result.items).toHaveLength(1);
      expect(mockQueryBuilder.andWhere).toHaveBeenCalled();
    });
  });

  describe('findOne', () => {
    it('应该成功获取账目详情', async () => {
      mockLedgerRepository.findOne.mockResolvedValue({
        ...mockLedger,
        group: mockGroup,
        creator: mockUser,
      });

      const result = await service.findOne('ledger-1');

      expect(result).toHaveProperty('id');
      expect(result.id).toBe('ledger-1');
    });

    it('应该在账目不存在时抛出异常', async () => {
      mockLedgerRepository.findOne.mockResolvedValue(null);

      await expect(service.findOne('ledger-1')).rejects.toThrow(
        NotFoundException,
      );
    });
  });

  describe('update', () => {
    it('应该成功更新账目', async () => {
      mockLedgerRepository.findOne
        .mockResolvedValueOnce(mockLedger)
        .mockResolvedValueOnce({
          ...mockLedger,
          amount: 200,
          group: mockGroup,
          creator: mockUser,
        });
      mockGroupMemberRepository.findOne.mockResolvedValue({
        ...mockMembership,
        role: 'admin',
      });
      mockLedgerRepository.save.mockResolvedValue({
        ...mockLedger,
        amount: 200,
      });

      const result = await service.update('user-1', 'ledger-1', {
        amount: 200,
      });

      expect(result.amount).toBe(200);
    });

    it('应该在账目不存在时抛出异常', async () => {
      mockLedgerRepository.findOne.mockResolvedValue(null);

      await expect(
        service.update('user-1', 'ledger-1', { amount: 200 }),
      ).rejects.toThrow(NotFoundException);
    });

    it('应该在无权限时抛出异常', async () => {
      mockLedgerRepository.findOne.mockResolvedValue(mockLedger);
      mockGroupMemberRepository.findOne.mockResolvedValue({
        ...mockMembership,
        role: 'member',
      });

      await expect(
        service.update('user-2', 'ledger-1', { amount: 200 }),
      ).rejects.toThrow(ForbiddenException);
    });
  });

  describe('remove', () => {
    it('应该成功删除账目', async () => {
      mockLedgerRepository.findOne.mockResolvedValue(mockLedger);
      mockGroupMemberRepository.findOne.mockResolvedValue({
        ...mockMembership,
        role: 'admin',
      });
      mockLedgerRepository.remove.mockResolvedValue(mockLedger);

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

      expect(result).toHaveProperty('message');
    });

    it('应该在无权限时抛出异常', async () => {
      mockLedgerRepository.findOne.mockResolvedValue(mockLedger);
      mockGroupMemberRepository.findOne.mockResolvedValue({
        ...mockMembership,
        role: 'member',
      });

      await expect(
        service.remove('user-2', 'ledger-1'),
      ).rejects.toThrow(ForbiddenException);
    });
  });

  describe('getMonthlyStatistics', () => {
    it('应该成功获取月度统计', async () => {
      const mockQueryBuilder = {
        where: jest.fn().mockReturnThis(),
        andWhere: jest.fn().mockReturnThis(),
        getMany: jest.fn().mockResolvedValue([
          { ...mockLedger, type: LedgerType.INCOME, amount: 100 },
          { ...mockLedger, type: LedgerType.EXPENSE, amount: 50 },
        ]),
      };

      mockLedgerRepository.createQueryBuilder.mockReturnValue(mockQueryBuilder);
      mockGroupMemberRepository.findOne.mockResolvedValue(mockMembership);

      const result = await service.getMonthlyStatistics('user-1', {
        groupId: 'group-1',
        year: 2024,
        month: 1,
      });

      expect(result).toHaveProperty('income');
      expect(result).toHaveProperty('expense');
      expect(result).toHaveProperty('balance');
      expect(result).toHaveProperty('categories');
    });

    it('应该在用户不在小组时抛出异常', async () => {
      mockGroupMemberRepository.findOne.mockResolvedValue(null);

      await expect(
        service.getMonthlyStatistics('user-1', {
          groupId: 'group-1',
          year: 2024,
          month: 1,
        }),
      ).rejects.toThrow(ForbiddenException);
    });
  });

  describe('getHierarchicalSummary', () => {
    it('应该成功获取层级汇总', async () => {
      const childGroup = { id: 'group-2', name: '子小组', parentId: 'group-1' };
      const mockQueryBuilder = {
        where: jest.fn().mockReturnThis(),
        getMany: jest.fn().mockResolvedValue([mockLedger]),
      };

      mockGroupRepository.findOne.mockResolvedValue(mockGroup);
      mockGroupMemberRepository.findOne.mockResolvedValue(mockMembership);
      mockGroupRepository.find.mockResolvedValue([childGroup]);
      mockLedgerRepository.createQueryBuilder.mockReturnValue(mockQueryBuilder);

      const result = await service.getHierarchicalSummary('user-1', 'group-1');

      expect(result).toHaveProperty('groupId');
      expect(result).toHaveProperty('income');
      expect(result).toHaveProperty('expense');
      expect(result).toHaveProperty('balance');
    });
  });
});