consumer: fix test case data race

Test must wait for the shard to react to the assignment and update its status, to avoid a data race when modifying WatchApplyDelay again.
gazette · Feb 14, 2022 · 443ab90 · 443ab90
1 parent a21a5b8
commit 443ab90
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/consumer/resolver_test.go b/consumer/resolver_test.go
@@ -288,6 +288,8 @@ func TestResolverShardTransitions(t *testing.T) {
 	tf.ks.WatchApplyDelay = 3 * time.Millisecond
 	tf.allocateShardNoWait(makeShard(shardB))
 	tf.allocateShard(makeShard(shardB), localID)
+
+	expectStatusCode(t, tf.state, pc.ReplicaStatus_PRIMARY)
 	tf.ks.WatchApplyDelay = 0
 
 	// Expect a new shard |newB| is created, and the prior |sB| is cancelled.