6

如何使用 EasyMock 修改模拟方法的可变方法参数？

例如，我有一个使用 BlockingQueue 的类。我想模拟 BlockingQueue 成员进行单元测试。我的班级调用方法 queue.drainTo(Collection c)。调用此方法会从队列中移除元素并将它们添加到集合中。我将如何使用 EasyMock 模拟这种行为？例子会很棒。

4

2 回答 2

13

您可以使用andAnswer和getCurrentArguments：

public void testDrainToQueue() {
  BlockingQueue<Foo> queue = EasyMock.createMock(BlockingQueue.class);
  EasyMock.expect(queue.drainTo(EasyMock.isA(List.class)))
      .andAnswer(new IAnswer<Integer>() {
        public Integer answer() {
          ((List) EasyMock.getCurrentArguments()[0]).add(new Foo(123));
          return 1; // 1 element drained
        }
      });
  EasyMock.replay(queue);
  ...
}

它有时有助于提取帮助类或方法：

private static IAnswer<Integer> fakeDrainReturning(final List drainedElements) {
  return new IAnswer<Integer() {
    @Override public Integer answer() {
      ((List) EasyMock.getCurrentArguments()[0]).addAll(drainedElements);
      return drainedElements.size();
    }
  };
}

然后你可以这样做：

List<Foo> drainedElements = Arrays.asList(new Foo(123), new Foo(42));
EasyMock.expect(queue.drainTo(EasyMock.isA(List.class)))
    .andAnswer(fakeDrainReturning(drainedElements));

在您希望从队列中删除数据的方法之前，最好使用实数BlockingQueue并找到一种将所需值插入队列的方法。

于 2010-09-08T08:37:34.137 回答
0

很难准确地说出您的代码是什么样的。如果我知道您要测试的代码，可以为您提供更好的帮助。但假设您要测试的代码如下所示：

private BlockingQueue<String> queue;
private List<String>  myList = new ArrayList<String> ():

private void setBlockingQueue( BlockingQueue<String>  queue ) { 
    this.queue = queue;
}

public List<String> getMyList() {
    return myList;
}

public void setMyList( List<String> myList) {
    this.myList = myList;
}

public void doWork() {
    System.out.println("blah");
    queue.drainTo( myList );
}

一个测试将是

public void testDoWork() {
    List<String> stuffToDrain = new ArrayList<String>();
    stuffToDrain.add( "foo" );
    stuffToDrain.add( "bar" );
    myTestingClass.setMyList( stuffToTrain );


    BlockingQueue<String> queue = EasyMock.createMock( BlockingQueue.class  );
    myTestingClass.setBlockingQueue( queue );
    queue.drainTo( stuffToDrain );
    EasyMock.replay( queue );
    myTestingClass.doWork();
    EasyMock.verify( queue );



}

抱歉，如果那不正确，但是很难建议对我看不到的代码进行测试...

编辑 - 我们不能真正断言可变参数会因为我们使用模拟的方式而改变。我们所能做的就是断言 drainTo 方法被调用。如果 drainTo 做了我们想做的事情，就必须在其他地方进行测试.. 即在 BlockingQueue.class 的测试中

编辑 2 - 我们可以更具体地了解我们期望调用该方法的列表。

于 2010-08-25T19:36:30.543 回答