-
Notifications
You must be signed in to change notification settings - Fork 4
/
mock_sqs_client_test.go
75 lines (63 loc) · 2.12 KB
/
mock_sqs_client_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
// Code generated by mockery v2.7.5. DO NOT EDIT.
package asyncsqs
import (
context "context"
sqs "github.com/aws/aws-sdk-go-v2/service/sqs"
mock "github.com/stretchr/testify/mock"
)
// mockSQSClient is an autogenerated mock type for the SQSClient type
type mockSQSClient struct {
mock.Mock
}
// DeleteMessageBatch provides a mock function with given fields: _a0, _a1, _a2
func (_m *mockSQSClient) DeleteMessageBatch(_a0 context.Context, _a1 *sqs.DeleteMessageBatchInput, _a2 ...func(*sqs.Options)) (*sqs.DeleteMessageBatchOutput, error) {
_va := make([]interface{}, len(_a2))
for _i := range _a2 {
_va[_i] = _a2[_i]
}
var _ca []interface{}
_ca = append(_ca, _a0, _a1)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *sqs.DeleteMessageBatchOutput
if rf, ok := ret.Get(0).(func(context.Context, *sqs.DeleteMessageBatchInput, ...func(*sqs.Options)) *sqs.DeleteMessageBatchOutput); ok {
r0 = rf(_a0, _a1, _a2...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*sqs.DeleteMessageBatchOutput)
}
}
var r1 error
if rf, ok := ret.Get(1).(func(context.Context, *sqs.DeleteMessageBatchInput, ...func(*sqs.Options)) error); ok {
r1 = rf(_a0, _a1, _a2...)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// SendMessageBatch provides a mock function with given fields: _a0, _a1, _a2
func (_m *mockSQSClient) SendMessageBatch(_a0 context.Context, _a1 *sqs.SendMessageBatchInput, _a2 ...func(*sqs.Options)) (*sqs.SendMessageBatchOutput, error) {
_va := make([]interface{}, len(_a2))
for _i := range _a2 {
_va[_i] = _a2[_i]
}
var _ca []interface{}
_ca = append(_ca, _a0, _a1)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *sqs.SendMessageBatchOutput
if rf, ok := ret.Get(0).(func(context.Context, *sqs.SendMessageBatchInput, ...func(*sqs.Options)) *sqs.SendMessageBatchOutput); ok {
r0 = rf(_a0, _a1, _a2...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*sqs.SendMessageBatchOutput)
}
}
var r1 error
if rf, ok := ret.Get(1).(func(context.Context, *sqs.SendMessageBatchInput, ...func(*sqs.Options)) error); ok {
r1 = rf(_a0, _a1, _a2...)
} else {
r1 = ret.Error(1)
}
return r0, r1
}