如何模拟从抽象 class 继承的受保护的 subclass 方法？

Question

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

换句话说，我想在模拟 "doSomethingElse" 的同时测试 "doSomething" 方法。

摘要超级class

public abstract class TypeA {

    public void doSomething() {     

        // Calls for subclass behavior
        doSomethingElse();      
    }

    protected abstract String doSomethingElse();

}

子class实现

public class TypeB extends TypeA {

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

}

解决方案

此处给出的答案是正确的，如果涉及的 class 个在同一个包中，则将有效。

但是如果涉及不同的包，一种选择是使用 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 完成的测试。

Answer 1

您可以按以下方式使用 mockito 测试您的摘要 class

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

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

Answer 3

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

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

Answer 4

要模拟 returns 抽象无效的方法类我们可以使用：

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

我们可以根据需要用 Mockito.anyString() 等替换参数。

如何模拟从抽象 class 继承的受保护的 subclass 方法？

How to mock protected subclass method inherited from abstract class?

java

junit

unit-testing

mockito

powermock