Java单元测试：替换被测私有方法

Question

在运行 JUnit 测试时，有什么方法可以替换私有方法中的逻辑？

一点背景知识：我们有一些私有方法与 OSGi 容器中的包交互。这在单元测试中不可用，因此方法将失败。

我们已经查看了 JMockIt，但方法替换功能似乎要强制您替换类中相互调用的所有方法。

实现是这样的：

public final doSomething() {  
    firstThing();
    secondThing();
}
private firstThing() {  
    // normal code
}
private secondThing() {  
    // code which is unavailable in a unit test
}

并且单元测试会指定 secondThing() 的新实现：

// replace secondThing() in impl with this secondThing()

private secondThing() {  
    // dummy code
}

// run tests

Answer 1

您当然可以使用 JMockit 解决这种情况。 一种方法是定义一个“模拟”类，例如：

public class MyTest
{
    @Test
    public void testDoSomething()
    {
        new MockUp<ClassWhichDependsOnOtherBundles>()
        {
            @Mock
            void secondThing()
            {
               // do anything here
            }
        };

        new ClassWhichDependsOnOtherBundles().doSomething();
    }
}

只有模拟类中的secondThing() 方法会被JMockit 替换。 JMockit Expectations API 也可以通过部分模拟来使用。

Answer 2

您正在将您的实现与 osgi 对象的创建耦合（在 secondThing() 或类本身中执行它）。如果您从外部将实现传递给您的类，则可以在测试时使用存根/模拟。

Answer 3

我也认为依赖注入可以解决这个问题。 如果你不想在你的项目中使用另一个框架，而这是唯一会出现问题的地方，你可以为 secondThing 定义一个接口并为此设置两个实现，一个用于原始代码，一个用于单元测试。

Answer 4

我的建议 - 重新设计您的应用程序。如果你想改变 private 方法的行为：

• 将其设为 protected / public 并在模拟对象中覆盖它
• 将功能从方法中移到一个辅助类中，该类是可注入的（通过依赖注入）。然后模拟该助手并将模拟注入到被测类中，而不是原来的助手。

一种解决方法可能是一些字节码操作技术，但我不建议这样做。

Answer 5

有一个很好的存根模式

ProductionClass.java：

public class ProductionClass { ... //default visibility to make it available for stubbing void firstThing(){...} ... }

BlaTest.java（与生产类相同的包）：

public class BlaTest { @Test void test_xx(){ //use stubbed impl ProductionClass sut = new ProductionClassStubbed(); sut.doSomething(); } } class ProductionClassStubbed extends ProductionClass{ @Override void firstThing(){ //stub out fill in what you want (like dummy behaviour) } }

一个不同的东西。我在您的示例代码中看到了 final 修饰符。小心使用 final 修饰符。它们不利于可测试性。仅在确实需要时使用。