Выделенный частный метод с PowerMock, но базовый метод все еще называется
Я пытаюсь смоделировать частный метод, который делает вызов JNDI. Когда этот метод вызывается из модульного теста, он генерирует исключение ^. Я хотел бы макетировать этот метод для целей тестирования. Я использовал пример кода из другого ответа на вопрос, и пока тест проходит, кажется, что базовый метод все еще вызывается. Я вставил System.err.println()
в метод doTheGamble()
, и он распечатывается на моей консоли.
Довольно интересно, если я закомментирую первый assertThat
, тест пройден.? :(
Итак, как мне смоделировать закрытый метод, чтобы он не вызывался?
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;
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;
import java.util.Random;
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(CodeWithPrivateMethod.class)
public class PowerMock_Test {
static boolean gambleCalled = false;
@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);
/* 1 */ assertThat( PowerMock_Test.gambleCalled, is(false) );
spy.meaningfulPublicApi();
/* 2 */ assertThat( PowerMock_Test.gambleCalled, is(false) );
}
}
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();
System.err.println( "\n>>> GAMBLE CALLED <<<\n" );
PowerMock_Test.gambleCalled = true;
return gamble;
}
}
^ понятно, поскольку мое рабочее пространство не поддерживает JNDI, только производственная среда поддерживает
Я использую последние версии всех библиотек, JUnit 4.10, Mockito 1.8.5, Hamcrest 1.1, Javassist 3.15.0 и PowerMock 1.4.10.
Ответы
Ответ 1
Из Пример приватного метода PowerMock:
@RunWith(PowerMockRunner.class)
// We prepare PartialMockClass for test because it final or we need to mock private or static methods
@PrepareForTest(PartialMockClass.class)
public class YourTestCase {
@Test
public void privatePartialMockingWithPowerMock() {
PartialMockClass classUnderTest = PowerMockito.spy(new PartialMockClass());
// use PowerMockito to set up your expectation
PowerMockito.doReturn(value).when(classUnderTest, "methodToMock", "parameter1");
// execute your test
classUnderTest.execute();
// Use PowerMockito.verify() to verify result
PowerMockito.verifyPrivate(classUnderTest, times(2)).invoke("methodToMock", "parameter1");
}
Итак, чтобы применить это к вашему коду, я думаю, что он может стать:
@RunWith(PowerMockRunner.class)
@PrepareForTest(CodeWithPrivateMethod.class)
public class PowerMock_Test {
@Test(expected = RuntimeException.class)
public void when_gambling_is_true_then_always_explode() throws Exception {
CodeWithPrivateMethod spy = PowerMockito.spy(new CodeWithPrivateMethod());
PowerMockito.doReturn(true).when(spy, "doTheGamble", anyString(), anyInt());
/* 1 */ PowerMockito.verifyPrivate(spy, times(0)).invoke("doTheGamble", anyString(), anyInt());
spy.meaningfulPublicApi();
/* 2 */ PowerMockito.verifyPrivate(spy, times(2)).invoke("doTheGamble", anyString(), anyInt());
}
}
Я просто закодировал это в редакторе. На самом деле тесты не выполнялись, и никакие ошибки не пострадали при разработке этого кода.
Ответ 2
ArtB,
Просто вставьте полный код, который отлично работает в моей Eclipse IDE. Я только изменил ожидания, сказанные в моем последнем посте. Удачи.
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;
import static org.mockito.Matchers.anyInt;
import static org.mockito.Matchers.anyString;
import static org.powermock.api.support.membermodification.MemberMatcher.method;
import java.util.Random;
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(CodeWithPrivateMethod.class)
public class PowerMock_Test {
static boolean gambleCalled = false;
@Test(expected = RuntimeException.class)
public void when_gambling_is_true_then_always_explode() throws Exception {
CodeWithPrivateMethod spy = PowerMockito.spy(new CodeWithPrivateMethod());
// PowerMockito.doReturn(true).when(spy, "doTheGamble", anyString(), anyInt());
PowerMockito.doReturn(true).when(spy,
method(CodeWithPrivateMethod.class, "doTheGamble", String.class, int.class))
.withArguments(anyString(), anyInt());
assertThat( PowerMock_Test.gambleCalled, is(false) );
spy.meaningfulPublicApi();
assertThat( PowerMock_Test.gambleCalled, is(false) );
}
}
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();
System.err.println( "\n>>> GAMBLE CALLED <<<\n" );
PowerMock_Test.gambleCalled = true;
return gamble;
}
}
Ответ 3
ArtB,
Вы уверены, что ваш код не работает (или) я что-то здесь не вижу? Я просто заменил ваше математическое ожидание следующим образом, как предложил Майк, и он отлично работает:
PowerMockito.doReturn(true).when(spy,
method(CodeWithPrivateMethod.class, "doTheGamble", String.class, int.class))
.withArguments(anyString(), anyInt());
Я никогда не использовал Powermockito, но раньше использовал Mockito.
Ответ 4
когда вы используете spy
для создания макетного объекта, его реальный наполовину поддерживаемый объект. Я бы предпочел избавиться от spy
. Работа над чистым фальшивым объектом.