(t *testing.T)
| 31 | ) |
| 32 | |
| 33 | func TestForEach(t *testing.T) { |
| 34 | producer := utils.NewIncrementalDispatcher[int]() |
| 35 | go func() { |
| 36 | defer producer.Done() |
| 37 | producer.Add(1) |
| 38 | producer.Add(2) |
| 39 | producer.Add(3) |
| 40 | }() |
| 41 | |
| 42 | sum := 0 |
| 43 | producer.ForEach(func(item int) { |
| 44 | sum += item |
| 45 | }) |
| 46 | |
| 47 | require.Equal(t, 6, sum) |
| 48 | } |
| 49 | |
| 50 | func TestConcurrentConsumption(t *testing.T) { |
| 51 | producer := utils.NewIncrementalDispatcher[int]() |