Function TestInjectContextWithMiddlewareError

internal/command/inject_test.go:96–132

(t *testing.T)

}

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

	fnCalled := false

	type contextKey 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)
		fnCalled = true
		return nil
	}

	var middlewareError = errors.New("a middleware error")

	middleware := []func(context.Context) (context.Context, error){
		func(context.Context) (context.Context, error) {
			assert.True(t, ctx.Value(contextKey{}).(bool))
			return nil, middlewareError
		},
	}

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

	assert.NotNil(t, got)

	// execute the function and verify middleware resulted in error
	err := got(new(cli.Context))
	assert.Error(t, err)
	assert.Equal(t, middlewareError, err)
	assert.False(t, fnCalled) // fn is not called because of middleware error
}