web-dev-qa-db-ja.com

保護されたメソッドのモック

継承された保護されたメソッドをモックしたい。このメソッドをJavaコードから直接呼び出すことはできません。別のパッケージのクラスから継承されているためです。このメソッドをwhen(...)

package a;

public class A() {
    protected int m() {}
}

package b;

public class B extends a.A {
    // this class currently does not override m method from a.A
    public asd() {}
}

// test
package b;

class BTest {
    @Test
    public void testClass() {
        B instance = PowerMockito.spy(new B());
        PowerMockito.when(instance, <specify a method m>).thenReturn(123);
        //PowerMockito.when(instance.m()).thenReturn(123); -- obviously does not work
    }
}

私は見た PowerMockito.whenオーバーライドし、これはすべてプライベートメソッド専用であるようです!

保護されたメソッドを指定する方法は?

16

簡単に言うと、スパイをスタブするために常にwhenを使用できるわけではありません。 doReturnを使用します。

spyおよびdoReturn(両方PowerMockito)の静的インポートを想定すると、

_@RunWith(PowerMockRunner.class)
@PrepareForTest(B.class)
public class BTest {
    @Test public void testClass() throws Exception {
        B b = spy(new B());
        doReturn(42).when(b, "m");
        b.asd();
    }
}
_

@PrepareForTest(A.class)を使用して、when(a, "m")doReturnを設定することもできます。どちらがより意味があるかは、実際のテストによって異なります。

25
Dave Newton