package jnpf.dmsService;

import jnpf.dmsEntity.fenfa.DmsWenjianFenfaRenwuEntity;
import jnpf.dmsMapper.DmsWenjianFenfaMapper;
import org.junit.jupiter.api.*;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.springframework.core.io.ByteArrayResource;
import org.springframework.core.io.ClassPathResource;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.DriverManagerDataSource;
import java.nio.charset.StandardCharsets;
import java.util.*;
import static org.junit.jupiter.api.Assertions.*;

/** 使用随机隔离表验证部分打印可发放，但不会在剩余份数打印前结束分发。 */
@EnabledIfEnvironmentVariable(named="DMS_PRINT_TEST_JDBC_URL", matches=".+")
class DmsPartialPrintDistributionDatabaseTest {
    private final String prefix = "tpi_" + UUID.randomUUID().toString().replace("-", "").substring(0, 8) + "_";
    private final List<String> tables = List.of("dms_wenjian_dayin_renwu", "dms_wenjian_dayin_jilu", "dms_wenjian_fenfa_renwu");
    private JdbcTemplate jdbc;
    private DmsWenjianFenfaMapper mapper;
    private String isolated(String sql) {
        for (String table : tables) sql = sql.replace(table, prefix + table);
        return sql;
    }
    private void sql(String sql) { jdbc.execute(isolated(sql)); }
    @BeforeEach void setup() throws Exception {
        var ds = new DriverManagerDataSource(System.getenv("DMS_PRINT_TEST_JDBC_URL"),
                System.getenv("DMS_PRINT_TEST_DB_USER"), System.getenv("DMS_PRINT_TEST_DB_PASSWORD"));
        jdbc = new JdbcTemplate(ds);
        sql("CREATE TABLE dms_wenjian_dayin_renwu(f_id text, f_tenant_id text, biz_status text, f_delete_mark int)");
        sql("CREATE TABLE dms_wenjian_dayin_jilu(f_id text, f_tenant_id text, dayin_renwu_id text, biz_status text, f_delete_mark int)");
        sql("CREATE TABLE dms_wenjian_fenfa_renwu(f_id text PRIMARY KEY, f_tenant_id text, dayin_renwu_id text, fenfa_bianhao text, fafang_leixing text, biz_status text, f_creator_user_id text, f_creator_time timestamp, f_delete_mark int, jieshou_ren_id text, fafang_ren_id text, fafang_bumen_id text, fafang_shijian timestamp, beizhu_shuoming text, f_last_modify_user_id text, f_last_modify_time timestamp)");
        String xml;
        try (var in = new ClassPathResource("mapper/DmsWenjianFenfaMapper.xml").getInputStream()) {
            xml = new String(in.readAllBytes(), StandardCharsets.UTF_8);
        }
        var factory = new SqlSessionFactoryBean(); factory.setDataSource(ds);
        factory.setMapperLocations(new ByteArrayResource(isolated(xml).getBytes(StandardCharsets.UTF_8)));
        mapper = new SqlSessionTemplate(Objects.requireNonNull(factory.getObject())).getMapper(DmsWenjianFenfaMapper.class);
    }
    @AfterEach void cleanup() {
        if (jdbc != null) for (String table : tables) jdbc.execute("DROP TABLE IF EXISTS " + prefix + table);
    }
    @Test void partialCopiesCanBeIssuedAndOnlyFinalReceiptCompletesDistribution() {
        sql("INSERT INTO dms_wenjian_dayin_renwu VALUES ('task','tenant','bufendayin',NULL)");
        sql("INSERT INTO dms_wenjian_dayin_jilu VALUES ('first','tenant','task','daifafang',NULL)");
        var d = new DmsWenjianFenfaRenwuEntity();
        d.setId("distribution"); d.setTenantId("tenant"); d.setDayinRenwuId("task");
        d.setFenfaBianhao("number"); d.setFafangLeixing("zidongfenfa"); d.setBizStatus("daifafang");
        d.setCreatorUserId("issuer"); d.setCreatorTime(new Date());
        assertEquals(1, mapper.insertDistributionTaskIfAbsent(d));
        assertEquals(0, mapper.insertDistributionTaskIfAbsent(d));
        sql("UPDATE dms_wenjian_dayin_jilu SET biz_status='daihuishou'");
        assertEquals(0, finish(), "首批已签收，但尚有未打印份数，不得结束分发");
        sql("INSERT INTO dms_wenjian_dayin_jilu VALUES ('last','tenant','task','daifafang',NULL)");
        sql("UPDATE dms_wenjian_dayin_renwu SET biz_status='yidayin'");
        assertEquals(0, finish(), "全部打印完成但末批未签收");
        sql("UPDATE dms_wenjian_dayin_jilu SET biz_status='daihuishou'");
        assertEquals(1, finish());
        assertEquals(0, finish(), "完成操作保持幂等");
    }
    private int finish() {
        return mapper.completeDistributionTaskHandoverIfFinished("tenant", "distribution", "task",
                "receiver", "issuer", "department", null, new Date());
    }
}