github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/db/dbfakes/fake_tx.go (about) 1 // Code generated by counterfeiter. DO NOT EDIT. 2 package dbfakes 3 4 import ( 5 "context" 6 "database/sql" 7 "sync" 8 9 "github.com/Masterminds/squirrel" 10 "github.com/pf-qiu/concourse/v6/atc/db" 11 "github.com/pf-qiu/concourse/v6/atc/db/encryption" 12 ) 13 14 type FakeTx struct { 15 CommitStub func() error 16 commitMutex sync.RWMutex 17 commitArgsForCall []struct { 18 } 19 commitReturns struct { 20 result1 error 21 } 22 commitReturnsOnCall map[int]struct { 23 result1 error 24 } 25 EncryptionStrategyStub func() encryption.Strategy 26 encryptionStrategyMutex sync.RWMutex 27 encryptionStrategyArgsForCall []struct { 28 } 29 encryptionStrategyReturns struct { 30 result1 encryption.Strategy 31 } 32 encryptionStrategyReturnsOnCall map[int]struct { 33 result1 encryption.Strategy 34 } 35 ExecStub func(string, ...interface{}) (sql.Result, error) 36 execMutex sync.RWMutex 37 execArgsForCall []struct { 38 arg1 string 39 arg2 []interface{} 40 } 41 execReturns struct { 42 result1 sql.Result 43 result2 error 44 } 45 execReturnsOnCall map[int]struct { 46 result1 sql.Result 47 result2 error 48 } 49 ExecContextStub func(context.Context, string, ...interface{}) (sql.Result, error) 50 execContextMutex sync.RWMutex 51 execContextArgsForCall []struct { 52 arg1 context.Context 53 arg2 string 54 arg3 []interface{} 55 } 56 execContextReturns struct { 57 result1 sql.Result 58 result2 error 59 } 60 execContextReturnsOnCall map[int]struct { 61 result1 sql.Result 62 result2 error 63 } 64 PrepareStub func(string) (*sql.Stmt, error) 65 prepareMutex sync.RWMutex 66 prepareArgsForCall []struct { 67 arg1 string 68 } 69 prepareReturns struct { 70 result1 *sql.Stmt 71 result2 error 72 } 73 prepareReturnsOnCall map[int]struct { 74 result1 *sql.Stmt 75 result2 error 76 } 77 PrepareContextStub func(context.Context, string) (*sql.Stmt, error) 78 prepareContextMutex sync.RWMutex 79 prepareContextArgsForCall []struct { 80 arg1 context.Context 81 arg2 string 82 } 83 prepareContextReturns struct { 84 result1 *sql.Stmt 85 result2 error 86 } 87 prepareContextReturnsOnCall map[int]struct { 88 result1 *sql.Stmt 89 result2 error 90 } 91 QueryStub func(string, ...interface{}) (*sql.Rows, error) 92 queryMutex sync.RWMutex 93 queryArgsForCall []struct { 94 arg1 string 95 arg2 []interface{} 96 } 97 queryReturns struct { 98 result1 *sql.Rows 99 result2 error 100 } 101 queryReturnsOnCall map[int]struct { 102 result1 *sql.Rows 103 result2 error 104 } 105 QueryContextStub func(context.Context, string, ...interface{}) (*sql.Rows, error) 106 queryContextMutex sync.RWMutex 107 queryContextArgsForCall []struct { 108 arg1 context.Context 109 arg2 string 110 arg3 []interface{} 111 } 112 queryContextReturns struct { 113 result1 *sql.Rows 114 result2 error 115 } 116 queryContextReturnsOnCall map[int]struct { 117 result1 *sql.Rows 118 result2 error 119 } 120 QueryRowStub func(string, ...interface{}) squirrel.RowScanner 121 queryRowMutex sync.RWMutex 122 queryRowArgsForCall []struct { 123 arg1 string 124 arg2 []interface{} 125 } 126 queryRowReturns struct { 127 result1 squirrel.RowScanner 128 } 129 queryRowReturnsOnCall map[int]struct { 130 result1 squirrel.RowScanner 131 } 132 QueryRowContextStub func(context.Context, string, ...interface{}) squirrel.RowScanner 133 queryRowContextMutex sync.RWMutex 134 queryRowContextArgsForCall []struct { 135 arg1 context.Context 136 arg2 string 137 arg3 []interface{} 138 } 139 queryRowContextReturns struct { 140 result1 squirrel.RowScanner 141 } 142 queryRowContextReturnsOnCall map[int]struct { 143 result1 squirrel.RowScanner 144 } 145 RollbackStub func() error 146 rollbackMutex sync.RWMutex 147 rollbackArgsForCall []struct { 148 } 149 rollbackReturns struct { 150 result1 error 151 } 152 rollbackReturnsOnCall map[int]struct { 153 result1 error 154 } 155 StmtStub func(*sql.Stmt) *sql.Stmt 156 stmtMutex sync.RWMutex 157 stmtArgsForCall []struct { 158 arg1 *sql.Stmt 159 } 160 stmtReturns struct { 161 result1 *sql.Stmt 162 } 163 stmtReturnsOnCall map[int]struct { 164 result1 *sql.Stmt 165 } 166 invocations map[string][][]interface{} 167 invocationsMutex sync.RWMutex 168 } 169 170 func (fake *FakeTx) Commit() error { 171 fake.commitMutex.Lock() 172 ret, specificReturn := fake.commitReturnsOnCall[len(fake.commitArgsForCall)] 173 fake.commitArgsForCall = append(fake.commitArgsForCall, struct { 174 }{}) 175 fake.recordInvocation("Commit", []interface{}{}) 176 fake.commitMutex.Unlock() 177 if fake.CommitStub != nil { 178 return fake.CommitStub() 179 } 180 if specificReturn { 181 return ret.result1 182 } 183 fakeReturns := fake.commitReturns 184 return fakeReturns.result1 185 } 186 187 func (fake *FakeTx) CommitCallCount() int { 188 fake.commitMutex.RLock() 189 defer fake.commitMutex.RUnlock() 190 return len(fake.commitArgsForCall) 191 } 192 193 func (fake *FakeTx) CommitCalls(stub func() error) { 194 fake.commitMutex.Lock() 195 defer fake.commitMutex.Unlock() 196 fake.CommitStub = stub 197 } 198 199 func (fake *FakeTx) CommitReturns(result1 error) { 200 fake.commitMutex.Lock() 201 defer fake.commitMutex.Unlock() 202 fake.CommitStub = nil 203 fake.commitReturns = struct { 204 result1 error 205 }{result1} 206 } 207 208 func (fake *FakeTx) CommitReturnsOnCall(i int, result1 error) { 209 fake.commitMutex.Lock() 210 defer fake.commitMutex.Unlock() 211 fake.CommitStub = nil 212 if fake.commitReturnsOnCall == nil { 213 fake.commitReturnsOnCall = make(map[int]struct { 214 result1 error 215 }) 216 } 217 fake.commitReturnsOnCall[i] = struct { 218 result1 error 219 }{result1} 220 } 221 222 func (fake *FakeTx) EncryptionStrategy() encryption.Strategy { 223 fake.encryptionStrategyMutex.Lock() 224 ret, specificReturn := fake.encryptionStrategyReturnsOnCall[len(fake.encryptionStrategyArgsForCall)] 225 fake.encryptionStrategyArgsForCall = append(fake.encryptionStrategyArgsForCall, struct { 226 }{}) 227 fake.recordInvocation("EncryptionStrategy", []interface{}{}) 228 fake.encryptionStrategyMutex.Unlock() 229 if fake.EncryptionStrategyStub != nil { 230 return fake.EncryptionStrategyStub() 231 } 232 if specificReturn { 233 return ret.result1 234 } 235 fakeReturns := fake.encryptionStrategyReturns 236 return fakeReturns.result1 237 } 238 239 func (fake *FakeTx) EncryptionStrategyCallCount() int { 240 fake.encryptionStrategyMutex.RLock() 241 defer fake.encryptionStrategyMutex.RUnlock() 242 return len(fake.encryptionStrategyArgsForCall) 243 } 244 245 func (fake *FakeTx) EncryptionStrategyCalls(stub func() encryption.Strategy) { 246 fake.encryptionStrategyMutex.Lock() 247 defer fake.encryptionStrategyMutex.Unlock() 248 fake.EncryptionStrategyStub = stub 249 } 250 251 func (fake *FakeTx) EncryptionStrategyReturns(result1 encryption.Strategy) { 252 fake.encryptionStrategyMutex.Lock() 253 defer fake.encryptionStrategyMutex.Unlock() 254 fake.EncryptionStrategyStub = nil 255 fake.encryptionStrategyReturns = struct { 256 result1 encryption.Strategy 257 }{result1} 258 } 259 260 func (fake *FakeTx) EncryptionStrategyReturnsOnCall(i int, result1 encryption.Strategy) { 261 fake.encryptionStrategyMutex.Lock() 262 defer fake.encryptionStrategyMutex.Unlock() 263 fake.EncryptionStrategyStub = nil 264 if fake.encryptionStrategyReturnsOnCall == nil { 265 fake.encryptionStrategyReturnsOnCall = make(map[int]struct { 266 result1 encryption.Strategy 267 }) 268 } 269 fake.encryptionStrategyReturnsOnCall[i] = struct { 270 result1 encryption.Strategy 271 }{result1} 272 } 273 274 func (fake *FakeTx) Exec(arg1 string, arg2 ...interface{}) (sql.Result, error) { 275 fake.execMutex.Lock() 276 ret, specificReturn := fake.execReturnsOnCall[len(fake.execArgsForCall)] 277 fake.execArgsForCall = append(fake.execArgsForCall, struct { 278 arg1 string 279 arg2 []interface{} 280 }{arg1, arg2}) 281 fake.recordInvocation("Exec", []interface{}{arg1, arg2}) 282 fake.execMutex.Unlock() 283 if fake.ExecStub != nil { 284 return fake.ExecStub(arg1, arg2...) 285 } 286 if specificReturn { 287 return ret.result1, ret.result2 288 } 289 fakeReturns := fake.execReturns 290 return fakeReturns.result1, fakeReturns.result2 291 } 292 293 func (fake *FakeTx) ExecCallCount() int { 294 fake.execMutex.RLock() 295 defer fake.execMutex.RUnlock() 296 return len(fake.execArgsForCall) 297 } 298 299 func (fake *FakeTx) ExecCalls(stub func(string, ...interface{}) (sql.Result, error)) { 300 fake.execMutex.Lock() 301 defer fake.execMutex.Unlock() 302 fake.ExecStub = stub 303 } 304 305 func (fake *FakeTx) ExecArgsForCall(i int) (string, []interface{}) { 306 fake.execMutex.RLock() 307 defer fake.execMutex.RUnlock() 308 argsForCall := fake.execArgsForCall[i] 309 return argsForCall.arg1, argsForCall.arg2 310 } 311 312 func (fake *FakeTx) ExecReturns(result1 sql.Result, result2 error) { 313 fake.execMutex.Lock() 314 defer fake.execMutex.Unlock() 315 fake.ExecStub = nil 316 fake.execReturns = struct { 317 result1 sql.Result 318 result2 error 319 }{result1, result2} 320 } 321 322 func (fake *FakeTx) ExecReturnsOnCall(i int, result1 sql.Result, result2 error) { 323 fake.execMutex.Lock() 324 defer fake.execMutex.Unlock() 325 fake.ExecStub = nil 326 if fake.execReturnsOnCall == nil { 327 fake.execReturnsOnCall = make(map[int]struct { 328 result1 sql.Result 329 result2 error 330 }) 331 } 332 fake.execReturnsOnCall[i] = struct { 333 result1 sql.Result 334 result2 error 335 }{result1, result2} 336 } 337 338 func (fake *FakeTx) ExecContext(arg1 context.Context, arg2 string, arg3 ...interface{}) (sql.Result, error) { 339 fake.execContextMutex.Lock() 340 ret, specificReturn := fake.execContextReturnsOnCall[len(fake.execContextArgsForCall)] 341 fake.execContextArgsForCall = append(fake.execContextArgsForCall, struct { 342 arg1 context.Context 343 arg2 string 344 arg3 []interface{} 345 }{arg1, arg2, arg3}) 346 fake.recordInvocation("ExecContext", []interface{}{arg1, arg2, arg3}) 347 fake.execContextMutex.Unlock() 348 if fake.ExecContextStub != nil { 349 return fake.ExecContextStub(arg1, arg2, arg3...) 350 } 351 if specificReturn { 352 return ret.result1, ret.result2 353 } 354 fakeReturns := fake.execContextReturns 355 return fakeReturns.result1, fakeReturns.result2 356 } 357 358 func (fake *FakeTx) ExecContextCallCount() int { 359 fake.execContextMutex.RLock() 360 defer fake.execContextMutex.RUnlock() 361 return len(fake.execContextArgsForCall) 362 } 363 364 func (fake *FakeTx) ExecContextCalls(stub func(context.Context, string, ...interface{}) (sql.Result, error)) { 365 fake.execContextMutex.Lock() 366 defer fake.execContextMutex.Unlock() 367 fake.ExecContextStub = stub 368 } 369 370 func (fake *FakeTx) ExecContextArgsForCall(i int) (context.Context, string, []interface{}) { 371 fake.execContextMutex.RLock() 372 defer fake.execContextMutex.RUnlock() 373 argsForCall := fake.execContextArgsForCall[i] 374 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3 375 } 376 377 func (fake *FakeTx) ExecContextReturns(result1 sql.Result, result2 error) { 378 fake.execContextMutex.Lock() 379 defer fake.execContextMutex.Unlock() 380 fake.ExecContextStub = nil 381 fake.execContextReturns = struct { 382 result1 sql.Result 383 result2 error 384 }{result1, result2} 385 } 386 387 func (fake *FakeTx) ExecContextReturnsOnCall(i int, result1 sql.Result, result2 error) { 388 fake.execContextMutex.Lock() 389 defer fake.execContextMutex.Unlock() 390 fake.ExecContextStub = nil 391 if fake.execContextReturnsOnCall == nil { 392 fake.execContextReturnsOnCall = make(map[int]struct { 393 result1 sql.Result 394 result2 error 395 }) 396 } 397 fake.execContextReturnsOnCall[i] = struct { 398 result1 sql.Result 399 result2 error 400 }{result1, result2} 401 } 402 403 func (fake *FakeTx) Prepare(arg1 string) (*sql.Stmt, error) { 404 fake.prepareMutex.Lock() 405 ret, specificReturn := fake.prepareReturnsOnCall[len(fake.prepareArgsForCall)] 406 fake.prepareArgsForCall = append(fake.prepareArgsForCall, struct { 407 arg1 string 408 }{arg1}) 409 fake.recordInvocation("Prepare", []interface{}{arg1}) 410 fake.prepareMutex.Unlock() 411 if fake.PrepareStub != nil { 412 return fake.PrepareStub(arg1) 413 } 414 if specificReturn { 415 return ret.result1, ret.result2 416 } 417 fakeReturns := fake.prepareReturns 418 return fakeReturns.result1, fakeReturns.result2 419 } 420 421 func (fake *FakeTx) PrepareCallCount() int { 422 fake.prepareMutex.RLock() 423 defer fake.prepareMutex.RUnlock() 424 return len(fake.prepareArgsForCall) 425 } 426 427 func (fake *FakeTx) PrepareCalls(stub func(string) (*sql.Stmt, error)) { 428 fake.prepareMutex.Lock() 429 defer fake.prepareMutex.Unlock() 430 fake.PrepareStub = stub 431 } 432 433 func (fake *FakeTx) PrepareArgsForCall(i int) string { 434 fake.prepareMutex.RLock() 435 defer fake.prepareMutex.RUnlock() 436 argsForCall := fake.prepareArgsForCall[i] 437 return argsForCall.arg1 438 } 439 440 func (fake *FakeTx) PrepareReturns(result1 *sql.Stmt, result2 error) { 441 fake.prepareMutex.Lock() 442 defer fake.prepareMutex.Unlock() 443 fake.PrepareStub = nil 444 fake.prepareReturns = struct { 445 result1 *sql.Stmt 446 result2 error 447 }{result1, result2} 448 } 449 450 func (fake *FakeTx) PrepareReturnsOnCall(i int, result1 *sql.Stmt, result2 error) { 451 fake.prepareMutex.Lock() 452 defer fake.prepareMutex.Unlock() 453 fake.PrepareStub = nil 454 if fake.prepareReturnsOnCall == nil { 455 fake.prepareReturnsOnCall = make(map[int]struct { 456 result1 *sql.Stmt 457 result2 error 458 }) 459 } 460 fake.prepareReturnsOnCall[i] = struct { 461 result1 *sql.Stmt 462 result2 error 463 }{result1, result2} 464 } 465 466 func (fake *FakeTx) PrepareContext(arg1 context.Context, arg2 string) (*sql.Stmt, error) { 467 fake.prepareContextMutex.Lock() 468 ret, specificReturn := fake.prepareContextReturnsOnCall[len(fake.prepareContextArgsForCall)] 469 fake.prepareContextArgsForCall = append(fake.prepareContextArgsForCall, struct { 470 arg1 context.Context 471 arg2 string 472 }{arg1, arg2}) 473 fake.recordInvocation("PrepareContext", []interface{}{arg1, arg2}) 474 fake.prepareContextMutex.Unlock() 475 if fake.PrepareContextStub != nil { 476 return fake.PrepareContextStub(arg1, arg2) 477 } 478 if specificReturn { 479 return ret.result1, ret.result2 480 } 481 fakeReturns := fake.prepareContextReturns 482 return fakeReturns.result1, fakeReturns.result2 483 } 484 485 func (fake *FakeTx) PrepareContextCallCount() int { 486 fake.prepareContextMutex.RLock() 487 defer fake.prepareContextMutex.RUnlock() 488 return len(fake.prepareContextArgsForCall) 489 } 490 491 func (fake *FakeTx) PrepareContextCalls(stub func(context.Context, string) (*sql.Stmt, error)) { 492 fake.prepareContextMutex.Lock() 493 defer fake.prepareContextMutex.Unlock() 494 fake.PrepareContextStub = stub 495 } 496 497 func (fake *FakeTx) PrepareContextArgsForCall(i int) (context.Context, string) { 498 fake.prepareContextMutex.RLock() 499 defer fake.prepareContextMutex.RUnlock() 500 argsForCall := fake.prepareContextArgsForCall[i] 501 return argsForCall.arg1, argsForCall.arg2 502 } 503 504 func (fake *FakeTx) PrepareContextReturns(result1 *sql.Stmt, result2 error) { 505 fake.prepareContextMutex.Lock() 506 defer fake.prepareContextMutex.Unlock() 507 fake.PrepareContextStub = nil 508 fake.prepareContextReturns = struct { 509 result1 *sql.Stmt 510 result2 error 511 }{result1, result2} 512 } 513 514 func (fake *FakeTx) PrepareContextReturnsOnCall(i int, result1 *sql.Stmt, result2 error) { 515 fake.prepareContextMutex.Lock() 516 defer fake.prepareContextMutex.Unlock() 517 fake.PrepareContextStub = nil 518 if fake.prepareContextReturnsOnCall == nil { 519 fake.prepareContextReturnsOnCall = make(map[int]struct { 520 result1 *sql.Stmt 521 result2 error 522 }) 523 } 524 fake.prepareContextReturnsOnCall[i] = struct { 525 result1 *sql.Stmt 526 result2 error 527 }{result1, result2} 528 } 529 530 func (fake *FakeTx) Query(arg1 string, arg2 ...interface{}) (*sql.Rows, error) { 531 fake.queryMutex.Lock() 532 ret, specificReturn := fake.queryReturnsOnCall[len(fake.queryArgsForCall)] 533 fake.queryArgsForCall = append(fake.queryArgsForCall, struct { 534 arg1 string 535 arg2 []interface{} 536 }{arg1, arg2}) 537 fake.recordInvocation("Query", []interface{}{arg1, arg2}) 538 fake.queryMutex.Unlock() 539 if fake.QueryStub != nil { 540 return fake.QueryStub(arg1, arg2...) 541 } 542 if specificReturn { 543 return ret.result1, ret.result2 544 } 545 fakeReturns := fake.queryReturns 546 return fakeReturns.result1, fakeReturns.result2 547 } 548 549 func (fake *FakeTx) QueryCallCount() int { 550 fake.queryMutex.RLock() 551 defer fake.queryMutex.RUnlock() 552 return len(fake.queryArgsForCall) 553 } 554 555 func (fake *FakeTx) QueryCalls(stub func(string, ...interface{}) (*sql.Rows, error)) { 556 fake.queryMutex.Lock() 557 defer fake.queryMutex.Unlock() 558 fake.QueryStub = stub 559 } 560 561 func (fake *FakeTx) QueryArgsForCall(i int) (string, []interface{}) { 562 fake.queryMutex.RLock() 563 defer fake.queryMutex.RUnlock() 564 argsForCall := fake.queryArgsForCall[i] 565 return argsForCall.arg1, argsForCall.arg2 566 } 567 568 func (fake *FakeTx) QueryReturns(result1 *sql.Rows, result2 error) { 569 fake.queryMutex.Lock() 570 defer fake.queryMutex.Unlock() 571 fake.QueryStub = nil 572 fake.queryReturns = struct { 573 result1 *sql.Rows 574 result2 error 575 }{result1, result2} 576 } 577 578 func (fake *FakeTx) QueryReturnsOnCall(i int, result1 *sql.Rows, result2 error) { 579 fake.queryMutex.Lock() 580 defer fake.queryMutex.Unlock() 581 fake.QueryStub = nil 582 if fake.queryReturnsOnCall == nil { 583 fake.queryReturnsOnCall = make(map[int]struct { 584 result1 *sql.Rows 585 result2 error 586 }) 587 } 588 fake.queryReturnsOnCall[i] = struct { 589 result1 *sql.Rows 590 result2 error 591 }{result1, result2} 592 } 593 594 func (fake *FakeTx) QueryContext(arg1 context.Context, arg2 string, arg3 ...interface{}) (*sql.Rows, error) { 595 fake.queryContextMutex.Lock() 596 ret, specificReturn := fake.queryContextReturnsOnCall[len(fake.queryContextArgsForCall)] 597 fake.queryContextArgsForCall = append(fake.queryContextArgsForCall, struct { 598 arg1 context.Context 599 arg2 string 600 arg3 []interface{} 601 }{arg1, arg2, arg3}) 602 fake.recordInvocation("QueryContext", []interface{}{arg1, arg2, arg3}) 603 fake.queryContextMutex.Unlock() 604 if fake.QueryContextStub != nil { 605 return fake.QueryContextStub(arg1, arg2, arg3...) 606 } 607 if specificReturn { 608 return ret.result1, ret.result2 609 } 610 fakeReturns := fake.queryContextReturns 611 return fakeReturns.result1, fakeReturns.result2 612 } 613 614 func (fake *FakeTx) QueryContextCallCount() int { 615 fake.queryContextMutex.RLock() 616 defer fake.queryContextMutex.RUnlock() 617 return len(fake.queryContextArgsForCall) 618 } 619 620 func (fake *FakeTx) QueryContextCalls(stub func(context.Context, string, ...interface{}) (*sql.Rows, error)) { 621 fake.queryContextMutex.Lock() 622 defer fake.queryContextMutex.Unlock() 623 fake.QueryContextStub = stub 624 } 625 626 func (fake *FakeTx) QueryContextArgsForCall(i int) (context.Context, string, []interface{}) { 627 fake.queryContextMutex.RLock() 628 defer fake.queryContextMutex.RUnlock() 629 argsForCall := fake.queryContextArgsForCall[i] 630 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3 631 } 632 633 func (fake *FakeTx) QueryContextReturns(result1 *sql.Rows, result2 error) { 634 fake.queryContextMutex.Lock() 635 defer fake.queryContextMutex.Unlock() 636 fake.QueryContextStub = nil 637 fake.queryContextReturns = struct { 638 result1 *sql.Rows 639 result2 error 640 }{result1, result2} 641 } 642 643 func (fake *FakeTx) QueryContextReturnsOnCall(i int, result1 *sql.Rows, result2 error) { 644 fake.queryContextMutex.Lock() 645 defer fake.queryContextMutex.Unlock() 646 fake.QueryContextStub = nil 647 if fake.queryContextReturnsOnCall == nil { 648 fake.queryContextReturnsOnCall = make(map[int]struct { 649 result1 *sql.Rows 650 result2 error 651 }) 652 } 653 fake.queryContextReturnsOnCall[i] = struct { 654 result1 *sql.Rows 655 result2 error 656 }{result1, result2} 657 } 658 659 func (fake *FakeTx) QueryRow(arg1 string, arg2 ...interface{}) squirrel.RowScanner { 660 fake.queryRowMutex.Lock() 661 ret, specificReturn := fake.queryRowReturnsOnCall[len(fake.queryRowArgsForCall)] 662 fake.queryRowArgsForCall = append(fake.queryRowArgsForCall, struct { 663 arg1 string 664 arg2 []interface{} 665 }{arg1, arg2}) 666 fake.recordInvocation("QueryRow", []interface{}{arg1, arg2}) 667 fake.queryRowMutex.Unlock() 668 if fake.QueryRowStub != nil { 669 return fake.QueryRowStub(arg1, arg2...) 670 } 671 if specificReturn { 672 return ret.result1 673 } 674 fakeReturns := fake.queryRowReturns 675 return fakeReturns.result1 676 } 677 678 func (fake *FakeTx) QueryRowCallCount() int { 679 fake.queryRowMutex.RLock() 680 defer fake.queryRowMutex.RUnlock() 681 return len(fake.queryRowArgsForCall) 682 } 683 684 func (fake *FakeTx) QueryRowCalls(stub func(string, ...interface{}) squirrel.RowScanner) { 685 fake.queryRowMutex.Lock() 686 defer fake.queryRowMutex.Unlock() 687 fake.QueryRowStub = stub 688 } 689 690 func (fake *FakeTx) QueryRowArgsForCall(i int) (string, []interface{}) { 691 fake.queryRowMutex.RLock() 692 defer fake.queryRowMutex.RUnlock() 693 argsForCall := fake.queryRowArgsForCall[i] 694 return argsForCall.arg1, argsForCall.arg2 695 } 696 697 func (fake *FakeTx) QueryRowReturns(result1 squirrel.RowScanner) { 698 fake.queryRowMutex.Lock() 699 defer fake.queryRowMutex.Unlock() 700 fake.QueryRowStub = nil 701 fake.queryRowReturns = struct { 702 result1 squirrel.RowScanner 703 }{result1} 704 } 705 706 func (fake *FakeTx) QueryRowReturnsOnCall(i int, result1 squirrel.RowScanner) { 707 fake.queryRowMutex.Lock() 708 defer fake.queryRowMutex.Unlock() 709 fake.QueryRowStub = nil 710 if fake.queryRowReturnsOnCall == nil { 711 fake.queryRowReturnsOnCall = make(map[int]struct { 712 result1 squirrel.RowScanner 713 }) 714 } 715 fake.queryRowReturnsOnCall[i] = struct { 716 result1 squirrel.RowScanner 717 }{result1} 718 } 719 720 func (fake *FakeTx) QueryRowContext(arg1 context.Context, arg2 string, arg3 ...interface{}) squirrel.RowScanner { 721 fake.queryRowContextMutex.Lock() 722 ret, specificReturn := fake.queryRowContextReturnsOnCall[len(fake.queryRowContextArgsForCall)] 723 fake.queryRowContextArgsForCall = append(fake.queryRowContextArgsForCall, struct { 724 arg1 context.Context 725 arg2 string 726 arg3 []interface{} 727 }{arg1, arg2, arg3}) 728 fake.recordInvocation("QueryRowContext", []interface{}{arg1, arg2, arg3}) 729 fake.queryRowContextMutex.Unlock() 730 if fake.QueryRowContextStub != nil { 731 return fake.QueryRowContextStub(arg1, arg2, arg3...) 732 } 733 if specificReturn { 734 return ret.result1 735 } 736 fakeReturns := fake.queryRowContextReturns 737 return fakeReturns.result1 738 } 739 740 func (fake *FakeTx) QueryRowContextCallCount() int { 741 fake.queryRowContextMutex.RLock() 742 defer fake.queryRowContextMutex.RUnlock() 743 return len(fake.queryRowContextArgsForCall) 744 } 745 746 func (fake *FakeTx) QueryRowContextCalls(stub func(context.Context, string, ...interface{}) squirrel.RowScanner) { 747 fake.queryRowContextMutex.Lock() 748 defer fake.queryRowContextMutex.Unlock() 749 fake.QueryRowContextStub = stub 750 } 751 752 func (fake *FakeTx) QueryRowContextArgsForCall(i int) (context.Context, string, []interface{}) { 753 fake.queryRowContextMutex.RLock() 754 defer fake.queryRowContextMutex.RUnlock() 755 argsForCall := fake.queryRowContextArgsForCall[i] 756 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3 757 } 758 759 func (fake *FakeTx) QueryRowContextReturns(result1 squirrel.RowScanner) { 760 fake.queryRowContextMutex.Lock() 761 defer fake.queryRowContextMutex.Unlock() 762 fake.QueryRowContextStub = nil 763 fake.queryRowContextReturns = struct { 764 result1 squirrel.RowScanner 765 }{result1} 766 } 767 768 func (fake *FakeTx) QueryRowContextReturnsOnCall(i int, result1 squirrel.RowScanner) { 769 fake.queryRowContextMutex.Lock() 770 defer fake.queryRowContextMutex.Unlock() 771 fake.QueryRowContextStub = nil 772 if fake.queryRowContextReturnsOnCall == nil { 773 fake.queryRowContextReturnsOnCall = make(map[int]struct { 774 result1 squirrel.RowScanner 775 }) 776 } 777 fake.queryRowContextReturnsOnCall[i] = struct { 778 result1 squirrel.RowScanner 779 }{result1} 780 } 781 782 func (fake *FakeTx) Rollback() error { 783 fake.rollbackMutex.Lock() 784 ret, specificReturn := fake.rollbackReturnsOnCall[len(fake.rollbackArgsForCall)] 785 fake.rollbackArgsForCall = append(fake.rollbackArgsForCall, struct { 786 }{}) 787 fake.recordInvocation("Rollback", []interface{}{}) 788 fake.rollbackMutex.Unlock() 789 if fake.RollbackStub != nil { 790 return fake.RollbackStub() 791 } 792 if specificReturn { 793 return ret.result1 794 } 795 fakeReturns := fake.rollbackReturns 796 return fakeReturns.result1 797 } 798 799 func (fake *FakeTx) RollbackCallCount() int { 800 fake.rollbackMutex.RLock() 801 defer fake.rollbackMutex.RUnlock() 802 return len(fake.rollbackArgsForCall) 803 } 804 805 func (fake *FakeTx) RollbackCalls(stub func() error) { 806 fake.rollbackMutex.Lock() 807 defer fake.rollbackMutex.Unlock() 808 fake.RollbackStub = stub 809 } 810 811 func (fake *FakeTx) RollbackReturns(result1 error) { 812 fake.rollbackMutex.Lock() 813 defer fake.rollbackMutex.Unlock() 814 fake.RollbackStub = nil 815 fake.rollbackReturns = struct { 816 result1 error 817 }{result1} 818 } 819 820 func (fake *FakeTx) RollbackReturnsOnCall(i int, result1 error) { 821 fake.rollbackMutex.Lock() 822 defer fake.rollbackMutex.Unlock() 823 fake.RollbackStub = nil 824 if fake.rollbackReturnsOnCall == nil { 825 fake.rollbackReturnsOnCall = make(map[int]struct { 826 result1 error 827 }) 828 } 829 fake.rollbackReturnsOnCall[i] = struct { 830 result1 error 831 }{result1} 832 } 833 834 func (fake *FakeTx) Stmt(arg1 *sql.Stmt) *sql.Stmt { 835 fake.stmtMutex.Lock() 836 ret, specificReturn := fake.stmtReturnsOnCall[len(fake.stmtArgsForCall)] 837 fake.stmtArgsForCall = append(fake.stmtArgsForCall, struct { 838 arg1 *sql.Stmt 839 }{arg1}) 840 fake.recordInvocation("Stmt", []interface{}{arg1}) 841 fake.stmtMutex.Unlock() 842 if fake.StmtStub != nil { 843 return fake.StmtStub(arg1) 844 } 845 if specificReturn { 846 return ret.result1 847 } 848 fakeReturns := fake.stmtReturns 849 return fakeReturns.result1 850 } 851 852 func (fake *FakeTx) StmtCallCount() int { 853 fake.stmtMutex.RLock() 854 defer fake.stmtMutex.RUnlock() 855 return len(fake.stmtArgsForCall) 856 } 857 858 func (fake *FakeTx) StmtCalls(stub func(*sql.Stmt) *sql.Stmt) { 859 fake.stmtMutex.Lock() 860 defer fake.stmtMutex.Unlock() 861 fake.StmtStub = stub 862 } 863 864 func (fake *FakeTx) StmtArgsForCall(i int) *sql.Stmt { 865 fake.stmtMutex.RLock() 866 defer fake.stmtMutex.RUnlock() 867 argsForCall := fake.stmtArgsForCall[i] 868 return argsForCall.arg1 869 } 870 871 func (fake *FakeTx) StmtReturns(result1 *sql.Stmt) { 872 fake.stmtMutex.Lock() 873 defer fake.stmtMutex.Unlock() 874 fake.StmtStub = nil 875 fake.stmtReturns = struct { 876 result1 *sql.Stmt 877 }{result1} 878 } 879 880 func (fake *FakeTx) StmtReturnsOnCall(i int, result1 *sql.Stmt) { 881 fake.stmtMutex.Lock() 882 defer fake.stmtMutex.Unlock() 883 fake.StmtStub = nil 884 if fake.stmtReturnsOnCall == nil { 885 fake.stmtReturnsOnCall = make(map[int]struct { 886 result1 *sql.Stmt 887 }) 888 } 889 fake.stmtReturnsOnCall[i] = struct { 890 result1 *sql.Stmt 891 }{result1} 892 } 893 894 func (fake *FakeTx) Invocations() map[string][][]interface{} { 895 fake.invocationsMutex.RLock() 896 defer fake.invocationsMutex.RUnlock() 897 fake.commitMutex.RLock() 898 defer fake.commitMutex.RUnlock() 899 fake.encryptionStrategyMutex.RLock() 900 defer fake.encryptionStrategyMutex.RUnlock() 901 fake.execMutex.RLock() 902 defer fake.execMutex.RUnlock() 903 fake.execContextMutex.RLock() 904 defer fake.execContextMutex.RUnlock() 905 fake.prepareMutex.RLock() 906 defer fake.prepareMutex.RUnlock() 907 fake.prepareContextMutex.RLock() 908 defer fake.prepareContextMutex.RUnlock() 909 fake.queryMutex.RLock() 910 defer fake.queryMutex.RUnlock() 911 fake.queryContextMutex.RLock() 912 defer fake.queryContextMutex.RUnlock() 913 fake.queryRowMutex.RLock() 914 defer fake.queryRowMutex.RUnlock() 915 fake.queryRowContextMutex.RLock() 916 defer fake.queryRowContextMutex.RUnlock() 917 fake.rollbackMutex.RLock() 918 defer fake.rollbackMutex.RUnlock() 919 fake.stmtMutex.RLock() 920 defer fake.stmtMutex.RUnlock() 921 copiedInvocations := map[string][][]interface{}{} 922 for key, value := range fake.invocations { 923 copiedInvocations[key] = value 924 } 925 return copiedInvocations 926 } 927 928 func (fake *FakeTx) recordInvocation(key string, args []interface{}) { 929 fake.invocationsMutex.Lock() 930 defer fake.invocationsMutex.Unlock() 931 if fake.invocations == nil { 932 fake.invocations = map[string][][]interface{}{} 933 } 934 if fake.invocations[key] == nil { 935 fake.invocations[key] = [][]interface{}{} 936 } 937 fake.invocations[key] = append(fake.invocations[key], args) 938 } 939 940 var _ db.Tx = new(FakeTx)