单元测试中的计数方法调用

时间:2011-10-08 05:26:49

标签: unit-testing junit4 mockito jmockit

在单元测试中计算方法调用的最佳方法是什么。任何测试框架都允许这样做吗?

7 个答案:

答案 0 :(得分:53)

听起来你可能想要使用模拟框架通常提供的.expects(1)类型方法。

使用mockito,如果您正在测试List并想要验证clear已被调用3次并且使用这些参数至少调用了一次,则执行以下操作:

List mock = mock(List.class);        

someCodeThatInteractsWithMock();                 

verify(mock, times(3)).clear();
verify(mock, atLeastOnce()).add(anyObject());      

(来自http://code.google.com/p/mockito/wiki/MockitoVSEasyMock

答案 1 :(得分:19)

In Mockito you can do something like this:

YourService serviceMock = Mockito.mock(YourService.class);

// code using YourService

// details of all invocations including methods and arguments
Collection<Invocation> invocations = Mockito.mockingDetails(serviceMock).getInvocations();
// just a number of calls of any mock's methods
int numberOfCalls = invocations.size();

答案 2 :(得分:7)

您可以使用Mockito中的接口Answer来计算方法调用次数。

ConnectionPool mockedConnectionPool = mock(ConnectionPool.class);

    final int[] counter = new int[1];

    when(mockedConnectionPool.getConnection()).then(new Answer<Connection>() {

        @Override
        public Connection answer(InvocationOnMock invocation) throws Throwable {
            counter[0]++;
            return conn;
        }

    }); 
    // some your code

    assertTrue(counter[0] == 1);

答案 3 :(得分:3)

根据您想要计算的方法,您可以创建一个测试配置,其@Before建议与您的类/包/方法匹配:

import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;

@Aspect
public class MethodCounterAspect {

    private int counter = 0 // or inject the Counter object into this aspect

    @Pointcut( "execution( * com.sample.your.package.*.*(..) )" )
    public void methodsToCount() {}

    @Before("methodsToCount()")
    public void execute() throws Throwable {
        counter++; // or update the counter injected into this aspect..
    }

    // get the counter
}

如果您觉得更容易,可以使用上面的通用AspectJ或Spring AOP或XML配置。

如果需要,您可以创建不同的切入点/方面。

答案 4 :(得分:3)

给定一个示例类“RoleRepository”,其中包含一个返回角色的方法“getRole(String user)”。

假设您已将此对象声明为Mock或Spy,并且您想要检查是否一次调用方法getRole(String)。

您可以执行以下操作:Mockito.verify(roleRepository, Mockito.times(1)).getRole(Mockito.anyString());

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.mockito.Spy;
import org.mockito.junit.MockitoJUnitRunner;

@RunWith(MockitoJUnitRunner.class)
public class RoleRepositoryTest {

    @Spy
    private RoleRepository roleRepository = new RoleRepository();

    @Test
    public void test() {
        roleRepository.getRole("toto");
        Mockito.verify(roleRepository, Mockito.times(1)).getRole(Mockito.anyString());
    }

    public static class RoleRepository {

        public String getRole(String user) {
            return "MyRole";
        }
    }
}

答案 5 :(得分:2)

听起来你可能需要test spy。例如,请参阅Mockito.spy()

答案 6 :(得分:1)

你有几个选择

1)添加一些特殊的代码来计算函数中的调用。它会起作用,但它不是一个很好的解决方案。

2)运行单元测试后,检查代码覆盖率。大多数覆盖工具都会计算调用次数,但它们实际上是为后期处理而设计的。

3)使用分析器。分析器将让您计算调用函数的次数。这是一个非常手动的过程,因此它并非真正用于单元测试。

更好的解决方案是检查输出是否符合预期,而不是检查内部的工作方式。