database/tidb
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ddl/table_split_test.go: migrate test-infra to testify (#29960)

Browse Source
This commit is contained in:
wangggong
2021-11-23 13:49:50 +08:00
committed by GitHub
parent 805a887380
commit 2f119e30a4
1 changed files with 23 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
ddl/table_split_test.go
View File

@ -17,35 +17,36 @@ package ddl_test
import (
"context"
"sync/atomic"
"testing"
"time"
. "github.com/pingcap/check"
"github.com/pingcap/tidb/ddl"
"github.com/pingcap/tidb/parser/model"
"github.com/pingcap/tidb/session"
"github.com/pingcap/tidb/store/mockstore"
"github.com/pingcap/tidb/tablecodec"
"github.com/pingcap/tidb/util/testkit"
"github.com/pingcap/tidb/testkit"
"github.com/tikv/client-go/v2/tikv"
"github.com/stretchr/testify/require"
)
type testDDLTableSplitSuite struct{}
func TestTableSplit(t *testing.T) {
t.Parallel()
var _ = Suite(&testDDLTableSplitSuite{})
func (s *testDDLTableSplitSuite) TestTableSplit(c *C) {
store, err := mockstore.NewMockStore()
c.Assert(err, IsNil)
require.NoError(t, err)
defer func() {
err := store.Close()
c.Assert(err, IsNil)
require.NoError(t, err)
}()
session.SetSchemaLease(100 * time.Millisecond)
session.DisableStats4Test()
atomic.StoreUint32(&ddl.EnableSplitTableRegion, 1)
dom, err := session.BootstrapSession(store)
c.Assert(err, IsNil)
tk := testkit.NewTestKit(c, store)
require.NoError(t, err)
tk := testkit.NewTestKit(t, store)
tk.MustExec("use test")
// Synced split table region.
tk.MustExec("set global tidb_scatter_region = 1")
@ -56,17 +57,17 @@ func (s *testDDLTableSplitSuite) TestTableSplit(c *C) {
defer dom.Close()
atomic.StoreUint32(&ddl.EnableSplitTableRegion, 0)
infoSchema := dom.InfoSchema()
c.Assert(infoSchema, NotNil)
t, err := infoSchema.TableByName(model.NewCIStr("mysql"), model.NewCIStr("tidb"))
c.Assert(err, IsNil)
checkRegionStartWithTableID(c, t.Meta().ID, store.(kvStore))
require.NotNil(t, infoSchema)
tbl, err := infoSchema.TableByName(model.NewCIStr("mysql"), model.NewCIStr("tidb"))
require.NoError(t, err)
checkRegionStartWithTableID(t, tbl.Meta().ID, store.(kvStore))
t, err = infoSchema.TableByName(model.NewCIStr("test"), model.NewCIStr("t_part"))
c.Assert(err, IsNil)
pi := t.Meta().GetPartitionInfo()
c.Assert(pi, NotNil)
tbl, err = infoSchema.TableByName(model.NewCIStr("test"), model.NewCIStr("t_part"))
require.NoError(t, err)
pi := tbl.Meta().GetPartitionInfo()
require.NotNil(t, pi)
for _, def := range pi.Definitions {
checkRegionStartWithTableID(c, def.ID, store.(kvStore))
checkRegionStartWithTableID(t, def.ID, store.(kvStore))
}
}
@ -74,14 +75,14 @@ type kvStore interface {
GetRegionCache() *tikv.RegionCache
}
func checkRegionStartWithTableID(c *C, id int64, store kvStore) {
func checkRegionStartWithTableID(t *testing.T, id int64, store kvStore) {
regionStartKey := tablecodec.EncodeTablePrefix(id)
var loc *tikv.KeyLocation
var err error
cache := store.GetRegionCache()
loc, err = cache.LocateKey(tikv.NewBackoffer(context.Background(), 5000), regionStartKey)
c.Assert(err, IsNil)
require.NoError(t, err)
// Region cache may be out of date, so we need to drop this expired region and load it again.
cache.InvalidateCachedRegion(loc.Region)
c.Assert(loc.StartKey, BytesEquals, []byte(regionStartKey))
require.Equal(t, []byte(regionStartKey), loc.StartKey)
}