Question

我有一个类使用new jdbcTemplate(dataSource)创建它自己的jdbcTemplate，并在需要时使用新的NameParameterJdbcTemplate(jdbcTemplate)创建自己的NamedParameterJdbcTemplate。我如何使用Mockito来制作它，以便在单元测试模板时模拟模板，我可以控制返回值？

Answer 1

您可以创建特殊的包级别方法，该方法将创建这些对象，然后使用@Spy功能模拟它们：

1）创建专门的方法：

public class MyClass{

    private Datasource ds;

    public void methodUnderTest(){
         JdbcTemplate template = getTemplateInstance(ds);

         ...
    }

    JdbcTemplate getTemplateInstance(Datasource ds){
        return new JdbcTemplate(ds);
    }
}

2）监视类并模拟实例创建方法：

public class MyClassTest{

     @Spy
     @InjectMocks
     private MyClass myClassSpy;

     @Mock
     private Datasource dsMock;

     @Mock
     private JdbcTemplate templateMock;

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

     @Test
     public void test(){
          // Arrange
          doReturn(templateMock).when(myClassSpy).getTemplateInstance(dsMock);

          // Act and Assert ...
     }
}

由类

1 个答案: