xuans
e6e67197f6
识别DELIMITER客户端指令，保留复合语句及可执行注释，词法格式错误时拒绝执行部分备份。新增双版本官方导出对照与Scert恢复实库测试，验证触发器顺序、过程、函数、事件调度、权限失败及格式错误保护。仅覆盖本批恢复场景，自有导出对象完整性与生产验收仍未完成。
68 lines
2.1 KiB
Go
68 lines
2.1 KiB
Go
package cli
|
||
|
||
import (
|
||
"reflect"
|
||
"testing"
|
||
)
|
||
|
||
func TestSplitMySQLStatementsDelimiter(t *testing.T) {
|
||
for _, item := range []struct {
|
||
name string
|
||
script string
|
||
want []string
|
||
}{
|
||
{
|
||
name: "procedure",
|
||
script: "DELIMITER ;;\nCREATE PROCEDURE p() BEGIN SELECT 1; SELECT '值;;分号'; END;;\nDELIMITER ;\nSELECT 2;",
|
||
want: []string{"CREATE PROCEDURE p() BEGIN SELECT 1; SELECT '值;;分号'; END", "SELECT 2"},
|
||
},
|
||
{
|
||
name: "quoted_crlf_bom",
|
||
script: "\uFEFF-- 注释\r\n dElImItEr '$$'\r\nSELECT '$$', `a$$b`, 'it''s';$$\r\nDELIMITER ;\r\nSELECT 2;",
|
||
want: []string{"SELECT '$$', `a$$b`, 'it''s';", "SELECT 2"},
|
||
},
|
||
{
|
||
name: "comments",
|
||
script: "DELIMITER //\n/* // */ SELECT 1 # //\n//\nDELIMITER ;\nSELECT/**/2;",
|
||
want: []string{"SELECT 1", "SELECT 2"},
|
||
},
|
||
{
|
||
name: "executable_comments",
|
||
script: "DELIMITER ;;\n/*!50003 CREATE*/ /*!50003 TRIGGER t BEFORE INSERT ON x FOR EACH ROW BEGIN SET NEW.x = ';;'; END */;;\nDELIMITER ;",
|
||
want: []string{"/*!50003 CREATE*/ /*!50003 TRIGGER t BEFORE INSERT ON x FOR EACH ROW BEGIN SET NEW.x = ';;'; END */"},
|
||
},
|
||
{
|
||
name: "keyword_is_not_command",
|
||
script: "SELECT delimiter FROM x;\nSELECT 'DELIMITER $$';\n-- DELIMITER $$\nSELECT 2;",
|
||
want: []string{"SELECT delimiter FROM x", "SELECT 'DELIMITER $$'", "SELECT 2"},
|
||
},
|
||
} {
|
||
t.Run(item.name, func(t *testing.T) {
|
||
got, err := splitMySQLStatements(item.script)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if !reflect.DeepEqual(got, item.want) {
|
||
t.Fatalf("分隔符解析不正确：得到 %#v，期望 %#v", got, item.want)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestSplitMySQLStatementsRejectsMalformedScript(t *testing.T) {
|
||
for _, script := range []string{
|
||
"SELECT 1;\nDELIMITER\nSELECT 2;",
|
||
"SELECT 1;\nDELIMITER a b\nSELECT 2;",
|
||
"SELECT 1;\nDELIMITER " + string(rune(92)) + "\nSELECT 2;",
|
||
"SELECT 1; SELECT '未闭合",
|
||
"SELECT 1; SELECT `未闭合",
|
||
"SELECT 1; /* 未闭合",
|
||
"SELECT 1; /*!50003 未闭合",
|
||
} {
|
||
got, err := splitMySQLStatements(script)
|
||
if err == nil || got != nil {
|
||
t.Fatalf("格式错误时不应返回可执行的部分语句：%#v，%v", got, err)
|
||
}
|
||
}
|
||
}
|