| 298 | } |
| 299 | |
| 300 | func TestIteratorContextCancelled(t *testing.T) { |
| 301 | t.Run("already cancelled", func(t *testing.T) { |
| 302 | testWalBlock(t, func(w *walBlock, _ []common.ID, _ []*tempopb.Trace) { |
| 303 | ctx, cancel := context.WithCancel(context.Background()) |
| 304 | cancel() |
| 305 | |
| 306 | _, err := w.Iterator(ctx) |
| 307 | require.Error(t, err) |
| 308 | require.ErrorIs(t, err, context.Canceled) |
| 309 | }) |
| 310 | }) |
| 311 | |
| 312 | t.Run("cancelled after creation", func(t *testing.T) { |
| 313 | testWalBlock(t, func(w *walBlock, _ []common.ID, _ []*tempopb.Trace) { |
| 314 | ctx, cancel := context.WithCancel(context.Background()) |
| 315 | |
| 316 | iter, err := w.Iterator(ctx) |
| 317 | require.NoError(t, err) |
| 318 | defer iter.Close() |
| 319 | |
| 320 | // Cancel the context after iterator creation. Subsequent Next calls |
| 321 | // go through walReaderAt.ReadAt which checks ctx.Err(). |
| 322 | cancel() |
| 323 | |
| 324 | _, _, err = iter.Next(ctx) |
| 325 | require.Error(t, err) |
| 326 | require.ErrorIs(t, err, context.Canceled) |
| 327 | }) |
| 328 | }) |
| 329 | } |
| 330 | |
| 331 | func TestWalBlockRaceConditionCheck(t *testing.T) { |
| 332 | meta := backend.NewBlockMeta("fake", uuid.New(), VersionString) |