Skip to content

Commit

Permalink
[chore] Remove redundant wait group from a test
Browse files Browse the repository at this point in the history
  • Loading branch information
dmitryax committed May 30, 2024
1 parent ff6f029 commit 65d9f82
Showing 1 changed file with 7 additions and 16 deletions.
23 changes: 7 additions & 16 deletions exporter/exporterhelper/batch_sender_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -458,18 +458,11 @@ func TestBatchSender_ShutdownDeadlock(t *testing.T) {

sink := newFakeRequestSink()

// Send 10 concurrent requests and wait for them to start
startWG := sync.WaitGroup{}
for i := 0; i < 10; i++ {
startWG.Add(1)
go func() {
startWG.Done()
require.NoError(t, be.send(context.Background(), &fakeRequest{items: 4, sink: sink}))
}()
}
startWG.Wait()
// Send 2 concurrent requests
go func() { require.NoError(t, be.send(context.Background(), &fakeRequest{items: 4, sink: sink})) }()
go func() { require.NoError(t, be.send(context.Background(), &fakeRequest{items: 4, sink: sink})) }()

// Wait for at least one batch to enter the merge function
// Wait for the requests to enter the merge function
<-waitMerge

// Initiate the exporter shutdown, unblock the batch merge function to catch possible deadlocks,
Expand All @@ -485,12 +478,10 @@ func TestBatchSender_ShutdownDeadlock(t *testing.T) {
close(blockMerge)
<-doneShutdown

// The exporter should have sent only one "merged" batch, in some cases it might send two if the shutdown
// happens before the batch is fully merged.
assert.LessOrEqual(t, uint64(1), sink.requestsCount.Load())
assert.EqualValues(t, 1, sink.requestsCount.Load())

// blockedBatchMergeFunc just returns the first request, so the items count should be 4 times the requests count.
assert.Equal(t, sink.requestsCount.Load()*4, sink.itemsCount.Load())
// blockedBatchMergeFunc just returns the first request, so the items count should be 4.
assert.EqualValues(t, 4, sink.itemsCount.Load())
}

func queueBatchExporter(t *testing.T, batchOption Option) *baseExporter {
Expand Down

0 comments on commit 65d9f82

Please sign in to comment.