JMockit 返回集合

Question

我有以下代码：

public void someMethod() {
     Set<Foo> fooSet = bar.getFoos();

     for(Foo foo: fooSet) {
         foo.doSomething();
     }
}

我想使用 JMockit 测试它，但不确定如何返回特定类型和大小的集合。

当尝试将 foo 添加到 foo 集合时，我的代码的以下测试会引发哈希码的空指针异常。

@Test
public void someTestMethod()
{
     new Expectations()
     {
         @Mocked Bar bar;
         @Mocked Foo foo;


         Set<Foo> foos = new HashSet<Foo>();
         foos.add(foo);

         bar.getFoos(); returns(foos);
         foo.doSomething();
     };

     new SomeClass().someMethod();
}

这应该怎么做呢？

Answer 1

我不太确定如何回答你的问题，因为我不明白你要测试什么，但我相信你想要这样的东西：

@Test
public void someTestMethod(@Mocked(methods="getFoos")final Bar mockedBar
                           @Mocked(methods="doSomething")final Foo mockedFoo {

   final Set<Foo> foos = new HashSet<Foo>();
   foos.add(new Foo()); 

   new Expectations() {
      {
         mockedBar.getFoos(); returns(foos);
         mockedFoo.doSomething();
      }
   };

   new SomeClass().someMethod();
}

使用这个，JMockit 将模拟对 getFoos 的调用并返回设置foos。如果您查看我传入的参数，就会发现我正在对 Bar 和 Foo 类进行部分模拟（我只是模拟对 getFoos 和 doSomething 方法的调用）。我还注意到您在 new Expectations 块中缺少一组大括号，因此这肯定会给您带来一些问题。您必须记住的另一问题是，如果您在 Set foos 中放入多个对象，则使用 E​​xpectations 而不是 NonStrictExpectations 会导致错误，因为它只期望对 进行一次调用做某事。如果您创建一个测试用例，其中 foos 包含多个对象，您可以使用 NonStrictExpectations 或使用 minTimes 和 maxTimes 来指定调用计数约束