package jnpf.dmsMapper;
|
|
import org.apache.ibatis.builder.xml.XMLMapperBuilder;
|
import org.apache.ibatis.mapping.Environment;
|
import org.apache.ibatis.session.*;
|
import org.junit.jupiter.api.*;
|
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
|
import org.mybatis.spring.SqlSessionTemplate;
|
import org.mybatis.spring.transaction.SpringManagedTransactionFactory;
|
import org.springframework.jdbc.datasource.SingleConnectionDataSource;
|
import java.io.StringReader;
|
import java.nio.charset.StandardCharsets;
|
import java.sql.*;
|
import java.util.*;
|
import static org.junit.jupiter.api.Assertions.*;
|
|
/** Real shared usage predicate and native picker; only session-private TEMP tables are written. */
|
@EnabledIfEnvironmentVariable(named="DMS_OCCUPANCY_TEST_URL",matches=".+")
|
class DmsNumberUsagePostgresTest {
|
private Connection connection;
|
private SingleConnectionDataSource source;
|
private SqlSessionTemplate sessions;
|
private DmsWenjianBianhaoMapper mapper;
|
private final String tenant="number-usage-"+UUID.randomUUID();
|
@BeforeEach void setup() throws Exception {
|
connection=DriverManager.getConnection(System.getenv("DMS_OCCUPANCY_TEST_URL"),
|
System.getenv("DMS_OCCUPANCY_TEST_USER"),System.getenv("DMS_OCCUPANCY_TEST_PASSWORD"));
|
source=new SingleConnectionDataSource(connection,true);
|
for(String table:List.of("dms_wenjian_bianhao","dms_wenjian_ziyuan","dms_wenjian_xinzeng_shenhe",
|
"dms_documents","dms_wenjian_mingcheng_biangeng_shenqing")) {
|
try(var statement=connection.createStatement()) {
|
statement.execute("CREATE TEMP TABLE "+table+" AS SELECT * FROM public."+table+" WITH NO DATA");
|
}
|
}
|
Configuration config=new Configuration(new Environment("test",new SpringManagedTransactionFactory(),source));
|
for(String resource:List.of("mapper/BusinessFileOccupancyMapper.xml","mapper/DmsWenjianBianhaoMapper.xml")) {
|
try(var in=getClass().getClassLoader().getResourceAsStream(resource)) {
|
String xml=new String(Objects.requireNonNull(in).readAllBytes(),StandardCharsets.UTF_8).replace("public.","pg_temp.");
|
new XMLMapperBuilder(new StringReader(xml),config,resource,config.getSqlFragments()).parse();
|
}
|
}
|
sessions=new SqlSessionTemplate(new SqlSessionFactoryBuilder().build(config));
|
mapper=sessions.getMapper(DmsWenjianBianhaoMapper.class);
|
execute("INSERT INTO dms_wenjian_bianhao(f_id,f_tenant_id,wenjian_bianhao,biz_status) VALUES ('number',?,'N-001','using')",tenant);
|
}
|
@AfterEach void close() throws Exception {if(source!=null)source.destroy();else if(connection!=null)connection.close();}
|
private void execute(String sql,Object...values) throws SQLException {
|
try(var p=connection.prepareStatement(sql)){for(int i=0;i<values.length;i++)p.setObject(i+1,values[i]);p.execute();}
|
}
|
private List<?> candidates(String id) {
|
return sessions.selectList(DmsWenjianBianhaoMapper.class.getName()+".selectNameChangeAvailablePage",
|
Map.of("tenantId",tenant,"bianhaoId",id,"keyword",""));
|
}
|
@Test void unusedNumberIsAvailableButSavedReviewDraftBlocksEvenExactIdLookup() throws Exception {
|
assertFalse(mapper.numberAlreadyUsed(tenant,"number"));assertEquals(1,candidates("").size());
|
execute("INSERT INTO dms_wenjian_xinzeng_shenhe(f_id,f_tenant_id,wenjian_bianhao_id,f_flow_state) VALUES ('draft',?,'number',0)",tenant);
|
assertTrue(mapper.numberAlreadyUsed(tenant,"number"));assertTrue(candidates("").isEmpty());assertTrue(candidates("number").isEmpty());
|
}
|
@Test void ExactNumberUseAlsoBlocksAndOtherTenantDoesNot() throws Exception {
|
execute("INSERT INTO dms_wenjian_xinzeng_shenhe(f_id,f_tenant_id,wenjian_bianhao,f_flow_state) VALUES ('draft',?,'N-001',0)","other-tenant");
|
assertFalse(mapper.numberAlreadyUsed(tenant,"number"));
|
execute("UPDATE dms_wenjian_xinzeng_shenhe SET f_tenant_id=?",tenant);
|
assertTrue(mapper.numberAlreadyUsed(tenant,"number"));
|
}
|
@Test void historicalResourceStillCountsAsUsed() throws Exception {
|
execute("INSERT INTO dms_wenjian_ziyuan(f_id,f_tenant_id,wenjian_bianhao,biz_status,f_delete_mark) VALUES ('file',?,'N-001','expired',1)",tenant);
|
assertTrue(mapper.numberAlreadyUsed(tenant,"number"));assertTrue(candidates("").isEmpty());
|
}
|
@Test void centerDocumentCountsAsUsed() throws Exception {
|
execute("INSERT INTO dms_documents(id,tenant_id,document_no) VALUES (?::uuid,?,'N-001')",UUID.randomUUID().toString(),tenant);
|
assertTrue(mapper.numberAlreadyUsed(tenant,"number"));assertTrue(candidates("").isEmpty());
|
}
|
}
|
