console/command_test.go

package console_test

import (
	"context"
	"strings"
	"sync/atomic"
	"testing"
	"time"

	"gitoa.ru/go-4devs/console"
	"gitoa.ru/go-4devs/console/example/pkg/command"
	"gitoa.ru/go-4devs/console/input"
	"gitoa.ru/go-4devs/console/input/argument"
	"gitoa.ru/go-4devs/console/input/option"
	"gitoa.ru/go-4devs/console/output"
)

//nolint: gochecknoinits
func init() {
	console.MustRegister(Command().With(console.WithName("fdevs:console:test")))
	console.MustRegister(command.Args())
}

func Command() *console.Command {
	return &console.Command{
		Name:        "test:command",
		Description: "test command",
		Execute: func(ctx context.Context, in input.Input, out output.Output) error {
			out.Print(ctx,
				"test argument:", in.Argument(ctx, "test_argument").String(), "\n",
				"bool option:", in.Option(ctx, "bool").Bool(), "\n",
				"duration option with default:", in.Option(ctx, "duration").Duration(), "\n",
				"array string:[", strings.Join(in.Option(ctx, "string").Strings(), ","), "]\n",
			)

			return nil
		},
		Configure: func(ctx context.Context, def *input.Definition) error {
			def.
				SetArguments(
					argument.New("test_argument", "test argument"),
				).
				SetOptions(
					option.New("string", "array string", option.Array),
					option.Bool("bool", "test bool option"),
					option.Duration("duration", "test duration with default", option.Default(time.Second)),
				)

			return nil
		},
	}
}

func TestChainPrepare(t *testing.T) {
	var cnt int32

	ctx := context.Background()
	def := input.NewDefinition()

	prepare := func(ctx context.Context, def *input.Definition, n console.Configure) error {
		atomic.AddInt32(&cnt, 1)

		return n(ctx, def)
	}
	configure := func(context.Context, *input.Definition) error {
		return nil
	}

	for i := range []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10} {
		prepares := make([]console.Prepare, i)
		for p := 0; p < i; p++ {
			prepares[p] = prepare
		}

		cnt = 0
		chain := console.ChainPrepare(prepares...)

		if err := chain(ctx, def, configure); err != nil {
			t.Errorf("expected nil err, got: %s", err)
		}

		if cnt != int32(i) {
			t.Fatalf("expected: call prepare 1, got: %d ", cnt)
		}
	}
}

func TestChainHandle(t *testing.T) {
	var cnt int32

	ctx := context.Background()
	in := &input.Array{}
	out := output.Stdout()

	handle := func(ctx context.Context, in input.Input, out output.Output, next console.Action) error {
		atomic.AddInt32(&cnt, 1)

		return next(ctx, in, out)
	}
	action := func(context.Context, input.Input, output.Output) error {
		return nil
	}

	for i := range []int{0, 1, 2, 30, 40, 50} {
		handles := make([]console.Handle, i)
		for p := 0; p < i; p++ {
			handles[p] = handle
		}

		cnt = 0
		chain := console.ChainHandle(handles...)

		if err := chain(ctx, in, out, action); err != nil {
			t.Errorf("expected nil err, got: %s", err)
		}

		if cnt != int32(i) {
			t.Fatalf("expected: call prepare 1, got: %d ", cnt)
		}
	}
}