Mockito模拟接口在Before中返回null

时间:2016-03-09 22:19:04

标签: java junit mockito

我正在尝试使用Mockito API运行JUnit测试。

我有以下CacheTest类

public class CacheTest {

    public static final String KEY_1 = "key1";
    public static final long VALUE_1 = 1L;
    public static final long VALUE_2 = 2136L;
    private static final String KEY_2 = "key2";
    private Cache<String, Long> objectUnderTest;

    @Mock
    private CacheLoader<String, Long> cacheLoader;

    @Before
    public void setUp() {
        objectUnderTest = new Cache<>(1000L, cacheLoader); //cacheLoader is null here
        when(cacheLoader.load(Matchers.eq(KEY_1))).thenReturn(VALUE_1); //nullpointer when trying to load from null
        when(cacheLoader.load(Matchers.eq(KEY_2))).thenReturn(VALUE_2);
    }

    @Test
    public void shouldLoadElement() {
        // when
        Long value = objectUnderTest.get(KEY_1);

        // then
        assertThat(value).isEqualTo(VALUE_1);
        verify(cacheLoader, Mockito.times(1)).load(Matchers.eq(KEY_1));
    }
}

CacheLoader接口

public interface CacheLoader<KEY, VALUE> {

    VALUE load(KEY key);
}

缓存类,它将模拟的cacheLoader作为参数

public class Cache<K,V> {

    private long evictionTimeMillis;

    private CacheLoader<K,V> cacheLoader;

    public Cache(final long evictionTimeMillis, final CacheLoader<K, V> cacheLoader) {
        this.evictionTimeMillis = evictionTimeMillis;
        this.cacheLoader = cacheLoader;
    }

    public V get(K key) {
        // TODO
        return this.cacheLoader.load(key);
    }


    public void cleanUp() {
        //TODO
    }

}

和CacheLoader实现

public class CacheLoaderImpl implements CacheLoader<String, Long>{

    private HashMap<String,Long> map = new HashMap<String, Long>() {
        /**
         * 
         */
        private static final long serialVersionUID = -7591912310318794715L;

        {
            map.put("key1", 1L);
            map.put("key2", 2136L);
        }
    };

    @Override
    public Long load(String key) {
        return map.get(key);
    }

}

问题是,正如CacheTest类中的注释所暗示的,当执行@Before方法时,模拟的接口为null,所以当我尝试加载时,我得到一个nullpointer异常。

我是Mockito的新手,如果这是一个noobie错误,请原谅。

我在SO中看到过类似的问题,但没有一个能真正给我答案。

我希望我的问题很清楚,如果没有,请告诉我如何改进它。

1 个答案:

答案 0 :(得分:8)

您需要使用@RunWith(MockitoJUnitRunner.class)注释测试类。库存JUnit运行器不会初始化模拟。另请参阅@RunWith(MockitoJUnitRunner.class) vs MockitoAnnotations.initMocks(this)