-
Notifications
You must be signed in to change notification settings - Fork 0
/
send_test.go
62 lines (47 loc) · 1.63 KB
/
send_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
package mediator_test
import (
"context"
"github.com/Oleexo/mediator-go"
"github.com/stretchr/testify/assert"
"testing"
)
type TestRequest struct {
Value string
}
func (t TestRequest) String() string {
return t.Value
}
type TestRequestHandler struct {
}
func (t TestRequestHandler) Handle(ctx context.Context, request *TestRequest) (string, error) {
return request.Value, nil
}
type TestPipelineBehavior struct {
}
func (t TestPipelineBehavior) Handle(ctx context.Context, request mediator.BaseRequest, next mediator.RequestHandlerFunc) (interface{}, error) {
request.(*TestRequest).Value = "pipeline"
return next()
}
func TestSend(t *testing.T) {
t.Run("should send a request to a single handler", func(t *testing.T) {
handler := TestRequestHandler{}
container := mediator.NewSendContainer(
mediator.WithRequestDefinitionHandler(mediator.NewRequestHandlerDefinition[*TestRequest, string](handler)),
)
request := TestRequest{Value: "test"}
response, err := mediator.Send[*TestRequest, string](context.Background(), container, &request)
assert.NoError(t, err)
assert.Equal(t, request.Value, response)
})
t.Run("should apply pipeline behavior", func(t *testing.T) {
handler := TestRequestHandler{}
container := mediator.NewSendContainer(
mediator.WithRequestDefinitionHandler(mediator.NewRequestHandlerDefinition[*TestRequest, string](handler)),
mediator.WithPipelineBehavior(TestPipelineBehavior{}),
)
request := TestRequest{Value: "test"}
response, err := mediator.Send[*TestRequest, string](context.Background(), container, &request)
assert.NoError(t, err)
assert.Equal(t, "pipeline", response)
})
}