test: Migrate WaitMembersForLeader to V3 API

This commit is contained in:
Marek Siarkowicz 2022-01-24 13:10:45 +01:00
parent 81490c25d1
commit c59694d41e

View File

@ -419,13 +419,14 @@ func (c *Cluster) WaitMembersForLeader(t testutil.TB, membs []*Member) int {
for _, m := range membs { for _, m := range membs {
possibleLead[uint64(m.Server.ID())] = true possibleLead[uint64(m.Server.ID())] = true
} }
cc := MustNewHTTPClient(t, getMembersURLs(membs), nil) cc, err := c.ClusterClient()
kapi := client.NewKeysAPI(cc) if err != nil {
t.Fatal(err)
}
// ensure leader is up via linearizable get // ensure leader is up via linearizable get
for { for {
ctx, cancel := context.WithTimeout(context.Background(), 10*TickDuration+time.Second) ctx, cancel := context.WithTimeout(context.Background(), 10*TickDuration+time.Second)
_, err := kapi.Get(ctx, "0", &client.GetOptions{Quorum: true}) _, err := cc.Get(ctx, "0")
cancel() cancel()
if err == nil || strings.Contains(err.Error(), "Key not found") { if err == nil || strings.Contains(err.Error(), "Key not found") {
break break