@@ -30,6 +30,8 @@ func (r *RootCmd) scaletestDynamicParameters() *serpent.Command {
3030numEvals int64
3131tracingFlags = & scaletestTracingFlags {}
3232prometheusFlags = & scaletestPrometheusFlags {}
33+ // This test requires unlimited concurrency
34+ timeoutStrategy = & timeoutFlags {}
3335)
3436orgContext := NewOrganizationContext ()
3537output := & scaletestOutputFlags {}
@@ -102,7 +104,10 @@ func (r *RootCmd) scaletestDynamicParameters() *serpent.Command {
102104return xerrors .Errorf ("setup dynamic parameters partitions: %w" ,err )
103105}
104106
105- th := harness .NewTestHarness (harness.ConcurrentExecutionStrategy {}, harness.ConcurrentExecutionStrategy {})
107+ th := harness .NewTestHarness (
108+ timeoutStrategy .wrapStrategy (harness.ConcurrentExecutionStrategy {}),
109+ // there is no cleanup since it's just a connection that we sever.
110+ nil )
106111
107112for i ,part := range partitions {
108113for j := range part .ConcurrentEvaluations {
@@ -123,7 +128,9 @@ func (r *RootCmd) scaletestDynamicParameters() *serpent.Command {
123128}
124129}
125130
126- err = th .Run (ctx )
131+ testCtx ,testCancel := timeoutStrategy .toContext (ctx )
132+ defer testCancel ()
133+ err = th .Run (testCtx )
127134if err != nil {
128135return xerrors .Errorf ("run test harness: %w" ,err )
129136}
@@ -158,5 +165,6 @@ func (r *RootCmd) scaletestDynamicParameters() *serpent.Command {
158165output .attach (& cmd .Options )
159166tracingFlags .attach (& cmd .Options )
160167prometheusFlags .attach (& cmd .Options )
168+ timeoutStrategy .attach (& cmd .Options )
161169return cmd
162170}