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<TourOrder> 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");
|
}
|
}
|