fix testcase

pingcap · Jun 23, 2022 · ce5522b · ce5522b
1 parent efbba84
commit ce5522b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/infoschema/infoschema.go b/infoschema/infoschema.go
@@ -487,6 +487,11 @@ func (is *LocalTemporaryTables) RemoveTable(schema, table model.CIStr) (exist bo
 	return true
 }
 
+// Count gets the count of the temporary tables.
+func (is *LocalTemporaryTables) Count() int {
+	return len(is.idx2table)
+}
+
 // SchemaByTable get a table's schema name
 func (is *LocalTemporaryTables) SchemaByTable(tableInfo *model.TableInfo) (*model.DBInfo, bool) {
 	if tableInfo == nil {

diff --git a/table/temptable/ddl_test.go b/table/temptable/ddl_test.go
@@ -164,7 +164,7 @@ func TestTruncateLocalTemporaryTable(t *testing.T) {
 	// truncate when empty
 	err := ddl.TruncateLocalTemporaryTable(model.NewCIStr("db1"), model.NewCIStr("t1"))
 	require.True(t, infoschema.ErrTableNotExists.Equal(err))
-	require.Nil(t, sessVars.LocalTemporaryTables)
+	require.Equal(t, 0, sessVars.LocalTemporaryTables.(*infoschema.LocalTemporaryTables).Count())
 	require.Nil(t, sessVars.TemporaryTableData)
 
 	// add one table