[#16] Fix TestHealthyReweight

Signed-off-by: Denis Kirillov <denis@nspcc.ru>
2021-07-20 11:06:45 +03:00 · 2021-07-20 11:06:45 +03:00 · 7a0128e6ff
parent edc1382178
commit 7a0128e6ff
1 changed files with 1 additions and 0 deletions
--- a/pkg/pool/sampler_test.go
+++ b/pkg/pool/sampler_test.go
@ -94,6 +94,7 @@ func TestHealthyReweight(t *testing.T) {
 	p.lock.Unlock()

 	updateNodesHealth(context.TODO(), p, options, buffer)
+	p.sampler = NewSampler(weights, rand.NewSource(0))

 	connection0, _, err = p.Connection()
 	require.NoError(t, err)