This repository has been archived by the owner on Mar 21, 2024. It is now read-only.
/
Publisher.go
98 lines (78 loc) · 2.75 KB
/
Publisher.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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
// Code generated by mockery v2.32.4. DO NOT EDIT.
package mocks
import mock "github.com/stretchr/testify/mock"
// Publisher is an autogenerated mock type for the Publisher type
type Publisher[T interface{}] struct {
mock.Mock
}
type Publisher_Expecter[T interface{}] struct {
mock *mock.Mock
}
func (_m *Publisher[T]) EXPECT() *Publisher_Expecter[T] {
return &Publisher_Expecter[T]{mock: &_m.Mock}
}
// Register provides a mock function with given fields: _a0
func (_m *Publisher[T]) Register(_a0 chan T) {
_m.Called(_a0)
}
// Publisher_Register_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Register'
type Publisher_Register_Call[T interface{}] struct {
*mock.Call
}
// Register is a helper method to define mock.On call
// - _a0 chan T
func (_e *Publisher_Expecter[T]) Register(_a0 interface{}) *Publisher_Register_Call[T] {
return &Publisher_Register_Call[T]{Call: _e.mock.On("Register", _a0)}
}
func (_c *Publisher_Register_Call[T]) Run(run func(_a0 chan T)) *Publisher_Register_Call[T] {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(chan T))
})
return _c
}
func (_c *Publisher_Register_Call[T]) Return() *Publisher_Register_Call[T] {
_c.Call.Return()
return _c
}
func (_c *Publisher_Register_Call[T]) RunAndReturn(run func(chan T)) *Publisher_Register_Call[T] {
_c.Call.Return(run)
return _c
}
// Unregister provides a mock function with given fields: _a0
func (_m *Publisher[T]) Unregister(_a0 chan T) {
_m.Called(_a0)
}
// Publisher_Unregister_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Unregister'
type Publisher_Unregister_Call[T interface{}] struct {
*mock.Call
}
// Unregister is a helper method to define mock.On call
// - _a0 chan T
func (_e *Publisher_Expecter[T]) Unregister(_a0 interface{}) *Publisher_Unregister_Call[T] {
return &Publisher_Unregister_Call[T]{Call: _e.mock.On("Unregister", _a0)}
}
func (_c *Publisher_Unregister_Call[T]) Run(run func(_a0 chan T)) *Publisher_Unregister_Call[T] {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(chan T))
})
return _c
}
func (_c *Publisher_Unregister_Call[T]) Return() *Publisher_Unregister_Call[T] {
_c.Call.Return()
return _c
}
func (_c *Publisher_Unregister_Call[T]) RunAndReturn(run func(chan T)) *Publisher_Unregister_Call[T] {
_c.Call.Return(run)
return _c
}
// NewPublisher creates a new instance of Publisher. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
// The first argument is typically a *testing.T value.
func NewPublisher[T interface{}](t interface {
mock.TestingT
Cleanup(func())
}) *Publisher[T] {
mock := &Publisher[T]{}
mock.Mock.Test(t)
t.Cleanup(func() { mock.AssertExpectations(t) })
return mock
}