89 lines
1.7 KiB
Go
89 lines
1.7 KiB
Go
package command_test
|
|
|
|
import (
|
|
"context"
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"gitoa.ru/go-4devs/config"
|
|
"gitoa.ru/go-4devs/config/definition"
|
|
"gitoa.ru/go-4devs/config/provider/memory"
|
|
"gitoa.ru/go-4devs/console/command"
|
|
"gitoa.ru/go-4devs/console/output"
|
|
)
|
|
|
|
func TestChainPrepare(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var cnt int64
|
|
|
|
ctx := context.Background()
|
|
def := definition.New()
|
|
|
|
prepare := func(ctx context.Context, def config.Definition, n command.ConfigureFn) error {
|
|
atomic.AddInt64(&cnt, 1)
|
|
|
|
return n(ctx, def)
|
|
}
|
|
configure := func(context.Context, config.Definition) error {
|
|
return nil
|
|
}
|
|
|
|
for i := range []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10} {
|
|
prepares := make([]command.PrepareFn, i)
|
|
for p := range i {
|
|
prepares[p] = prepare
|
|
}
|
|
|
|
cnt = 0
|
|
chain := command.ChainPrepare(prepares...)
|
|
|
|
err := chain(ctx, def, configure)
|
|
if err != nil {
|
|
t.Errorf("expected nil err, got: %s", err)
|
|
}
|
|
|
|
if cnt != int64(i) {
|
|
t.Fatalf("expected: call prepare 1, got: %d ", cnt)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestChainHandle(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var cnt int64
|
|
|
|
ctx := context.Background()
|
|
in := &memory.Map{}
|
|
out := output.Stdout()
|
|
|
|
handle := func(ctx context.Context, in config.Provider, out output.Output, next command.ExecuteFn) error {
|
|
atomic.AddInt64(&cnt, 1)
|
|
|
|
return next(ctx, in, out)
|
|
}
|
|
action := func(context.Context, config.Provider, output.Output) error {
|
|
return nil
|
|
}
|
|
|
|
for i := range []int{0, 1, 2, 30, 40, 50} {
|
|
handles := make([]command.HandleFn, i)
|
|
for p := range i {
|
|
handles[p] = handle
|
|
}
|
|
|
|
cnt = 0
|
|
chain := command.ChainHandle(handles...)
|
|
|
|
err := chain(ctx, in, out, action)
|
|
if err != nil {
|
|
t.Errorf("expected nil err, got: %s", err)
|
|
}
|
|
|
|
if cnt != int64(i) {
|
|
t.Fatalf("expected: call prepare 1, got: %d ", cnt)
|
|
}
|
|
}
|
|
}
|