feature (server): add additional option for downstream workers DAG (#2278)

internal/server/options.go

@@ -40,6 +40,7 @@ type options struct {
 	withTestPkiWorkerKeyId             *string
 	withWorkerType                     WorkerType
 	withRoot                           string
+	withStopAfter                      uint
 }
 
 func getDefaultOptions() options {
@@ -180,3 +181,10 @@ func WithRoot(workerId string) Option {
 		o.withRoot = workerId
 	}
 }
+
+// WithStopAfter provides an optional stop after count
+func WithStopAfter(stopAfter uint) Option {
+	return func(o *options) {
+		o.withStopAfter = stopAfter
+	}
+}

internal/server/options_test.go

@@ -170,4 +170,10 @@ func Test_GetOpts(t *testing.T) {
 		opts = getOpts(WithRoot("a"))
 		assert.Equal(t, "a", opts.withRoot)
 	})
+	t.Run("WithStopAfter", func(t *testing.T) {
+		opts := getDefaultOptions()
+		assert.Empty(t, opts.withStopAfter)
+		opts = getOpts(WithStopAfter(10))
+		assert.Equal(t, uint(10), opts.withStopAfter)
+	})
 }