mirror of
https://github.com/Syngnat/GoNavi.git
synced 2026-08-10 00:33:28 +08:00
1b3137d947
- 区分匿名 BEGIN...END 块与显式事务控制语句 - 统一前后端方言判定并挂起 Oracle、达梦和 SQL Server 块内 DML - 补充编辑器、事务控制器及 SQL Server 回滚链路回归测试
107 lines
3.6 KiB
Go
107 lines
3.6 KiB
Go
package app
|
|
|
|
import "testing"
|
|
|
|
func TestShouldUseManagedSQLTransaction_UnsupportedTypesUsePlainExecution(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cases := []struct {
|
|
dbType string
|
|
query string
|
|
}{
|
|
{dbType: "trino", query: "UPDATE hive.default.orders SET status = 'done'"},
|
|
{dbType: "tdengine", query: "INSERT INTO meters(ts, current) VALUES (NOW, 10.2)"},
|
|
{dbType: "clickhouse", query: `INSERT INTO events FORMAT JSONEachRow {"id":1}`},
|
|
{dbType: "iotdb", query: "INSERT INTO root.ln.wf01.wt01(timestamp,status) VALUES(1,true)"},
|
|
}
|
|
for _, tc := range cases {
|
|
tc := tc
|
|
t.Run(tc.dbType, func(t *testing.T) {
|
|
t.Parallel()
|
|
if shouldUseManagedSQLTransaction(tc.dbType, tc.query) {
|
|
t.Fatalf("expected %s DML to skip SQL editor managed transactions", tc.dbType)
|
|
}
|
|
if shouldUseManagedSQLTransaction(tc.dbType, "BEGIN; "+tc.query+"; COMMIT;") {
|
|
t.Fatalf("expected %s explicit transactions to stay unmanaged", tc.dbType)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestShouldUseManagedSQLTransaction_OracleAnonymousBlockWithDMLUsesManagedTransaction(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
query := `BEGIN
|
|
UPDATE users SET name = 'new' WHERE id = 1;
|
|
DELETE FROM audit_logs WHERE user_id = 1;
|
|
END;`
|
|
if !shouldUseManagedSQLTransaction("oracle", query) {
|
|
t.Fatal("expected Oracle anonymous block with DML to use SQL editor managed transaction")
|
|
}
|
|
}
|
|
|
|
func TestShouldUseManagedSQLTransaction_OracleReadOnlyAnonymousBlockStaysUnmanaged(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
query := `BEGIN
|
|
NULL;
|
|
END;`
|
|
if shouldUseManagedSQLTransaction("oracle", query) {
|
|
t.Fatal("expected Oracle read-only anonymous block to stay unmanaged")
|
|
}
|
|
}
|
|
|
|
func TestShouldUseManagedSQLTransaction_SQLServerBeginEndWithDMLUsesManagedTransaction(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
query := `BEGIN
|
|
PRINT 'the word DELETE here is only text';
|
|
-- INSERT INTO audit_logs(id) VALUES (1);
|
|
UPDATE users SET name = 'new' WHERE id = 1;
|
|
END;`
|
|
if !shouldUseManagedSQLTransaction("sqlserver", query) {
|
|
t.Fatal("expected SQL Server BEGIN...END block with DML to use SQL editor managed transaction")
|
|
}
|
|
}
|
|
|
|
func TestShouldUseManagedSQLTransaction_SQLServerBeginEndWithoutExecutableDMLStaysUnmanaged(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
query := `BEGIN
|
|
PRINT 'UPDATE users SET name = ''new''';
|
|
-- DELETE FROM users WHERE id = 1;
|
|
/* INSERT INTO audit_logs(id) VALUES (1); */
|
|
END;`
|
|
if shouldUseManagedSQLTransaction("sqlserver", query) {
|
|
t.Fatal("expected SQL Server BEGIN...END block with DML keywords only in comments or strings to stay unmanaged")
|
|
}
|
|
}
|
|
|
|
func TestShouldUseManagedSQLTransaction_SQLServerExplicitBeginStaysUnmanaged(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
for _, query := range []string{
|
|
"BEGIN TRANSACTION; UPDATE users SET name = 'new' WHERE id = 1; COMMIT TRANSACTION;",
|
|
"BEGIN TRAN; UPDATE users SET name = 'new' WHERE id = 1; COMMIT TRAN;",
|
|
"BEGIN WORK; UPDATE users SET name = 'new' WHERE id = 1; COMMIT WORK;",
|
|
} {
|
|
if shouldUseManagedSQLTransaction("sqlserver", query) {
|
|
t.Fatalf("expected explicit SQL Server transaction to stay unmanaged: %q", query)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestShouldUseManagedSQLTransaction_UsesDialectCommentRules(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
if !shouldUseManagedSQLTransaction("mysql", "DELETE FROM users WHERE id = 1; -- pending") {
|
|
t.Fatal("expected a valid MySQL trailing comment to preserve the managed transaction")
|
|
}
|
|
if shouldUseManagedSQLTransaction("mysql", "DELETE FROM users WHERE id = 1;--comment") {
|
|
t.Fatal("expected compact MySQL double-dash text to remain an executable statement")
|
|
}
|
|
if !shouldUseManagedSQLTransaction("postgres", "DELETE FROM users WHERE id = 1; /*! MySQL-only comment */") {
|
|
t.Fatal("expected PostgreSQL to ignore a MySQL-only block comment")
|
|
}
|
|
}
|