script/runner_test.go

package script_test

import (
	"context"
	"reflect"
	"testing"
	"time"

	"code.uint32.ru/dmitry/script"
)

var echoProcessor = func(_ context.Context, in []string) ([]string, error) {
	return in, nil
}

func TestBasicRun(t *testing.T) {
	t.Parallel()
	input := [][]string{
		{"hello", "world"},
	}

	r := script.NewMemReader(input)
	w := script.NewMemWriter()

	conf := script.RunConfig{
		Input:     r,
		Output:    w,
		Processor: echoProcessor,
	}

	if err := script.Run(t.Context(), conf); err != nil {
		t.Fatal(err)
	}

	output := w.Output()

	if !reflect.DeepEqual(input, output) {
		t.Errorf("incorrect output, want: %v, got: %v", input, output)
	}
}

type infiniteReader struct{}

func (ir *infiniteReader) Read() ([]string, error) {
	return []string{"infinity", "looks", "like", "this"}, nil
}

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

	r := &infiniteReader{}
	w := script.NewNopWriter()

	conf := script.RunConfig{
		Input:     r,
		Output:    w,
		Processor: echoProcessor,
	}

	ctx, cancel := context.WithTimeout(t.Context(), time.Millisecond)
	defer cancel()

	if err := script.Run(ctx, conf); err != nil {
		t.Fatal(err)
	}
}