Mockito间谍测试

时间:2012-04-23 20:27:15

标签: java spring mockito

我正在使用Mockito编写代码测试。但是我被困在以下情景中 - A类有2个方法,method1()和method2()。我尝试使用ArgumentCaptor来捕获发送到method2()的值。但是,由于我使用@Spy,我不能使用Matchers。

如何测试method1()?

class A{
    B b;
    method1(arg1, arg2){
       //some logic
       method2(arg1, arg2, ....argN);
    }

   method2(arg1, arg2,....argN){
       //some logic
       b.method3(arg1, arg2...);
   }
}

如何验证method2是否收到相同的参数值? 以下是我写的测试类:

Class TestA{

@Mock
B b;

@Spy
@InjectMocks   //required else b is null
A a = new A();

@Test
public void testMethod1(){

 a.method1(arg1, arg2);

  //How to verify method2 receives same argument values (arg1, arg2)????
  //verify(a, times(1)).method2(.......);   
}

}

3 个答案:

答案 0 :(得分:9)

我对这篇文章以及@David留下的评论很感兴趣所以我决定为像我一样的人编写一个有效的例子

/*
 * class to test
 */
public class A {

    public void methodA(String str) {
        this.methodB(str);
    }

    protected void methodB(String str) {
        // some implementation
    }
}

我们想断言传递给methodB()的值是我们所期望的。阅读ArgumentCaptor让我发现了等效的Captor Annotation

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.verify;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.Spy;
import org.mockito.runners.MockitoJUnitRunner;

@RunWith(MockitoJUnitRunner.class)
public class MultipleMethodCallTest {

    @Spy
    A a = new A();

    @Captor ArgumentCaptor<String> captor;

    @Test
    public void captureSecondMethodCallArgument() throws Exception {

        // EXPECTED

        String greeting = "hello world";

        // PERFORM TEST

        a.methodA(greeting);

        // ASSERT

        verify(a).methodB(captor.capture());

        assertEquals(greeting, captor.getValue());
    }
}

此示例已使用

进行了测试
  • 的Mockito-全1.8.5.jar
  • 的junit-4.8.2.jar

答案 1 :(得分:3)

你不能,你必须通过B的method3调用验证它。如果你对方法2的args对方法3没有影响,那么这些args可能毫无用处?!

答案 2 :(得分:2)

您可以使用具有间谍的匹配器;这很好用。我不知道为什么你认为你不能。

我拿了你的源代码并编辑它以使其编译。然后我添加了对MockitoAnnotations.initMocks的调用 - 你需要这个来创建间谍和模拟,并注入模拟(除非你使用MockitoJUnitRunner,它为你做initMocks) 。我将调用的verify重新放入method2。这很好用。

因此,与Omnaest的回答相反,您不需要使用B method3来验证这一点。我怀疑你唯一的问题是你忘记了initMocks

祝你好运,如果你需要更多帮助,请随时再发帖。