java - 使用 PowerMock 模拟时避免 javax/ejb/EJBLocalObject 异常

Question

我正在尝试对需要依赖项的类进行单元测试。直接创建依赖类是不可能的，因为依赖的构造函数具有只需要在运行时可用的对象的逻辑。

但是，当我尝试模拟依赖类时，出现“java.lang.NoClassDefFoundError: javax/ejb/EJBLocalObject”异常。我该如何解决这个问题？目前我能想到的唯一解决方案是将 ClassToTest 更改为使用接口(interface)而不是实际的具体依赖类。

一些代码来说明我目前是如何尝试模拟依赖的:

package mockTest;

import org.junit.Test;
import org.powermock.api.mockito.PowerMockito;

public class MockTest {

    @Test
    public void performTest() {
        // Mock the dependency and create the class to test
        Dependency dependency = PowerMockito.mock(Dependency.class);
        ClassToTest classToTest = new ClassToTest(dependency);

       // Invoke a method in classToTest, assert..
    }
}

进一步说明:

问:Dependent类是接口(interface)的impl吗？

ClassToTest(依赖类)是一个具体类并且没有实现任何接口(interface)，尽管它很容易实现——我控制源代码。 Dependency 类是一个不实现接口(interface)的具体类，我无法控制源代码。

问:您是否只传入“依赖”对象，因为它是在普通构造函数中创建的，所以通过提供一个对象，您尝试私有(private)化工作 Mock 而不是创建的 Mock？

是的。 ClassToTest 使用 Dependency 的方法导致 Dependency 例如进行 JDBC 调用。我希望能够传入实际的依赖项(来自实现代码)或模拟(来自测试代码)。

问:依赖是静态的吗？

不，Dependency 或 Dependency 中的方法不是静态的。

Answer 1

拥有一个需要大量依赖对象的类构造函数(运行时的单例静态等)在创建模拟测试时总是很痛苦。我能想到的最佳方法是结合使用 Powermockito.whenNew() 调用和 Mockito.any(Class.class) 调用。这样，当您的构造函数被调用时，您可以 Hook 所需的对象。如果您在依赖对象上遵循类似的单例模式，则可以模拟这些类 getInstance() 调用以返回您在测试中实例化的模拟副本。

示例:

Dependency dependMock = PowerMockito.mock(Dependency.class);
PowerMockito.whenNew(Dependency.class).withNoArguments().thenReturn(dependMock);

或者，如果 Dependency 需要为您的二级依赖传入参数:

Dependency dependMock = PowerMockito.mock(Dependency.class);
PowerMockito.whenNew(Dependency.class)
  .withArguments(Mockito.any(LevelTwoDependency.class)).thenReturn(dependMock);

请注意，当使用 whenNew 时，您需要为您正在测试的类设置一个钩子(Hook)，以便代码被触发。这可以通过为测试类添加一些模拟注释来完成。

@RunWith(PowerMockRunner.class)
@PrepareForTest({ClassYouAreTesting.class})
public class ClassTest{
    // code    
}

这允许您使用 Mocks 完全控制您的二级依赖对象。通过将它们从逻辑中移除(通过提供 Mocked 功能)，您可以进入测试的核心，最终测试您所讨论的类的纯功能。通过从这些依赖类中删除任何奇怪的行为，您可以消除代码中存在错误的风险，这些错误最终会破坏并为使用这些依赖类的任何类提供错误的测试结果。

通过将这些 Mock 链接在一起，您基本上可以使用其他模拟来模拟主依赖类及其所有子依赖类。根据构造函数的工作方式和您为这些对象设置私有(private)变量，它可能会变得有点困惑。