fix unit test TestPTOSC

actiontech · Jan 24, 2025 · a30e8b1 · a30e8b1
1 parent cd510cb
commit a30e8b1
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 1 deletion.
diff --git a/sqle/driver/mysql/pt_osc_test.go b/sqle/driver/mysql/pt_osc_test.go
@@ -19,7 +19,7 @@ func TestPTOSC(t *testing.T) {
 		plocale.Bundle.LocalizeMsgByLang(language.Chinese, plocale.PTOSCAvoidNoDefaultValueOnNotNullColumn))
 
 	runOSCCase(t, "not pk and unique key",
-		"alter table exist_tb_3 add column v3 varchar(255);",
+		"alter table exist_tb_13 add column v4 varchar(255);",
 		plocale.Bundle.LocalizeMsgByLang(language.Chinese, plocale.PTOSCNoUniqueIndexOrPrimaryKey))
 
 	runOSCCase(t, "rename table",

diff --git a/sqle/driver/mysql/session/mock_context.go b/sqle/driver/mysql/session/mock_context.go
@@ -97,6 +97,12 @@ func NewMockContext(e *executor.Executor) *Context {
 						Size:          1,
 						OriginalTable: getTestCreateTableStmt12(),
 					},
+					"exist_tb_13": {
+						sizeLoad:      true,
+						isLoad:        true,
+						Size:          1,
+						OriginalTable: getTestCreateTableStmt13(),
+					},
 				},
 			},
 			"myisam_utf8_db": {
@@ -286,6 +292,12 @@ func NewMockContextForTestTableSize(e *executor.Executor, tableSize map[string]
 						Size:          float64(tableSize["exist_tb_12"]),
 						OriginalTable: getTestCreateTableStmt12(),
 					},
+					"exist_tb_13": {
+						sizeLoad:      true,
+						isLoad:        true,
+						Size:          float64(tableSize["exist_tb_13"]),
+						OriginalTable: getTestCreateTableStmt13(),
+					},
 				},
 			},
 			"myisam_utf8_db": {
@@ -550,6 +562,21 @@ PRIMARY KEY (id) USING BTREE
 	stmt, _ := node.(*ast.CreateTableStmt)
 	return stmt
 }
+func getTestCreateTableStmt13() *ast.CreateTableStmt {
+	baseCreateQuery := `
+CREATE TABLE exist_db.exist_tb_13 (
+id bigint(10) unsigned NOT NULL AUTO_INCREMENT COMMENT "unit test",
+v1 blob,
+v2 int
+)ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT="unit test";
+`
+	node, err := util.ParseOneSql(baseCreateQuery)
+	if err != nil {
+		panic(err)
+	}
+	stmt, _ := node.(*ast.CreateTableStmt)
+	return stmt
+}
 
 type AIMockContext struct {
 	createContextSqls []string