Function TestInjectContextWithMiddleware

internal/command/inject_test.go:55–94

(t *testing.T)

}

func TestInjectContextWithMiddleware(t *testing.T) {
	t.Parallel()

	fnCalled := false

	type contextKey struct{}
	type firstMiddlewareContextKey struct{}
	type secondMiddlewareContextKey struct{}

	ctx := context.Background()
	ctx = context.WithValue(ctx, contextKey{}, true)

	fn := func(ctx context.Context) error {
		assert.True(t, ctx.Value(contextKey{}).(bool))
		clictx := CLIContextFromContext(ctx)
		assert.NotNil(t, clictx)
		assert.Equal(t, 42, ctx.Value(firstMiddlewareContextKey{}).(int))
		assert.Equal(t, "value", ctx.Value(secondMiddlewareContextKey{}).(string))
		fnCalled = true
		return nil
	}

	middleware := []func(context.Context) (context.Context, error){
		func(ctx context.Context) (context.Context, error) {
			return context.WithValue(ctx, firstMiddlewareContextKey{}, 42), nil
		},
		func(ctx context.Context) (context.Context, error) {
			return context.WithValue(ctx, secondMiddlewareContextKey{}, "value"), nil
		},
	}

	got := InjectContext(ctx, fn, middleware...)

	assert.NotNil(t, got)

	// execute the function to ensure injected context is OK
	err := got(new(cli.Context))
	assert.NoError(t, err)
	assert.True(t, fnCalled)
}

func TestInjectContextWithMiddlewareError(t *testing.T) {
	t.Parallel()