java - 如何模拟从抽象类继承的 protected 子类方法？

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

换句话说，我想在模拟“doSomethingElse”时测试“doSomething”方法。

抽象超类

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);

    }
}

注意:使用Java 5，jUnit 4.11，Mockito 1.9.0和PowerMock 1.4.12完成的测试。

最佳答案

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

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

或者，您可以使用间谍直接在TypeB上进行测试:

TypeB typeB = spy(new TypeB());
when(typeB.doSomethingElse()).thenReturn("Hello");
typeB.doSomething();