Mockito更改方法的返回值

时间:2019-01-07 21:58:54

标签: mockito junit4

在测试类中,尝试更改从方法调用返回的结果。

在测试method1时,从测试类中调用isExistStudentInList(...)时尝试返回true。

尝试了以下内容,但实际上是在执行该方法,而不是返回真实值。

关于如何做到这一点的任何建议?

public class School { 
    public static void main(String[] s) {
        ..........
    }

    public void method1(List<Students> lstStudents) {
        int ilCounter       = 0; 
        .....
        while(rs.next()) { 
            ilCounter++;

            Students voObj = new Students();
            voObj.setName(rs.getString(1));
            voObj.setDepartment(rs.getString(2));
                .....           

            boolean existStu = isExistStudentInList(lstStudents, voObj);
            if(elementId == 0 && existStu) {
                ilCounter--;    
                .....
            } 
        }
    }

    public boolean isExistStudentInList(List<Students> lstJobElements, Students voObj) {
        boolean checkStudent;
        .........
        return checkStudent;
    }
}

public class SchoolTest { 
    @InjectMocks
    School school;

    .....
    @Before
    public void setUp() {
        ........
    }

    @Test
    public void testMethod1() throws SQLException {
        ........    
        when(school.isExistStudentInList(getSampleDataForStudents(), (Students)getSampleDataForStudents().get(0))).thenReturn(true);
        school.method1(getSampleDataForStudents());         
        ....
    }

    private List<Students> getSampleDataForStudents() {
        List<Students> lstStudents = new ArrayList<Students>();

                Students student1 = new Students();
        student1 .setName("aaaa");
        student1 .setDepartment("1222");
        .....       

        Students student2 = new Students();
        student1 .setName("bbbb");
        student1 .setDepartment("1222");
        .....

        lstStudents.add(student1);
        lstStudents.add(student2);

        return lstStudents;
    }
}

1 个答案:

答案 0 :(得分:1)

您正在使用@InjectMocks创建一个类的实例,并注入使用@Mock@Spy创建的模拟。但是School既不是模拟也不是间谍。因此,令您惊讶的是,当您在其上调用when()时,您的代码没有引发任何异常。

您可以通过让School成为间谍来完成您想做的事情。这将允许调用底层方法,除非将其存根。

School school = Mockito.spy(new School());

但是,我认为您应该真正改变测试代码或抽象某些实现的方式,因为您不必模拟/监视正在测试的类。