package jnpf.dmsMapper;
|
|
import org.apache.ibatis.builder.xml.XMLMapperBuilder;
|
import org.apache.ibatis.io.Resources;
|
import org.apache.ibatis.mapping.MappedStatement;
|
import org.apache.ibatis.mapping.ResultMapping;
|
import org.apache.ibatis.session.Configuration;
|
import org.junit.jupiter.api.Test;
|
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.io.Reader;
|
import java.util.Arrays;
|
import java.util.List;
|
|
import static org.junit.jupiter.api.Assertions.assertFalse;
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
|
class DmsWenjianBanbenMapperXmlTest {
|
|
private static final List<String> MAPPERS = Arrays.asList(
|
"DmsWenjianBanbenKongzhiMapper",
|
"DmsWenjianXinzengGuanliMapper",
|
"DmsWenjianXinzengShenheMapper",
|
"DmsWenjianXiudingShenqingMapper",
|
"DmsWenjianXiudingShenheMapper");
|
|
@Test
|
void allVersionMappersParseAndAvoidSelectStar() throws IOException {
|
for (String mapper : MAPPERS) {
|
String resource = resource(mapper);
|
String xml;
|
try (Reader reader = Resources.getResourceAsReader(resource)) {
|
xml = read(reader);
|
}
|
assertFalse(xml.toUpperCase().contains("SELECT *"), resource);
|
parse(resource);
|
}
|
}
|
|
@Test
|
void flowAndVersionFieldsHaveExplicitMappings() throws IOException {
|
Configuration newAudit = parse(resource("DmsWenjianXinzengShenheMapper"));
|
assertMapping(newAudit,
|
DmsWenjianXinzengShenheMapper.class.getName() + ".selectByFlowForUpdate",
|
"f_flow_state", "flowState");
|
assertMapping(newAudit,
|
DmsWenjianXinzengShenheMapper.class.getName() + ".selectByFlowForUpdate",
|
"shengxiao_banben_xuhao", "shengxiaoBanbenXuhao");
|
|
Configuration control = parse(resource("DmsWenjianBanbenKongzhiMapper"));
|
assertMapping(control,
|
DmsWenjianBanbenKongzhiMapper.class.getName() + ".selectByFileForUpdate",
|
"biangeng_xuhao", "biangengXuhao");
|
}
|
|
@Test
|
void controlAdvanceChecksBothExpectedSequences() throws IOException {
|
Configuration configuration = parse(resource("DmsWenjianBanbenKongzhiMapper"));
|
MappedStatement statement = configuration.getMappedStatement(
|
DmsWenjianBanbenKongzhiMapper.class.getName() + ".advance");
|
String sql = statement.getBoundSql(null).getSql().replaceAll("\\s+", " ");
|
assertTrue(sql.contains("banben_xuhao = ?"));
|
assertTrue(sql.contains("biangeng_xuhao = ?"));
|
}
|
|
private Configuration parse(String resource) throws IOException {
|
Configuration configuration = new Configuration();
|
try (InputStream input = Resources.getResourceAsStream(resource)) {
|
XMLMapperBuilder builder = new XMLMapperBuilder(
|
input, configuration, resource, configuration.getSqlFragments());
|
builder.parse();
|
}
|
return configuration;
|
}
|
|
private void assertMapping(Configuration configuration, String statementId,
|
String column, String property) {
|
MappedStatement statement = configuration.getMappedStatement(statementId);
|
boolean mapped = statement.getResultMaps().stream()
|
.flatMap(resultMap -> resultMap.getResultMappings().stream())
|
.anyMatch(mapping -> matches(mapping, column, property));
|
assertTrue(mapped, column + " must map explicitly to " + property);
|
}
|
|
private boolean matches(ResultMapping mapping, String column, String property) {
|
return column.equals(mapping.getColumn()) && property.equals(mapping.getProperty());
|
}
|
|
private String read(Reader reader) throws IOException {
|
StringBuilder result = new StringBuilder();
|
char[] buffer = new char[2048];
|
int length;
|
while ((length = reader.read(buffer)) >= 0) {
|
result.append(buffer, 0, length);
|
}
|
return result.toString();
|
}
|
|
private String resource(String mapper) {
|
return "mapper/" + mapper + ".xml";
|
}
|
}
|
