单元测试如何使用Mockito模拟存储库
我在清除存储库时遇到问题。有人建议我只创建另一个Applation.Properties(我还没有这样做),并使用像h2这样的内存数据库。不过,我想知道是否可以只将调用存根,这样在调用myDataService.findById(Id)时,而不是试图从数据库中获取时,只会返回一个模拟对象?
我刚开始为我的单元测试和Spring Boot编写模拟代码,所以我可能遗漏了一些东西。下面的代码(试图简化名称并使其成为通用名称以便在此处发布)。
我的测试类
public class MyServiceImplTest
{
private MyDataService myDataService;
private NyService myService;
private MyRepository myRepository;
@Before
public void setUp() {
myDataService = Mockito.mock(MyDataServiceImpl.class);
myService = new MyServiceImpl(myDataService);
}
@Test
public void getById_ValidId() {
doReturn(MyMockData.getMyObject()).when(myDataService).findById("1");
when(myService.getById("1")).thenReturn(MyMockData.getMyObject());
MyObject myObject = myService.getById("1");
//Whatever asserts need to be done on the object myObject
}
}
用于对数据层进行服务调用的类
@Service
public class MyServiceImpl implements MyService {
MyDataService myDataService;
@Autowired
public MyServiceImpl(MyDataService myDataService) {
this.myDataService = myDataService;
}
@Override
public MyObject getById(String id) {
if(id == null || id == "") {
throw new InvalidRequestException("Invalid Identifier");
}
MyObject myObj;
try {
myObj = myDataService.findById(id);
}catch(Exception ex) {
throw new SystemException("Internal Server Error");
}
return myObj;
}
}
这就是我在测试中遇到问题的地方。当调用findById()方法时,变量存储库为空,因此当尝试执行repository.findOne(Id)时,它抛出异常n。这就是我尝试模拟的内容,但存储库给我带来了问题。
@Repository
@Qualifier("MyRepo")
public class MyDataServiceImpl {
@PersistenceContext
private EntityManager em;
private MyRepository repository;
@Autowired
public MyDataServiceImpl(MyRepository repository) {
super(repository);
this.repository = repository;
}
public MyObject findById(String id) {
P persitentObject = repository.findOne(id);
//Calls to map what persitentObject holds to MyObject and returns a MyObject
}
}
此处MyRepository的代码只是为了显示它是一个扩展CrudRepository的空接口
public interface MyRepository extends CrudRepository<MyObjectPO, String>, JpaSpecificationExecutor<MyObjectPO> {
}
解决方案
首先我要说的是,通过使用构造函数注入而不是字段注入(这使得使用模拟编写测试要简单得多),您走上了正确的道路。
public class MyServiceImplTest
{
private MyDataService myDataService;
private NyService myService;
@Mock
private MyRepository myRepository;
@Before
public void setUp() {
MockitoAnnotations.initMocks(this); // this is needed for inititalizytion of mocks, if you use @Mock
myDataService = new MyDataServiceImpl(myRepository);
myService = new MyServiceImpl(myDataService);
}
@Test
public void getById_ValidId() {
doReturn(someMockData).when(myRepository).findOne("1");
MyObject myObject = myService.getById("1");
//Whatever asserts need to be done on the object myObject
}
}
从您的服务-->dataService-->开始调用。但只模拟您的存储库调用。通过这种方式,您可以控制和测试类的所有其他部分(包括服务和数据服务),并且只模拟存储库调用。
相关文章