有 Java 编程相关的问题?

你可以在下面搜索框中键入要查询的问题!

java如何模拟从抽象类继承的受保护子类方法?

如何使用Mockito或PowerMock来模拟由子类实现但从抽象超类继承的受保护方法

换句话说,我想测试“doSomething”方法,同时模拟“doSomethingElse”

抽象超类

public abstract class TypeA {

    public void doSomething() {     

        // Calls for subclass behavior
        doSomethingElse();      
    }

    protected abstract String doSomethingElse();

}

子类实现

public class TypeB extends TypeA {

    @Override
    protected String doSomethingElse() {
        return "this method needs to be mocked";
    }

}

解决方案

这里给出的答案是正确的,如果所涉及的课程在同一个包中,那么这些答案将起作用

但是,如果涉及不同的包,一个选择是使用PowerMock。下面的例子对我很有用。当然,可能还有其他方法,这是一种有效的方法

import static org.junit.Assert.assertEquals;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

@RunWith(PowerMockRunner.class)
@PrepareForTest({ TypeB.class })
public class TestAbstract {

    @Test
    public void test_UsingPowerMock() throws Exception {
        // Spy a subclass using PowerMock
        TypeB b = PowerMockito.spy(new TypeB());
        String expected = "some value for mock";

        // Mock a method by its name using PowerMock again
        PowerMockito.doReturn(expected).when(b, "doSomethingElse");

        // Calls the 
        String actual = b.doSomething();
        assertEquals(expected, actual);     

    }
}

注意:测试使用Java5、JUnit4.11、Mockito 1.9.0和PowerMock 1.4.12完成


共 (4) 个答案

  1. # 1 楼答案

    要模拟在抽象类中返回void的方法,我们可以使用:

    MyAbstractClass abs = Mockito.mock(MyAbstractClass.class);
    Mockito.doNothing().when(abs).myMethod(arg1,arg2....));
    

    我们可以用Mockito替换参数。anyString()等符合要求

  2. # 2 楼答案

    我建议使用Mockito:

    // Create a new Mock
    final TypeA a = Mockito.mock(TypeA.class, Mockito.CALLS_REAL_METHODS);
    
    // Call the method
    a.doSomething();
    
    // Now verify that our mocked class' method was called
    Mockito.verify(a, Mockito.times(1)).doSomethingElse();
    
  3. # 3 楼答案

    模拟抽象方法时可以使用Mockito.CALLS_REAL_METHODS。这将调用类的原始方法,您可以自己模拟所有抽象方法

    TypeA typeA = mock(TypeA.class, Mockito.CALLS_REAL_METHODS);
    when(typeA.doSomethingElse()).thenReturn("Hello");
    typeA.doSomething();
    

    或者直接在TypeB上使用spy进行测试:

    TypeB typeB = spy(new TypeB());
    when(typeB.doSomethingElse()).thenReturn("Hello");
    typeB.doSomething();
    
  4. # 4 楼答案

    您可以通过以下方式使用mockito测试abtract类

    TypeA typA = Mockito.mock(TypeA.class, Mockito.CALLS_REAL_METHODS);
    when(typA.doSomethingElse()).thenReturn("doSomethingElse");
    Assert.assertSomething(typeA.doSomething());