-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathexternal_workflow_test.go
84 lines (71 loc) · 2.4 KB
/
external_workflow_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
package temporalex
import (
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"go.temporal.io/sdk/testsuite"
"go.temporal.io/sdk/workflow"
"testing"
)
func TestExternalWorkflow_DoChild(t *testing.T) {
suite := &testsuite.WorkflowTestSuite{}
env := suite.NewTestWorkflowEnvironment()
wflow := ExternalWorkflow[FakeInput, string]{
Name: "fake-workflow",
TaskQueue: "X",
HandleResult: func(wctx workflow.Context, result string, err error) (string, error) {
assert.Equal(t, "sub-workflow", result)
return "handled-result", nil
},
}
mainWorkflow := func(wctx workflow.Context, input FakeInput) (string, error) {
return wflow.DoChild(wctx, input)
}
subWorkflow := func(wctx workflow.Context, input FakeInput) (string, error) {
return "sub-workflow", nil
}
env.RegisterWorkflow(mainWorkflow)
env.RegisterWorkflowWithOptions(subWorkflow, workflow.RegisterOptions{
Name: wflow.Name,
DisableAlreadyRegisteredCheck: false,
})
input := FakeInput{}
want := "handled-result"
env.ExecuteWorkflow(mainWorkflow, input)
require.True(t, env.IsWorkflowCompleted())
require.NoError(t, env.GetWorkflowError())
var result string
require.NoError(t, env.GetWorkflowResult(&result))
require.Equal(t, want, result)
}
func TestExternalWorkflow_DoChildAsync(t *testing.T) {
suite := &testsuite.WorkflowTestSuite{}
env := suite.NewTestWorkflowEnvironment()
wflow := ExternalWorkflow[FakeInput, string]{
Name: "fake-workflow",
TaskQueue: "X",
HandleResult: func(wctx workflow.Context, result string, err error) (string, error) {
assert.Equal(t, "sub-workflow", result)
return "handled-result", nil
},
}
mainWorkflow := func(wctx workflow.Context, input FakeInput) (string, error) {
future := wflow.DoChildAsync(wctx, input)
return future.GetTyped(wctx)
}
subWorkflow := func(wctx workflow.Context, input FakeInput) (string, error) {
return "sub-workflow", nil
}
env.RegisterWorkflow(mainWorkflow)
env.RegisterWorkflowWithOptions(subWorkflow, workflow.RegisterOptions{
Name: wflow.Name,
DisableAlreadyRegisteredCheck: false,
})
input := FakeInput{}
want := "handled-result"
env.ExecuteWorkflow(mainWorkflow, input)
require.True(t, env.IsWorkflowCompleted())
require.NoError(t, env.GetWorkflowError())
var result string
require.NoError(t, env.GetWorkflowResult(&result))
require.Equal(t, want, result)
}