mocked_connection_info_test.go 14 KB

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