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

Question

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

换句话说，在mock“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进行测试。

- Evandro Pomatti

---

这种方法在Java 8、JUnit 4.12和PowerMock 1.6.4中会导致NullPointerException。 - Mack

@ Mack，你解决了吗？ - ceyun

@ceyun，非常抱歉，因为已经过了很长时间，我不记得是否解决了这个问题，如果我解决了，也不记得怎么解决的。也许其中一个点赞我的评论的人还记得？ - Mack

Answer 1

您可以在对抽象方法进行mock时使用Mockito.CALLS_REAL_METHODS。这将调用类的原始方法，您可以自己mock所有的抽象方法。

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

- mszalbach

---

第一种选择失败了，因为TypeA是抽象的。第二个选项在使用spy()时有效，但当我尝试将其应用到我的生产代码中时，它会给我返回“来自类型<子类名称>的方法<方法名称>不可见”的错误。我正在努力找出原因。 - Evandro Pomatti

实际上你的测试是正确的，但是我的类在不同的包中，这就是为什么我之前无法做到的原因。当方法被保护时，它们不能从其他包中访问。 - Evandro Pomatti

第一种选项也有效吗？我已经使用Mockito 1.10.19和最新的beta版本进行了测试，两者都按预期工作。 - mszalbach

确实可以工作。我忘了告诉你，我的Mockito版本非常古老，是1.9.0，因为我使用的是Java 5。 - Evandro Pomatti

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

- Florian Schaetz

---

它可以工作，但我刚意识到我的生产代码中有不同包中的类。受保护的方法在这种情况下是不可见的。 - Evandro Pomatti

这会成为一个问题。一个可能的解决方案是使用PowerMock，它应该能够验证私有方法，据我所知。尽管我从未使用过。 - Florian Schaetz

Answer 3

你可以使用mockito来测试你的抽象类，具体方法如下。

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

- jozzy

Answer 4

为了模拟抽象类中返回 void 的方法，我们可以使用：

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

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

- Aftab