fix race in drain integration tests
This commit is contained in:
parent
c401d5a098
commit
640ebdaef6
|
@ -60,7 +60,13 @@ func allocPromoter(t *testing.T, ctx context.Context,
|
||||||
WriteRequest: structs.WriteRequest{Region: "global"},
|
WriteRequest: structs.WriteRequest{Region: "global"},
|
||||||
}
|
}
|
||||||
var resp structs.NodeAllocsResponse
|
var resp structs.NodeAllocsResponse
|
||||||
require.Nil(t, msgpackrpc.CallWithCodec(codec, "Node.UpdateAlloc", req, &resp))
|
if err := msgpackrpc.CallWithCodec(codec, "Node.UpdateAlloc", req, &resp); err != nil {
|
||||||
|
if ctx.Err() == context.Canceled {
|
||||||
|
return
|
||||||
|
} else {
|
||||||
|
require.Nil(t, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue