connection.go 29 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112
  1. // Code generated by counterfeiter. DO NOT EDIT.
  2. package mocks
  3. import (
  4. "context"
  5. "net"
  6. "sync"
  7. "time"
  8. "github.com/syncthing/syncthing/lib/protocol"
  9. )
  10. type Connection struct {
  11. CloseStub func(error)
  12. closeMutex sync.RWMutex
  13. closeArgsForCall []struct {
  14. arg1 error
  15. }
  16. ClosedStub func() bool
  17. closedMutex sync.RWMutex
  18. closedArgsForCall []struct {
  19. }
  20. closedReturns struct {
  21. result1 bool
  22. }
  23. closedReturnsOnCall map[int]struct {
  24. result1 bool
  25. }
  26. ClusterConfigStub func(protocol.ClusterConfig)
  27. clusterConfigMutex sync.RWMutex
  28. clusterConfigArgsForCall []struct {
  29. arg1 protocol.ClusterConfig
  30. }
  31. CryptoStub func() string
  32. cryptoMutex sync.RWMutex
  33. cryptoArgsForCall []struct {
  34. }
  35. cryptoReturns struct {
  36. result1 string
  37. }
  38. cryptoReturnsOnCall map[int]struct {
  39. result1 string
  40. }
  41. DownloadProgressStub func(context.Context, string, []protocol.FileDownloadProgressUpdate)
  42. downloadProgressMutex sync.RWMutex
  43. downloadProgressArgsForCall []struct {
  44. arg1 context.Context
  45. arg2 string
  46. arg3 []protocol.FileDownloadProgressUpdate
  47. }
  48. EstablishedAtStub func() time.Time
  49. establishedAtMutex sync.RWMutex
  50. establishedAtArgsForCall []struct {
  51. }
  52. establishedAtReturns struct {
  53. result1 time.Time
  54. }
  55. establishedAtReturnsOnCall map[int]struct {
  56. result1 time.Time
  57. }
  58. IDStub func() protocol.DeviceID
  59. iDMutex sync.RWMutex
  60. iDArgsForCall []struct {
  61. }
  62. iDReturns struct {
  63. result1 protocol.DeviceID
  64. }
  65. iDReturnsOnCall map[int]struct {
  66. result1 protocol.DeviceID
  67. }
  68. IndexStub func(context.Context, string, []protocol.FileInfo) error
  69. indexMutex sync.RWMutex
  70. indexArgsForCall []struct {
  71. arg1 context.Context
  72. arg2 string
  73. arg3 []protocol.FileInfo
  74. }
  75. indexReturns struct {
  76. result1 error
  77. }
  78. indexReturnsOnCall map[int]struct {
  79. result1 error
  80. }
  81. IndexUpdateStub func(context.Context, string, []protocol.FileInfo) error
  82. indexUpdateMutex sync.RWMutex
  83. indexUpdateArgsForCall []struct {
  84. arg1 context.Context
  85. arg2 string
  86. arg3 []protocol.FileInfo
  87. }
  88. indexUpdateReturns struct {
  89. result1 error
  90. }
  91. indexUpdateReturnsOnCall map[int]struct {
  92. result1 error
  93. }
  94. PriorityStub func() int
  95. priorityMutex sync.RWMutex
  96. priorityArgsForCall []struct {
  97. }
  98. priorityReturns struct {
  99. result1 int
  100. }
  101. priorityReturnsOnCall map[int]struct {
  102. result1 int
  103. }
  104. RemoteAddrStub func() net.Addr
  105. remoteAddrMutex sync.RWMutex
  106. remoteAddrArgsForCall []struct {
  107. }
  108. remoteAddrReturns struct {
  109. result1 net.Addr
  110. }
  111. remoteAddrReturnsOnCall map[int]struct {
  112. result1 net.Addr
  113. }
  114. RequestStub func(context.Context, string, string, int, int64, int, []byte, uint32, bool) ([]byte, error)
  115. requestMutex sync.RWMutex
  116. requestArgsForCall []struct {
  117. arg1 context.Context
  118. arg2 string
  119. arg3 string
  120. arg4 int
  121. arg5 int64
  122. arg6 int
  123. arg7 []byte
  124. arg8 uint32
  125. arg9 bool
  126. }
  127. requestReturns struct {
  128. result1 []byte
  129. result2 error
  130. }
  131. requestReturnsOnCall map[int]struct {
  132. result1 []byte
  133. result2 error
  134. }
  135. StartStub func()
  136. startMutex sync.RWMutex
  137. startArgsForCall []struct {
  138. }
  139. StatisticsStub func() protocol.Statistics
  140. statisticsMutex sync.RWMutex
  141. statisticsArgsForCall []struct {
  142. }
  143. statisticsReturns struct {
  144. result1 protocol.Statistics
  145. }
  146. statisticsReturnsOnCall map[int]struct {
  147. result1 protocol.Statistics
  148. }
  149. StringStub func() string
  150. stringMutex sync.RWMutex
  151. stringArgsForCall []struct {
  152. }
  153. stringReturns struct {
  154. result1 string
  155. }
  156. stringReturnsOnCall map[int]struct {
  157. result1 string
  158. }
  159. TransportStub func() string
  160. transportMutex sync.RWMutex
  161. transportArgsForCall []struct {
  162. }
  163. transportReturns struct {
  164. result1 string
  165. }
  166. transportReturnsOnCall map[int]struct {
  167. result1 string
  168. }
  169. TypeStub func() string
  170. typeMutex sync.RWMutex
  171. typeArgsForCall []struct {
  172. }
  173. typeReturns struct {
  174. result1 string
  175. }
  176. typeReturnsOnCall map[int]struct {
  177. result1 string
  178. }
  179. invocations map[string][][]interface{}
  180. invocationsMutex sync.RWMutex
  181. }
  182. func (fake *Connection) Close(arg1 error) {
  183. fake.closeMutex.Lock()
  184. fake.closeArgsForCall = append(fake.closeArgsForCall, struct {
  185. arg1 error
  186. }{arg1})
  187. stub := fake.CloseStub
  188. fake.recordInvocation("Close", []interface{}{arg1})
  189. fake.closeMutex.Unlock()
  190. if stub != nil {
  191. fake.CloseStub(arg1)
  192. }
  193. }
  194. func (fake *Connection) CloseCallCount() int {
  195. fake.closeMutex.RLock()
  196. defer fake.closeMutex.RUnlock()
  197. return len(fake.closeArgsForCall)
  198. }
  199. func (fake *Connection) CloseCalls(stub func(error)) {
  200. fake.closeMutex.Lock()
  201. defer fake.closeMutex.Unlock()
  202. fake.CloseStub = stub
  203. }
  204. func (fake *Connection) CloseArgsForCall(i int) error {
  205. fake.closeMutex.RLock()
  206. defer fake.closeMutex.RUnlock()
  207. argsForCall := fake.closeArgsForCall[i]
  208. return argsForCall.arg1
  209. }
  210. func (fake *Connection) Closed() bool {
  211. fake.closedMutex.Lock()
  212. ret, specificReturn := fake.closedReturnsOnCall[len(fake.closedArgsForCall)]
  213. fake.closedArgsForCall = append(fake.closedArgsForCall, struct {
  214. }{})
  215. stub := fake.ClosedStub
  216. fakeReturns := fake.closedReturns
  217. fake.recordInvocation("Closed", []interface{}{})
  218. fake.closedMutex.Unlock()
  219. if stub != nil {
  220. return stub()
  221. }
  222. if specificReturn {
  223. return ret.result1
  224. }
  225. return fakeReturns.result1
  226. }
  227. func (fake *Connection) ClosedCallCount() int {
  228. fake.closedMutex.RLock()
  229. defer fake.closedMutex.RUnlock()
  230. return len(fake.closedArgsForCall)
  231. }
  232. func (fake *Connection) ClosedCalls(stub func() bool) {
  233. fake.closedMutex.Lock()
  234. defer fake.closedMutex.Unlock()
  235. fake.ClosedStub = stub
  236. }
  237. func (fake *Connection) ClosedReturns(result1 bool) {
  238. fake.closedMutex.Lock()
  239. defer fake.closedMutex.Unlock()
  240. fake.ClosedStub = nil
  241. fake.closedReturns = struct {
  242. result1 bool
  243. }{result1}
  244. }
  245. func (fake *Connection) ClosedReturnsOnCall(i int, result1 bool) {
  246. fake.closedMutex.Lock()
  247. defer fake.closedMutex.Unlock()
  248. fake.ClosedStub = nil
  249. if fake.closedReturnsOnCall == nil {
  250. fake.closedReturnsOnCall = make(map[int]struct {
  251. result1 bool
  252. })
  253. }
  254. fake.closedReturnsOnCall[i] = struct {
  255. result1 bool
  256. }{result1}
  257. }
  258. func (fake *Connection) ClusterConfig(arg1 protocol.ClusterConfig) {
  259. fake.clusterConfigMutex.Lock()
  260. fake.clusterConfigArgsForCall = append(fake.clusterConfigArgsForCall, struct {
  261. arg1 protocol.ClusterConfig
  262. }{arg1})
  263. stub := fake.ClusterConfigStub
  264. fake.recordInvocation("ClusterConfig", []interface{}{arg1})
  265. fake.clusterConfigMutex.Unlock()
  266. if stub != nil {
  267. fake.ClusterConfigStub(arg1)
  268. }
  269. }
  270. func (fake *Connection) ClusterConfigCallCount() int {
  271. fake.clusterConfigMutex.RLock()
  272. defer fake.clusterConfigMutex.RUnlock()
  273. return len(fake.clusterConfigArgsForCall)
  274. }
  275. func (fake *Connection) ClusterConfigCalls(stub func(protocol.ClusterConfig)) {
  276. fake.clusterConfigMutex.Lock()
  277. defer fake.clusterConfigMutex.Unlock()
  278. fake.ClusterConfigStub = stub
  279. }
  280. func (fake *Connection) ClusterConfigArgsForCall(i int) protocol.ClusterConfig {
  281. fake.clusterConfigMutex.RLock()
  282. defer fake.clusterConfigMutex.RUnlock()
  283. argsForCall := fake.clusterConfigArgsForCall[i]
  284. return argsForCall.arg1
  285. }
  286. func (fake *Connection) Crypto() string {
  287. fake.cryptoMutex.Lock()
  288. ret, specificReturn := fake.cryptoReturnsOnCall[len(fake.cryptoArgsForCall)]
  289. fake.cryptoArgsForCall = append(fake.cryptoArgsForCall, struct {
  290. }{})
  291. stub := fake.CryptoStub
  292. fakeReturns := fake.cryptoReturns
  293. fake.recordInvocation("Crypto", []interface{}{})
  294. fake.cryptoMutex.Unlock()
  295. if stub != nil {
  296. return stub()
  297. }
  298. if specificReturn {
  299. return ret.result1
  300. }
  301. return fakeReturns.result1
  302. }
  303. func (fake *Connection) CryptoCallCount() int {
  304. fake.cryptoMutex.RLock()
  305. defer fake.cryptoMutex.RUnlock()
  306. return len(fake.cryptoArgsForCall)
  307. }
  308. func (fake *Connection) CryptoCalls(stub func() string) {
  309. fake.cryptoMutex.Lock()
  310. defer fake.cryptoMutex.Unlock()
  311. fake.CryptoStub = stub
  312. }
  313. func (fake *Connection) CryptoReturns(result1 string) {
  314. fake.cryptoMutex.Lock()
  315. defer fake.cryptoMutex.Unlock()
  316. fake.CryptoStub = nil
  317. fake.cryptoReturns = struct {
  318. result1 string
  319. }{result1}
  320. }
  321. func (fake *Connection) CryptoReturnsOnCall(i int, result1 string) {
  322. fake.cryptoMutex.Lock()
  323. defer fake.cryptoMutex.Unlock()
  324. fake.CryptoStub = nil
  325. if fake.cryptoReturnsOnCall == nil {
  326. fake.cryptoReturnsOnCall = make(map[int]struct {
  327. result1 string
  328. })
  329. }
  330. fake.cryptoReturnsOnCall[i] = struct {
  331. result1 string
  332. }{result1}
  333. }
  334. func (fake *Connection) DownloadProgress(arg1 context.Context, arg2 string, arg3 []protocol.FileDownloadProgressUpdate) {
  335. var arg3Copy []protocol.FileDownloadProgressUpdate
  336. if arg3 != nil {
  337. arg3Copy = make([]protocol.FileDownloadProgressUpdate, len(arg3))
  338. copy(arg3Copy, arg3)
  339. }
  340. fake.downloadProgressMutex.Lock()
  341. fake.downloadProgressArgsForCall = append(fake.downloadProgressArgsForCall, struct {
  342. arg1 context.Context
  343. arg2 string
  344. arg3 []protocol.FileDownloadProgressUpdate
  345. }{arg1, arg2, arg3Copy})
  346. stub := fake.DownloadProgressStub
  347. fake.recordInvocation("DownloadProgress", []interface{}{arg1, arg2, arg3Copy})
  348. fake.downloadProgressMutex.Unlock()
  349. if stub != nil {
  350. fake.DownloadProgressStub(arg1, arg2, arg3)
  351. }
  352. }
  353. func (fake *Connection) DownloadProgressCallCount() int {
  354. fake.downloadProgressMutex.RLock()
  355. defer fake.downloadProgressMutex.RUnlock()
  356. return len(fake.downloadProgressArgsForCall)
  357. }
  358. func (fake *Connection) DownloadProgressCalls(stub func(context.Context, string, []protocol.FileDownloadProgressUpdate)) {
  359. fake.downloadProgressMutex.Lock()
  360. defer fake.downloadProgressMutex.Unlock()
  361. fake.DownloadProgressStub = stub
  362. }
  363. func (fake *Connection) DownloadProgressArgsForCall(i int) (context.Context, string, []protocol.FileDownloadProgressUpdate) {
  364. fake.downloadProgressMutex.RLock()
  365. defer fake.downloadProgressMutex.RUnlock()
  366. argsForCall := fake.downloadProgressArgsForCall[i]
  367. return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
  368. }
  369. func (fake *Connection) EstablishedAt() time.Time {
  370. fake.establishedAtMutex.Lock()
  371. ret, specificReturn := fake.establishedAtReturnsOnCall[len(fake.establishedAtArgsForCall)]
  372. fake.establishedAtArgsForCall = append(fake.establishedAtArgsForCall, struct {
  373. }{})
  374. stub := fake.EstablishedAtStub
  375. fakeReturns := fake.establishedAtReturns
  376. fake.recordInvocation("EstablishedAt", []interface{}{})
  377. fake.establishedAtMutex.Unlock()
  378. if stub != nil {
  379. return stub()
  380. }
  381. if specificReturn {
  382. return ret.result1
  383. }
  384. return fakeReturns.result1
  385. }
  386. func (fake *Connection) EstablishedAtCallCount() int {
  387. fake.establishedAtMutex.RLock()
  388. defer fake.establishedAtMutex.RUnlock()
  389. return len(fake.establishedAtArgsForCall)
  390. }
  391. func (fake *Connection) EstablishedAtCalls(stub func() time.Time) {
  392. fake.establishedAtMutex.Lock()
  393. defer fake.establishedAtMutex.Unlock()
  394. fake.EstablishedAtStub = stub
  395. }
  396. func (fake *Connection) EstablishedAtReturns(result1 time.Time) {
  397. fake.establishedAtMutex.Lock()
  398. defer fake.establishedAtMutex.Unlock()
  399. fake.EstablishedAtStub = nil
  400. fake.establishedAtReturns = struct {
  401. result1 time.Time
  402. }{result1}
  403. }
  404. func (fake *Connection) EstablishedAtReturnsOnCall(i int, result1 time.Time) {
  405. fake.establishedAtMutex.Lock()
  406. defer fake.establishedAtMutex.Unlock()
  407. fake.EstablishedAtStub = nil
  408. if fake.establishedAtReturnsOnCall == nil {
  409. fake.establishedAtReturnsOnCall = make(map[int]struct {
  410. result1 time.Time
  411. })
  412. }
  413. fake.establishedAtReturnsOnCall[i] = struct {
  414. result1 time.Time
  415. }{result1}
  416. }
  417. func (fake *Connection) ID() protocol.DeviceID {
  418. fake.iDMutex.Lock()
  419. ret, specificReturn := fake.iDReturnsOnCall[len(fake.iDArgsForCall)]
  420. fake.iDArgsForCall = append(fake.iDArgsForCall, struct {
  421. }{})
  422. stub := fake.IDStub
  423. fakeReturns := fake.iDReturns
  424. fake.recordInvocation("ID", []interface{}{})
  425. fake.iDMutex.Unlock()
  426. if stub != nil {
  427. return stub()
  428. }
  429. if specificReturn {
  430. return ret.result1
  431. }
  432. return fakeReturns.result1
  433. }
  434. func (fake *Connection) IDCallCount() int {
  435. fake.iDMutex.RLock()
  436. defer fake.iDMutex.RUnlock()
  437. return len(fake.iDArgsForCall)
  438. }
  439. func (fake *Connection) IDCalls(stub func() protocol.DeviceID) {
  440. fake.iDMutex.Lock()
  441. defer fake.iDMutex.Unlock()
  442. fake.IDStub = stub
  443. }
  444. func (fake *Connection) IDReturns(result1 protocol.DeviceID) {
  445. fake.iDMutex.Lock()
  446. defer fake.iDMutex.Unlock()
  447. fake.IDStub = nil
  448. fake.iDReturns = struct {
  449. result1 protocol.DeviceID
  450. }{result1}
  451. }
  452. func (fake *Connection) IDReturnsOnCall(i int, result1 protocol.DeviceID) {
  453. fake.iDMutex.Lock()
  454. defer fake.iDMutex.Unlock()
  455. fake.IDStub = nil
  456. if fake.iDReturnsOnCall == nil {
  457. fake.iDReturnsOnCall = make(map[int]struct {
  458. result1 protocol.DeviceID
  459. })
  460. }
  461. fake.iDReturnsOnCall[i] = struct {
  462. result1 protocol.DeviceID
  463. }{result1}
  464. }
  465. func (fake *Connection) Index(arg1 context.Context, arg2 string, arg3 []protocol.FileInfo) error {
  466. var arg3Copy []protocol.FileInfo
  467. if arg3 != nil {
  468. arg3Copy = make([]protocol.FileInfo, len(arg3))
  469. copy(arg3Copy, arg3)
  470. }
  471. fake.indexMutex.Lock()
  472. ret, specificReturn := fake.indexReturnsOnCall[len(fake.indexArgsForCall)]
  473. fake.indexArgsForCall = append(fake.indexArgsForCall, struct {
  474. arg1 context.Context
  475. arg2 string
  476. arg3 []protocol.FileInfo
  477. }{arg1, arg2, arg3Copy})
  478. stub := fake.IndexStub
  479. fakeReturns := fake.indexReturns
  480. fake.recordInvocation("Index", []interface{}{arg1, arg2, arg3Copy})
  481. fake.indexMutex.Unlock()
  482. if stub != nil {
  483. return stub(arg1, arg2, arg3)
  484. }
  485. if specificReturn {
  486. return ret.result1
  487. }
  488. return fakeReturns.result1
  489. }
  490. func (fake *Connection) IndexCallCount() int {
  491. fake.indexMutex.RLock()
  492. defer fake.indexMutex.RUnlock()
  493. return len(fake.indexArgsForCall)
  494. }
  495. func (fake *Connection) IndexCalls(stub func(context.Context, string, []protocol.FileInfo) error) {
  496. fake.indexMutex.Lock()
  497. defer fake.indexMutex.Unlock()
  498. fake.IndexStub = stub
  499. }
  500. func (fake *Connection) IndexArgsForCall(i int) (context.Context, string, []protocol.FileInfo) {
  501. fake.indexMutex.RLock()
  502. defer fake.indexMutex.RUnlock()
  503. argsForCall := fake.indexArgsForCall[i]
  504. return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
  505. }
  506. func (fake *Connection) IndexReturns(result1 error) {
  507. fake.indexMutex.Lock()
  508. defer fake.indexMutex.Unlock()
  509. fake.IndexStub = nil
  510. fake.indexReturns = struct {
  511. result1 error
  512. }{result1}
  513. }
  514. func (fake *Connection) IndexReturnsOnCall(i int, result1 error) {
  515. fake.indexMutex.Lock()
  516. defer fake.indexMutex.Unlock()
  517. fake.IndexStub = nil
  518. if fake.indexReturnsOnCall == nil {
  519. fake.indexReturnsOnCall = make(map[int]struct {
  520. result1 error
  521. })
  522. }
  523. fake.indexReturnsOnCall[i] = struct {
  524. result1 error
  525. }{result1}
  526. }
  527. func (fake *Connection) IndexUpdate(arg1 context.Context, arg2 string, arg3 []protocol.FileInfo) error {
  528. var arg3Copy []protocol.FileInfo
  529. if arg3 != nil {
  530. arg3Copy = make([]protocol.FileInfo, len(arg3))
  531. copy(arg3Copy, arg3)
  532. }
  533. fake.indexUpdateMutex.Lock()
  534. ret, specificReturn := fake.indexUpdateReturnsOnCall[len(fake.indexUpdateArgsForCall)]
  535. fake.indexUpdateArgsForCall = append(fake.indexUpdateArgsForCall, struct {
  536. arg1 context.Context
  537. arg2 string
  538. arg3 []protocol.FileInfo
  539. }{arg1, arg2, arg3Copy})
  540. stub := fake.IndexUpdateStub
  541. fakeReturns := fake.indexUpdateReturns
  542. fake.recordInvocation("IndexUpdate", []interface{}{arg1, arg2, arg3Copy})
  543. fake.indexUpdateMutex.Unlock()
  544. if stub != nil {
  545. return stub(arg1, arg2, arg3)
  546. }
  547. if specificReturn {
  548. return ret.result1
  549. }
  550. return fakeReturns.result1
  551. }
  552. func (fake *Connection) IndexUpdateCallCount() int {
  553. fake.indexUpdateMutex.RLock()
  554. defer fake.indexUpdateMutex.RUnlock()
  555. return len(fake.indexUpdateArgsForCall)
  556. }
  557. func (fake *Connection) IndexUpdateCalls(stub func(context.Context, string, []protocol.FileInfo) error) {
  558. fake.indexUpdateMutex.Lock()
  559. defer fake.indexUpdateMutex.Unlock()
  560. fake.IndexUpdateStub = stub
  561. }
  562. func (fake *Connection) IndexUpdateArgsForCall(i int) (context.Context, string, []protocol.FileInfo) {
  563. fake.indexUpdateMutex.RLock()
  564. defer fake.indexUpdateMutex.RUnlock()
  565. argsForCall := fake.indexUpdateArgsForCall[i]
  566. return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
  567. }
  568. func (fake *Connection) IndexUpdateReturns(result1 error) {
  569. fake.indexUpdateMutex.Lock()
  570. defer fake.indexUpdateMutex.Unlock()
  571. fake.IndexUpdateStub = nil
  572. fake.indexUpdateReturns = struct {
  573. result1 error
  574. }{result1}
  575. }
  576. func (fake *Connection) IndexUpdateReturnsOnCall(i int, result1 error) {
  577. fake.indexUpdateMutex.Lock()
  578. defer fake.indexUpdateMutex.Unlock()
  579. fake.IndexUpdateStub = nil
  580. if fake.indexUpdateReturnsOnCall == nil {
  581. fake.indexUpdateReturnsOnCall = make(map[int]struct {
  582. result1 error
  583. })
  584. }
  585. fake.indexUpdateReturnsOnCall[i] = struct {
  586. result1 error
  587. }{result1}
  588. }
  589. func (fake *Connection) Priority() int {
  590. fake.priorityMutex.Lock()
  591. ret, specificReturn := fake.priorityReturnsOnCall[len(fake.priorityArgsForCall)]
  592. fake.priorityArgsForCall = append(fake.priorityArgsForCall, struct {
  593. }{})
  594. stub := fake.PriorityStub
  595. fakeReturns := fake.priorityReturns
  596. fake.recordInvocation("Priority", []interface{}{})
  597. fake.priorityMutex.Unlock()
  598. if stub != nil {
  599. return stub()
  600. }
  601. if specificReturn {
  602. return ret.result1
  603. }
  604. return fakeReturns.result1
  605. }
  606. func (fake *Connection) PriorityCallCount() int {
  607. fake.priorityMutex.RLock()
  608. defer fake.priorityMutex.RUnlock()
  609. return len(fake.priorityArgsForCall)
  610. }
  611. func (fake *Connection) PriorityCalls(stub func() int) {
  612. fake.priorityMutex.Lock()
  613. defer fake.priorityMutex.Unlock()
  614. fake.PriorityStub = stub
  615. }
  616. func (fake *Connection) PriorityReturns(result1 int) {
  617. fake.priorityMutex.Lock()
  618. defer fake.priorityMutex.Unlock()
  619. fake.PriorityStub = nil
  620. fake.priorityReturns = struct {
  621. result1 int
  622. }{result1}
  623. }
  624. func (fake *Connection) PriorityReturnsOnCall(i int, result1 int) {
  625. fake.priorityMutex.Lock()
  626. defer fake.priorityMutex.Unlock()
  627. fake.PriorityStub = nil
  628. if fake.priorityReturnsOnCall == nil {
  629. fake.priorityReturnsOnCall = make(map[int]struct {
  630. result1 int
  631. })
  632. }
  633. fake.priorityReturnsOnCall[i] = struct {
  634. result1 int
  635. }{result1}
  636. }
  637. func (fake *Connection) RemoteAddr() net.Addr {
  638. fake.remoteAddrMutex.Lock()
  639. ret, specificReturn := fake.remoteAddrReturnsOnCall[len(fake.remoteAddrArgsForCall)]
  640. fake.remoteAddrArgsForCall = append(fake.remoteAddrArgsForCall, struct {
  641. }{})
  642. stub := fake.RemoteAddrStub
  643. fakeReturns := fake.remoteAddrReturns
  644. fake.recordInvocation("RemoteAddr", []interface{}{})
  645. fake.remoteAddrMutex.Unlock()
  646. if stub != nil {
  647. return stub()
  648. }
  649. if specificReturn {
  650. return ret.result1
  651. }
  652. return fakeReturns.result1
  653. }
  654. func (fake *Connection) RemoteAddrCallCount() int {
  655. fake.remoteAddrMutex.RLock()
  656. defer fake.remoteAddrMutex.RUnlock()
  657. return len(fake.remoteAddrArgsForCall)
  658. }
  659. func (fake *Connection) RemoteAddrCalls(stub func() net.Addr) {
  660. fake.remoteAddrMutex.Lock()
  661. defer fake.remoteAddrMutex.Unlock()
  662. fake.RemoteAddrStub = stub
  663. }
  664. func (fake *Connection) RemoteAddrReturns(result1 net.Addr) {
  665. fake.remoteAddrMutex.Lock()
  666. defer fake.remoteAddrMutex.Unlock()
  667. fake.RemoteAddrStub = nil
  668. fake.remoteAddrReturns = struct {
  669. result1 net.Addr
  670. }{result1}
  671. }
  672. func (fake *Connection) RemoteAddrReturnsOnCall(i int, result1 net.Addr) {
  673. fake.remoteAddrMutex.Lock()
  674. defer fake.remoteAddrMutex.Unlock()
  675. fake.RemoteAddrStub = nil
  676. if fake.remoteAddrReturnsOnCall == nil {
  677. fake.remoteAddrReturnsOnCall = make(map[int]struct {
  678. result1 net.Addr
  679. })
  680. }
  681. fake.remoteAddrReturnsOnCall[i] = struct {
  682. result1 net.Addr
  683. }{result1}
  684. }
  685. func (fake *Connection) Request(arg1 context.Context, arg2 string, arg3 string, arg4 int, arg5 int64, arg6 int, arg7 []byte, arg8 uint32, arg9 bool) ([]byte, error) {
  686. var arg7Copy []byte
  687. if arg7 != nil {
  688. arg7Copy = make([]byte, len(arg7))
  689. copy(arg7Copy, arg7)
  690. }
  691. fake.requestMutex.Lock()
  692. ret, specificReturn := fake.requestReturnsOnCall[len(fake.requestArgsForCall)]
  693. fake.requestArgsForCall = append(fake.requestArgsForCall, struct {
  694. arg1 context.Context
  695. arg2 string
  696. arg3 string
  697. arg4 int
  698. arg5 int64
  699. arg6 int
  700. arg7 []byte
  701. arg8 uint32
  702. arg9 bool
  703. }{arg1, arg2, arg3, arg4, arg5, arg6, arg7Copy, arg8, arg9})
  704. stub := fake.RequestStub
  705. fakeReturns := fake.requestReturns
  706. fake.recordInvocation("Request", []interface{}{arg1, arg2, arg3, arg4, arg5, arg6, arg7Copy, arg8, arg9})
  707. fake.requestMutex.Unlock()
  708. if stub != nil {
  709. return stub(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9)
  710. }
  711. if specificReturn {
  712. return ret.result1, ret.result2
  713. }
  714. return fakeReturns.result1, fakeReturns.result2
  715. }
  716. func (fake *Connection) RequestCallCount() int {
  717. fake.requestMutex.RLock()
  718. defer fake.requestMutex.RUnlock()
  719. return len(fake.requestArgsForCall)
  720. }
  721. func (fake *Connection) RequestCalls(stub func(context.Context, string, string, int, int64, int, []byte, uint32, bool) ([]byte, error)) {
  722. fake.requestMutex.Lock()
  723. defer fake.requestMutex.Unlock()
  724. fake.RequestStub = stub
  725. }
  726. func (fake *Connection) RequestArgsForCall(i int) (context.Context, string, string, int, int64, int, []byte, uint32, bool) {
  727. fake.requestMutex.RLock()
  728. defer fake.requestMutex.RUnlock()
  729. argsForCall := fake.requestArgsForCall[i]
  730. return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4, argsForCall.arg5, argsForCall.arg6, argsForCall.arg7, argsForCall.arg8, argsForCall.arg9
  731. }
  732. func (fake *Connection) RequestReturns(result1 []byte, result2 error) {
  733. fake.requestMutex.Lock()
  734. defer fake.requestMutex.Unlock()
  735. fake.RequestStub = nil
  736. fake.requestReturns = struct {
  737. result1 []byte
  738. result2 error
  739. }{result1, result2}
  740. }
  741. func (fake *Connection) RequestReturnsOnCall(i int, result1 []byte, result2 error) {
  742. fake.requestMutex.Lock()
  743. defer fake.requestMutex.Unlock()
  744. fake.RequestStub = nil
  745. if fake.requestReturnsOnCall == nil {
  746. fake.requestReturnsOnCall = make(map[int]struct {
  747. result1 []byte
  748. result2 error
  749. })
  750. }
  751. fake.requestReturnsOnCall[i] = struct {
  752. result1 []byte
  753. result2 error
  754. }{result1, result2}
  755. }
  756. func (fake *Connection) Start() {
  757. fake.startMutex.Lock()
  758. fake.startArgsForCall = append(fake.startArgsForCall, struct {
  759. }{})
  760. stub := fake.StartStub
  761. fake.recordInvocation("Start", []interface{}{})
  762. fake.startMutex.Unlock()
  763. if stub != nil {
  764. fake.StartStub()
  765. }
  766. }
  767. func (fake *Connection) StartCallCount() int {
  768. fake.startMutex.RLock()
  769. defer fake.startMutex.RUnlock()
  770. return len(fake.startArgsForCall)
  771. }
  772. func (fake *Connection) StartCalls(stub func()) {
  773. fake.startMutex.Lock()
  774. defer fake.startMutex.Unlock()
  775. fake.StartStub = stub
  776. }
  777. func (fake *Connection) Statistics() protocol.Statistics {
  778. fake.statisticsMutex.Lock()
  779. ret, specificReturn := fake.statisticsReturnsOnCall[len(fake.statisticsArgsForCall)]
  780. fake.statisticsArgsForCall = append(fake.statisticsArgsForCall, struct {
  781. }{})
  782. stub := fake.StatisticsStub
  783. fakeReturns := fake.statisticsReturns
  784. fake.recordInvocation("Statistics", []interface{}{})
  785. fake.statisticsMutex.Unlock()
  786. if stub != nil {
  787. return stub()
  788. }
  789. if specificReturn {
  790. return ret.result1
  791. }
  792. return fakeReturns.result1
  793. }
  794. func (fake *Connection) StatisticsCallCount() int {
  795. fake.statisticsMutex.RLock()
  796. defer fake.statisticsMutex.RUnlock()
  797. return len(fake.statisticsArgsForCall)
  798. }
  799. func (fake *Connection) StatisticsCalls(stub func() protocol.Statistics) {
  800. fake.statisticsMutex.Lock()
  801. defer fake.statisticsMutex.Unlock()
  802. fake.StatisticsStub = stub
  803. }
  804. func (fake *Connection) StatisticsReturns(result1 protocol.Statistics) {
  805. fake.statisticsMutex.Lock()
  806. defer fake.statisticsMutex.Unlock()
  807. fake.StatisticsStub = nil
  808. fake.statisticsReturns = struct {
  809. result1 protocol.Statistics
  810. }{result1}
  811. }
  812. func (fake *Connection) StatisticsReturnsOnCall(i int, result1 protocol.Statistics) {
  813. fake.statisticsMutex.Lock()
  814. defer fake.statisticsMutex.Unlock()
  815. fake.StatisticsStub = nil
  816. if fake.statisticsReturnsOnCall == nil {
  817. fake.statisticsReturnsOnCall = make(map[int]struct {
  818. result1 protocol.Statistics
  819. })
  820. }
  821. fake.statisticsReturnsOnCall[i] = struct {
  822. result1 protocol.Statistics
  823. }{result1}
  824. }
  825. func (fake *Connection) String() string {
  826. fake.stringMutex.Lock()
  827. ret, specificReturn := fake.stringReturnsOnCall[len(fake.stringArgsForCall)]
  828. fake.stringArgsForCall = append(fake.stringArgsForCall, struct {
  829. }{})
  830. stub := fake.StringStub
  831. fakeReturns := fake.stringReturns
  832. fake.recordInvocation("String", []interface{}{})
  833. fake.stringMutex.Unlock()
  834. if stub != nil {
  835. return stub()
  836. }
  837. if specificReturn {
  838. return ret.result1
  839. }
  840. return fakeReturns.result1
  841. }
  842. func (fake *Connection) StringCallCount() int {
  843. fake.stringMutex.RLock()
  844. defer fake.stringMutex.RUnlock()
  845. return len(fake.stringArgsForCall)
  846. }
  847. func (fake *Connection) StringCalls(stub func() string) {
  848. fake.stringMutex.Lock()
  849. defer fake.stringMutex.Unlock()
  850. fake.StringStub = stub
  851. }
  852. func (fake *Connection) StringReturns(result1 string) {
  853. fake.stringMutex.Lock()
  854. defer fake.stringMutex.Unlock()
  855. fake.StringStub = nil
  856. fake.stringReturns = struct {
  857. result1 string
  858. }{result1}
  859. }
  860. func (fake *Connection) StringReturnsOnCall(i int, result1 string) {
  861. fake.stringMutex.Lock()
  862. defer fake.stringMutex.Unlock()
  863. fake.StringStub = nil
  864. if fake.stringReturnsOnCall == nil {
  865. fake.stringReturnsOnCall = make(map[int]struct {
  866. result1 string
  867. })
  868. }
  869. fake.stringReturnsOnCall[i] = struct {
  870. result1 string
  871. }{result1}
  872. }
  873. func (fake *Connection) Transport() string {
  874. fake.transportMutex.Lock()
  875. ret, specificReturn := fake.transportReturnsOnCall[len(fake.transportArgsForCall)]
  876. fake.transportArgsForCall = append(fake.transportArgsForCall, struct {
  877. }{})
  878. stub := fake.TransportStub
  879. fakeReturns := fake.transportReturns
  880. fake.recordInvocation("Transport", []interface{}{})
  881. fake.transportMutex.Unlock()
  882. if stub != nil {
  883. return stub()
  884. }
  885. if specificReturn {
  886. return ret.result1
  887. }
  888. return fakeReturns.result1
  889. }
  890. func (fake *Connection) TransportCallCount() int {
  891. fake.transportMutex.RLock()
  892. defer fake.transportMutex.RUnlock()
  893. return len(fake.transportArgsForCall)
  894. }
  895. func (fake *Connection) TransportCalls(stub func() string) {
  896. fake.transportMutex.Lock()
  897. defer fake.transportMutex.Unlock()
  898. fake.TransportStub = stub
  899. }
  900. func (fake *Connection) TransportReturns(result1 string) {
  901. fake.transportMutex.Lock()
  902. defer fake.transportMutex.Unlock()
  903. fake.TransportStub = nil
  904. fake.transportReturns = struct {
  905. result1 string
  906. }{result1}
  907. }
  908. func (fake *Connection) TransportReturnsOnCall(i int, result1 string) {
  909. fake.transportMutex.Lock()
  910. defer fake.transportMutex.Unlock()
  911. fake.TransportStub = nil
  912. if fake.transportReturnsOnCall == nil {
  913. fake.transportReturnsOnCall = make(map[int]struct {
  914. result1 string
  915. })
  916. }
  917. fake.transportReturnsOnCall[i] = struct {
  918. result1 string
  919. }{result1}
  920. }
  921. func (fake *Connection) Type() string {
  922. fake.typeMutex.Lock()
  923. ret, specificReturn := fake.typeReturnsOnCall[len(fake.typeArgsForCall)]
  924. fake.typeArgsForCall = append(fake.typeArgsForCall, struct {
  925. }{})
  926. stub := fake.TypeStub
  927. fakeReturns := fake.typeReturns
  928. fake.recordInvocation("Type", []interface{}{})
  929. fake.typeMutex.Unlock()
  930. if stub != nil {
  931. return stub()
  932. }
  933. if specificReturn {
  934. return ret.result1
  935. }
  936. return fakeReturns.result1
  937. }
  938. func (fake *Connection) TypeCallCount() int {
  939. fake.typeMutex.RLock()
  940. defer fake.typeMutex.RUnlock()
  941. return len(fake.typeArgsForCall)
  942. }
  943. func (fake *Connection) TypeCalls(stub func() string) {
  944. fake.typeMutex.Lock()
  945. defer fake.typeMutex.Unlock()
  946. fake.TypeStub = stub
  947. }
  948. func (fake *Connection) TypeReturns(result1 string) {
  949. fake.typeMutex.Lock()
  950. defer fake.typeMutex.Unlock()
  951. fake.TypeStub = nil
  952. fake.typeReturns = struct {
  953. result1 string
  954. }{result1}
  955. }
  956. func (fake *Connection) TypeReturnsOnCall(i int, result1 string) {
  957. fake.typeMutex.Lock()
  958. defer fake.typeMutex.Unlock()
  959. fake.TypeStub = nil
  960. if fake.typeReturnsOnCall == nil {
  961. fake.typeReturnsOnCall = make(map[int]struct {
  962. result1 string
  963. })
  964. }
  965. fake.typeReturnsOnCall[i] = struct {
  966. result1 string
  967. }{result1}
  968. }
  969. func (fake *Connection) Invocations() map[string][][]interface{} {
  970. fake.invocationsMutex.RLock()
  971. defer fake.invocationsMutex.RUnlock()
  972. fake.closeMutex.RLock()
  973. defer fake.closeMutex.RUnlock()
  974. fake.closedMutex.RLock()
  975. defer fake.closedMutex.RUnlock()
  976. fake.clusterConfigMutex.RLock()
  977. defer fake.clusterConfigMutex.RUnlock()
  978. fake.cryptoMutex.RLock()
  979. defer fake.cryptoMutex.RUnlock()
  980. fake.downloadProgressMutex.RLock()
  981. defer fake.downloadProgressMutex.RUnlock()
  982. fake.establishedAtMutex.RLock()
  983. defer fake.establishedAtMutex.RUnlock()
  984. fake.iDMutex.RLock()
  985. defer fake.iDMutex.RUnlock()
  986. fake.indexMutex.RLock()
  987. defer fake.indexMutex.RUnlock()
  988. fake.indexUpdateMutex.RLock()
  989. defer fake.indexUpdateMutex.RUnlock()
  990. fake.priorityMutex.RLock()
  991. defer fake.priorityMutex.RUnlock()
  992. fake.remoteAddrMutex.RLock()
  993. defer fake.remoteAddrMutex.RUnlock()
  994. fake.requestMutex.RLock()
  995. defer fake.requestMutex.RUnlock()
  996. fake.startMutex.RLock()
  997. defer fake.startMutex.RUnlock()
  998. fake.statisticsMutex.RLock()
  999. defer fake.statisticsMutex.RUnlock()
  1000. fake.stringMutex.RLock()
  1001. defer fake.stringMutex.RUnlock()
  1002. fake.transportMutex.RLock()
  1003. defer fake.transportMutex.RUnlock()
  1004. fake.typeMutex.RLock()
  1005. defer fake.typeMutex.RUnlock()
  1006. copiedInvocations := map[string][][]interface{}{}
  1007. for key, value := range fake.invocations {
  1008. copiedInvocations[key] = value
  1009. }
  1010. return copiedInvocations
  1011. }
  1012. func (fake *Connection) recordInvocation(key string, args []interface{}) {
  1013. fake.invocationsMutex.Lock()
  1014. defer fake.invocationsMutex.Unlock()
  1015. if fake.invocations == nil {
  1016. fake.invocations = map[string][][]interface{}{}
  1017. }
  1018. if fake.invocations[key] == nil {
  1019. fake.invocations[key] = [][]interface{}{}
  1020. }
  1021. fake.invocations[key] = append(fake.invocations[key], args)
  1022. }
  1023. var _ protocol.Connection = new(Connection)