ddl: migrate test-infra to testify for ddl/table_test.go

Signed-off-by: Weizhen Wang <wangweizhen@pingcap.com>

ddl/restart_test.go

@@ -22,6 +22,7 @@ import (
 	"time"
 
 	. "github.com/pingcap/check"
+	"github.com/pingcap/tidb/kv"
 	"github.com/pingcap/tidb/parser/model"
 	"github.com/pingcap/tidb/parser/terror"
 	"github.com/pingcap/tidb/util/mock"
@@ -201,6 +202,38 @@ LOOP:
 	}
 }
 
+var _ = Suite(&testTableSuite{})
+
+type testTableSuite struct {
+	store  kv.Storage
+	dbInfo *model.DBInfo
+
+	d *ddl
+}
+
+func (s *testTableSuite) SetUpSuite(c *C) {
+	s.store = testCreateStore(c, "test_table")
+	ddl, err := testNewDDLAndStart(
+		context.Background(),
+		WithStore(s.store),
+		WithLease(testLease),
+	)
+	c.Assert(err, IsNil)
+	s.d = ddl
+
+	s.dbInfo, err = testSchemaInfo(s.d, "test_table")
+	c.Assert(err, IsNil)
+	testCreateSchema(c, testNewContext(s.d), s.d, s.dbInfo)
+}
+
+func (s *testTableSuite) TearDownSuite(c *C) {
+	testDropSchema(c, testNewContext(s.d), s.d, s.dbInfo)
+	err := s.d.Stop()
+	c.Assert(err, IsNil)
+	err = s.store.Close()
+	c.Assert(err, IsNil)
+}
+
 func (s *testTableSuite) TestTableResume(c *C) {
 	d := s.d
 

ddl/table_test.go

@@ -35,15 +35,6 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
-var _ = Suite(&testTableSuite{})
-
-type testTableSuite struct {
-	store  kv.Storage
-	dbInfo *model.DBInfo
-
-	d *ddl
-}
-
 func testTableInfoWith2IndexOnFirstColumn(c *C, d *ddl, name string, num int) *model.TableInfo {
 	normalInfo, err := testTableInfo(d, name, num)
 	c.Assert(err, IsNil)