package stepper

import (
	"context"
	"errors"
	"testing"
)

func TestBasicSteps(t *testing.T) {

	ctx := context.Background()
	state := make(map[string]bool)

	step1 := func(ctx context.Context, cancel context.CancelFunc, state interface{}) error {
		st, ok := state.(map[string]bool)
		if !ok {
			t.Error(ErrUnsupportedState)
		}
		st["step1"] = true
		return nil
	}

	step2 := func(ctx context.Context, cancel context.CancelFunc, state interface{}) error {
		st, ok := state.(map[string]bool)
		if !ok {
			t.Error(ErrUnsupportedState)
		}
		st["step2"] = true
		return nil
	}

	status, err := Run(ctx, state, step1, step2)
	if err != nil {
		t.Error(err)
	}

	if g, e := status, Completed; g != e {
		t.Errorf("status: expected '%v', got '%v'", e, g)
	}

	if g, e := state["step1"], true; g != e {
		t.Errorf("state[\"step1\"]: expected '%v', got '%v'", e, g)
	}

	if g, e := state["step2"], true; g != e {
		t.Errorf("state[\"step2\"]: expected '%v', got '%v'", e, g)
	}

}

func TestStepsCancel(t *testing.T) {

	ctx := context.Background()
	state := make(map[string]bool)

	step1 := func(ctx context.Context, cancel context.CancelFunc, state interface{}) error {
		st, ok := state.(map[string]bool)
		if !ok {
			t.Error(ErrUnsupportedState)
		}
		st["step1"] = true
		return nil
	}

	step2 := func(ctx context.Context, cancel context.CancelFunc, state interface{}) error {
		cancel()
		return nil
	}

	step3 := func(ctx context.Context, cancel context.CancelFunc, state interface{}) error {
		st, ok := state.(map[string]bool)
		if !ok {
			t.Error(ErrUnsupportedState)
		}
		st["step3"] = true
		return nil
	}

	status, err := Run(ctx, state, step1, step2, step3)
	if err != nil {
		t.Error(err)
	}

	if g, e := status, Canceled; g != e {
		t.Errorf("status: expected '%v', got '%v'", e, g)
	}

	if g, e := state["step1"], true; g != e {
		t.Errorf("state[\"step1\"]: expected '%v', got '%v'", e, g)
	}

	if g, e := state["step3"], false; g != e {
		t.Errorf("state[\"step3\"]: expected '%v', got '%v'", e, g)
	}

}

func TestStepsError(t *testing.T) {

	stepErr := errors.New("test error")
	ctx := context.Background()
	state := make(map[string]bool)

	step1 := func(ctx context.Context, cancel context.CancelFunc, state interface{}) error {
		st, ok := state.(map[string]bool)
		if !ok {
			t.Error(ErrUnsupportedState)
		}
		st["step1"] = true
		return nil
	}

	step2 := func(ctx context.Context, cancel context.CancelFunc, state interface{}) error {
		return stepErr
	}

	step3 := func(ctx context.Context, cancel context.CancelFunc, state interface{}) error {
		st, ok := state.(map[string]bool)
		if !ok {
			t.Error(ErrUnsupportedState)
		}
		st["step3"] = true
		return nil
	}

	status, err := Run(ctx, state, step1, step2, step3)
	if err == nil {
		t.Errorf("err should not be nil")
	}
	if g, e := err, stepErr; g != e {
		t.Errorf("err: expected '%v', got '%v'", e, g)
	}

	if g, e := status, Error; g != e {
		t.Errorf("status: expected '%v', got '%v'", e, g)
	}

	if g, e := state["step1"], true; g != e {
		t.Errorf("state[\"step1\"]: expected '%v', got '%v'", e, g)
	}

	if g, e := state["step3"], false; g != e {
		t.Errorf("state[\"step3\"]: expected '%v', got '%v'", e, g)
	}

}