如何在构造函数中禁止私有方法调用?

时间:2014-07-03 18:03:36

标签: java unit-testing mocking powermock

我有一个非常简单的类,它有一个私有方法。问题是如何抑制此方法调用?

这是我的代码:

public class Example { 
    private int value;

    public Example(int value) {
        this.value = value;
        method();
    }

    private void method() {
        throw new RuntimeException();
    }

    public int getValue() {
        return value;
    }
}

测试代码(至少尝试):

public void test() throws Exception {

    PowerMockito.doNothing().when(Example.class, "method");
    final int EXPECTED_VALUE = 1;
    Example example = new Example(EXPECTED_VALUE);
    int RETRIEVED_VALUE = example.getValue();

    assertEquals(RETRIEVED_VALUE, EXPECTED_VALUE);
    verifyPrivate(Example.class, times(1)).invoke("method");
}

UPD

对我来说,遵守这两个术语非常重要:

  1. PowerMockito.doNothing().when(Example.class, "method");
  2. PowerMockito.verifyPrivate(Example.class, times(1)).invoke("method");

1 个答案:

答案 0 :(得分:0)

由于您无法修改测试中的代码。我认为没有一个完美的解决方案。您需要部分模拟 Example实例。

List list = new LinkedList();
List spy = spy(list);
//You have to use doReturn() for stubbing
doReturn("foo").when(spy).get(0);

但是你不能这样做,因为你必须首先实例化你的对象。


所以我建议由两个测试组成的以下解决方法。第一个测试从类中删除私有method,实例化Example并验证Example是否已正确初始化。 第二个测试实例化Example并验证RuntimeException(私有method副作用)。

import static org.junit.Assert.assertEquals;
import static org.powermock.api.support.membermodification.MemberMatcher.method;
import static org.powermock.api.support.membermodification.MemberModifier.suppress;

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(Example.class)
public class ExampleTest {
    @Test
    public void constructor_should_initialize_the_v2alue() throws Exception {
        suppress(method(Example.class, "method"));

        final int EXPECTED_VALUE = 1;
        Example example = PowerMockito.spy(new Example(EXPECTED_VALUE));
        int RETRIEVED_VALUE = example.getValue();

        assertEquals(RETRIEVED_VALUE, EXPECTED_VALUE);
    }

    @Test(expected=RuntimeException.class)
    public void constructor_should_invoke_the_private_method() throws Exception {
        new Example(1);
    }
}