package workflow

import "testing"

func getTestWorkflow() *Workflow {
	w := NewWorkflow("initial")
	w.AddTransition("From initial to A", []Place{"initial"}, "A")
	w.AddTransition("From initial to B", []Place{"initial"}, "B")
	w.AddTransition("From A to C", []Place{"A"}, "C")
	w.AddTransition("From B,C to D", []Place{"B", "C"}, "D")
	w.AddTransition("From C,D to Finish", []Place{"C", "D"}, "Finish")
	return w
}

type testObject struct {
	place Place
}

func (t *testObject) GetPlace() Place {
	return t.place
}

func (t *testObject) SetPlace(p Place) error {
	t.place = p
	return nil
}

func TestWorkflow_Can(t *testing.T) {
	o := new(testObject)
	w := getTestWorkflow()
	if err := w.Can(o, "From initial to A"); err != nil {
		t.Error("Must has transition")
	}
	if err := w.Can(o, "From A to C"); err == nil {
		t.Error("Must has no transition")
	}
}

func TestWorkflow_GetEnabledTransitions(t *testing.T) {
	w:=getTestWorkflow()
	o := new(testObject)
	if len(w.GetEnabledTransitions(o)) != 2 {
		t.Error("Must be exactly 2 transitions from initial")
	}
}

func TestWorkflow_Apply(t *testing.T) {
	o := new(testObject)
	w := getTestWorkflow()
	if err := w.Apply(o, "From initial to A"); err != nil {
		t.Error(err)
	}
	if o.GetPlace() != "A" {
		t.Error("Must be at A place")
	}
	if err := w.Apply(o, "From B,C to D"); err != ErrCantApply {
		t.Error("Must be cant move")
	}
	if err := w.Apply(o, "From A to D"); err != ErrTransitionNotFound {
		t.Error("Must be transition not found")
	}
	if err := w.Apply(o, "From A to C"); err != nil {
		t.Error(err)
	}
	if o.GetPlace() != "C" {
		t.Error("Must be at C place")
	}
}

func TestWorkflow_DumpToDot(t *testing.T) {
	dump := getTestWorkflow().DumpToDot()
	if len(dump) != 288 {
		t.Error("Len must be 288")
	}
}