mirror of
https://github.com/Syngnat/GoNavi.git
synced 2026-08-18 04:44:18 +08:00
53d0480058
- 移除 Oracle 元数据附带的 ALTER TRIGGER ENABLE 语句 - 保留触发器主体与 DISABLE 状态，避免禁用触发器被误启用 - 补充触发器状态语句回归测试 Fixes #825
249 lines
8.8 KiB
Go
249 lines
8.8 KiB
Go
package db
|
||
|
||
import (
|
||
"database/sql/driver"
|
||
"slices"
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
func TestOracleGetTriggersUsesDBMSMetadataDDL(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
dbConn, state := openOracleRecordingDB(t)
|
||
triggerListQuery := buildOracleTriggersQuery("H2", "T_MEMCARD_CASH")
|
||
triggerDDLQuery := `SELECT DBMS_METADATA.GET_DDL('TRIGGER', 'TR_T_MEMCARD_CASH', 'H2') as ddl FROM DUAL`
|
||
metadataDDL := `CREATE OR REPLACE TRIGGER "H2"."TR_T_MEMCARD_CASH"
|
||
BEFORE INSERT ON "H2"."T_MEMCARD_CASH"
|
||
BEGIN
|
||
NULL;
|
||
END;`
|
||
|
||
state.mu.Lock()
|
||
state.queryResults[triggerListQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"OWNER", "TABLE_OWNER", "TABLE_NAME", "TRIGGER_NAME", "TRIGGER_TYPE", "TRIGGERING_EVENT", "WHEN_CLAUSE", "TRIGGER_BODY"},
|
||
rows: [][]driver.Value{
|
||
{"H2", "H2", "T_MEMCARD_CASH", "TR_T_MEMCARD_CASH", "BEFORE EACH ROW", "INSERT", nil, "SOURCE HIDDEN"},
|
||
},
|
||
}
|
||
state.queryResults[triggerDDLQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"DDL"},
|
||
rows: [][]driver.Value{
|
||
{metadataDDL},
|
||
},
|
||
}
|
||
state.mu.Unlock()
|
||
|
||
oracleDB := &OracleDB{conn: dbConn}
|
||
triggers, err := oracleDB.GetTriggers("H2", "T_MEMCARD_CASH")
|
||
if err != nil {
|
||
t.Fatalf("GetTriggers 返回错误: %v", err)
|
||
}
|
||
if len(triggers) != 1 {
|
||
t.Fatalf("期望返回 1 个触发器，实际 %#v", triggers)
|
||
}
|
||
if !strings.Contains(triggers[0].Statement, `CREATE OR REPLACE TRIGGER "H2"."TR_T_MEMCARD_CASH"`) {
|
||
t.Fatalf("期望返回 DBMS_METADATA 完整 DDL，实际: %s", triggers[0].Statement)
|
||
}
|
||
if strings.Contains(triggers[0].Statement, "SOURCE HIDDEN") {
|
||
t.Fatalf("触发器语句不应继续返回 SOURCE HIDDEN: %s", triggers[0].Statement)
|
||
}
|
||
if queries := state.snapshotQueries(); !slices.Contains(queries, triggerDDLQuery) {
|
||
t.Fatalf("期望查询 DBMS_METADATA 获取触发器 DDL，实际 queries=%v", queries)
|
||
}
|
||
}
|
||
|
||
func TestOracleGetTriggersOmitsMetadataEnableStatement(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
dbConn, state := openOracleRecordingDB(t)
|
||
triggerListQuery := buildOracleTriggersQuery("H2", "T_MEMCARD_CASH")
|
||
triggerDDLQuery := `SELECT DBMS_METADATA.GET_DDL('TRIGGER', 'TR_T_MEMCARD_CASH', 'H2') as ddl FROM DUAL`
|
||
metadataDDL := `CREATE OR REPLACE TRIGGER "H2"."TR_T_MEMCARD_CASH"
|
||
BEFORE INSERT ON "H2"."T_MEMCARD_CASH"
|
||
BEGIN
|
||
NULL;
|
||
END;
|
||
ALTER TRIGGER "H2"."TR_T_MEMCARD_CASH" ENABLE;`
|
||
|
||
state.mu.Lock()
|
||
state.queryResults[triggerListQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"OWNER", "TABLE_OWNER", "TABLE_NAME", "TRIGGER_NAME", "TRIGGER_TYPE", "TRIGGERING_EVENT", "WHEN_CLAUSE", "TRIGGER_BODY"},
|
||
rows: [][]driver.Value{
|
||
{"H2", "H2", "T_MEMCARD_CASH", "TR_T_MEMCARD_CASH", "BEFORE EACH ROW", "INSERT", nil, "SOURCE HIDDEN"},
|
||
},
|
||
}
|
||
state.queryResults[triggerDDLQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"DDL"},
|
||
rows: [][]driver.Value{
|
||
{metadataDDL},
|
||
},
|
||
}
|
||
state.mu.Unlock()
|
||
|
||
oracleDB := &OracleDB{conn: dbConn}
|
||
triggers, err := oracleDB.GetTriggers("H2", "T_MEMCARD_CASH")
|
||
if err != nil {
|
||
t.Fatalf("GetTriggers 返回错误: %v", err)
|
||
}
|
||
if len(triggers) != 1 {
|
||
t.Fatalf("期望返回 1 个触发器，实际 %#v", triggers)
|
||
}
|
||
|
||
statement := triggers[0].Statement
|
||
if strings.Contains(strings.ToUpper(statement), "ALTER TRIGGER") || strings.Contains(strings.ToUpper(statement), " ENABLE") {
|
||
t.Fatalf("触发器定义不应包含 Oracle 自动追加的启用语句: %s", statement)
|
||
}
|
||
if !strings.Contains(statement, `CREATE OR REPLACE TRIGGER "H2"."TR_T_MEMCARD_CASH"`) {
|
||
t.Fatalf("触发器定义主体丢失: %s", statement)
|
||
}
|
||
}
|
||
|
||
func TestStripOracleTriggerEnableStatementPreservesDisableState(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
ddl := `CREATE OR REPLACE TRIGGER "H2"."TR_T_MEMCARD_CASH"
|
||
BEFORE INSERT ON "H2"."T_MEMCARD_CASH"
|
||
BEGIN
|
||
NULL;
|
||
END;
|
||
ALTER TRIGGER "H2"."TR_T_MEMCARD_CASH" DISABLE;`
|
||
|
||
if got := stripOracleTriggerEnableStatement(ddl); got != ddl {
|
||
t.Fatalf("清理 ENABLE 语句不应改写 DISABLE 状态，got=%q", got)
|
||
}
|
||
}
|
||
|
||
func TestOracleGetTriggersDoesNotTruncateMetadataCLOB(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
dbConn, state := openOracleRecordingDB(t)
|
||
triggerListQuery := buildOracleTriggersQuery("H2", "T_MEMCARD_REG")
|
||
triggerDDLQuery := `SELECT DBMS_METADATA.GET_DDL('TRIGGER', 'TR_T_MEMCARD_REG', 'H2') as ddl FROM DUAL`
|
||
fullDDL := `CREATE OR REPLACE TRIGGER "H2"."TR_T_MEMCARD_REG"
|
||
BEFORE INSERT OR UPDATE ON "H2"."T_MEMCARD_REG"
|
||
FOR EACH ROW
|
||
BEGIN
|
||
` + strings.Repeat(" NULL;\n", 700) + ` -- FULL_TRIGGER_DDL_TAIL
|
||
END;`
|
||
|
||
state.mu.Lock()
|
||
state.queryResults[triggerListQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"OWNER", "TABLE_OWNER", "TABLE_NAME", "TRIGGER_NAME", "TRIGGER_TYPE", "TRIGGERING_EVENT", "WHEN_CLAUSE", "TRIGGER_BODY"},
|
||
rows: [][]driver.Value{
|
||
{"H2", "H2", "T_MEMCARD_REG", "TR_T_MEMCARD_REG", "BEFORE EACH ROW", "INSERT OR UPDATE", nil, "SOURCE HIDDEN"},
|
||
},
|
||
}
|
||
state.queryResults[triggerDDLQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"DDL"},
|
||
columnTypes: []string{"OCICLOBLOCATOR"},
|
||
rows: [][]driver.Value{
|
||
{fullDDL},
|
||
},
|
||
}
|
||
state.mu.Unlock()
|
||
|
||
oracleDB := &OracleDB{conn: dbConn}
|
||
triggers, err := oracleDB.GetTriggers("H2", "T_MEMCARD_REG")
|
||
if err != nil {
|
||
t.Fatalf("GetTriggers 返回错误: %v", err)
|
||
}
|
||
if len(triggers) != 1 {
|
||
t.Fatalf("期望返回 1 个触发器，实际 %#v", triggers)
|
||
}
|
||
|
||
statement := triggers[0].Statement
|
||
if strings.Contains(statement, "[CLOB preview:") {
|
||
t.Fatalf("Oracle 触发器 DDL 不应包含交互式 CLOB 预览标记: %q", statement[:min(len(statement), 96)])
|
||
}
|
||
if !strings.Contains(statement, "FULL_TRIGGER_DDL_TAIL") {
|
||
t.Fatalf("Oracle 触发器 DDL 被截断，长度=%d", len(statement))
|
||
}
|
||
}
|
||
|
||
func TestOracleGetTriggersDoesNotTruncateTriggerBodyFallback(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
dbConn, state := openOracleRecordingDB(t)
|
||
triggerListQuery := buildOracleTriggersQuery("H2", "T_MEMCARD_REG")
|
||
triggerDDLQuery := `SELECT DBMS_METADATA.GET_DDL('TRIGGER', 'TR_T_MEMCARD_REG', 'H2') as ddl FROM DUAL`
|
||
fullBody := "BEGIN\n" + strings.Repeat(" NULL;\n", 700) + " -- FULL_TRIGGER_BODY_TAIL\nEND;"
|
||
|
||
state.mu.Lock()
|
||
state.queryResults[triggerListQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"OWNER", "TABLE_OWNER", "TABLE_NAME", "TRIGGER_NAME", "TRIGGER_TYPE", "TRIGGERING_EVENT", "WHEN_CLAUSE", "TRIGGER_BODY"},
|
||
columnTypes: []string{"VARCHAR2", "VARCHAR2", "VARCHAR2", "VARCHAR2", "VARCHAR2", "VARCHAR2", "VARCHAR2", "LONG"},
|
||
rows: [][]driver.Value{
|
||
{"H2", "H2", "T_MEMCARD_REG", "TR_T_MEMCARD_REG", "BEFORE EACH ROW", "INSERT OR UPDATE", nil, fullBody},
|
||
},
|
||
}
|
||
state.queryResults[triggerDDLQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"DDL"},
|
||
rows: [][]driver.Value{},
|
||
}
|
||
state.mu.Unlock()
|
||
|
||
oracleDB := &OracleDB{conn: dbConn}
|
||
triggers, err := oracleDB.GetTriggers("H2", "T_MEMCARD_REG")
|
||
if err != nil {
|
||
t.Fatalf("GetTriggers 返回错误: %v", err)
|
||
}
|
||
if len(triggers) != 1 {
|
||
t.Fatalf("期望返回 1 个触发器，实际 %#v", triggers)
|
||
}
|
||
|
||
statement := triggers[0].Statement
|
||
if strings.Contains(statement, "[CLOB preview:") {
|
||
t.Fatalf("Oracle 触发器正文不应包含交互式 CLOB 预览标记: %q", statement[:min(len(statement), 96)])
|
||
}
|
||
if !strings.Contains(statement, "FULL_TRIGGER_BODY_TAIL") {
|
||
t.Fatalf("Oracle 触发器正文被截断，长度=%d", len(statement))
|
||
}
|
||
}
|
||
|
||
func TestOracleGetTriggersRebuildsDDLFromTriggerBodyWhenMetadataDDLIsEmpty(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
dbConn, state := openOracleRecordingDB(t)
|
||
triggerListQuery := buildOracleTriggersQuery("H2", "T_MEMCARD_CASH")
|
||
triggerDDLQuery := `SELECT DBMS_METADATA.GET_DDL('TRIGGER', 'TR_T_MEMCARD_CASH', 'H2') as ddl FROM DUAL`
|
||
|
||
state.mu.Lock()
|
||
state.queryResults[triggerListQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"OWNER", "TABLE_OWNER", "TABLE_NAME", "TRIGGER_NAME", "TRIGGER_TYPE", "TRIGGERING_EVENT", "WHEN_CLAUSE", "TRIGGER_BODY"},
|
||
rows: [][]driver.Value{
|
||
{"H2", "H2", "T_MEMCARD_CASH", "TR_T_MEMCARD_CASH", "BEFORE EACH ROW", "INSERT OR UPDATE", "NEW.ID IS NOT NULL", "BEGIN\n :NEW.UPDATED_AT := SYSDATE;\nEND;"},
|
||
},
|
||
}
|
||
state.queryResults[triggerDDLQuery] = oracleRecordingQueryResult{
|
||
columns: []string{"DDL"},
|
||
rows: [][]driver.Value{},
|
||
}
|
||
state.mu.Unlock()
|
||
|
||
oracleDB := &OracleDB{conn: dbConn}
|
||
triggers, err := oracleDB.GetTriggers("H2", "T_MEMCARD_CASH")
|
||
if err != nil {
|
||
t.Fatalf("GetTriggers 返回错误: %v", err)
|
||
}
|
||
if len(triggers) != 1 {
|
||
t.Fatalf("期望返回 1 个触发器，实际 %#v", triggers)
|
||
}
|
||
|
||
statement := triggers[0].Statement
|
||
for _, want := range []string{
|
||
`CREATE OR REPLACE TRIGGER "H2"."TR_T_MEMCARD_CASH"`,
|
||
`BEFORE INSERT OR UPDATE ON "H2"."T_MEMCARD_CASH"`,
|
||
`FOR EACH ROW`,
|
||
`WHEN (NEW.ID IS NOT NULL)`,
|
||
`:NEW.UPDATED_AT := SYSDATE;`,
|
||
} {
|
||
if !strings.Contains(statement, want) {
|
||
t.Fatalf("期望重建后的触发器 DDL 包含 %q，实际: %s", want, statement)
|
||
}
|
||
}
|
||
if strings.Contains(statement, "SOURCE HIDDEN") {
|
||
t.Fatalf("触发器语句不应继续返回 SOURCE HIDDEN: %s", statement)
|
||
}
|
||
}
|