-
Notifications
You must be signed in to change notification settings - Fork 14
/
mock_worker.go
71 lines (57 loc) · 2.04 KB
/
mock_worker.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
package temporal_tester
import (
"go.temporal.io/sdk/activity"
"go.temporal.io/sdk/client"
"go.temporal.io/sdk/converter"
"go.temporal.io/sdk/testsuite"
"go.temporal.io/sdk/worker"
"go.temporal.io/sdk/workflow"
)
var _ worker.Worker = &MockWorker{}
type MockWorker struct {
actEnv *testsuite.TestActivityEnvironment
mockClient *MockClient
}
func MockWorkerBuilder(client client.Client, taskQueue string, options worker.Options) worker.Worker {
mockClient, is := client.(*MockClient)
if !is {
panic("MockWorker needs MockClient")
}
suite := &testsuite.WorkflowTestSuite{}
mockClient.Env = mockClient.Env.SetWorkerOptions(options)
return &MockWorker{
actEnv: suite.NewTestActivityEnvironment().SetWorkerOptions(options),
mockClient: mockClient,
}
}
func (mw *MockWorker) Start() error { return nil }
func (mw *MockWorker) Run(interruptCh <-chan interface{}) error { return nil }
func (mw *MockWorker) Stop() {}
func (mw *MockWorker) RegisterWorkflow(w interface{}) {
mw.mockClient.Env.RegisterWorkflow(w)
}
func (mw *MockWorker) RegisterWorkflowWithOptions(w interface{}, options workflow.RegisterOptions) {
mw.mockClient.Env.RegisterWorkflowWithOptions(w, options)
}
func (mw *MockWorker) RegisterActivity(a interface{}) {
mw.mockClient.Env.RegisterActivity(a)
}
func (mw *MockWorker) RegisterActivityWithOptions(a interface{}, options activity.RegisterOptions) {
mw.mockClient.Env.RegisterActivityWithOptions(a, options)
}
func (mw *MockWorker) GetTestActivityEnv() *testsuite.TestActivityEnvironment {
return mw.actEnv
}
// TestFuture is a wrapper of the test result of executing test activity.
type TestFuture[Resp any] struct {
converter.EncodedValue
}
func (t *TestFuture[Resp]) Get() (Resp, error) {
var v Resp
err := t.EncodedValue.Get(&v)
return v, err
}
func ExecuteTestActivity[Resp any](env *testsuite.TestActivityEnvironment, name string, params ...any) (*TestFuture[Resp], error) {
r, err := env.ExecuteActivity(name, params...)
return &TestFuture[Resp]{r}, err
}