web-dev-qa-db-ja.com

PowerMockを使用してプライベートメソッドをテストする方法は?

プライベートメソッドを呼び出すパブリックメソッドでテストしたいクラスがあります。私はプライベートメソッドが正しく機能すると仮定したいと思います。たとえば、doReturn....when...のようなものが欲しいです。 PowerMockを使用した可能な解決策 があることがわかりましたが、この解決策はうまくいきません。どうすればできますか?誰もこの問題を抱えていましたか?

59

ここには問題はありません。 Mockito APIを使用した次のコードで、私はまさにそれを行うことができました。

public class CodeWithPrivateMethod {

    public void meaningfulPublicApi() {
        if (doTheGamble("Whatever", 1 << 3)) {
            throw new RuntimeException("boom");
        }
    }

    private boolean doTheGamble(String whatever, int binary) {
        Random random = new Random(System.nanoTime());
        boolean gamble = random.nextBoolean();
        return gamble;
    }
}

そして、これがJUnitテストです。

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;
import static org.mockito.Matchers.anyInt;
import static org.mockito.Matchers.anyString;
import static org.powermock.api.mockito.PowerMockito.when;
import static org.powermock.api.support.membermodification.MemberMatcher.method;

@RunWith(PowerMockRunner.class)
@PrepareForTest(CodeWithPrivateMethod.class)
public class CodeWithPrivateMethodTest {

    @Test(expected = RuntimeException.class)
    public void when_gambling_is_true_then_always_explode() throws Exception {
        CodeWithPrivateMethod spy = PowerMockito.spy(new CodeWithPrivateMethod());

        when(spy, method(CodeWithPrivateMethod.class, "doTheGamble", String.class, int.class))
                .withArguments(anyString(), anyInt())
                .thenReturn(true);

        spy.meaningfulPublicApi();
    }
}
92
Brice

テストフレームワーク(ifクラスがnon _final)で動作する一般的なソリューションは、独自のモックを手動で作成することです。

  1. プライベートメソッドをprotectedに変更します。
  2. テストクラスでクラスを拡張します
  3. 以前にプライベートなメソッドをオーバーライドして、必要な定数を返します

これはフレームワークを使用しないため、それほどエレガントではありませんが、PowerMockがなくても常に機能します。または、ステップ#1を既に実行している場合は、Mockitoを使用してステップ#2および#3を実行できます。

プライベートメソッドを直接モックするには、 other answer に示すようにPowerMockを使用する必要があります。

21
Tobogganski

なんらかの理由で、Briceの答えはうまくいきません。動作させるために少し操作することができました。 PowerMockの新しいバージョンを持っているからかもしれません。 1.6.5を使用しています。

import Java.util.Random;

public class CodeWithPrivateMethod {

    public void meaningfulPublicApi() {
        if (doTheGamble("Whatever", 1 << 3)) {
            throw new RuntimeException("boom");
        }
    }

    private boolean doTheGamble(String whatever, int binary) {
        Random random = new Random(System.nanoTime());
        boolean gamble = random.nextBoolean();
        return gamble;
    }
}

テストクラスは次のようになります。

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;

import static org.mockito.Matchers.anyInt;
import static org.mockito.Matchers.anyString;
import static org.powermock.api.mockito.PowerMockito.doReturn;

@RunWith(PowerMockRunner.class)
@PrepareForTest(CodeWithPrivateMethod.class)
public class CodeWithPrivateMethodTest {
    private CodeWithPrivateMethod classToTest;

    @Test(expected = RuntimeException.class)
    public void when_gambling_is_true_then_always_explode() throws Exception {
        classToTest = PowerMockito.spy(classToTest);

        doReturn(true).when(classToTest, "doTheGamble", anyString(), anyInt());

        classToTest.meaningfulPublicApi();
    }
}
1
gaoagong

私はあなたがプライベート関数をmockitoでテストすることができる方法nyを知っています

@Test
    public  void  commandEndHandlerTest() throws  Exception
    {
        Method retryClientDetail_privateMethod =yourclass.class.getDeclaredMethod("Your_function_name",null);
        retryClientDetail_privateMethod.setAccessible(true);
        retryClientDetail_privateMethod.invoke(yourclass.class, null);
    }
1
Arun Yadav