[#11] fetcher: Fix flaky TestFetcherCancel
All checks were successful
/ Builds (1.21) (pull_request) Successful in 1m42s
/ Builds (1.22) (pull_request) Successful in 1m30s
/ DCO (pull_request) Successful in 1m54s
/ Vulncheck (pull_request) Successful in 2m13s
/ Lint (pull_request) Successful in 2m46s
/ Tests (1.21) (pull_request) Successful in 1m32s
/ Tests (1.22) (pull_request) Successful in 1m30s

Signed-off-by: Denis Kirillov <d.kirillov@yadro.com>
This commit is contained in:
Denis Kirillov 2024-07-25 16:57:55 +03:00
parent 84ed2f55e3
commit 0190bbbde3

View file

@ -205,12 +205,13 @@ func TestFetcherCancel(t *testing.T) {
key, err := keys.NewPrivateKey()
require.NoError(t, err)
mocks, err := initMocks(1, 3)
mocks, err := initMocks(1, 1)
require.NoError(t, err)
epochCh := make(chan uint64)
go func() {
epochCh <- 1
epochCh <- 2
close(epochCh)
}()
@ -228,13 +229,8 @@ func TestFetcherCancel(t *testing.T) {
f := NewJobProvider(ctx, cfg)
ch := f.Jobs()
res := []Job{<-ch}
cancel()
<-ctx.Done()
for job := range ch {
var res []Job
for job := range f.Jobs() {
res = append(res, job)
}