package jnpf.dmsMapper;
|
|
import jnpf.dmsEntity.storage.approval.DmsApprovedBatch;
|
import org.apache.ibatis.builder.xml.XMLMapperBuilder;
|
import org.apache.ibatis.io.Resources;
|
import org.apache.ibatis.session.Configuration;
|
import org.junit.jupiter.api.Test;
|
import java.util.UUID;
|
import static org.junit.jupiter.api.Assertions.*;
|
|
class DmsApprovedBatchMapperXmlTest {
|
@Test void receiptStatementsParseAndScannerOnlySelectsWaitingBatches() throws Exception {
|
Configuration config = new Configuration();
|
String path = "mapper/DmsApprovedBatchMapper.xml";
|
try (java.io.InputStream stream = Resources.getResourceAsStream(path)) {
|
new XMLMapperBuilder(stream, config, path, config.getSqlFragments()).parse();
|
}
|
String prefix = DmsApprovedBatchMapper.class.getName() + ".";
|
org.apache.ibatis.mapping.BoundSql insert = config.getMappedStatement(prefix + "insertBatch").getBoundSql(new DmsApprovedBatch());
|
assertEquals(14, insert.getParameterMappings().size());
|
String insertItem = config.getMappedStatement(prefix + "insertItem")
|
.getBoundSql(new jnpf.dmsEntity.storage.approval.DmsApprovedItem()).getSql();
|
assertTrue(insertItem.contains("file_id"));
|
assertTrue(insertItem.contains("business_version_id"));
|
assertTrue(insertItem.contains("shifou_shoukong"));
|
assertTrue(config.getResultMap(prefix + "Item").getResultMappings().stream()
|
.anyMatch(r -> "shifou_shoukong".equals(r.getColumn())
|
&& "shifouShoukong".equals(r.getProperty())));
|
String due = config.getMappedStatement(prefix + "due").getBoundSql(null).getSql();
|
assertTrue(due.contains("status = 'WAITING'")); assertFalse(due.contains("FAILED")); assertFalse(due.contains("EFFECTIVE"));
|
assertFalse(due.contains("REGISTERING"));
|
String registered = config.getMappedStatement(prefix + "registered").getBoundSql(null).getSql();
|
assertTrue(registered.contains("SET status = 'WAITING'"));
|
assertTrue(registered.contains("AND status = 'REGISTERING'"));
|
String activate = config.getMappedStatement(prefix + "activate").getBoundSql(null).getSql();
|
assertTrue(activate.contains("AND status IN ('WAITING','FAILED')"));
|
String checksum = config.getMappedStatement(prefix + "storeChecksum").getBoundSql(null).getSql();
|
assertTrue(checksum.contains("tenant_id = ? AND batch_id = ? AND source_key = ?"));
|
assertTrue(checksum.contains("AND checksum_sha256 IS NULL"));
|
assertTrue(config.getResultMap(prefix + "Batch").getResultMappings().stream()
|
.anyMatch(r -> "source_digest".equals(r.getColumn()) && "sourceDigest".equals(r.getProperty())));
|
assertTrue(config.getResultMap(prefix + "Batch").getResultMappings().stream()
|
.anyMatch(r -> "approval_render_snapshot".equals(r.getColumn())
|
&& "approvalRenderSnapshot".equals(r.getProperty())));
|
assertTrue(config.getResultMap(prefix + "Item").getResultMappings().stream()
|
.anyMatch(r -> "storage_receipt".equals(r.getColumn())
|
&& "storageReceipt".equals(r.getProperty())));
|
}
|
|
@Test void disablingBackoffOnlyRemovesRetryTimeFromBothSelectionAndClaim() throws Exception {
|
Configuration config = new Configuration();
|
String path = "mapper/DmsApprovedBatchMapper.xml";
|
try (java.io.InputStream stream = Resources.getResourceAsStream(path)) {
|
new XMLMapperBuilder(stream, config, path, config.getSqlFragments()).parse();
|
}
|
for (boolean enabled : new boolean[]{true,false}) {
|
java.util.Map<String,Object> params = new java.util.HashMap<>();
|
params.put("retryBackoffEnabled",enabled);
|
params.put("includeDisable",false);
|
for (String name : new String[]{"retryCandidates","claim"}) {
|
String sql = config.getMappedStatement(DmsApprovedBatchMapper.class.getName()+"."+name)
|
.getBoundSql(params).getSql().replaceAll("\\s+"," ");
|
assertEquals(enabled,sql.contains("next_attempt_at <= CURRENT_TIMESTAMP"));
|
assertTrue(sql.contains("status <> 'APPLIED'"));
|
assertTrue(sql.contains("lease_until IS NULL OR lease_until <= CURRENT_TIMESTAMP"));
|
assertTrue(sql.contains("status <> 'WAITING' OR scheduled_at <= CURRENT_TIMESTAMP"));
|
}
|
}
|
}
|
|
@Test void uuidAllocationBypassesMybatisTransactionLocalCache() throws Exception {
|
Configuration config = new Configuration();
|
String path = "mapper/DmsApprovedBatchMapper.xml";
|
try (java.io.InputStream stream = Resources.getResourceAsStream(path)) {
|
new XMLMapperBuilder(stream, config, path, config.getSqlFragments()).parse();
|
}
|
org.apache.ibatis.mapping.MappedStatement statement = config.getMappedStatement(
|
DmsApprovedBatchMapper.class.getName() + ".nextUuidV7");
|
org.apache.ibatis.executor.BaseExecutor executor = new org.apache.ibatis.executor.BaseExecutor(config, null) {
|
protected int doUpdate(org.apache.ibatis.mapping.MappedStatement s, Object p) { throw new UnsupportedOperationException(); }
|
protected java.util.List<org.apache.ibatis.executor.BatchResult> doFlushStatements(boolean rollback) { return java.util.List.of(); }
|
@SuppressWarnings("unchecked")
|
protected <E> java.util.List<E> doQuery(org.apache.ibatis.mapping.MappedStatement s, Object p,
|
org.apache.ibatis.session.RowBounds r, org.apache.ibatis.session.ResultHandler h,
|
org.apache.ibatis.mapping.BoundSql b) { return (java.util.List<E>) java.util.List.of(UUID.randomUUID().toString()); }
|
protected <E> org.apache.ibatis.cursor.Cursor<E> doQueryCursor(org.apache.ibatis.mapping.MappedStatement s,
|
Object p, org.apache.ibatis.session.RowBounds r, org.apache.ibatis.mapping.BoundSql b) { throw new UnsupportedOperationException(); }
|
};
|
java.util.List<Object> first = executor.query(statement, null, org.apache.ibatis.session.RowBounds.DEFAULT, null);
|
java.util.List<Object> second = executor.query(statement, null, org.apache.ibatis.session.RowBounds.DEFAULT, null);
|
assertNotEquals(first, second, "A second allocation in the same transaction must execute the query again");
|
assertFalse(statement.isUseCache(), "Allocation must also bypass second-level caching");
|
}
|
|
@Test void sqlSessionFactoryParsesApprovalMappersBeforeUuidPostProcessorRuns() throws Exception {
|
com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean factory =
|
new com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean();
|
factory.setDataSource(org.mockito.Mockito.mock(javax.sql.DataSource.class));
|
factory.setMapperLocations(
|
new org.springframework.core.io.ClassPathResource("mapper/DmsApprovedBatchMapper.xml"),
|
new org.springframework.core.io.ClassPathResource("mapper/DmsVersionStateMapper.xml"));
|
// Production registers UUID globally only after the SqlSessionFactory has been built.
|
Configuration config = factory.getObject().getConfiguration();
|
for (String map : new String[]{DmsApprovedBatchMapper.class.getName() + ".Batch",
|
DmsApprovedBatchMapper.class.getName() + ".Item", DmsVersionStateMapper.class.getName() + ".State"}) {
|
for (org.apache.ibatis.mapping.ResultMapping mapping : config.getResultMap(map).getResultMappings()) {
|
if (UUID.class.equals(mapping.getJavaType()))
|
assertInstanceOf(jnpf.dmsDatabase.DmsUuidTypeHandler.class, mapping.getTypeHandler());
|
}
|
}
|
}
|
}
|
