青年IT男

个人从事金融行业,就职过易极付、思建科技等重庆一流技术团队,目前就职于某网约车平台负责整个支付系统建设。自身对金融行业有强烈的爱好。同时也实践大数据、数据存储、自动化集成和部署、分布式微服务、人工智能等领域。

JMockit高级用法 -在Mock时做AOP

JMockit高级用法 -在Mock时做AOP

在学习MockUp(API:MockUp)时, MockUp可以自定义方法体,覆盖类原有的方法。 JMockit给我们提供了一个特别的方法,用于拦截所有方法,可以让我们在基于被Mock的方法做AOP。 比如我们想测试每个方法的时间性能。


//通过在mock时做AOP测试方法的时间性能 public class MethodCostPerformanceTest { // 测试SayHello类每个方法的时间性能 @Test public void testSayHelloCostPerformance() { // 把方法的调用时间记录到costMap中。key是方法名称,value是平均调用时间 Map<String, Long> costMap = new HashMap<String, Long>(); new MockUp<SayHello>() { @Mock public Object $advice(Invocation invocation) { long a = System.currentTimeMillis(); Object result = invocation.proceed(); long cost = System.currentTimeMillis() - a; // 把某方法的平均调用时间记录下来 String methodName = invocation.getInvokedMember().getName(); Long preCost = costMap.get(methodName); if (preCost == null) { costMap.put(methodName, cost); } else { costMap.put(methodName, (preCost + cost) / 2); } return result; } }; SayHello sayHello = new SayHello(); sayHello.sayHello("david", ISayHello.MALE); sayHello.sayHello("lucy", ISayHello.FEMALE); for (Iterator<String> iterator = costMap.keySet().iterator(); iterator.hasNext();) { String methodName = (String) iterator.next(); // 期望每个方法的调用时间不超过20ms Assert.isTrue(costMap.get(methodName) < 20); } } }
0
1028826685@qq.com