Annotation注解自定义+测试（2）

继上一篇：Annotation注解自定义+测试：Annotation注解自定义+测试_上弦暮光的博客-CSDN博客

newTest——进行测试

AnnotationKit——要被测试的工具类

import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.Collection;
import java.util.List;

import org.springframework.web.method.HandlerMethod;

import com.br.common.annotation.RequirePermission;
import com.br.kit.JavaKit;
import com.br.kit.collection.ListKit;


public class AnnotationKit {

    public static <T extends Annotation> T getAnnotation(Object handler,Class<T> clazz) {
        if(handler instanceof HandlerMethod) {
            return getAnnotation((HandlerMethod)handler,clazz);
        }
        return null;
    }
    /**
     * 从方法上获得注解，如果没有，尝试从类上获得
     * @param <T>
     * @param handlerMethod
     * @param clazz
     * @return
     */
    public static <T extends Annotation> T getAnnotation(HandlerMethod handlerMethod,Class<T> clazz) {
        final Method method = handlerMethod.getMethod();
        T annotation = method.getAnnotation(clazz);
        if(annotation==null) {
            final Class<?> declaringClass = method.getDeclaringClass();
            annotation = declaringClass.getAnnotation(clazz);
        }
        return annotation;
    }
    
    /**
     * 从方法上获得注解RequirePermission，如果没有，尝试从类上获得
     * @param <T>
     * @param handlerMethod
     * @param clazz
     * @return
     */
    public static RequirePermission getRequirePermission(HandlerMethod handlerMethod) {
        return getAnnotation(handlerMethod,RequirePermission.class);
    }

    /**
     * 是否有权限
     * @param needPermission  需要的权限，比如"permission",或"permission1,permission2"
     * @param modulePermissionList  模块权限，比如：菜单url等前端用的，模块code，后端用的
     * @return
     */
    public static boolean hasPermission(RequirePermission needPermission, Collection<String> modulePermissionList) {
        if(ListKit.isEmpty(modulePermissionList)){
            return false;
        }
        if(needPermission==null) {
            return true;
        }
        String permissionName = needPermission.value();
        List<String> perssmionNameList = JavaKit.splitAsStrList(permissionName);
        for (String onePerssmionName : perssmionNameList) {
            final boolean hasPermission = modulePermissionList.contains(onePerssmionName);
            return hasPermission;
        }
        return false;
    }

    /**
     * 是否有权限
     * 
     * @param needPermission
     *            需要的权限，比如"permission1,permission2"。只要url权限和code权限，有其中1个（目前支持or），就判定有权限
     * @param moduleUrlList
     *            模块权限，比如：菜单url，前端用的
     * @param moduleCodeList
     *            模块权限，比如code，后端用的
     * @return
     */
    public static boolean hasPermission(RequirePermission needPermission, Collection<String> moduleUrlList,
        Collection<String> moduleCodeList) {
        return hasPermission(needPermission, moduleUrlList) || hasPermission(needPermission, moduleCodeList);
    }
}

newTest：主要测试hasPermission函数

import java.util.ArrayList;
import java.util.Collection;

import org.junit.Test;
import org.springframework.web.method.HandlerMethod;

import com.br.common.annotation.RequirePermission;

import io.swagger.annotations.ApiModelProperty;

public class newTest {
    @Test
    public void getAnnotation1() throws NoSuchMethodException {

        AnnotationTestVo annotationTestVo = new AnnotationTestVo();
        HandlerMethod handlerMethod = new HandlerMethod(annotationTestVo, "getName", String.class, int.class);

        ApiModelProperty apiModelProperty = AnnotationKit.getAnnotation(handlerMethod, ApiModelProperty.class);
        System.out.println(apiModelProperty.value());
        System.out.println(apiModelProperty.name());
    }

    @Test
    public void hasPermissionTest() throws NoSuchMethodException {

        Collection<String> stringCollection = new ArrayList<>();
        String s1 = "123456";
        String s2 = "qweasd";
        stringCollection.add(s1);
        stringCollection.add(s2);

        RequirePermission requirePermission = null;

        boolean b = AnnotationKit.hasPermission(requirePermission, stringCollection);
        System.out.println(b);
    }

    @Test
    public void hasPermissionTest2() throws NoSuchMethodException {

        AnnotationTestVo annotationTestVo = new AnnotationTestVo();
        HandlerMethod handlerMethod = new HandlerMethod(annotationTestVo, "getName", String.class, int.class);

        RequirePermission requirePermission = AnnotationKit.getAnnotation(handlerMethod, RequirePermission.class);

        System.out.println(requirePermission.value());

        Collection<String> stringCollection = new ArrayList<>();
        String s1 = "123456";
        String s2 = "qweasd";
        stringCollection.add(s1);
        stringCollection.add(s2);

        boolean b = AnnotationKit.hasPermission(requirePermission, stringCollection);
        System.out.println(b);
    }
}