Function WorkContext

rivertest/rivertest.go:616–618

WorkContext returns a realistic context that can be used to test JobArgs.Work implementations. In particular, adds a client to the context so that river.ClientFromContext is usable in the test suite.

(ctx context.Context, client *river.Client[TTx])

// In particular, adds a client to the context so that river.ClientFromContext is
// usable in the test suite.
func WorkContext[TTx any](ctx context.Context, client *river.Client[TTx]) context.Context {
	return context.WithValue(ctx, rivercommon.ContextKeyClient{}, client)
}