fix test

Signed-off-by: Zheng Xiangsheng <hundundm@gmail.com>
tikv · Jan 4, 2021 · 4d6d298 · 4d6d298
1 parent ba5257e
commit 4d6d298
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/server/id/id_test.go b/tests/server/id/id_test.go
@@ -183,6 +183,7 @@ func (s *testAllocIDSuite) TestPDRestart(c *C) {
 
 	c.Assert(leaderServer.Stop(), IsNil)
 	c.Assert(leaderServer.Run(), IsNil)
+	cluster.WaitLeader()
 
 	for i := uint64(0); i < 10; i++ {
 		id, err := leaderServer.GetAllocator().Alloc()