64 lines
2.1 KiB
Go
64 lines
2.1 KiB
Go
package fsm
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
func TestFSMStart(t *testing.T) {
|
|
ctx := context.TODO()
|
|
buf := bytes.NewBuffer(nil)
|
|
pfb := func(_ context.Context, state string, _ interface{}) {
|
|
fmt.Fprintf(buf, "before state %s\n", state)
|
|
}
|
|
pfa := func(_ context.Context, state string, _ interface{}) {
|
|
fmt.Fprintf(buf, "after state %s\n", state)
|
|
}
|
|
f := New(InitialState("1"), HookBefore(pfb), HookAfter(pfa))
|
|
f1 := func(_ context.Context, req interface{}, _ ...StateOption) (string, interface{}, error) {
|
|
args := req.(map[string]interface{})
|
|
if v, ok := args["request"].(string); !ok || v == "" {
|
|
return "", nil, fmt.Errorf("empty request")
|
|
}
|
|
return "2", map[string]interface{}{"response": "test2"}, nil
|
|
}
|
|
f2 := func(_ context.Context, req interface{}, _ ...StateOption) (string, interface{}, error) {
|
|
args := req.(map[string]interface{})
|
|
if v, ok := args["response"].(string); !ok || v == "" {
|
|
return "", nil, fmt.Errorf("empty response")
|
|
}
|
|
return "", map[string]interface{}{"response": "test"}, nil
|
|
}
|
|
f3 := func(_ context.Context, req interface{}, _ ...StateOption) (string, interface{}, error) {
|
|
args := req.(map[string]interface{})
|
|
if v, ok := args["response"].(string); !ok || v == "" {
|
|
return "", nil, fmt.Errorf("empty response")
|
|
}
|
|
return StateEnd, map[string]interface{}{"response": "test_last"}, nil
|
|
}
|
|
f.State("1", f1)
|
|
f.State("2", f2)
|
|
f.State("3", f3)
|
|
rsp, err := f.Start(ctx, map[string]interface{}{"request": "test1"})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
args := rsp.(map[string]interface{})
|
|
if v, ok := args["response"].(string); !ok || v == "" {
|
|
t.Fatalf("nil rsp: %#+v", args)
|
|
} else if v != "test_last" {
|
|
t.Fatalf("invalid rsp %#+v", args)
|
|
}
|
|
|
|
if !bytes.Contains(buf.Bytes(), []byte(`before state 1`)) ||
|
|
!bytes.Contains(buf.Bytes(), []byte(`before state 2`)) ||
|
|
!bytes.Contains(buf.Bytes(), []byte(`after state 1`)) ||
|
|
!bytes.Contains(buf.Bytes(), []byte(`after state 2`)) ||
|
|
!bytes.Contains(buf.Bytes(), []byte(`after state 3`)) ||
|
|
!bytes.Contains(buf.Bytes(), []byte(`after state 3`)) {
|
|
t.Fatalf("fsm not works properly or hooks error, buf: %s", buf.Bytes())
|
|
}
|
|
}
|