fix all unit test.

pingcap · Nov 21, 2023 · 60b4b36 · 60b4b36
1 parent 16ad5a7
commit 60b4b36
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/sink/codec/canal/type_test.go b/pkg/sink/codec/canal/type_test.go
@@ -76,6 +76,7 @@ func TestGetMySQLType4IntTypes(t *testing.T) {
 	require.Equal(t, internal.JavaSQLTypeSMALLINT, javaType)
 
 	columnInfo, ok = tableInfo.GetColumnInfo(colInfos[3].ID)
+	require.True(t, ok)
 	flag = tableInfo.ColumnsFlag[colInfos[3].ID]
 	mysqlType = getMySQLType(columnInfo, false)
 	require.Equal(t, "mediumint", mysqlType)
@@ -86,6 +87,7 @@ func TestGetMySQLType4IntTypes(t *testing.T) {
 	require.Equal(t, internal.JavaSQLTypeINTEGER, javaType)
 
 	columnInfo, ok = tableInfo.GetColumnInfo(colInfos[4].ID)
+	require.True(t, ok)
 	flag = tableInfo.ColumnsFlag[colInfos[4].ID]
 	mysqlType = getMySQLType(columnInfo, false)
 	require.Equal(t, "bigint", mysqlType)
@@ -107,6 +109,7 @@ func TestGetMySQLType4IntTypes(t *testing.T) {
 	_, _, colInfos = tableInfo.GetRowColInfos()
 
 	columnInfo, ok = tableInfo.GetColumnInfo(colInfos[0].ID)
+	require.True(t, ok)
 	flag = tableInfo.ColumnsFlag[colInfos[0].ID]
 	mysqlType = getMySQLType(columnInfo, false)
 	require.Equal(t, "int unsigned", mysqlType)