liujie
2026-09-11 4f6f15b818704079ff2e4e4719b798f48d68a335
backend/src/test/java/com/tourguide/modules/finance/service/GuideWithdrawableBalanceServiceTest.java
@@ -1,6 +1,8 @@
package com.tourguide.modules.finance.service;
import com.tourguide.modules.user.mapper.GuideUserMapper;
import com.tourguide.modules.user.mapper.GuideTeamMapper;
import com.tourguide.modules.user.entity.GuideTeam;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
@@ -13,12 +15,14 @@
class GuideWithdrawableBalanceServiceTest {
    private GuideUserMapper guideUserMapper;
    private GuideTeamMapper teamMapper;
    private GuideWithdrawableBalanceService service;
    @BeforeEach
    void setUp() {
        guideUserMapper = mock(GuideUserMapper.class);
        service = new GuideWithdrawableBalanceService(guideUserMapper);
        teamMapper = mock(GuideTeamMapper.class);
        service = new GuideWithdrawableBalanceService(guideUserMapper, teamMapper);
    }
    @Test
@@ -35,4 +39,16 @@
        assertThat(service.debit(8L, new BigDecimal("80"))).isTrue();
        verify(guideUserMapper).decreaseBalance(8L, new BigDecimal("80.00"));
    }
    @Test
    void shouldReadAndAtomicallyDebitTeamBalance() {
        GuideTeam team = new GuideTeam();
        team.setBalance(new BigDecimal("504.30"));
        when(teamMapper.selectById(6L)).thenReturn(team);
        when(teamMapper.decreaseBalance(6L, new BigDecimal("80.00"))).thenReturn(1);
        assertThat(service.getAvailableTeamAmount(6L)).isEqualByComparingTo("504.30");
        assertThat(service.debitTeam(6L, new BigDecimal("80"))).isTrue();
        verify(teamMapper).decreaseBalance(6L, new BigDecimal("80.00"));
    }
}