1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package com.iailab.framework.datapermission.core.aop;
 
import cn.hutool.core.collection.CollUtil;
import com.iailab.framework.datapermission.core.annotation.DataPermission;
import com.iailab.framework.test.core.ut.BaseMockitoUnitTest;
import org.aopalliance.intercept.MethodInvocation;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.InjectMocks;
import org.mockito.Mock;
 
import java.lang.reflect.Method;
 
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.when;
 
/**
 * {@link DataPermissionAnnotationInterceptor} 的单元测试
 *
 * @author iailab
 */
public class DataPermissionAnnotationInterceptorTest extends BaseMockitoUnitTest {
 
    @InjectMocks
    private DataPermissionAnnotationInterceptor interceptor;
 
    @Mock
    private MethodInvocation methodInvocation;
 
    @BeforeEach
    public void setUp() {
        interceptor.getDataPermissionCache().clear();
    }
 
    @Test // 无 @DataPermission 注解
    public void testInvoke_none() throws Throwable {
        // 参数
        mockMethodInvocation(TestNone.class);
 
        // 调用
        Object result = interceptor.invoke(methodInvocation);
        // 断言
        assertEquals("none", result);
        assertEquals(1, interceptor.getDataPermissionCache().size());
        assertTrue(CollUtil.getFirst(interceptor.getDataPermissionCache().values()).enable());
    }
 
    @Test // 在 Method 上有 @DataPermission 注解
    public void testInvoke_method() throws Throwable {
        // 参数
        mockMethodInvocation(TestMethod.class);
 
        // 调用
        Object result = interceptor.invoke(methodInvocation);
        // 断言
        assertEquals("method", result);
        assertEquals(1, interceptor.getDataPermissionCache().size());
        assertFalse(CollUtil.getFirst(interceptor.getDataPermissionCache().values()).enable());
    }
 
    @Test // 在 Class 上有 @DataPermission 注解
    public void testInvoke_class() throws Throwable {
        // 参数
        mockMethodInvocation(TestClass.class);
 
        // 调用
        Object result = interceptor.invoke(methodInvocation);
        // 断言
        assertEquals("class", result);
        assertEquals(1, interceptor.getDataPermissionCache().size());
        assertFalse(CollUtil.getFirst(interceptor.getDataPermissionCache().values()).enable());
    }
 
    private void mockMethodInvocation(Class<?> clazz) throws Throwable {
        Object targetObject = clazz.newInstance();
        Method method = targetObject.getClass().getMethod("echo");
        when(methodInvocation.getThis()).thenReturn(targetObject);
        when(methodInvocation.getMethod()).thenReturn(method);
        when(methodInvocation.proceed()).then(invocationOnMock -> method.invoke(targetObject));
    }
 
    static class TestMethod {
 
        @DataPermission(enable = false)
        public String echo() {
            return "method";
        }
 
    }
 
    @DataPermission(enable = false)
    static class TestClass {
 
        public String echo() {
            return "class";
        }
 
    }
 
    static class TestNone {
 
        public String echo() {
            return "none";
        }
 
    }
 
}