Junit for Spring Cache Manager

时间:2016-09-13 17:11:04

标签: java spring junit mockito spring-cache

我在我的应用程序中使用spring缓存层,我在使用 Mockito 编写单元测试以测试弹簧缓存层时遇到了问题。

请参阅以下代码以解决我的问题:

服务层:

    public CustomerServiceImpl implements CustomerService {
    @Autowired
    private CacheManager cacheManager;

    @Autowired
    private CustomerRepository customerRepository;//Repository is simple JPA repository interface which contains findByCustomerName()

    @Override
    @CachePut(value = "#customer", key = "#customer.customerName")
    public Customer insertOrUpdate(Customer customer) {
        return customerRepository.save(customer);
    }

    @Cacheable(value="customersCache", key = "#customerName")
    public Customer findByCustomerName(String customerName) {
        Customer customer = customerRepository.findByCustomerName(customerName);
        return customer;
    }
}

服务层的JUnit测试代码:

@RunWith(PowerMockRunner.class)
@PrepareForTest(CustomerServiceImplTest.class)
public class CustomerServiceImplTest {

    @Spy
    CacheManager cacheManager = new ConcurrentMapCacheManager("customersCache");

    @Mock
    CustomerRepository CustomerRepository;

    @InjectMocks
    CustomerServiceImpl customerServiceImpl = new CustomerServiceImpl();

    @Before
    public void setup() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testCacheForFindByCustomerName() {
        Customer customer1 = new Customer();
        customer1.setId("1");
        customer1.setName("John");
        Customer customer2 = new Customer();
        customer2.setId("2");
        customer2.setName("Sara");

        //this should save to cache   
        Mockito.when(CustomerRepository.save(customer1))
        .thenReturn(customer1);
        customerServiceImpl.insertOrUpdate(customer1);

        //Now it should retreive from cache, but not able to
      Mockito.when(CustomerRepository.findByCustomerName(Mockito.any(String.class)))
                .thenReturn(customer1, customer2);

        Customer result = customerServiceImpl.findByCustomerName("John");
        assertThat(result, is(customer1));

        result = customerServiceImpl.findByCustomerName("John");
        assertThat(result, is(customer1));
    }
}

例外:

我得到了一个" java.lang.AssertionError:"因为缓存层不起作用,并且调用已经传递给存储库对象(两次),该对象已经返回了' customer2'上面的模拟对象,即通过传递服务层为相同的密钥调用了两次存储库方法。

另外,请注意我正在使用" Mockito"我的测试框架。

我曾试图谷歌进行弹簧缓存的单元测试,并且还引用了下面的URL,它几乎使用相同的概念,但它不适用于我的上述代码。

How to test Spring's declarative caching support on Spring Data repositories?

您能帮忙解决上述异常吗?

1 个答案:

答案 0 :(得分:3)

Spring Cache Manager依赖于Spring管理应用程序。您无法使用PowerMockRunner获得该信息,您需要使用SpringJUnit4Runner。您仍然可以通过编程方式使用PowerMock或Mockito,但不能作为Runner使用。

通常,您会将单元测试转换为Spring样式的集成测试,如下所示:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration
public class SpringTest {

    @Configuration
    @EnableCaching
    static class SpringConfig{
        @Bean
        public CustomerService customerService(){
            return new CustomerServiceImpl(customerRepository());
        }
        @Bean
        public CustomerRepository customerRepository(){
            return Mockito.mock(CustomerRepository.class);
        }
    }

    @Autowired
    CustomerService customerService; // this will contain a proper managed cache

    @Autowired
    CustomerRepository customerRepository; // this is a mockito mock you can fine-tune
}