func rebalanceLeases(t testing.TB, tc *testcluster.TestCluster) {
kvDB := tc.Server(0).KVClient().(*client.DB)
txn := client.NewTxn(context.Background(), *kvDB)
rangeDescs, err := sql.AllRangeDescriptors(txn)
if err != nil {
t.Fatal(err)
}
for _, r := range rangeDescs {
target := tc.Target(int(r.RangeID) % tc.NumServers())
if err := tc.TransferRangeLease(r, target); err != nil {
t.Fatal(err)
}
}
}
// verifyTables ensures that the correct number of tables were created and that
// they all correspond to individual table descriptor IDs in the correct range
// of values.
func verifyTables(
t *testing.T,
tc *testcluster.TestCluster,
completed chan int,
expectedNumOfTables int,
descIDStart int64,
) {
descIDEnd := descIDStart + int64(expectedNumOfTables)
usedTableIDs := make(map[sqlbase.ID]string)
var count int
for id := range completed {
count++
tableName := fmt.Sprintf("table_%d", id)
kvDB := tc.Servers[count%tc.NumServers()].KVClient().(*client.DB)
tableDesc := sqlbase.GetTableDescriptor(kvDB, "test", tableName)
if int64(tableDesc.ID) < descIDStart || int64(tableDesc.ID) >= descIDEnd {
t.Fatalf(
"table %s's ID %d is not within the expected range of %d to %d",
tableName,
tableDesc.ID,
descIDStart,
descIDEnd,
)
}
usedTableIDs[tableDesc.ID] = tableName
}
if e, a := expectedNumOfTables, len(usedTableIDs); e != a {
t.Fatalf("expected %d tables created, only got %d", e, a)
}
kvDB := tc.Servers[count%tc.NumServers()].KVClient().(*client.DB)
if descID, err := kvDB.Get(context.Background(), keys.DescIDGenerator); err != nil {
t.Fatal(err)
} else {
if e, a := descIDEnd, descID.ValueInt(); e != a {
t.Fatalf("expected next descriptor ID to be %d, got %d", e, a)
}
}
}
