刘光辉
13 小时以前 34981c30a78e8bbd7791131059a9210f9928b62c
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
package jnpf.dmsMapper;
 
import jnpf.dmsDatabase.DmsUuidTypeHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.session.Configuration;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
 
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
 
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;
 
class DmsFilePermissionMapperSqlTest {
    private Configuration configuration;
 
    @BeforeEach
    void setUp() {
        configuration = new Configuration();
        configuration.getTypeHandlerRegistry().register(UUID.class, DmsUuidTypeHandler.class);
        configuration.addMapper(DmsFilePermissionMapper.class);
    }
 
    @Test
    void rendersMixedActionsAndResourcesAsOneCandidateQuery() {
        Map<String, Object> parameters = new HashMap<>();
        parameters.put("tenantId", "tenant-1");
        parameters.put("actionCodes", Arrays.asList("READ", "UPDATE"));
        parameters.put("resourceIds", Arrays.asList(UUID.randomUUID(), UUID.randomUUID()));
 
        String sql = sql("selectCandidates", parameters);
 
        assertTrue(sql.startsWith("SELECT id, tenant_id"));
        assertTrue(sql.contains("action_code IN ( ? , ? )"));
        assertTrue(sql.contains("resource_id IN ( ? , ? )"));
        assertEquals(1, occurrences(sql, "FROM public.dms_file_permissions"));
    }
 
    private String sql(String method, Map<String, Object> parameters) {
        MappedStatement statement = configuration.getMappedStatement(
                DmsFilePermissionMapper.class.getName() + "." + method);
        BoundSql boundSql = statement.getBoundSql(parameters);
        return boundSql.getSql().replaceAll("\\s+", " ").trim();
    }
 
    private int occurrences(String value, String token) {
        int result = 0;
        int offset = 0;
        while ((offset = value.indexOf(token, offset)) >= 0) {
            result++;
            offset += token.length();
        }
        return result;
    }
}