| 414 | } |
| 415 | |
| 416 | func TestChunkQueue_Retry(t *testing.T) { |
| 417 | queue, teardown := setupChunkQueue(t) |
| 418 | defer teardown() |
| 419 | |
| 420 | // Allocate and add all chunks to the queue |
| 421 | for i := uint32(0); i < queue.Size(); i++ { |
| 422 | _, err := queue.Allocate() |
| 423 | require.NoError(t, err) |
| 424 | _, err = queue.Add(&chunk{Height: 3, Format: 1, Index: i, Chunk: []byte{byte(i)}}) |
| 425 | require.NoError(t, err) |
| 426 | _, err = queue.Next() |
| 427 | require.NoError(t, err) |
| 428 | } |
| 429 | |
| 430 | // Retrying a couple of chunks makes Next() return them, but they are not allocatable |
| 431 | queue.Retry(3) |
| 432 | queue.Retry(1) |
| 433 | |
| 434 | _, err := queue.Allocate() |
| 435 | assert.Equal(t, errDone, err) |
| 436 | |
| 437 | chunk, err := queue.Next() |
| 438 | require.NoError(t, err) |
| 439 | assert.EqualValues(t, 1, chunk.Index) |
| 440 | |
| 441 | chunk, err = queue.Next() |
| 442 | require.NoError(t, err) |
| 443 | assert.EqualValues(t, 3, chunk.Index) |
| 444 | |
| 445 | _, err = queue.Next() |
| 446 | assert.Equal(t, errDone, err) |
| 447 | } |
| 448 | |
| 449 | func TestChunkQueue_RetryAll(t *testing.T) { |
| 450 | queue, teardown := setupChunkQueue(t) |