package com.tourguide.modules.order.service; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.tourguide.modules.order.dto.GuideEarningsResponse; import com.tourguide.modules.order.entity.TourOrder; import com.tourguide.modules.order.mapper.TourOrderMapper; import com.tourguide.modules.user.entity.GuideUser; import com.tourguide.modules.user.service.GuideUserService; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; import java.math.BigDecimal; import java.util.Collections; import static org.assertj.core.api.Assertions.assertThat; import static org.mockito.ArgumentMatchers.any; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; class GuideEarningsServiceTest { private TourOrderMapper orderMapper; private GuideEarningsService service; @BeforeEach void setUp() { orderMapper = mock(TourOrderMapper.class); GuideUserService guideUserService = mock(GuideUserService.class); GuideUser guide = new GuideUser(); guide.setId(8L); guide.setStatus(1); when(guideUserService.findEnabledById(8L)).thenReturn(guide); service = new GuideEarningsService(orderMapper, guideUserService); } @Test void shouldUseGuideEarningSnapshotForSummaryAndDetails() { TourOrder order = new TourOrder(); order.setId(10L); order.setPaymentAmount(new BigDecimal("100.00")); order.setGuideEarningAmount(new BigDecimal("90.00")); Page page = new Page<>(1, 50); page.setRecords(Collections.singletonList(order)); page.setTotal(1); when(orderMapper.selectGuideEarningSum(any(), any(), any())) .thenReturn(new BigDecimal("90.00")); when(orderMapper.selectPage(any(Page.class), any())).thenReturn(page); when(orderMapper.selectGuideEarningScenics(8L)).thenReturn(Collections.emptyList()); GuideEarningsResponse response = service.get(8L, null, null, null, 1, 50); assertThat(response.getTodayEarnings()).isEqualByComparingTo("90.00"); assertThat(response.getMonthEarnings()).isEqualByComparingTo("90.00"); assertThat(response.getRecords()).hasSize(1); assertThat(response.getRecords().get(0).getOrderAmount()) .isEqualByComparingTo("100.00"); assertThat(response.getRecords().get(0).getEarningAmount()) .isEqualByComparingTo("90.00"); } }