package com.battery.business.runtime;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertFalse;
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
|
import java.io.InputStream;
|
import java.nio.charset.StandardCharsets;
|
import java.util.UUID;
|
import org.apache.ibatis.datasource.unpooled.UnpooledDataSource;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.core.io.ClassPathResource;
|
import org.springframework.jdbc.core.JdbcTemplate;
|
import org.springframework.jdbc.datasource.init.ScriptUtils;
|
|
/**
|
* 循环记录来源类别的迁移契约：设备没有循环汇总行时循环统计由工步层聚合得出，
|
* 该来源事实必须落在库里，界面与导出才能明确标注"这个数是推导的"。
|
*/
|
class CycleRecordSourceKindMigrationContractTest
|
{
|
@Test
|
void migrationAddsSourceKindColumnWithDeviceAsHistoricalDefault() throws Exception
|
{
|
String path = "db/migration-final/V23__cycle_record_source_kind.sql";
|
String migration = resource(path);
|
assertTrue(migration.contains("ADD COLUMN source_kind VARCHAR(16) NOT NULL DEFAULT 'DEVICE'"));
|
assertTrue(migration.contains("AFTER source_record_id"));
|
// 迁移只加列：不改写既有循环事实，也不写入任何数据。
|
assertFalse(migration.matches("(?s).*\\b(INSERT|UPDATE|DELETE|DROP)\\b.*"));
|
|
var source = new UnpooledDataSource("org.h2.Driver",
|
"jdbc:h2:mem:" + UUID.randomUUID() + ";MODE=MySQL;DATABASE_TO_LOWER=TRUE", "sa", "");
|
try (var connection = source.getConnection())
|
{
|
JdbcTemplate jdbc = new JdbcTemplate(source);
|
jdbc.execute("create table t_test_cycle_record(cycle_record_id bigint auto_increment primary key,"
|
+ "source_record_id varchar(128))");
|
// 迁移之前写入的循环记录按设备事实处理：语言上的默认值与存量一致。
|
jdbc.update("insert into t_test_cycle_record(source_record_id) values(?)", "CYCLE-1");
|
ScriptUtils.executeSqlScript(connection, new ClassPathResource(path));
|
|
assertEquals("DEVICE", jdbc.queryForObject(
|
"select source_kind from t_test_cycle_record where source_record_id = 'CYCLE-1'", String.class));
|
// 聚合来源是显式写入的事实，不是从别处推断出来的。
|
jdbc.update("insert into t_test_cycle_record(source_record_id, source_kind) values(?,?)",
|
"neware:442:cycle:3", "AGGREGATED");
|
assertEquals(1, jdbc.queryForObject(
|
"select count(*) from t_test_cycle_record where source_kind = 'AGGREGATED'", Integer.class));
|
}
|
}
|
|
private static String resource(String path)
|
{
|
try (InputStream stream = CycleRecordSourceKindMigrationContractTest.class.getClassLoader()
|
.getResourceAsStream(path))
|
{
|
if (stream == null) throw new AssertionError("缺少资源 " + path);
|
return new String(stream.readAllBytes(), StandardCharsets.UTF_8);
|
}
|
catch (Exception exception)
|
{
|
throw new AssertionError("无法读取资源 " + path, exception);
|
}
|
}
|
}
|
