diff --git a/ddl/restart_test.go b/ddl/restart_test.go index 617e073635c0f..e1cf6d15fee84 100644 --- a/ddl/restart_test.go +++ b/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 diff --git a/ddl/table_test.go b/ddl/table_test.go index 6dcdc3021eb7a..a7311a15e45cc 100644 --- a/ddl/table_test.go +++ b/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)