代码之家 › 专栏 › 技术社区 › Evandro Pomatti

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

powermock mockito junit unit-testing java

Evandro Pomatti · 技术社区 · 9 年前

如何使用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);     

    }
}

注: 使用Java 5、jUnit 4.11、Mockito 1.9.0和PowerMock 1.4.12进行测试。

4 回复 | 直到 9 年前

mszalbach 9 年前

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

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

Florian Schaetz 9 年前

我建议使用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();

jozzy 9 年前

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

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

Aftab 8 年前

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

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

我们可以用Mockito替换参数。anyString()等。