(七)JMockit 的MockUp+@Mock--基础篇

一种常见的mock方式:MockUp + @Mock

import mockit.Mock;
import mockit.MockUp;
import org.junit.Assert;
import org.junit.Test;
import java.util.Calendar;
import java.util.Locale;

//(Mockup + @Mock) 的mock方式
public class MockUpTest {
    @Test
    public void testMockUp(){
        /*
        * 对java自带类Calendar的get方法进行定制
        * 只需要把Calendar类传入MockUp类的构造函数即可
        * */
        new MockUp(Calendar.class){
            //想mock哪个方法,就给哪个方法加上@Mock,没有@Mock的方法,不受影响
            @Mock
            public int get(int unit){
                if (unit == Calendar.YEAR){
                    return 2017;
                }
                if (unit == Calendar.MONDAY){
                    return 12;
                }
                if (unit == Calendar.DAY_OF_MONTH){
                    return 25;
                }
                if (unit == Calendar.HOUR_OF_DAY){
                    return 7;
                }
                return 0;
            }
        };
        // 从此Calendar的get方法,就沿用你定制过的逻辑,而不是它原先的逻辑。
        Calendar cal = Calendar.getInstance(Locale.FRANCE);
        Assert.assertTrue(cal.get(Calendar.YEAR) == 2017);
        Assert.assertTrue(cal.get(Calendar.MONDAY) == 12);
        Assert.assertTrue(cal.get(Calendar.DAY_OF_MONTH) == 25);
        Assert.assertTrue(cal.get(Calendar.HOUR_OF_DAY) == 7);
        // Calendar的其它方法,不受影响
        Assert.assertTrue((cal.getFirstDayOfWeek() == Calendar.MONDAY));
    }
}

Mock方式是不是很简单,直接 ? 难怪很多程序员们都喜欢用MockUp & @Mock了。

那是不是我们只需要掌握MockUp & @Mock就可以了?就不需要了解其它Mock API了?
当然不是! 掌握了MockUp & @Mock的确能帮我们解决大部分的Mock场景,因为其使用方式最直接嘛。

比如下面的场景是MockUp & @Mock做不到的。

  • 一个类有多个实例。只对其中某1个实例进行mock。
    最新版的JMockit已经让MockUp不再支持对实例的Mock了。1.19之前的老版本仍支持。
  • AOP动态生成类的Mock。
  • 对类的所有方法都需要Mock时,书写MockUp的代码量太大。
    比如web程序中,经常需要对HttpSession进行Mock。若用MockUp你要写大量的代码,可是用@Mocked就一行代码就可以搞定。

MockUp & @Mock的确是好东西,在实际Mock场景中,我们需要灵活运用JMockit其它的Mock API。让我们的Mock程序简单,高效。

MockUp & @Mock比较适合于一个项目中,用于对一些通用类的Mock,以减少大量重复的**new Exceptations{{}}**代码。

你可能感兴趣的:(JMockit单元测试)