Skip to content

fix(test): eliminate data race in TestBatchWorkflowV2_TuneSignal #7892

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
officialasishkumar wants to merge 2 commits into
base: master
Choose a base branch
from
Open

fix(test): eliminate data race in TestBatchWorkflowV2_TuneSignal #7892

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
929d190
fix(test): eliminate data race in TestBatchWorkflowV2_TuneSignal
officialasishkumar Apr 5, 2026
7376ab7
test(batcher): drop redundant post-wait lock
officialasishkumar Apr 6, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 18 additions & 10 deletions service/worker/batcher/workflow_v2_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -150,23 +150,28 @@ func TestBatchWorkflowV2_TuneSignal(t *testing.T) {
var mu sync.Mutex
var capturedParams []BatchParams

// firstActivityDone is kept open while the test runs, which prevents the
// first activity mock goroutine from returning before the workflow has had
// a chance to process the tune signal. The Cadence test framework delivers
// CanceledError to the activity future independently of the goroutine, so
// blocking here does not prevent the workflow from completing. The channel
// is closed by t.Cleanup once the assertions are done.
// firstActivityDone gates the first activity mock goroutine so it does not
// return before the workflow has had a chance to process the tune signal.
// The Cadence test framework delivers CanceledError to the activity future
// independently of the goroutine, so blocking here does not prevent the
// workflow from completing.
firstActivityDone := make(chan struct{})
t.Cleanup(func() { close(firstActivityDone) })

// activityWG tracks running activity mock goroutines so we can wait for
// them to finish before reading capturedParams, avoiding a data race.
var activityWG sync.WaitGroup

env.OnActivity(batchActivityV2Name, mock.Anything, mock.Anything).
Return(func(_ context.Context, p BatchParams) (HeartBeatDetails, error) {
activityWG.Add(1)
defer activityWG.Done()
Comment on lines +166 to +167
Copy link
Copy Markdown

@gitar-bot gitar-bot Bot Apr 6, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

💡 Bug: WaitGroup.Add(1) called inside goroutine instead of before it

At line 166, activityWG.Add(1) is called inside the activity callback goroutine rather than before it starts. Per the sync.WaitGroup documentation, "calls with a positive delta that occur when the counter is zero must happen before a Wait." If the Go scheduler delays the goroutine start, activityWG.Wait() (line 204) could observe a zero counter and return before Add(1) has executed, reading capturedParams while the goroutine is still writing to it.

In practice this is very unlikely because the Cadence test framework invokes the second activity synchronously during ExecuteWorkflow (which completes before Wait is called), and the first activity must have started to block on the channel. Still, it violates the documented contract and could theoretically cause the race detector to fire under extreme scheduling.

A cleaner approach: add both counts eagerly before the mock is invoked, or use activityWG.Add(1) inside a wrapper that is called synchronously by the framework before spawning the goroutine.

Suggested fix:

// Before registering the mock, add the expected count:
activityWG.Add(2) // we expect exactly 2 invocations

env.OnActivity(batchActivityV2Name, mock.Anything, mock.Anything).
    Return(func(_ context.Context, p BatchParams) (HeartBeatDetails, error) {
        defer activityWG.Done()
        // ... rest unchanged

Was this helpful? React with 👍 / 👎 | Reply gitar fix to apply this suggestion


mu.Lock()
capturedParams = append(capturedParams, p)
n := len(capturedParams)
mu.Unlock()
if n == 1 {
// Block until the test is done. This prevents the future from
// Block until the workflow is done. This prevents the future from
// resolving with nil before the workflow processes the tune signal,
// which would cause the workflow to take the early-exit path.
<-firstActivityDone
Expand All @@ -193,10 +198,13 @@ func TestBatchWorkflowV2_TuneSignal(t *testing.T) {
assert.Equal(t, 8, result.SuccessCount)
assert.Equal(t, 3, result.CurrentPage)

mu.Lock()
// Release the first activity goroutine and wait for all activity goroutines
// to finish before reading capturedParams.
close(firstActivityDone)
activityWG.Wait()

captured := make([]BatchParams, len(capturedParams))
copy(captured, capturedParams)
mu.Unlock()

require.Len(t, captured, 2, "activity must be invoked twice (cancelled then restarted)")
// Second invocation must carry the updated RPS and Concurrency from the signal.
Expand Down
Loading