-
Notifications
You must be signed in to change notification settings - Fork 0
/
fake_messenger.go
144 lines (128 loc) · 3.79 KB
/
fake_messenger.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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
// This file was generated by counterfeiter
package cowbullfakes
import (
"sync"
"github.com/Bo0mer/cowbull"
)
type FakeMessenger struct {
IDStub func() string
iDMutex sync.RWMutex
iDArgsForCall []struct{}
iDReturns struct {
result1 string
}
OnMessageStub func(kind string, action func(data string))
onMessageMutex sync.RWMutex
onMessageArgsForCall []struct {
kind string
action func(data string)
}
SendMessageStub func(kind string, data string) error
sendMessageMutex sync.RWMutex
sendMessageArgsForCall []struct {
kind string
data string
}
sendMessageReturns struct {
result1 error
}
invocations map[string][][]interface{}
invocationsMutex sync.RWMutex
}
func (fake *FakeMessenger) ID() string {
fake.iDMutex.Lock()
fake.iDArgsForCall = append(fake.iDArgsForCall, struct{}{})
fake.recordInvocation("ID", []interface{}{})
fake.iDMutex.Unlock()
if fake.IDStub != nil {
return fake.IDStub()
} else {
return fake.iDReturns.result1
}
}
func (fake *FakeMessenger) IDCallCount() int {
fake.iDMutex.RLock()
defer fake.iDMutex.RUnlock()
return len(fake.iDArgsForCall)
}
func (fake *FakeMessenger) IDReturns(result1 string) {
fake.IDStub = nil
fake.iDReturns = struct {
result1 string
}{result1}
}
func (fake *FakeMessenger) OnMessage(kind string, action func(data string)) {
fake.onMessageMutex.Lock()
fake.onMessageArgsForCall = append(fake.onMessageArgsForCall, struct {
kind string
action func(data string)
}{kind, action})
fake.recordInvocation("OnMessage", []interface{}{kind, action})
fake.onMessageMutex.Unlock()
if fake.OnMessageStub != nil {
fake.OnMessageStub(kind, action)
}
}
func (fake *FakeMessenger) OnMessageCallCount() int {
fake.onMessageMutex.RLock()
defer fake.onMessageMutex.RUnlock()
return len(fake.onMessageArgsForCall)
}
func (fake *FakeMessenger) OnMessageArgsForCall(i int) (string, func(data string)) {
fake.onMessageMutex.RLock()
defer fake.onMessageMutex.RUnlock()
return fake.onMessageArgsForCall[i].kind, fake.onMessageArgsForCall[i].action
}
func (fake *FakeMessenger) SendMessage(kind string, data string) error {
fake.sendMessageMutex.Lock()
fake.sendMessageArgsForCall = append(fake.sendMessageArgsForCall, struct {
kind string
data string
}{kind, data})
fake.recordInvocation("SendMessage", []interface{}{kind, data})
fake.sendMessageMutex.Unlock()
if fake.SendMessageStub != nil {
return fake.SendMessageStub(kind, data)
} else {
return fake.sendMessageReturns.result1
}
}
func (fake *FakeMessenger) SendMessageCallCount() int {
fake.sendMessageMutex.RLock()
defer fake.sendMessageMutex.RUnlock()
return len(fake.sendMessageArgsForCall)
}
func (fake *FakeMessenger) SendMessageArgsForCall(i int) (string, string) {
fake.sendMessageMutex.RLock()
defer fake.sendMessageMutex.RUnlock()
return fake.sendMessageArgsForCall[i].kind, fake.sendMessageArgsForCall[i].data
}
func (fake *FakeMessenger) SendMessageReturns(result1 error) {
fake.SendMessageStub = nil
fake.sendMessageReturns = struct {
result1 error
}{result1}
}
func (fake *FakeMessenger) Invocations() map[string][][]interface{} {
fake.invocationsMutex.RLock()
defer fake.invocationsMutex.RUnlock()
fake.iDMutex.RLock()
defer fake.iDMutex.RUnlock()
fake.onMessageMutex.RLock()
defer fake.onMessageMutex.RUnlock()
fake.sendMessageMutex.RLock()
defer fake.sendMessageMutex.RUnlock()
return fake.invocations
}
func (fake *FakeMessenger) recordInvocation(key string, args []interface{}) {
fake.invocationsMutex.Lock()
defer fake.invocationsMutex.Unlock()
if fake.invocations == nil {
fake.invocations = map[string][][]interface{}{}
}
if fake.invocations[key] == nil {
fake.invocations[key] = [][]interface{}{}
}
fake.invocations[key] = append(fake.invocations[key], args)
}
var _ cowbull.Messenger = new(FakeMessenger)