Как издеваться над приватным методом для тестирования с помощью PowerMock? - PullRequest
59 голосов
/ 18 октября 2011

У меня есть класс, который я хотел бы протестировать с помощью открытого метода, который вызывает закрытый.Я хотел бы предположить, что закрытый метод работает правильно.Например, я бы хотел что-то вроде doReturn....when....Я обнаружил, что существует возможное решение с использованием PowerMock , но это решение не работает для меняКак это можно сделать?У кого-нибудь была эта проблема?

Ответы [ 4 ]

92 голосов
/ 18 октября 2011

Я не вижу здесь проблемы.С помощью следующего кода с использованием 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();
    }
}
21 голосов
/ 19 октября 2011

Универсальное решение, которое будет работать с любой платформой тестирования (, если ваш класс не является final), заключается в том, чтобы вручную создать свой собственный макет.

  1. Изменить свой личныйметод для защищенного.
  2. В вашем тестовом классе расширьте класс
  3. , переопределяя ранее закрытый метод, чтобы вернуть любую постоянную, какую вы хотите

Это не использует никакойфреймворк не такой элегантный, но он всегда будет работать: даже без PowerMock.Кроме того, вы можете использовать Mockito для выполнения шагов № 2 и № 3, если вы уже выполнили шаг № 1.

Чтобы напрямую смоделировать приватный метод, вам нужно использовать PowerMock, как показано в другом ответе .

1 голос
/ 06 апреля 2018

Почему-то ответ Брайса на меня не работает. Я смог немного манипулировать им, чтобы заставить его работать. Это может быть только потому, что у меня есть более новая версия 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 голос
/ 24 августа 2017

Я знаю способ, которым вы можете вызвать вашу личную функцию для тестирования в mockito

@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);
    }
...