Examples with IMethodInterceptor org.testng.IMethodInterceptor used on opensource projects

Example 1 with IMethodInterceptor

use of org.testng.IMethodInterceptor in project infinispan by infinispan.

the class ChainMethodInterceptor method intercept.

@Override
public List<IMethodInstance> intercept(List<IMethodInstance> methods, ITestContext context) {
    try {
        Set<Class<? extends IMethodInterceptor>> interceptorSet = new HashSet<>();
        List<Class<? extends IMethodInterceptor>> interceptorList = new ArrayList<>();
        Set<Class<? extends Predicate<IMethodInstance>>> filters = new HashSet<>();
        for (IMethodInstance method : methods) {
            findInterceptors(method.getInstance().getClass(), interceptorSet, interceptorList, filters);
        }
        if (!filters.isEmpty()) {
            List<? extends Predicate<IMethodInstance>> filterInstances = filters.stream().map(clazz -> {
                try {
                    return clazz.getConstructor().newInstance();
                } catch (Exception e) {
                    throw new IllegalStateException("Cannot construct filter", e);
                }
            }).collect(Collectors.toList());
            ArrayList<IMethodInstance> filteredMethods = new ArrayList<>(methods.size());
            METHODS: for (IMethodInstance m : methods) {
                for (Predicate<IMethodInstance> filter : filterInstances) {
                    if (hasFilter(m.getInstance().getClass(), filter.getClass()) && !filter.test(m))
                        continue METHODS;
                }
                filteredMethods.add(m);
            }
            methods = filteredMethods;
        }
        for (Class<? extends IMethodInterceptor> interceptor : interceptorList) {
            methods = interceptor.getConstructor().newInstance().intercept(methods, context);
        }
        return methods;
    } catch (Throwable t) {
        MethodInstance methodInstance = FakeTestClass.newFailureMethodInstance(new TestNGException(t), context.getCurrentXmlTest(), context, methods.get(0).getInstance());
        return Collections.singletonList(methodInstance);
    }
}