BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 0d08c814ec8ad6d367d0dd4b767bccbd3de89b2d

Authored by YunaiV
1 parent 1452f6b2

mall + trade：增加 TradeDiscountActivityPriceCalculator 的单元测试

Inline Side-by-side
Showing 4 changed files with 128 additions and 93 deletions
yudao-module-mall/yudao-module-promotion-biz/src/test/java/cn/iocoder/yudao/module/promotion/service/price/PriceServiceTest.java
  View file @0d08c81
@@ -48,99 +48,12 @@ public class PriceServiceTest extends BaseMockitoUnitTest { @@ -48,99 +48,12 @@ public class PriceServiceTest extends BaseMockitoUnitTest {
48 private PriceServiceImpl priceService; 48 private PriceServiceImpl priceService;
49 49
50 @Mock 50 @Mock
51 - private DiscountActivityService discountService;  
52 - @Mock  
53 private RewardActivityService rewardActivityService; 51 private RewardActivityService rewardActivityService;
54 @Mock 52 @Mock
55 private CouponService couponService; 53 private CouponService couponService;
56 @Mock 54 @Mock
57 private ProductSkuApi productSkuApi; 55 private ProductSkuApi productSkuApi;
58 56
59 - @Test  
60 - public void testCalculatePrice_discountActivity() {  
61 - // 准备参数  
62 - PriceCalculateReqDTO calculateReqDTO = new PriceCalculateReqDTO().setUserId(randomLongId())  
63 - .setItems(asList(new PriceCalculateReqDTO.Item().setSkuId(10L).setCount(2),  
64 - new PriceCalculateReqDTO.Item().setSkuId(20L).setCount(3)));  
65 - // mock 方法(商品 SKU 信息)  
66 - ProductSkuRespDTO productSku01 = randomPojo(ProductSkuRespDTO.class, o -> o.setId(10L).setPrice(100));  
67 - ProductSkuRespDTO productSku02 = randomPojo(ProductSkuRespDTO.class, o -> o.setId(20L).setPrice(50));  
68 - when(productSkuApi.getSkuList(eq(asSet(10L, 20L)))).thenReturn(asList(productSku01, productSku02));  
69 - // mock 方法（限时折扣 DiscountActivity 信息）  
70 - DiscountProductDetailBO discountProduct01 = randomPojo(DiscountProductDetailBO.class, o -> o.setActivityId(1000L)  
71 - .setActivityName("活动 1000 号").setSkuId(10L)  
72 - .setDiscountType(PromotionDiscountTypeEnum.PRICE.getType()).setDiscountPrice(40));  
73 - DiscountProductDetailBO discountProduct02 = randomPojo(DiscountProductDetailBO.class, o -> o.setActivityId(2000L)  
74 - .setActivityName("活动 2000 号").setSkuId(20L)  
75 - .setDiscountType(PromotionDiscountTypeEnum.PERCENT.getType()).setDiscountPercent(60));  
76 - when(discountService.getMatchDiscountProductList(eq(asSet(10L, 20L)))).thenReturn(  
77 - MapUtil.builder(10L, discountProduct01).put(20L, discountProduct02).map());  
78 -  
79 - // 10L: 100 * 2 - 40 * 2 = 120  
80 - // 20L：50 * 3 - 50 * 3 * 0.4 = 90  
81 -  
82 - // 调用  
83 - PriceCalculateRespDTO priceCalculate = priceService.calculatePrice(calculateReqDTO);  
84 - // 断言 Order 部分  
85 - PriceCalculateRespDTO.Order order = priceCalculate.getOrder();  
86 - assertEquals(order.getTotalPrice(), 350);  
87 - assertEquals(order.getDiscountPrice(), 0);  
88 - assertEquals(order.getPointPrice(), 0);  
89 - assertEquals(order.getDeliveryPrice(), 0);  
90 - assertEquals(order.getPayPrice(), 210);  
91 - assertNull(order.getCouponId());  
92 - // 断言 OrderItem 部分  
93 - assertEquals(order.getItems().size(), 2);  
94 - PriceCalculateRespDTO.OrderItem orderItem01 = order.getItems().get(0);  
95 - assertEquals(orderItem01.getSkuId(), 10L);  
96 - assertEquals(orderItem01.getCount(), 2);  
97 - assertEquals(orderItem01.getOriginalPrice(), 200);  
98 - assertEquals(orderItem01.getOriginalUnitPrice(), 100);  
99 - assertEquals(orderItem01.getDiscountPrice(), 80);  
100 - assertEquals(orderItem01.getPayPrice(), 120);  
101 - assertEquals(orderItem01.getOrderPartPrice(), 0);  
102 - assertEquals(orderItem01.getOrderDividePrice(), 120);  
103 - PriceCalculateRespDTO.OrderItem orderItem02 = order.getItems().get(1);  
104 - assertEquals(orderItem02.getSkuId(), 20L);  
105 - assertEquals(orderItem02.getCount(), 3);  
106 - assertEquals(orderItem02.getOriginalPrice(), 150);  
107 - assertEquals(orderItem02.getOriginalUnitPrice(), 50);  
108 - assertEquals(orderItem02.getDiscountPrice(), 60);  
109 - assertEquals(orderItem02.getPayPrice(), 90);  
110 - assertEquals(orderItem02.getOrderPartPrice(), 0);  
111 - assertEquals(orderItem02.getOrderDividePrice(), 90);  
112 - // 断言 Promotion 部分  
113 - assertEquals(priceCalculate.getPromotions().size(), 2);  
114 - PriceCalculateRespDTO.Promotion promotion01 = priceCalculate.getPromotions().get(0);  
115 - assertEquals(promotion01.getId(), 1000L);  
116 - assertEquals(promotion01.getName(), "活动 1000 号");  
117 - assertEquals(promotion01.getType(), PromotionTypeEnum.DISCOUNT_ACTIVITY.getType());  
118 - assertEquals(promotion01.getLevel(), PromotionLevelEnum.SKU.getLevel());  
119 - assertEquals(promotion01.getTotalPrice(), 200);  
120 - assertEquals(promotion01.getDiscountPrice(), 80);  
121 - assertTrue(promotion01.getMatch());  
122 - assertEquals(promotion01.getDescription(), "限时折扣：省 0.80 元");  
123 - PriceCalculateRespDTO.PromotionItem promotionItem01 = promotion01.getItems().get(0);  
124 - assertEquals(promotion01.getItems().size(), 1);  
125 - assertEquals(promotionItem01.getSkuId(), 10L);  
126 - assertEquals(promotionItem01.getOriginalPrice(), 200);  
127 - assertEquals(promotionItem01.getDiscountPrice(), 80);  
128 - PriceCalculateRespDTO.Promotion promotion02 = priceCalculate.getPromotions().get(1);  
129 - assertEquals(promotion02.getId(), 2000L);  
130 - assertEquals(promotion02.getName(), "活动 2000 号");  
131 - assertEquals(promotion02.getType(), PromotionTypeEnum.DISCOUNT_ACTIVITY.getType());  
132 - assertEquals(promotion02.getLevel(), PromotionLevelEnum.SKU.getLevel());  
133 - assertEquals(promotion02.getTotalPrice(), 150);  
134 - assertEquals(promotion02.getDiscountPrice(), 60);  
135 - assertTrue(promotion02.getMatch());  
136 - assertEquals(promotion02.getDescription(), "限时折扣：省 0.60 元");  
137 - PriceCalculateRespDTO.PromotionItem promotionItem02 = promotion02.getItems().get(0);  
138 - assertEquals(promotion02.getItems().size(), 1);  
139 - assertEquals(promotionItem02.getSkuId(), 20L);  
140 - assertEquals(promotionItem02.getOriginalPrice(), 150);  
141 - assertEquals(promotionItem02.getDiscountPrice(), 60);  
142 - }  
143 -  
144 /** 57 /**
145 * 测试满减送活动，匹配的情况 58 * 测试满减送活动，匹配的情况
146 */ 59 */
yudao-module-mall/yudao-module-trade-biz/src/main/java/cn/iocoder/yudao/module/trade/service/price/calculator/TradeDiscountActivityPriceCalculator.java
  View file @0d08c81
@@ -53,10 +53,13 @@ public class TradeDiscountActivityPriceCalculator implements TradePriceCalculato @@ -53,10 +53,13 @@ public class TradeDiscountActivityPriceCalculator implements TradePriceCalculato
53 Integer newDiscountPrice = orderItem.getPayPrice() - newPayPrice; 53 Integer newDiscountPrice = orderItem.getPayPrice() - newPayPrice;
54 54
55 // 3.1 记录优惠明细 55 // 3.1 记录优惠明细
56 - TradePriceCalculatorHelper.addPromotion(result, orderItem,  
57 - discountProduct.getActivityId(), discountProduct.getActivityName(), PromotionTypeEnum.DISCOUNT_ACTIVITY.getType(),  
58 - StrUtil.format("限时折扣：省 {} 元", formatPrice(newDiscountPrice)),  
59 - newDiscountPrice); 56 + if (orderItem.getSelected()) {
  57 + // 注意，只有在选中的情况下，才会记录到优惠明细。否则仅仅是更新 SKU 优惠金额，用于展示
  58 + TradePriceCalculatorHelper.addPromotion(result, orderItem,
  59 + discountProduct.getActivityId(), discountProduct.getActivityName(), PromotionTypeEnum.DISCOUNT_ACTIVITY.getType(),
  60 + StrUtil.format("限时折扣：省 {} 元", formatPrice(newDiscountPrice)),
  61 + newDiscountPrice);
  62 + }
60 // 3.2 更新 SKU 优惠金额 63 // 3.2 更新 SKU 优惠金额
61 orderItem.setDiscountPrice(orderItem.getDiscountPrice() + newDiscountPrice); 64 orderItem.setDiscountPrice(orderItem.getDiscountPrice() + newDiscountPrice);
62 TradePriceCalculatorHelper.recountPayPrice(orderItem); 65 TradePriceCalculatorHelper.recountPayPrice(orderItem);
yudao-module-mall/yudao-module-trade-biz/src/test/java/cn/iocoder/yudao/module/trade/service/price/calculator/TradeCouponPriceCalculatorTest.java
  View file @0d08c81
@@ -36,7 +36,7 @@ class TradeCouponPriceCalculatorTest extends BaseMockitoUnitTest { @@ -36,7 +36,7 @@ class TradeCouponPriceCalculatorTest extends BaseMockitoUnitTest {
36 private CouponApi couponApi; 36 private CouponApi couponApi;
37 37
38 @Test 38 @Test
39 - void calculate() { 39 + public void testCalculate() {
40 // 准备参数 40 // 准备参数
41 TradePriceCalculateReqBO param = new TradePriceCalculateReqBO() 41 TradePriceCalculateReqBO param = new TradePriceCalculateReqBO()
42 .setUserId(233L).setCouponId(1024L) 42 .setUserId(233L).setCouponId(1024L)
@@ -120,7 +120,7 @@ class TradeCouponPriceCalculatorTest extends BaseMockitoUnitTest { @@ -120,7 +120,7 @@ class TradeCouponPriceCalculatorTest extends BaseMockitoUnitTest {
120 assertEquals(orderItem04.getCouponPrice(), 0); 120 assertEquals(orderItem04.getCouponPrice(), 0);
121 assertEquals(orderItem04.getPointPrice(), 0); 121 assertEquals(orderItem04.getPointPrice(), 0);
122 assertEquals(orderItem04.getPayPrice(), 300); 122 assertEquals(orderItem04.getPayPrice(), 300);
123 - // 断言 Promotion 部分 123 + // 断言：Promotion 部分
124 assertEquals(result.getPromotions().size(), 1); 124 assertEquals(result.getPromotions().size(), 1);
125 TradePriceCalculateRespBO.Promotion promotion01 = result.getPromotions().get(0); 125 TradePriceCalculateRespBO.Promotion promotion01 = result.getPromotions().get(0);
126 assertEquals(promotion01.getId(), 1024L); 126 assertEquals(promotion01.getId(), 1024L);
@@ -140,4 +140,5 @@ class TradeCouponPriceCalculatorTest extends BaseMockitoUnitTest { @@ -140,4 +140,5 @@ class TradeCouponPriceCalculatorTest extends BaseMockitoUnitTest {
140 assertEquals(promotionItem012.getTotalPrice(), 150); 140 assertEquals(promotionItem012.getTotalPrice(), 150);
141 assertEquals(promotionItem012.getDiscountPrice(), 30); 141 assertEquals(promotionItem012.getDiscountPrice(), 30);
142 } 142 }
  143 +
143 } 144 }
yudao-module-mall/yudao-module-trade-biz/src/test/java/cn/iocoder/yudao/module/trade/service/price/calculator/TradeDiscountActivityPriceCalculatorTest.java 0 → 100644
  View file @0d08c81
  1 +package cn.iocoder.yudao.module.trade.service.price.calculator;
  2 +
  3 +import cn.iocoder.yudao.framework.test.core.ut.BaseMockitoUnitTest;
  4 +import cn.iocoder.yudao.module.promotion.api.discount.DiscountActivityApi;
  5 +import cn.iocoder.yudao.module.promotion.api.discount.dto.DiscountProductRespDTO;
  6 +import cn.iocoder.yudao.module.promotion.enums.common.PromotionDiscountTypeEnum;
  7 +import cn.iocoder.yudao.module.promotion.enums.common.PromotionTypeEnum;
  8 +import cn.iocoder.yudao.module.trade.service.price.bo.TradePriceCalculateReqBO;
  9 +import cn.iocoder.yudao.module.trade.service.price.bo.TradePriceCalculateRespBO;
  10 +import org.junit.jupiter.api.Test;
  11 +import org.mockito.InjectMocks;
  12 +import org.mockito.Mock;
  13 +
  14 +import java.util.ArrayList;
  15 +
  16 +import static cn.iocoder.yudao.framework.common.util.collection.SetUtils.asSet;
  17 +import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
  18 +import static java.util.Arrays.asList;
  19 +import static org.junit.jupiter.api.Assertions.*;
  20 +import static org.mockito.ArgumentMatchers.eq;
  21 +import static org.mockito.Mockito.when;
  22 +
  23 +/**
  24 + * {@link TradeDiscountActivityPriceCalculator} 的单元测试类
  25 + *
  26 + * @author 芋道源码
  27 + */
  28 +public class TradeDiscountActivityPriceCalculatorTest extends BaseMockitoUnitTest {
  29 +
  30 + @InjectMocks
  31 + private TradeDiscountActivityPriceCalculator tradeDiscountActivityPriceCalculator;
  32 +
  33 + @Mock
  34 + private DiscountActivityApi discountActivityApi;
  35 +
  36 + @Test
  37 + public void testCalculate() {
  38 + // 准备参数
  39 + TradePriceCalculateReqBO param = new TradePriceCalculateReqBO()
  40 + .setItems(asList(
  41 + new TradePriceCalculateReqBO.Item().setSkuId(10L).setCount(2).setSelected(true), // 匹配活动，且已选中
  42 + new TradePriceCalculateReqBO.Item().setSkuId(20L).setCount(3).setSelected(false) // 匹配活动，但未选中
  43 + ));
  44 + TradePriceCalculateRespBO result = new TradePriceCalculateRespBO()
  45 + .setPrice(new TradePriceCalculateRespBO.Price())
  46 + .setPromotions(new ArrayList<>())
  47 + .setItems(asList(
  48 + new TradePriceCalculateRespBO.OrderItem().setSkuId(10L).setCount(2).setSelected(true)
  49 + .setPrice(100),
  50 + new TradePriceCalculateRespBO.OrderItem().setSkuId(20L).setCount(3).setSelected(false)
  51 + .setPrice(50)
  52 + ));
  53 + // 保证价格被初始化上
  54 + TradePriceCalculatorHelper.recountPayPrice(result.getItems());
  55 + TradePriceCalculatorHelper.recountAllPrice(result);
  56 +
  57 + // mock 方法（限时折扣活动）
  58 + when(discountActivityApi.getMatchDiscountProductList(eq(asSet(10L, 20L)))).thenReturn(asList(
  59 + randomPojo(DiscountProductRespDTO.class, o -> o.setActivityId(1000L)
  60 + .setActivityName("活动 1000 号").setSkuId(10L)
  61 + .setDiscountType(PromotionDiscountTypeEnum.PRICE.getType()).setDiscountPrice(40)),
  62 + randomPojo(DiscountProductRespDTO.class, o -> o.setActivityId(2000L)
  63 + .setActivityName("活动 2000 号").setSkuId(20L)
  64 + .setDiscountType(PromotionDiscountTypeEnum.PERCENT.getType()).setDiscountPercent(60))
  65 + ));
  66 + // 10L: 100 * 2 - 40 * 2 = 120
  67 + // 20L：50 * 3 - 50 * 3 * 0.4 = 90
  68 +
  69 + // 调用
  70 + tradeDiscountActivityPriceCalculator.calculate(param, result);
  71 + // 断言：Price 部分
  72 + TradePriceCalculateRespBO.Price price = result.getPrice();
  73 + assertEquals(price.getTotalPrice(), 200);
  74 + assertEquals(price.getDiscountPrice(), 80);
  75 + assertEquals(price.getPointPrice(), 0);
  76 + assertEquals(price.getDeliveryPrice(), 0);
  77 + assertEquals(price.getCouponPrice(), 0);
  78 + assertEquals(price.getPayPrice(), 120);
  79 + assertNull(result.getCouponId());
  80 + // 断言：SKU 1
  81 + assertEquals(result.getItems().size(), 2);
  82 + TradePriceCalculateRespBO.OrderItem orderItem01 = result.getItems().get(0);
  83 + assertEquals(orderItem01.getSkuId(), 10L);
  84 + assertEquals(orderItem01.getCount(), 2);
  85 + assertEquals(orderItem01.getPrice(), 100);
  86 + assertEquals(orderItem01.getDiscountPrice(), 80);
  87 + assertEquals(orderItem01.getDeliveryPrice(), 0);
  88 + assertEquals(orderItem01.getCouponPrice(), 0);
  89 + assertEquals(orderItem01.getPointPrice(), 0);
  90 + assertEquals(orderItem01.getPayPrice(), 120);
  91 + // 断言：SKU 2
  92 + TradePriceCalculateRespBO.OrderItem orderItem02 = result.getItems().get(1);
  93 + assertEquals(orderItem02.getSkuId(), 20L);
  94 + assertEquals(orderItem02.getCount(), 3);
  95 + assertEquals(orderItem02.getPrice(), 50);
  96 + assertEquals(orderItem02.getDiscountPrice(), 60);
  97 + assertEquals(orderItem02.getDeliveryPrice(), 0);
  98 + assertEquals(orderItem02.getCouponPrice(), 0);
  99 + assertEquals(orderItem02.getPointPrice(), 0);
  100 + assertEquals(orderItem02.getPayPrice(), 90);
  101 + // 断言：Promotion 部分
  102 + assertEquals(result.getPromotions().size(), 1);
  103 + TradePriceCalculateRespBO.Promotion promotion01 = result.getPromotions().get(0);
  104 + assertEquals(promotion01.getId(), 1000L);
  105 + assertEquals(promotion01.getName(), "活动 1000 号");
  106 + assertEquals(promotion01.getType(), PromotionTypeEnum.DISCOUNT_ACTIVITY.getType());
  107 + assertEquals(promotion01.getTotalPrice(), 200);
  108 + assertEquals(promotion01.getDiscountPrice(), 80);
  109 + assertTrue(promotion01.getMatch());
  110 + assertEquals(promotion01.getDescription(), "限时折扣：省 0.80 元");
  111 + TradePriceCalculateRespBO.PromotionItem promotionItem01 = promotion01.getItems().get(0);
  112 + assertEquals(promotion01.getItems().size(), 1);
  113 + assertEquals(promotionItem01.getSkuId(), 10L);
  114 + assertEquals(promotionItem01.getTotalPrice(), 200);
  115 + assertEquals(promotionItem01.getDiscountPrice(), 80);
  116 + }
  117 +
  118 +}