package com.zhaoonline.coupen.generator.distribute;

import com.zhaoonline.coupen.bean.CommonCoupenEntity;
import com.zhaoonline.coupen.bean.CommonCoupenSeed;
import com.zhaoonline.coupen.bean.CoupenContext;
import org.junit.Test;
import org.junit.Assert;

import java.math.BigDecimal;

/**
 * Created by ZhaoOnline<br/>
 * User: yangyoupeng<br/>
 * Date: 2016/12/15<br/>
 * Time: 11:15<br/>
 * Description:please descript you class
 */
public class TestAverageStrategy {
    AverageStrategy strategy = new AverageStrategy();
    @Test
    public void testDistributeNullSeed() {
        CommonCoupenSeed seed =null;
        CoupenContext context = strategy.distribute(seed);
        Assert.assertNull(context.getSeed());
    }


    @Test
    public void testDistributeSeed() {

        Integer expectCoupenCount=new Integer(10);
        BigDecimal totalAmount= new BigDecimal(8880);

        CommonCoupenSeed seed = new CommonCoupenSeed();
        seed.setTotalAmount(totalAmount);
        seed.setCoupenCount(expectCoupenCount);
        CoupenContext<CommonCoupenEntity,CommonCoupenSeed> context = strategy.distribute(seed);
        Assert.assertEquals(expectCoupenCount.intValue(), context.getCoupens().size());
        Assert.assertEquals(strategy.amountPerCoupen(seed), context.getCoupens().get(0).getAmount());
    }
}