mocked_connection_info_test.go 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491
  1. // Code generated by counterfeiter. DO NOT EDIT.
  2. package protocol
  3. import (
  4. "net"
  5. "sync"
  6. "time"
  7. )
  8. type mockedConnectionInfo struct {
  9. CryptoStub func() string
  10. cryptoMutex sync.RWMutex
  11. cryptoArgsForCall []struct {
  12. }
  13. cryptoReturns struct {
  14. result1 string
  15. }
  16. cryptoReturnsOnCall map[int]struct {
  17. result1 string
  18. }
  19. EstablishedAtStub func() time.Time
  20. establishedAtMutex sync.RWMutex
  21. establishedAtArgsForCall []struct {
  22. }
  23. establishedAtReturns struct {
  24. result1 time.Time
  25. }
  26. establishedAtReturnsOnCall map[int]struct {
  27. result1 time.Time
  28. }
  29. PriorityStub func() int
  30. priorityMutex sync.RWMutex
  31. priorityArgsForCall []struct {
  32. }
  33. priorityReturns struct {
  34. result1 int
  35. }
  36. priorityReturnsOnCall map[int]struct {
  37. result1 int
  38. }
  39. RemoteAddrStub func() net.Addr
  40. remoteAddrMutex sync.RWMutex
  41. remoteAddrArgsForCall []struct {
  42. }
  43. remoteAddrReturns struct {
  44. result1 net.Addr
  45. }
  46. remoteAddrReturnsOnCall map[int]struct {
  47. result1 net.Addr
  48. }
  49. StringStub func() string
  50. stringMutex sync.RWMutex
  51. stringArgsForCall []struct {
  52. }
  53. stringReturns struct {
  54. result1 string
  55. }
  56. stringReturnsOnCall map[int]struct {
  57. result1 string
  58. }
  59. TransportStub func() string
  60. transportMutex sync.RWMutex
  61. transportArgsForCall []struct {
  62. }
  63. transportReturns struct {
  64. result1 string
  65. }
  66. transportReturnsOnCall map[int]struct {
  67. result1 string
  68. }
  69. TypeStub func() string
  70. typeMutex sync.RWMutex
  71. typeArgsForCall []struct {
  72. }
  73. typeReturns struct {
  74. result1 string
  75. }
  76. typeReturnsOnCall map[int]struct {
  77. result1 string
  78. }
  79. invocations map[string][][]interface{}
  80. invocationsMutex sync.RWMutex
  81. }
  82. func (fake *mockedConnectionInfo) Crypto() string {
  83. fake.cryptoMutex.Lock()
  84. ret, specificReturn := fake.cryptoReturnsOnCall[len(fake.cryptoArgsForCall)]
  85. fake.cryptoArgsForCall = append(fake.cryptoArgsForCall, struct {
  86. }{})
  87. stub := fake.CryptoStub
  88. fakeReturns := fake.cryptoReturns
  89. fake.recordInvocation("Crypto", []interface{}{})
  90. fake.cryptoMutex.Unlock()
  91. if stub != nil {
  92. return stub()
  93. }
  94. if specificReturn {
  95. return ret.result1
  96. }
  97. return fakeReturns.result1
  98. }
  99. func (fake *mockedConnectionInfo) CryptoCallCount() int {
  100. fake.cryptoMutex.RLock()
  101. defer fake.cryptoMutex.RUnlock()
  102. return len(fake.cryptoArgsForCall)
  103. }
  104. func (fake *mockedConnectionInfo) CryptoCalls(stub func() string) {
  105. fake.cryptoMutex.Lock()
  106. defer fake.cryptoMutex.Unlock()
  107. fake.CryptoStub = stub
  108. }
  109. func (fake *mockedConnectionInfo) CryptoReturns(result1 string) {
  110. fake.cryptoMutex.Lock()
  111. defer fake.cryptoMutex.Unlock()
  112. fake.CryptoStub = nil
  113. fake.cryptoReturns = struct {
  114. result1 string
  115. }{result1}
  116. }
  117. func (fake *mockedConnectionInfo) CryptoReturnsOnCall(i int, result1 string) {
  118. fake.cryptoMutex.Lock()
  119. defer fake.cryptoMutex.Unlock()
  120. fake.CryptoStub = nil
  121. if fake.cryptoReturnsOnCall == nil {
  122. fake.cryptoReturnsOnCall = make(map[int]struct {
  123. result1 string
  124. })
  125. }
  126. fake.cryptoReturnsOnCall[i] = struct {
  127. result1 string
  128. }{result1}
  129. }
  130. func (fake *mockedConnectionInfo) EstablishedAt() time.Time {
  131. fake.establishedAtMutex.Lock()
  132. ret, specificReturn := fake.establishedAtReturnsOnCall[len(fake.establishedAtArgsForCall)]
  133. fake.establishedAtArgsForCall = append(fake.establishedAtArgsForCall, struct {
  134. }{})
  135. stub := fake.EstablishedAtStub
  136. fakeReturns := fake.establishedAtReturns
  137. fake.recordInvocation("EstablishedAt", []interface{}{})
  138. fake.establishedAtMutex.Unlock()
  139. if stub != nil {
  140. return stub()
  141. }
  142. if specificReturn {
  143. return ret.result1
  144. }
  145. return fakeReturns.result1
  146. }
  147. func (fake *mockedConnectionInfo) EstablishedAtCallCount() int {
  148. fake.establishedAtMutex.RLock()
  149. defer fake.establishedAtMutex.RUnlock()
  150. return len(fake.establishedAtArgsForCall)
  151. }
  152. func (fake *mockedConnectionInfo) EstablishedAtCalls(stub func() time.Time) {
  153. fake.establishedAtMutex.Lock()
  154. defer fake.establishedAtMutex.Unlock()
  155. fake.EstablishedAtStub = stub
  156. }
  157. func (fake *mockedConnectionInfo) EstablishedAtReturns(result1 time.Time) {
  158. fake.establishedAtMutex.Lock()
  159. defer fake.establishedAtMutex.Unlock()
  160. fake.EstablishedAtStub = nil
  161. fake.establishedAtReturns = struct {
  162. result1 time.Time
  163. }{result1}
  164. }
  165. func (fake *mockedConnectionInfo) EstablishedAtReturnsOnCall(i int, result1 time.Time) {
  166. fake.establishedAtMutex.Lock()
  167. defer fake.establishedAtMutex.Unlock()
  168. fake.EstablishedAtStub = nil
  169. if fake.establishedAtReturnsOnCall == nil {
  170. fake.establishedAtReturnsOnCall = make(map[int]struct {
  171. result1 time.Time
  172. })
  173. }
  174. fake.establishedAtReturnsOnCall[i] = struct {
  175. result1 time.Time
  176. }{result1}
  177. }
  178. func (fake *mockedConnectionInfo) Priority() int {
  179. fake.priorityMutex.Lock()
  180. ret, specificReturn := fake.priorityReturnsOnCall[len(fake.priorityArgsForCall)]
  181. fake.priorityArgsForCall = append(fake.priorityArgsForCall, struct {
  182. }{})
  183. stub := fake.PriorityStub
  184. fakeReturns := fake.priorityReturns
  185. fake.recordInvocation("Priority", []interface{}{})
  186. fake.priorityMutex.Unlock()
  187. if stub != nil {
  188. return stub()
  189. }
  190. if specificReturn {
  191. return ret.result1
  192. }
  193. return fakeReturns.result1
  194. }
  195. func (fake *mockedConnectionInfo) PriorityCallCount() int {
  196. fake.priorityMutex.RLock()
  197. defer fake.priorityMutex.RUnlock()
  198. return len(fake.priorityArgsForCall)
  199. }
  200. func (fake *mockedConnectionInfo) PriorityCalls(stub func() int) {
  201. fake.priorityMutex.Lock()
  202. defer fake.priorityMutex.Unlock()
  203. fake.PriorityStub = stub
  204. }
  205. func (fake *mockedConnectionInfo) PriorityReturns(result1 int) {
  206. fake.priorityMutex.Lock()
  207. defer fake.priorityMutex.Unlock()
  208. fake.PriorityStub = nil
  209. fake.priorityReturns = struct {
  210. result1 int
  211. }{result1}
  212. }
  213. func (fake *mockedConnectionInfo) PriorityReturnsOnCall(i int, result1 int) {
  214. fake.priorityMutex.Lock()
  215. defer fake.priorityMutex.Unlock()
  216. fake.PriorityStub = nil
  217. if fake.priorityReturnsOnCall == nil {
  218. fake.priorityReturnsOnCall = make(map[int]struct {
  219. result1 int
  220. })
  221. }
  222. fake.priorityReturnsOnCall[i] = struct {
  223. result1 int
  224. }{result1}
  225. }
  226. func (fake *mockedConnectionInfo) RemoteAddr() net.Addr {
  227. fake.remoteAddrMutex.Lock()
  228. ret, specificReturn := fake.remoteAddrReturnsOnCall[len(fake.remoteAddrArgsForCall)]
  229. fake.remoteAddrArgsForCall = append(fake.remoteAddrArgsForCall, struct {
  230. }{})
  231. stub := fake.RemoteAddrStub
  232. fakeReturns := fake.remoteAddrReturns
  233. fake.recordInvocation("RemoteAddr", []interface{}{})
  234. fake.remoteAddrMutex.Unlock()
  235. if stub != nil {
  236. return stub()
  237. }
  238. if specificReturn {
  239. return ret.result1
  240. }
  241. return fakeReturns.result1
  242. }
  243. func (fake *mockedConnectionInfo) RemoteAddrCallCount() int {
  244. fake.remoteAddrMutex.RLock()
  245. defer fake.remoteAddrMutex.RUnlock()
  246. return len(fake.remoteAddrArgsForCall)
  247. }
  248. func (fake *mockedConnectionInfo) RemoteAddrCalls(stub func() net.Addr) {
  249. fake.remoteAddrMutex.Lock()
  250. defer fake.remoteAddrMutex.Unlock()
  251. fake.RemoteAddrStub = stub
  252. }
  253. func (fake *mockedConnectionInfo) RemoteAddrReturns(result1 net.Addr) {
  254. fake.remoteAddrMutex.Lock()
  255. defer fake.remoteAddrMutex.Unlock()
  256. fake.RemoteAddrStub = nil
  257. fake.remoteAddrReturns = struct {
  258. result1 net.Addr
  259. }{result1}
  260. }
  261. func (fake *mockedConnectionInfo) RemoteAddrReturnsOnCall(i int, result1 net.Addr) {
  262. fake.remoteAddrMutex.Lock()
  263. defer fake.remoteAddrMutex.Unlock()
  264. fake.RemoteAddrStub = nil
  265. if fake.remoteAddrReturnsOnCall == nil {
  266. fake.remoteAddrReturnsOnCall = make(map[int]struct {
  267. result1 net.Addr
  268. })
  269. }
  270. fake.remoteAddrReturnsOnCall[i] = struct {
  271. result1 net.Addr
  272. }{result1}
  273. }
  274. func (fake *mockedConnectionInfo) String() string {
  275. fake.stringMutex.Lock()
  276. ret, specificReturn := fake.stringReturnsOnCall[len(fake.stringArgsForCall)]
  277. fake.stringArgsForCall = append(fake.stringArgsForCall, struct {
  278. }{})
  279. stub := fake.StringStub
  280. fakeReturns := fake.stringReturns
  281. fake.recordInvocation("String", []interface{}{})
  282. fake.stringMutex.Unlock()
  283. if stub != nil {
  284. return stub()
  285. }
  286. if specificReturn {
  287. return ret.result1
  288. }
  289. return fakeReturns.result1
  290. }
  291. func (fake *mockedConnectionInfo) StringCallCount() int {
  292. fake.stringMutex.RLock()
  293. defer fake.stringMutex.RUnlock()
  294. return len(fake.stringArgsForCall)
  295. }
  296. func (fake *mockedConnectionInfo) StringCalls(stub func() string) {
  297. fake.stringMutex.Lock()
  298. defer fake.stringMutex.Unlock()
  299. fake.StringStub = stub
  300. }
  301. func (fake *mockedConnectionInfo) StringReturns(result1 string) {
  302. fake.stringMutex.Lock()
  303. defer fake.stringMutex.Unlock()
  304. fake.StringStub = nil
  305. fake.stringReturns = struct {
  306. result1 string
  307. }{result1}
  308. }
  309. func (fake *mockedConnectionInfo) StringReturnsOnCall(i int, result1 string) {
  310. fake.stringMutex.Lock()
  311. defer fake.stringMutex.Unlock()
  312. fake.StringStub = nil
  313. if fake.stringReturnsOnCall == nil {
  314. fake.stringReturnsOnCall = make(map[int]struct {
  315. result1 string
  316. })
  317. }
  318. fake.stringReturnsOnCall[i] = struct {
  319. result1 string
  320. }{result1}
  321. }
  322. func (fake *mockedConnectionInfo) Transport() string {
  323. fake.transportMutex.Lock()
  324. ret, specificReturn := fake.transportReturnsOnCall[len(fake.transportArgsForCall)]
  325. fake.transportArgsForCall = append(fake.transportArgsForCall, struct {
  326. }{})
  327. stub := fake.TransportStub
  328. fakeReturns := fake.transportReturns
  329. fake.recordInvocation("Transport", []interface{}{})
  330. fake.transportMutex.Unlock()
  331. if stub != nil {
  332. return stub()
  333. }
  334. if specificReturn {
  335. return ret.result1
  336. }
  337. return fakeReturns.result1
  338. }
  339. func (fake *mockedConnectionInfo) TransportCallCount() int {
  340. fake.transportMutex.RLock()
  341. defer fake.transportMutex.RUnlock()
  342. return len(fake.transportArgsForCall)
  343. }
  344. func (fake *mockedConnectionInfo) TransportCalls(stub func() string) {
  345. fake.transportMutex.Lock()
  346. defer fake.transportMutex.Unlock()
  347. fake.TransportStub = stub
  348. }
  349. func (fake *mockedConnectionInfo) TransportReturns(result1 string) {
  350. fake.transportMutex.Lock()
  351. defer fake.transportMutex.Unlock()
  352. fake.TransportStub = nil
  353. fake.transportReturns = struct {
  354. result1 string
  355. }{result1}
  356. }
  357. func (fake *mockedConnectionInfo) TransportReturnsOnCall(i int, result1 string) {
  358. fake.transportMutex.Lock()
  359. defer fake.transportMutex.Unlock()
  360. fake.TransportStub = nil
  361. if fake.transportReturnsOnCall == nil {
  362. fake.transportReturnsOnCall = make(map[int]struct {
  363. result1 string
  364. })
  365. }
  366. fake.transportReturnsOnCall[i] = struct {
  367. result1 string
  368. }{result1}
  369. }
  370. func (fake *mockedConnectionInfo) Type() string {
  371. fake.typeMutex.Lock()
  372. ret, specificReturn := fake.typeReturnsOnCall[len(fake.typeArgsForCall)]
  373. fake.typeArgsForCall = append(fake.typeArgsForCall, struct {
  374. }{})
  375. stub := fake.TypeStub
  376. fakeReturns := fake.typeReturns
  377. fake.recordInvocation("Type", []interface{}{})
  378. fake.typeMutex.Unlock()
  379. if stub != nil {
  380. return stub()
  381. }
  382. if specificReturn {
  383. return ret.result1
  384. }
  385. return fakeReturns.result1
  386. }
  387. func (fake *mockedConnectionInfo) TypeCallCount() int {
  388. fake.typeMutex.RLock()
  389. defer fake.typeMutex.RUnlock()
  390. return len(fake.typeArgsForCall)
  391. }
  392. func (fake *mockedConnectionInfo) TypeCalls(stub func() string) {
  393. fake.typeMutex.Lock()
  394. defer fake.typeMutex.Unlock()
  395. fake.TypeStub = stub
  396. }
  397. func (fake *mockedConnectionInfo) TypeReturns(result1 string) {
  398. fake.typeMutex.Lock()
  399. defer fake.typeMutex.Unlock()
  400. fake.TypeStub = nil
  401. fake.typeReturns = struct {
  402. result1 string
  403. }{result1}
  404. }
  405. func (fake *mockedConnectionInfo) TypeReturnsOnCall(i int, result1 string) {
  406. fake.typeMutex.Lock()
  407. defer fake.typeMutex.Unlock()
  408. fake.TypeStub = nil
  409. if fake.typeReturnsOnCall == nil {
  410. fake.typeReturnsOnCall = make(map[int]struct {
  411. result1 string
  412. })
  413. }
  414. fake.typeReturnsOnCall[i] = struct {
  415. result1 string
  416. }{result1}
  417. }
  418. func (fake *mockedConnectionInfo) Invocations() map[string][][]interface{} {
  419. fake.invocationsMutex.RLock()
  420. defer fake.invocationsMutex.RUnlock()
  421. fake.cryptoMutex.RLock()
  422. defer fake.cryptoMutex.RUnlock()
  423. fake.establishedAtMutex.RLock()
  424. defer fake.establishedAtMutex.RUnlock()
  425. fake.priorityMutex.RLock()
  426. defer fake.priorityMutex.RUnlock()
  427. fake.remoteAddrMutex.RLock()
  428. defer fake.remoteAddrMutex.RUnlock()
  429. fake.stringMutex.RLock()
  430. defer fake.stringMutex.RUnlock()
  431. fake.transportMutex.RLock()
  432. defer fake.transportMutex.RUnlock()
  433. fake.typeMutex.RLock()
  434. defer fake.typeMutex.RUnlock()
  435. copiedInvocations := map[string][][]interface{}{}
  436. for key, value := range fake.invocations {
  437. copiedInvocations[key] = value
  438. }
  439. return copiedInvocations
  440. }
  441. func (fake *mockedConnectionInfo) recordInvocation(key string, args []interface{}) {
  442. fake.invocationsMutex.Lock()
  443. defer fake.invocationsMutex.Unlock()
  444. if fake.invocations == nil {
  445. fake.invocations = map[string][][]interface{}{}
  446. }
  447. if fake.invocations[key] == nil {
  448. fake.invocations[key] = [][]interface{}{}
  449. }
  450. fake.invocations[key] = append(fake.invocations[key], args)
  451. }