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

Question

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

Answer 1

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

使用mockito，如果您正在测试一个列表并想要验证clear被调用了3次并且add被使用这些参数至少调用了一次，您可以执行以下操作：

List mock = mock(List.class);        
someCodeThatInteractsWithMock();                 

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

Source - MockitoVsEasyMock

Answer 2

在 Mockito 中你可以这样做：

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();

Answer 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";
        }
    }
}

Answer 4

您可以使用 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);

Answer 5

根据您想要计数的方法，您可以创建一个测试配置，并使用与您的类/包/方法匹配的 @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
}

您可以通过上述或 XML 配置使用 vanilla AspectJ 或 Spring AOP，如果你会发现这更容易。

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

Answer 6

听起来您可能需要一个测试间谍。例如，请参见 Mockito.spy()。

Answer 7

您有几个选项

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

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

3）使用分析器。探查器可以让您计算函数被调用的次数。这是一个非常手动的过程，因此它并不是真正为单元测试而设计的。

更好的解决方案是检查输出是否符合您的预期，而不是检查其内部工作原理。