模拟注入SessionContext

Question

我将SessionContext作为资源注入到EJB (实现容器管理事务)：

@Stateless(name = "XXX", mappedName = "PPP-MMM-CCC")
@TransactionManagement(value=TransactionManagementType.CONTAINER)
public class Xxx implements ItsRemoteInterface {
   @Resource
   private SessionContext sctx;
   ....
}

我的单元测试失败了，因为"sctx“在运行(NullPointerException).时为空因此，我认为解决这个问题的唯一方法是创建一个实现FakeSessionContext的SessionContext类，然后我可以在测试期间使用它。

instance = new Xxx();
sessionContextResourceField = Xxx.class.getDeclaredField("sctx");
sessionContextResourceField.setAccessible(true);
sessionContextResourceField.set(instance, new FakeInitialContext());

但在我这么做之前，我想知道是否有一种更优雅的方式？除了创建一个FakeSessionContext类之外？就像工厂里的那种？

如果感兴趣，我使用的是jUnit 4.10和jmockit 0.999.15。

Answer 1

使用jmockit为您创建会话上下文的模拟版本，并使用jmockit版本的“when.返回”语句来确保模拟上下文返回所需的值。

Answer 2

为了其他人的利益，在jmockit中就是这样做的：

sessionContextResourceField.set(instance, 
            new MockUp<SessionContext>() {
                @Mock boolean getRollbackOnly() { return false; }
            }.getMockInstance());

虽然在我的例子中，不需要伪装getRollbackOnly (我在代码中使用)，但我还是这么做了，以防将来行为发生变化。