You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

121 lines
2.9 KiB

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)
}
}
}