Initial

1 files changed, 75 insertions, 0 deletions

diff --git a/workflow_test.go b/workflow_test.go
new file mode 100644
index 0000000..cb4fe3b
--- /dev/null
+++ b/workflow_test.go
@@ -0,0 +1,75 @@
+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")
+	}
+}