package jnpf.base.service;
|
|
import jnpf.base.mapper.UploadEditDraftMapper;
|
import org.apache.ibatis.builder.xml.XMLMapperBuilder;
|
import org.apache.ibatis.datasource.unpooled.UnpooledDataSource;
|
import org.apache.ibatis.mapping.Environment;
|
import org.apache.ibatis.session.*;
|
import org.apache.ibatis.transaction.jdbc.JdbcTransactionFactory;
|
import org.junit.*;
|
import java.util.List;
|
import static org.junit.Assert.*;
|
|
/** Executes the production lock SQL on connection-local copies of actual PostgreSQL columns. */
|
public class UploadEditDraftMapperPostgresTest {
|
private SqlSession session;
|
@Before public void setup() throws Exception {
|
Assume.assumeTrue(System.getenv("DMS_READ_TEST_URL") != null);
|
var source = new UnpooledDataSource("org.postgresql.Driver", System.getenv("DMS_READ_TEST_URL"),
|
System.getenv("DMS_READ_TEST_USER"), System.getenv("DMS_READ_TEST_PASSWORD"));
|
var config = new Configuration(new Environment("lock-test", new JdbcTransactionFactory(), source));
|
String resource = "mapper/UploadEditDraftMapper.xml";
|
try (var input = getClass().getClassLoader().getResourceAsStream(resource)) {
|
new XMLMapperBuilder(input, config, resource, config.getSqlFragments()).parse();
|
}
|
session = new SqlSessionFactoryBuilder().build(config).openSession(false);
|
for (String table : List.of("dms_wenjian_xinzeng_shenhe", "dms_wenjian_xiuding_shenhe")) {
|
try (var statement = session.getConnection().createStatement()) {
|
statement.execute("CREATE TEMP TABLE " + table + " ON COMMIT DROP AS SELECT f_id,f_tenant_id,f_delete_mark FROM public." + table + " WITH NO DATA");
|
statement.execute("INSERT INTO " + table + "(f_id,f_tenant_id,f_delete_mark) VALUES('record','test',NULL),('deleted','test',1)");
|
}
|
}
|
}
|
@After public void close() { if (session != null) { try { session.rollback(true); } finally { session.close(); } } }
|
@Test public void locksOnlyExistingRecordOfCorrectTenantInAllowedAuditTable() {
|
var mapper = session.getMapper(UploadEditDraftMapper.class);
|
for (String table : List.of("dms_wenjian_xinzeng_shenhe", "dms_wenjian_xiuding_shenhe")) {
|
assertEquals("record", mapper.lockDmsRecord("test", table, "record"));
|
assertNull(mapper.lockDmsRecord("other", table, "record"));
|
assertNull(mapper.lockDmsRecord("test", table, "missing"));
|
assertNull(mapper.lockDmsRecord("test", table, "deleted"));
|
}
|
assertNull(mapper.lockDmsRecord("test", "workflow_task", "record"));
|
}
|
}
|
