我在我的应用程序中使用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?
您能帮忙解决上述异常吗?
答案 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
}