github.com/braveheart12/insolar-09-08-19@v0.8.7/ledger/storage/jet_storage_mock.go (about) 1 package storage 2 3 /* 4 DO NOT EDIT! 5 This code was generated automatically using github.com/gojuno/minimock v1.9 6 The original interface "JetStorage" can be found in github.com/insolar/insolar/ledger/storage 7 */ 8 import ( 9 context "context" 10 "sync/atomic" 11 "time" 12 13 "github.com/gojuno/minimock" 14 core "github.com/insolar/insolar/core" 15 jet "github.com/insolar/insolar/ledger/storage/jet" 16 17 testify_assert "github.com/stretchr/testify/assert" 18 ) 19 20 //JetStorageMock implements github.com/insolar/insolar/ledger/storage.JetStorage 21 type JetStorageMock struct { 22 t minimock.Tester 23 24 AddJetsFunc func(p context.Context, p1 ...core.RecordID) (r error) 25 AddJetsCounter uint64 26 AddJetsPreCounter uint64 27 AddJetsMock mJetStorageMockAddJets 28 29 CloneJetTreeFunc func(p context.Context, p1 core.PulseNumber, p2 core.PulseNumber) (r *jet.Tree) 30 CloneJetTreeCounter uint64 31 CloneJetTreePreCounter uint64 32 CloneJetTreeMock mJetStorageMockCloneJetTree 33 34 DeleteJetTreeFunc func(p context.Context, p1 core.PulseNumber) 35 DeleteJetTreeCounter uint64 36 DeleteJetTreePreCounter uint64 37 DeleteJetTreeMock mJetStorageMockDeleteJetTree 38 39 FindJetFunc func(p context.Context, p1 core.PulseNumber, p2 core.RecordID) (r *core.RecordID, r1 bool) 40 FindJetCounter uint64 41 FindJetPreCounter uint64 42 FindJetMock mJetStorageMockFindJet 43 44 GetJetsFunc func(p context.Context) (r jet.IDSet, r1 error) 45 GetJetsCounter uint64 46 GetJetsPreCounter uint64 47 GetJetsMock mJetStorageMockGetJets 48 49 SplitJetTreeFunc func(p context.Context, p1 core.PulseNumber, p2 core.RecordID) (r *core.RecordID, r1 *core.RecordID, r2 error) 50 SplitJetTreeCounter uint64 51 SplitJetTreePreCounter uint64 52 SplitJetTreeMock mJetStorageMockSplitJetTree 53 54 UpdateJetTreeFunc func(p context.Context, p1 core.PulseNumber, p2 bool, p3 ...core.RecordID) 55 UpdateJetTreeCounter uint64 56 UpdateJetTreePreCounter uint64 57 UpdateJetTreeMock mJetStorageMockUpdateJetTree 58 } 59 60 //NewJetStorageMock returns a mock for github.com/insolar/insolar/ledger/storage.JetStorage 61 func NewJetStorageMock(t minimock.Tester) *JetStorageMock { 62 m := &JetStorageMock{t: t} 63 64 if controller, ok := t.(minimock.MockController); ok { 65 controller.RegisterMocker(m) 66 } 67 68 m.AddJetsMock = mJetStorageMockAddJets{mock: m} 69 m.CloneJetTreeMock = mJetStorageMockCloneJetTree{mock: m} 70 m.DeleteJetTreeMock = mJetStorageMockDeleteJetTree{mock: m} 71 m.FindJetMock = mJetStorageMockFindJet{mock: m} 72 m.GetJetsMock = mJetStorageMockGetJets{mock: m} 73 m.SplitJetTreeMock = mJetStorageMockSplitJetTree{mock: m} 74 m.UpdateJetTreeMock = mJetStorageMockUpdateJetTree{mock: m} 75 76 return m 77 } 78 79 type mJetStorageMockAddJets struct { 80 mock *JetStorageMock 81 mainExpectation *JetStorageMockAddJetsExpectation 82 expectationSeries []*JetStorageMockAddJetsExpectation 83 } 84 85 type JetStorageMockAddJetsExpectation struct { 86 input *JetStorageMockAddJetsInput 87 result *JetStorageMockAddJetsResult 88 } 89 90 type JetStorageMockAddJetsInput struct { 91 p context.Context 92 p1 []core.RecordID 93 } 94 95 type JetStorageMockAddJetsResult struct { 96 r error 97 } 98 99 //Expect specifies that invocation of JetStorage.AddJets is expected from 1 to Infinity times 100 func (m *mJetStorageMockAddJets) Expect(p context.Context, p1 ...core.RecordID) *mJetStorageMockAddJets { 101 m.mock.AddJetsFunc = nil 102 m.expectationSeries = nil 103 104 if m.mainExpectation == nil { 105 m.mainExpectation = &JetStorageMockAddJetsExpectation{} 106 } 107 m.mainExpectation.input = &JetStorageMockAddJetsInput{p, p1} 108 return m 109 } 110 111 //Return specifies results of invocation of JetStorage.AddJets 112 func (m *mJetStorageMockAddJets) Return(r error) *JetStorageMock { 113 m.mock.AddJetsFunc = nil 114 m.expectationSeries = nil 115 116 if m.mainExpectation == nil { 117 m.mainExpectation = &JetStorageMockAddJetsExpectation{} 118 } 119 m.mainExpectation.result = &JetStorageMockAddJetsResult{r} 120 return m.mock 121 } 122 123 //ExpectOnce specifies that invocation of JetStorage.AddJets is expected once 124 func (m *mJetStorageMockAddJets) ExpectOnce(p context.Context, p1 ...core.RecordID) *JetStorageMockAddJetsExpectation { 125 m.mock.AddJetsFunc = nil 126 m.mainExpectation = nil 127 128 expectation := &JetStorageMockAddJetsExpectation{} 129 expectation.input = &JetStorageMockAddJetsInput{p, p1} 130 m.expectationSeries = append(m.expectationSeries, expectation) 131 return expectation 132 } 133 134 func (e *JetStorageMockAddJetsExpectation) Return(r error) { 135 e.result = &JetStorageMockAddJetsResult{r} 136 } 137 138 //Set uses given function f as a mock of JetStorage.AddJets method 139 func (m *mJetStorageMockAddJets) Set(f func(p context.Context, p1 ...core.RecordID) (r error)) *JetStorageMock { 140 m.mainExpectation = nil 141 m.expectationSeries = nil 142 143 m.mock.AddJetsFunc = f 144 return m.mock 145 } 146 147 //AddJets implements github.com/insolar/insolar/ledger/storage.JetStorage interface 148 func (m *JetStorageMock) AddJets(p context.Context, p1 ...core.RecordID) (r error) { 149 counter := atomic.AddUint64(&m.AddJetsPreCounter, 1) 150 defer atomic.AddUint64(&m.AddJetsCounter, 1) 151 152 if len(m.AddJetsMock.expectationSeries) > 0 { 153 if counter > uint64(len(m.AddJetsMock.expectationSeries)) { 154 m.t.Fatalf("Unexpected call to JetStorageMock.AddJets. %v %v", p, p1) 155 return 156 } 157 158 input := m.AddJetsMock.expectationSeries[counter-1].input 159 testify_assert.Equal(m.t, *input, JetStorageMockAddJetsInput{p, p1}, "JetStorage.AddJets got unexpected parameters") 160 161 result := m.AddJetsMock.expectationSeries[counter-1].result 162 if result == nil { 163 m.t.Fatal("No results are set for the JetStorageMock.AddJets") 164 return 165 } 166 167 r = result.r 168 169 return 170 } 171 172 if m.AddJetsMock.mainExpectation != nil { 173 174 input := m.AddJetsMock.mainExpectation.input 175 if input != nil { 176 testify_assert.Equal(m.t, *input, JetStorageMockAddJetsInput{p, p1}, "JetStorage.AddJets got unexpected parameters") 177 } 178 179 result := m.AddJetsMock.mainExpectation.result 180 if result == nil { 181 m.t.Fatal("No results are set for the JetStorageMock.AddJets") 182 } 183 184 r = result.r 185 186 return 187 } 188 189 if m.AddJetsFunc == nil { 190 m.t.Fatalf("Unexpected call to JetStorageMock.AddJets. %v %v", p, p1) 191 return 192 } 193 194 return m.AddJetsFunc(p, p1...) 195 } 196 197 //AddJetsMinimockCounter returns a count of JetStorageMock.AddJetsFunc invocations 198 func (m *JetStorageMock) AddJetsMinimockCounter() uint64 { 199 return atomic.LoadUint64(&m.AddJetsCounter) 200 } 201 202 //AddJetsMinimockPreCounter returns the value of JetStorageMock.AddJets invocations 203 func (m *JetStorageMock) AddJetsMinimockPreCounter() uint64 { 204 return atomic.LoadUint64(&m.AddJetsPreCounter) 205 } 206 207 //AddJetsFinished returns true if mock invocations count is ok 208 func (m *JetStorageMock) AddJetsFinished() bool { 209 // if expectation series were set then invocations count should be equal to expectations count 210 if len(m.AddJetsMock.expectationSeries) > 0 { 211 return atomic.LoadUint64(&m.AddJetsCounter) == uint64(len(m.AddJetsMock.expectationSeries)) 212 } 213 214 // if main expectation was set then invocations count should be greater than zero 215 if m.AddJetsMock.mainExpectation != nil { 216 return atomic.LoadUint64(&m.AddJetsCounter) > 0 217 } 218 219 // if func was set then invocations count should be greater than zero 220 if m.AddJetsFunc != nil { 221 return atomic.LoadUint64(&m.AddJetsCounter) > 0 222 } 223 224 return true 225 } 226 227 type mJetStorageMockCloneJetTree struct { 228 mock *JetStorageMock 229 mainExpectation *JetStorageMockCloneJetTreeExpectation 230 expectationSeries []*JetStorageMockCloneJetTreeExpectation 231 } 232 233 type JetStorageMockCloneJetTreeExpectation struct { 234 input *JetStorageMockCloneJetTreeInput 235 result *JetStorageMockCloneJetTreeResult 236 } 237 238 type JetStorageMockCloneJetTreeInput struct { 239 p context.Context 240 p1 core.PulseNumber 241 p2 core.PulseNumber 242 } 243 244 type JetStorageMockCloneJetTreeResult struct { 245 r *jet.Tree 246 } 247 248 //Expect specifies that invocation of JetStorage.CloneJetTree is expected from 1 to Infinity times 249 func (m *mJetStorageMockCloneJetTree) Expect(p context.Context, p1 core.PulseNumber, p2 core.PulseNumber) *mJetStorageMockCloneJetTree { 250 m.mock.CloneJetTreeFunc = nil 251 m.expectationSeries = nil 252 253 if m.mainExpectation == nil { 254 m.mainExpectation = &JetStorageMockCloneJetTreeExpectation{} 255 } 256 m.mainExpectation.input = &JetStorageMockCloneJetTreeInput{p, p1, p2} 257 return m 258 } 259 260 //Return specifies results of invocation of JetStorage.CloneJetTree 261 func (m *mJetStorageMockCloneJetTree) Return(r *jet.Tree) *JetStorageMock { 262 m.mock.CloneJetTreeFunc = nil 263 m.expectationSeries = nil 264 265 if m.mainExpectation == nil { 266 m.mainExpectation = &JetStorageMockCloneJetTreeExpectation{} 267 } 268 m.mainExpectation.result = &JetStorageMockCloneJetTreeResult{r} 269 return m.mock 270 } 271 272 //ExpectOnce specifies that invocation of JetStorage.CloneJetTree is expected once 273 func (m *mJetStorageMockCloneJetTree) ExpectOnce(p context.Context, p1 core.PulseNumber, p2 core.PulseNumber) *JetStorageMockCloneJetTreeExpectation { 274 m.mock.CloneJetTreeFunc = nil 275 m.mainExpectation = nil 276 277 expectation := &JetStorageMockCloneJetTreeExpectation{} 278 expectation.input = &JetStorageMockCloneJetTreeInput{p, p1, p2} 279 m.expectationSeries = append(m.expectationSeries, expectation) 280 return expectation 281 } 282 283 func (e *JetStorageMockCloneJetTreeExpectation) Return(r *jet.Tree) { 284 e.result = &JetStorageMockCloneJetTreeResult{r} 285 } 286 287 //Set uses given function f as a mock of JetStorage.CloneJetTree method 288 func (m *mJetStorageMockCloneJetTree) Set(f func(p context.Context, p1 core.PulseNumber, p2 core.PulseNumber) (r *jet.Tree)) *JetStorageMock { 289 m.mainExpectation = nil 290 m.expectationSeries = nil 291 292 m.mock.CloneJetTreeFunc = f 293 return m.mock 294 } 295 296 //CloneJetTree implements github.com/insolar/insolar/ledger/storage.JetStorage interface 297 func (m *JetStorageMock) CloneJetTree(p context.Context, p1 core.PulseNumber, p2 core.PulseNumber) (r *jet.Tree) { 298 counter := atomic.AddUint64(&m.CloneJetTreePreCounter, 1) 299 defer atomic.AddUint64(&m.CloneJetTreeCounter, 1) 300 301 if len(m.CloneJetTreeMock.expectationSeries) > 0 { 302 if counter > uint64(len(m.CloneJetTreeMock.expectationSeries)) { 303 m.t.Fatalf("Unexpected call to JetStorageMock.CloneJetTree. %v %v %v", p, p1, p2) 304 return 305 } 306 307 input := m.CloneJetTreeMock.expectationSeries[counter-1].input 308 testify_assert.Equal(m.t, *input, JetStorageMockCloneJetTreeInput{p, p1, p2}, "JetStorage.CloneJetTree got unexpected parameters") 309 310 result := m.CloneJetTreeMock.expectationSeries[counter-1].result 311 if result == nil { 312 m.t.Fatal("No results are set for the JetStorageMock.CloneJetTree") 313 return 314 } 315 316 r = result.r 317 318 return 319 } 320 321 if m.CloneJetTreeMock.mainExpectation != nil { 322 323 input := m.CloneJetTreeMock.mainExpectation.input 324 if input != nil { 325 testify_assert.Equal(m.t, *input, JetStorageMockCloneJetTreeInput{p, p1, p2}, "JetStorage.CloneJetTree got unexpected parameters") 326 } 327 328 result := m.CloneJetTreeMock.mainExpectation.result 329 if result == nil { 330 m.t.Fatal("No results are set for the JetStorageMock.CloneJetTree") 331 } 332 333 r = result.r 334 335 return 336 } 337 338 if m.CloneJetTreeFunc == nil { 339 m.t.Fatalf("Unexpected call to JetStorageMock.CloneJetTree. %v %v %v", p, p1, p2) 340 return 341 } 342 343 return m.CloneJetTreeFunc(p, p1, p2) 344 } 345 346 //CloneJetTreeMinimockCounter returns a count of JetStorageMock.CloneJetTreeFunc invocations 347 func (m *JetStorageMock) CloneJetTreeMinimockCounter() uint64 { 348 return atomic.LoadUint64(&m.CloneJetTreeCounter) 349 } 350 351 //CloneJetTreeMinimockPreCounter returns the value of JetStorageMock.CloneJetTree invocations 352 func (m *JetStorageMock) CloneJetTreeMinimockPreCounter() uint64 { 353 return atomic.LoadUint64(&m.CloneJetTreePreCounter) 354 } 355 356 //CloneJetTreeFinished returns true if mock invocations count is ok 357 func (m *JetStorageMock) CloneJetTreeFinished() bool { 358 // if expectation series were set then invocations count should be equal to expectations count 359 if len(m.CloneJetTreeMock.expectationSeries) > 0 { 360 return atomic.LoadUint64(&m.CloneJetTreeCounter) == uint64(len(m.CloneJetTreeMock.expectationSeries)) 361 } 362 363 // if main expectation was set then invocations count should be greater than zero 364 if m.CloneJetTreeMock.mainExpectation != nil { 365 return atomic.LoadUint64(&m.CloneJetTreeCounter) > 0 366 } 367 368 // if func was set then invocations count should be greater than zero 369 if m.CloneJetTreeFunc != nil { 370 return atomic.LoadUint64(&m.CloneJetTreeCounter) > 0 371 } 372 373 return true 374 } 375 376 type mJetStorageMockDeleteJetTree struct { 377 mock *JetStorageMock 378 mainExpectation *JetStorageMockDeleteJetTreeExpectation 379 expectationSeries []*JetStorageMockDeleteJetTreeExpectation 380 } 381 382 type JetStorageMockDeleteJetTreeExpectation struct { 383 input *JetStorageMockDeleteJetTreeInput 384 } 385 386 type JetStorageMockDeleteJetTreeInput struct { 387 p context.Context 388 p1 core.PulseNumber 389 } 390 391 //Expect specifies that invocation of JetStorage.DeleteJetTree is expected from 1 to Infinity times 392 func (m *mJetStorageMockDeleteJetTree) Expect(p context.Context, p1 core.PulseNumber) *mJetStorageMockDeleteJetTree { 393 m.mock.DeleteJetTreeFunc = nil 394 m.expectationSeries = nil 395 396 if m.mainExpectation == nil { 397 m.mainExpectation = &JetStorageMockDeleteJetTreeExpectation{} 398 } 399 m.mainExpectation.input = &JetStorageMockDeleteJetTreeInput{p, p1} 400 return m 401 } 402 403 //Return specifies results of invocation of JetStorage.DeleteJetTree 404 func (m *mJetStorageMockDeleteJetTree) Return() *JetStorageMock { 405 m.mock.DeleteJetTreeFunc = nil 406 m.expectationSeries = nil 407 408 if m.mainExpectation == nil { 409 m.mainExpectation = &JetStorageMockDeleteJetTreeExpectation{} 410 } 411 412 return m.mock 413 } 414 415 //ExpectOnce specifies that invocation of JetStorage.DeleteJetTree is expected once 416 func (m *mJetStorageMockDeleteJetTree) ExpectOnce(p context.Context, p1 core.PulseNumber) *JetStorageMockDeleteJetTreeExpectation { 417 m.mock.DeleteJetTreeFunc = nil 418 m.mainExpectation = nil 419 420 expectation := &JetStorageMockDeleteJetTreeExpectation{} 421 expectation.input = &JetStorageMockDeleteJetTreeInput{p, p1} 422 m.expectationSeries = append(m.expectationSeries, expectation) 423 return expectation 424 } 425 426 //Set uses given function f as a mock of JetStorage.DeleteJetTree method 427 func (m *mJetStorageMockDeleteJetTree) Set(f func(p context.Context, p1 core.PulseNumber)) *JetStorageMock { 428 m.mainExpectation = nil 429 m.expectationSeries = nil 430 431 m.mock.DeleteJetTreeFunc = f 432 return m.mock 433 } 434 435 //DeleteJetTree implements github.com/insolar/insolar/ledger/storage.JetStorage interface 436 func (m *JetStorageMock) DeleteJetTree(p context.Context, p1 core.PulseNumber) { 437 counter := atomic.AddUint64(&m.DeleteJetTreePreCounter, 1) 438 defer atomic.AddUint64(&m.DeleteJetTreeCounter, 1) 439 440 if len(m.DeleteJetTreeMock.expectationSeries) > 0 { 441 if counter > uint64(len(m.DeleteJetTreeMock.expectationSeries)) { 442 m.t.Fatalf("Unexpected call to JetStorageMock.DeleteJetTree. %v %v", p, p1) 443 return 444 } 445 446 input := m.DeleteJetTreeMock.expectationSeries[counter-1].input 447 testify_assert.Equal(m.t, *input, JetStorageMockDeleteJetTreeInput{p, p1}, "JetStorage.DeleteJetTree got unexpected parameters") 448 449 return 450 } 451 452 if m.DeleteJetTreeMock.mainExpectation != nil { 453 454 input := m.DeleteJetTreeMock.mainExpectation.input 455 if input != nil { 456 testify_assert.Equal(m.t, *input, JetStorageMockDeleteJetTreeInput{p, p1}, "JetStorage.DeleteJetTree got unexpected parameters") 457 } 458 459 return 460 } 461 462 if m.DeleteJetTreeFunc == nil { 463 m.t.Fatalf("Unexpected call to JetStorageMock.DeleteJetTree. %v %v", p, p1) 464 return 465 } 466 467 m.DeleteJetTreeFunc(p, p1) 468 } 469 470 //DeleteJetTreeMinimockCounter returns a count of JetStorageMock.DeleteJetTreeFunc invocations 471 func (m *JetStorageMock) DeleteJetTreeMinimockCounter() uint64 { 472 return atomic.LoadUint64(&m.DeleteJetTreeCounter) 473 } 474 475 //DeleteJetTreeMinimockPreCounter returns the value of JetStorageMock.DeleteJetTree invocations 476 func (m *JetStorageMock) DeleteJetTreeMinimockPreCounter() uint64 { 477 return atomic.LoadUint64(&m.DeleteJetTreePreCounter) 478 } 479 480 //DeleteJetTreeFinished returns true if mock invocations count is ok 481 func (m *JetStorageMock) DeleteJetTreeFinished() bool { 482 // if expectation series were set then invocations count should be equal to expectations count 483 if len(m.DeleteJetTreeMock.expectationSeries) > 0 { 484 return atomic.LoadUint64(&m.DeleteJetTreeCounter) == uint64(len(m.DeleteJetTreeMock.expectationSeries)) 485 } 486 487 // if main expectation was set then invocations count should be greater than zero 488 if m.DeleteJetTreeMock.mainExpectation != nil { 489 return atomic.LoadUint64(&m.DeleteJetTreeCounter) > 0 490 } 491 492 // if func was set then invocations count should be greater than zero 493 if m.DeleteJetTreeFunc != nil { 494 return atomic.LoadUint64(&m.DeleteJetTreeCounter) > 0 495 } 496 497 return true 498 } 499 500 type mJetStorageMockFindJet struct { 501 mock *JetStorageMock 502 mainExpectation *JetStorageMockFindJetExpectation 503 expectationSeries []*JetStorageMockFindJetExpectation 504 } 505 506 type JetStorageMockFindJetExpectation struct { 507 input *JetStorageMockFindJetInput 508 result *JetStorageMockFindJetResult 509 } 510 511 type JetStorageMockFindJetInput struct { 512 p context.Context 513 p1 core.PulseNumber 514 p2 core.RecordID 515 } 516 517 type JetStorageMockFindJetResult struct { 518 r *core.RecordID 519 r1 bool 520 } 521 522 //Expect specifies that invocation of JetStorage.FindJet is expected from 1 to Infinity times 523 func (m *mJetStorageMockFindJet) Expect(p context.Context, p1 core.PulseNumber, p2 core.RecordID) *mJetStorageMockFindJet { 524 m.mock.FindJetFunc = nil 525 m.expectationSeries = nil 526 527 if m.mainExpectation == nil { 528 m.mainExpectation = &JetStorageMockFindJetExpectation{} 529 } 530 m.mainExpectation.input = &JetStorageMockFindJetInput{p, p1, p2} 531 return m 532 } 533 534 //Return specifies results of invocation of JetStorage.FindJet 535 func (m *mJetStorageMockFindJet) Return(r *core.RecordID, r1 bool) *JetStorageMock { 536 m.mock.FindJetFunc = nil 537 m.expectationSeries = nil 538 539 if m.mainExpectation == nil { 540 m.mainExpectation = &JetStorageMockFindJetExpectation{} 541 } 542 m.mainExpectation.result = &JetStorageMockFindJetResult{r, r1} 543 return m.mock 544 } 545 546 //ExpectOnce specifies that invocation of JetStorage.FindJet is expected once 547 func (m *mJetStorageMockFindJet) ExpectOnce(p context.Context, p1 core.PulseNumber, p2 core.RecordID) *JetStorageMockFindJetExpectation { 548 m.mock.FindJetFunc = nil 549 m.mainExpectation = nil 550 551 expectation := &JetStorageMockFindJetExpectation{} 552 expectation.input = &JetStorageMockFindJetInput{p, p1, p2} 553 m.expectationSeries = append(m.expectationSeries, expectation) 554 return expectation 555 } 556 557 func (e *JetStorageMockFindJetExpectation) Return(r *core.RecordID, r1 bool) { 558 e.result = &JetStorageMockFindJetResult{r, r1} 559 } 560 561 //Set uses given function f as a mock of JetStorage.FindJet method 562 func (m *mJetStorageMockFindJet) Set(f func(p context.Context, p1 core.PulseNumber, p2 core.RecordID) (r *core.RecordID, r1 bool)) *JetStorageMock { 563 m.mainExpectation = nil 564 m.expectationSeries = nil 565 566 m.mock.FindJetFunc = f 567 return m.mock 568 } 569 570 //FindJet implements github.com/insolar/insolar/ledger/storage.JetStorage interface 571 func (m *JetStorageMock) FindJet(p context.Context, p1 core.PulseNumber, p2 core.RecordID) (r *core.RecordID, r1 bool) { 572 counter := atomic.AddUint64(&m.FindJetPreCounter, 1) 573 defer atomic.AddUint64(&m.FindJetCounter, 1) 574 575 if len(m.FindJetMock.expectationSeries) > 0 { 576 if counter > uint64(len(m.FindJetMock.expectationSeries)) { 577 m.t.Fatalf("Unexpected call to JetStorageMock.FindJet. %v %v %v", p, p1, p2) 578 return 579 } 580 581 input := m.FindJetMock.expectationSeries[counter-1].input 582 testify_assert.Equal(m.t, *input, JetStorageMockFindJetInput{p, p1, p2}, "JetStorage.FindJet got unexpected parameters") 583 584 result := m.FindJetMock.expectationSeries[counter-1].result 585 if result == nil { 586 m.t.Fatal("No results are set for the JetStorageMock.FindJet") 587 return 588 } 589 590 r = result.r 591 r1 = result.r1 592 593 return 594 } 595 596 if m.FindJetMock.mainExpectation != nil { 597 598 input := m.FindJetMock.mainExpectation.input 599 if input != nil { 600 testify_assert.Equal(m.t, *input, JetStorageMockFindJetInput{p, p1, p2}, "JetStorage.FindJet got unexpected parameters") 601 } 602 603 result := m.FindJetMock.mainExpectation.result 604 if result == nil { 605 m.t.Fatal("No results are set for the JetStorageMock.FindJet") 606 } 607 608 r = result.r 609 r1 = result.r1 610 611 return 612 } 613 614 if m.FindJetFunc == nil { 615 m.t.Fatalf("Unexpected call to JetStorageMock.FindJet. %v %v %v", p, p1, p2) 616 return 617 } 618 619 return m.FindJetFunc(p, p1, p2) 620 } 621 622 //FindJetMinimockCounter returns a count of JetStorageMock.FindJetFunc invocations 623 func (m *JetStorageMock) FindJetMinimockCounter() uint64 { 624 return atomic.LoadUint64(&m.FindJetCounter) 625 } 626 627 //FindJetMinimockPreCounter returns the value of JetStorageMock.FindJet invocations 628 func (m *JetStorageMock) FindJetMinimockPreCounter() uint64 { 629 return atomic.LoadUint64(&m.FindJetPreCounter) 630 } 631 632 //FindJetFinished returns true if mock invocations count is ok 633 func (m *JetStorageMock) FindJetFinished() bool { 634 // if expectation series were set then invocations count should be equal to expectations count 635 if len(m.FindJetMock.expectationSeries) > 0 { 636 return atomic.LoadUint64(&m.FindJetCounter) == uint64(len(m.FindJetMock.expectationSeries)) 637 } 638 639 // if main expectation was set then invocations count should be greater than zero 640 if m.FindJetMock.mainExpectation != nil { 641 return atomic.LoadUint64(&m.FindJetCounter) > 0 642 } 643 644 // if func was set then invocations count should be greater than zero 645 if m.FindJetFunc != nil { 646 return atomic.LoadUint64(&m.FindJetCounter) > 0 647 } 648 649 return true 650 } 651 652 type mJetStorageMockGetJets struct { 653 mock *JetStorageMock 654 mainExpectation *JetStorageMockGetJetsExpectation 655 expectationSeries []*JetStorageMockGetJetsExpectation 656 } 657 658 type JetStorageMockGetJetsExpectation struct { 659 input *JetStorageMockGetJetsInput 660 result *JetStorageMockGetJetsResult 661 } 662 663 type JetStorageMockGetJetsInput struct { 664 p context.Context 665 } 666 667 type JetStorageMockGetJetsResult struct { 668 r jet.IDSet 669 r1 error 670 } 671 672 //Expect specifies that invocation of JetStorage.GetJets is expected from 1 to Infinity times 673 func (m *mJetStorageMockGetJets) Expect(p context.Context) *mJetStorageMockGetJets { 674 m.mock.GetJetsFunc = nil 675 m.expectationSeries = nil 676 677 if m.mainExpectation == nil { 678 m.mainExpectation = &JetStorageMockGetJetsExpectation{} 679 } 680 m.mainExpectation.input = &JetStorageMockGetJetsInput{p} 681 return m 682 } 683 684 //Return specifies results of invocation of JetStorage.GetJets 685 func (m *mJetStorageMockGetJets) Return(r jet.IDSet, r1 error) *JetStorageMock { 686 m.mock.GetJetsFunc = nil 687 m.expectationSeries = nil 688 689 if m.mainExpectation == nil { 690 m.mainExpectation = &JetStorageMockGetJetsExpectation{} 691 } 692 m.mainExpectation.result = &JetStorageMockGetJetsResult{r, r1} 693 return m.mock 694 } 695 696 //ExpectOnce specifies that invocation of JetStorage.GetJets is expected once 697 func (m *mJetStorageMockGetJets) ExpectOnce(p context.Context) *JetStorageMockGetJetsExpectation { 698 m.mock.GetJetsFunc = nil 699 m.mainExpectation = nil 700 701 expectation := &JetStorageMockGetJetsExpectation{} 702 expectation.input = &JetStorageMockGetJetsInput{p} 703 m.expectationSeries = append(m.expectationSeries, expectation) 704 return expectation 705 } 706 707 func (e *JetStorageMockGetJetsExpectation) Return(r jet.IDSet, r1 error) { 708 e.result = &JetStorageMockGetJetsResult{r, r1} 709 } 710 711 //Set uses given function f as a mock of JetStorage.GetJets method 712 func (m *mJetStorageMockGetJets) Set(f func(p context.Context) (r jet.IDSet, r1 error)) *JetStorageMock { 713 m.mainExpectation = nil 714 m.expectationSeries = nil 715 716 m.mock.GetJetsFunc = f 717 return m.mock 718 } 719 720 //GetJets implements github.com/insolar/insolar/ledger/storage.JetStorage interface 721 func (m *JetStorageMock) GetJets(p context.Context) (r jet.IDSet, r1 error) { 722 counter := atomic.AddUint64(&m.GetJetsPreCounter, 1) 723 defer atomic.AddUint64(&m.GetJetsCounter, 1) 724 725 if len(m.GetJetsMock.expectationSeries) > 0 { 726 if counter > uint64(len(m.GetJetsMock.expectationSeries)) { 727 m.t.Fatalf("Unexpected call to JetStorageMock.GetJets. %v", p) 728 return 729 } 730 731 input := m.GetJetsMock.expectationSeries[counter-1].input 732 testify_assert.Equal(m.t, *input, JetStorageMockGetJetsInput{p}, "JetStorage.GetJets got unexpected parameters") 733 734 result := m.GetJetsMock.expectationSeries[counter-1].result 735 if result == nil { 736 m.t.Fatal("No results are set for the JetStorageMock.GetJets") 737 return 738 } 739 740 r = result.r 741 r1 = result.r1 742 743 return 744 } 745 746 if m.GetJetsMock.mainExpectation != nil { 747 748 input := m.GetJetsMock.mainExpectation.input 749 if input != nil { 750 testify_assert.Equal(m.t, *input, JetStorageMockGetJetsInput{p}, "JetStorage.GetJets got unexpected parameters") 751 } 752 753 result := m.GetJetsMock.mainExpectation.result 754 if result == nil { 755 m.t.Fatal("No results are set for the JetStorageMock.GetJets") 756 } 757 758 r = result.r 759 r1 = result.r1 760 761 return 762 } 763 764 if m.GetJetsFunc == nil { 765 m.t.Fatalf("Unexpected call to JetStorageMock.GetJets. %v", p) 766 return 767 } 768 769 return m.GetJetsFunc(p) 770 } 771 772 //GetJetsMinimockCounter returns a count of JetStorageMock.GetJetsFunc invocations 773 func (m *JetStorageMock) GetJetsMinimockCounter() uint64 { 774 return atomic.LoadUint64(&m.GetJetsCounter) 775 } 776 777 //GetJetsMinimockPreCounter returns the value of JetStorageMock.GetJets invocations 778 func (m *JetStorageMock) GetJetsMinimockPreCounter() uint64 { 779 return atomic.LoadUint64(&m.GetJetsPreCounter) 780 } 781 782 //GetJetsFinished returns true if mock invocations count is ok 783 func (m *JetStorageMock) GetJetsFinished() bool { 784 // if expectation series were set then invocations count should be equal to expectations count 785 if len(m.GetJetsMock.expectationSeries) > 0 { 786 return atomic.LoadUint64(&m.GetJetsCounter) == uint64(len(m.GetJetsMock.expectationSeries)) 787 } 788 789 // if main expectation was set then invocations count should be greater than zero 790 if m.GetJetsMock.mainExpectation != nil { 791 return atomic.LoadUint64(&m.GetJetsCounter) > 0 792 } 793 794 // if func was set then invocations count should be greater than zero 795 if m.GetJetsFunc != nil { 796 return atomic.LoadUint64(&m.GetJetsCounter) > 0 797 } 798 799 return true 800 } 801 802 type mJetStorageMockSplitJetTree struct { 803 mock *JetStorageMock 804 mainExpectation *JetStorageMockSplitJetTreeExpectation 805 expectationSeries []*JetStorageMockSplitJetTreeExpectation 806 } 807 808 type JetStorageMockSplitJetTreeExpectation struct { 809 input *JetStorageMockSplitJetTreeInput 810 result *JetStorageMockSplitJetTreeResult 811 } 812 813 type JetStorageMockSplitJetTreeInput struct { 814 p context.Context 815 p1 core.PulseNumber 816 p2 core.RecordID 817 } 818 819 type JetStorageMockSplitJetTreeResult struct { 820 r *core.RecordID 821 r1 *core.RecordID 822 r2 error 823 } 824 825 //Expect specifies that invocation of JetStorage.SplitJetTree is expected from 1 to Infinity times 826 func (m *mJetStorageMockSplitJetTree) Expect(p context.Context, p1 core.PulseNumber, p2 core.RecordID) *mJetStorageMockSplitJetTree { 827 m.mock.SplitJetTreeFunc = nil 828 m.expectationSeries = nil 829 830 if m.mainExpectation == nil { 831 m.mainExpectation = &JetStorageMockSplitJetTreeExpectation{} 832 } 833 m.mainExpectation.input = &JetStorageMockSplitJetTreeInput{p, p1, p2} 834 return m 835 } 836 837 //Return specifies results of invocation of JetStorage.SplitJetTree 838 func (m *mJetStorageMockSplitJetTree) Return(r *core.RecordID, r1 *core.RecordID, r2 error) *JetStorageMock { 839 m.mock.SplitJetTreeFunc = nil 840 m.expectationSeries = nil 841 842 if m.mainExpectation == nil { 843 m.mainExpectation = &JetStorageMockSplitJetTreeExpectation{} 844 } 845 m.mainExpectation.result = &JetStorageMockSplitJetTreeResult{r, r1, r2} 846 return m.mock 847 } 848 849 //ExpectOnce specifies that invocation of JetStorage.SplitJetTree is expected once 850 func (m *mJetStorageMockSplitJetTree) ExpectOnce(p context.Context, p1 core.PulseNumber, p2 core.RecordID) *JetStorageMockSplitJetTreeExpectation { 851 m.mock.SplitJetTreeFunc = nil 852 m.mainExpectation = nil 853 854 expectation := &JetStorageMockSplitJetTreeExpectation{} 855 expectation.input = &JetStorageMockSplitJetTreeInput{p, p1, p2} 856 m.expectationSeries = append(m.expectationSeries, expectation) 857 return expectation 858 } 859 860 func (e *JetStorageMockSplitJetTreeExpectation) Return(r *core.RecordID, r1 *core.RecordID, r2 error) { 861 e.result = &JetStorageMockSplitJetTreeResult{r, r1, r2} 862 } 863 864 //Set uses given function f as a mock of JetStorage.SplitJetTree method 865 func (m *mJetStorageMockSplitJetTree) Set(f func(p context.Context, p1 core.PulseNumber, p2 core.RecordID) (r *core.RecordID, r1 *core.RecordID, r2 error)) *JetStorageMock { 866 m.mainExpectation = nil 867 m.expectationSeries = nil 868 869 m.mock.SplitJetTreeFunc = f 870 return m.mock 871 } 872 873 //SplitJetTree implements github.com/insolar/insolar/ledger/storage.JetStorage interface 874 func (m *JetStorageMock) SplitJetTree(p context.Context, p1 core.PulseNumber, p2 core.RecordID) (r *core.RecordID, r1 *core.RecordID, r2 error) { 875 counter := atomic.AddUint64(&m.SplitJetTreePreCounter, 1) 876 defer atomic.AddUint64(&m.SplitJetTreeCounter, 1) 877 878 if len(m.SplitJetTreeMock.expectationSeries) > 0 { 879 if counter > uint64(len(m.SplitJetTreeMock.expectationSeries)) { 880 m.t.Fatalf("Unexpected call to JetStorageMock.SplitJetTree. %v %v %v", p, p1, p2) 881 return 882 } 883 884 input := m.SplitJetTreeMock.expectationSeries[counter-1].input 885 testify_assert.Equal(m.t, *input, JetStorageMockSplitJetTreeInput{p, p1, p2}, "JetStorage.SplitJetTree got unexpected parameters") 886 887 result := m.SplitJetTreeMock.expectationSeries[counter-1].result 888 if result == nil { 889 m.t.Fatal("No results are set for the JetStorageMock.SplitJetTree") 890 return 891 } 892 893 r = result.r 894 r1 = result.r1 895 r2 = result.r2 896 897 return 898 } 899 900 if m.SplitJetTreeMock.mainExpectation != nil { 901 902 input := m.SplitJetTreeMock.mainExpectation.input 903 if input != nil { 904 testify_assert.Equal(m.t, *input, JetStorageMockSplitJetTreeInput{p, p1, p2}, "JetStorage.SplitJetTree got unexpected parameters") 905 } 906 907 result := m.SplitJetTreeMock.mainExpectation.result 908 if result == nil { 909 m.t.Fatal("No results are set for the JetStorageMock.SplitJetTree") 910 } 911 912 r = result.r 913 r1 = result.r1 914 r2 = result.r2 915 916 return 917 } 918 919 if m.SplitJetTreeFunc == nil { 920 m.t.Fatalf("Unexpected call to JetStorageMock.SplitJetTree. %v %v %v", p, p1, p2) 921 return 922 } 923 924 return m.SplitJetTreeFunc(p, p1, p2) 925 } 926 927 //SplitJetTreeMinimockCounter returns a count of JetStorageMock.SplitJetTreeFunc invocations 928 func (m *JetStorageMock) SplitJetTreeMinimockCounter() uint64 { 929 return atomic.LoadUint64(&m.SplitJetTreeCounter) 930 } 931 932 //SplitJetTreeMinimockPreCounter returns the value of JetStorageMock.SplitJetTree invocations 933 func (m *JetStorageMock) SplitJetTreeMinimockPreCounter() uint64 { 934 return atomic.LoadUint64(&m.SplitJetTreePreCounter) 935 } 936 937 //SplitJetTreeFinished returns true if mock invocations count is ok 938 func (m *JetStorageMock) SplitJetTreeFinished() bool { 939 // if expectation series were set then invocations count should be equal to expectations count 940 if len(m.SplitJetTreeMock.expectationSeries) > 0 { 941 return atomic.LoadUint64(&m.SplitJetTreeCounter) == uint64(len(m.SplitJetTreeMock.expectationSeries)) 942 } 943 944 // if main expectation was set then invocations count should be greater than zero 945 if m.SplitJetTreeMock.mainExpectation != nil { 946 return atomic.LoadUint64(&m.SplitJetTreeCounter) > 0 947 } 948 949 // if func was set then invocations count should be greater than zero 950 if m.SplitJetTreeFunc != nil { 951 return atomic.LoadUint64(&m.SplitJetTreeCounter) > 0 952 } 953 954 return true 955 } 956 957 type mJetStorageMockUpdateJetTree struct { 958 mock *JetStorageMock 959 mainExpectation *JetStorageMockUpdateJetTreeExpectation 960 expectationSeries []*JetStorageMockUpdateJetTreeExpectation 961 } 962 963 type JetStorageMockUpdateJetTreeExpectation struct { 964 input *JetStorageMockUpdateJetTreeInput 965 } 966 967 type JetStorageMockUpdateJetTreeInput struct { 968 p context.Context 969 p1 core.PulseNumber 970 p2 bool 971 p3 []core.RecordID 972 } 973 974 //Expect specifies that invocation of JetStorage.UpdateJetTree is expected from 1 to Infinity times 975 func (m *mJetStorageMockUpdateJetTree) Expect(p context.Context, p1 core.PulseNumber, p2 bool, p3 ...core.RecordID) *mJetStorageMockUpdateJetTree { 976 m.mock.UpdateJetTreeFunc = nil 977 m.expectationSeries = nil 978 979 if m.mainExpectation == nil { 980 m.mainExpectation = &JetStorageMockUpdateJetTreeExpectation{} 981 } 982 m.mainExpectation.input = &JetStorageMockUpdateJetTreeInput{p, p1, p2, p3} 983 return m 984 } 985 986 //Return specifies results of invocation of JetStorage.UpdateJetTree 987 func (m *mJetStorageMockUpdateJetTree) Return() *JetStorageMock { 988 m.mock.UpdateJetTreeFunc = nil 989 m.expectationSeries = nil 990 991 if m.mainExpectation == nil { 992 m.mainExpectation = &JetStorageMockUpdateJetTreeExpectation{} 993 } 994 995 return m.mock 996 } 997 998 //ExpectOnce specifies that invocation of JetStorage.UpdateJetTree is expected once 999 func (m *mJetStorageMockUpdateJetTree) ExpectOnce(p context.Context, p1 core.PulseNumber, p2 bool, p3 ...core.RecordID) *JetStorageMockUpdateJetTreeExpectation { 1000 m.mock.UpdateJetTreeFunc = nil 1001 m.mainExpectation = nil 1002 1003 expectation := &JetStorageMockUpdateJetTreeExpectation{} 1004 expectation.input = &JetStorageMockUpdateJetTreeInput{p, p1, p2, p3} 1005 m.expectationSeries = append(m.expectationSeries, expectation) 1006 return expectation 1007 } 1008 1009 //Set uses given function f as a mock of JetStorage.UpdateJetTree method 1010 func (m *mJetStorageMockUpdateJetTree) Set(f func(p context.Context, p1 core.PulseNumber, p2 bool, p3 ...core.RecordID)) *JetStorageMock { 1011 m.mainExpectation = nil 1012 m.expectationSeries = nil 1013 1014 m.mock.UpdateJetTreeFunc = f 1015 return m.mock 1016 } 1017 1018 //UpdateJetTree implements github.com/insolar/insolar/ledger/storage.JetStorage interface 1019 func (m *JetStorageMock) UpdateJetTree(p context.Context, p1 core.PulseNumber, p2 bool, p3 ...core.RecordID) { 1020 counter := atomic.AddUint64(&m.UpdateJetTreePreCounter, 1) 1021 defer atomic.AddUint64(&m.UpdateJetTreeCounter, 1) 1022 1023 if len(m.UpdateJetTreeMock.expectationSeries) > 0 { 1024 if counter > uint64(len(m.UpdateJetTreeMock.expectationSeries)) { 1025 m.t.Fatalf("Unexpected call to JetStorageMock.UpdateJetTree. %v %v %v %v", p, p1, p2, p3) 1026 return 1027 } 1028 1029 input := m.UpdateJetTreeMock.expectationSeries[counter-1].input 1030 testify_assert.Equal(m.t, *input, JetStorageMockUpdateJetTreeInput{p, p1, p2, p3}, "JetStorage.UpdateJetTree got unexpected parameters") 1031 1032 return 1033 } 1034 1035 if m.UpdateJetTreeMock.mainExpectation != nil { 1036 1037 input := m.UpdateJetTreeMock.mainExpectation.input 1038 if input != nil { 1039 testify_assert.Equal(m.t, *input, JetStorageMockUpdateJetTreeInput{p, p1, p2, p3}, "JetStorage.UpdateJetTree got unexpected parameters") 1040 } 1041 1042 return 1043 } 1044 1045 if m.UpdateJetTreeFunc == nil { 1046 m.t.Fatalf("Unexpected call to JetStorageMock.UpdateJetTree. %v %v %v %v", p, p1, p2, p3) 1047 return 1048 } 1049 1050 m.UpdateJetTreeFunc(p, p1, p2, p3...) 1051 } 1052 1053 //UpdateJetTreeMinimockCounter returns a count of JetStorageMock.UpdateJetTreeFunc invocations 1054 func (m *JetStorageMock) UpdateJetTreeMinimockCounter() uint64 { 1055 return atomic.LoadUint64(&m.UpdateJetTreeCounter) 1056 } 1057 1058 //UpdateJetTreeMinimockPreCounter returns the value of JetStorageMock.UpdateJetTree invocations 1059 func (m *JetStorageMock) UpdateJetTreeMinimockPreCounter() uint64 { 1060 return atomic.LoadUint64(&m.UpdateJetTreePreCounter) 1061 } 1062 1063 //UpdateJetTreeFinished returns true if mock invocations count is ok 1064 func (m *JetStorageMock) UpdateJetTreeFinished() bool { 1065 // if expectation series were set then invocations count should be equal to expectations count 1066 if len(m.UpdateJetTreeMock.expectationSeries) > 0 { 1067 return atomic.LoadUint64(&m.UpdateJetTreeCounter) == uint64(len(m.UpdateJetTreeMock.expectationSeries)) 1068 } 1069 1070 // if main expectation was set then invocations count should be greater than zero 1071 if m.UpdateJetTreeMock.mainExpectation != nil { 1072 return atomic.LoadUint64(&m.UpdateJetTreeCounter) > 0 1073 } 1074 1075 // if func was set then invocations count should be greater than zero 1076 if m.UpdateJetTreeFunc != nil { 1077 return atomic.LoadUint64(&m.UpdateJetTreeCounter) > 0 1078 } 1079 1080 return true 1081 } 1082 1083 //ValidateCallCounters checks that all mocked methods of the interface have been called at least once 1084 //Deprecated: please use MinimockFinish method or use Finish method of minimock.Controller 1085 func (m *JetStorageMock) ValidateCallCounters() { 1086 1087 if !m.AddJetsFinished() { 1088 m.t.Fatal("Expected call to JetStorageMock.AddJets") 1089 } 1090 1091 if !m.CloneJetTreeFinished() { 1092 m.t.Fatal("Expected call to JetStorageMock.CloneJetTree") 1093 } 1094 1095 if !m.DeleteJetTreeFinished() { 1096 m.t.Fatal("Expected call to JetStorageMock.DeleteJetTree") 1097 } 1098 1099 if !m.FindJetFinished() { 1100 m.t.Fatal("Expected call to JetStorageMock.FindJet") 1101 } 1102 1103 if !m.GetJetsFinished() { 1104 m.t.Fatal("Expected call to JetStorageMock.GetJets") 1105 } 1106 1107 if !m.SplitJetTreeFinished() { 1108 m.t.Fatal("Expected call to JetStorageMock.SplitJetTree") 1109 } 1110 1111 if !m.UpdateJetTreeFinished() { 1112 m.t.Fatal("Expected call to JetStorageMock.UpdateJetTree") 1113 } 1114 1115 } 1116 1117 //CheckMocksCalled checks that all mocked methods of the interface have been called at least once 1118 //Deprecated: please use MinimockFinish method or use Finish method of minimock.Controller 1119 func (m *JetStorageMock) CheckMocksCalled() { 1120 m.Finish() 1121 } 1122 1123 //Finish checks that all mocked methods of the interface have been called at least once 1124 //Deprecated: please use MinimockFinish or use Finish method of minimock.Controller 1125 func (m *JetStorageMock) Finish() { 1126 m.MinimockFinish() 1127 } 1128 1129 //MinimockFinish checks that all mocked methods of the interface have been called at least once 1130 func (m *JetStorageMock) MinimockFinish() { 1131 1132 if !m.AddJetsFinished() { 1133 m.t.Fatal("Expected call to JetStorageMock.AddJets") 1134 } 1135 1136 if !m.CloneJetTreeFinished() { 1137 m.t.Fatal("Expected call to JetStorageMock.CloneJetTree") 1138 } 1139 1140 if !m.DeleteJetTreeFinished() { 1141 m.t.Fatal("Expected call to JetStorageMock.DeleteJetTree") 1142 } 1143 1144 if !m.FindJetFinished() { 1145 m.t.Fatal("Expected call to JetStorageMock.FindJet") 1146 } 1147 1148 if !m.GetJetsFinished() { 1149 m.t.Fatal("Expected call to JetStorageMock.GetJets") 1150 } 1151 1152 if !m.SplitJetTreeFinished() { 1153 m.t.Fatal("Expected call to JetStorageMock.SplitJetTree") 1154 } 1155 1156 if !m.UpdateJetTreeFinished() { 1157 m.t.Fatal("Expected call to JetStorageMock.UpdateJetTree") 1158 } 1159 1160 } 1161 1162 //Wait waits for all mocked methods to be called at least once 1163 //Deprecated: please use MinimockWait or use Wait method of minimock.Controller 1164 func (m *JetStorageMock) Wait(timeout time.Duration) { 1165 m.MinimockWait(timeout) 1166 } 1167 1168 //MinimockWait waits for all mocked methods to be called at least once 1169 //this method is called by minimock.Controller 1170 func (m *JetStorageMock) MinimockWait(timeout time.Duration) { 1171 timeoutCh := time.After(timeout) 1172 for { 1173 ok := true 1174 ok = ok && m.AddJetsFinished() 1175 ok = ok && m.CloneJetTreeFinished() 1176 ok = ok && m.DeleteJetTreeFinished() 1177 ok = ok && m.FindJetFinished() 1178 ok = ok && m.GetJetsFinished() 1179 ok = ok && m.SplitJetTreeFinished() 1180 ok = ok && m.UpdateJetTreeFinished() 1181 1182 if ok { 1183 return 1184 } 1185 1186 select { 1187 case <-timeoutCh: 1188 1189 if !m.AddJetsFinished() { 1190 m.t.Error("Expected call to JetStorageMock.AddJets") 1191 } 1192 1193 if !m.CloneJetTreeFinished() { 1194 m.t.Error("Expected call to JetStorageMock.CloneJetTree") 1195 } 1196 1197 if !m.DeleteJetTreeFinished() { 1198 m.t.Error("Expected call to JetStorageMock.DeleteJetTree") 1199 } 1200 1201 if !m.FindJetFinished() { 1202 m.t.Error("Expected call to JetStorageMock.FindJet") 1203 } 1204 1205 if !m.GetJetsFinished() { 1206 m.t.Error("Expected call to JetStorageMock.GetJets") 1207 } 1208 1209 if !m.SplitJetTreeFinished() { 1210 m.t.Error("Expected call to JetStorageMock.SplitJetTree") 1211 } 1212 1213 if !m.UpdateJetTreeFinished() { 1214 m.t.Error("Expected call to JetStorageMock.UpdateJetTree") 1215 } 1216 1217 m.t.Fatalf("Some mocks were not called on time: %s", timeout) 1218 return 1219 default: 1220 time.Sleep(time.Millisecond) 1221 } 1222 } 1223 } 1224 1225 //AllMocksCalled returns true if all mocked methods were called before the execution of AllMocksCalled, 1226 //it can be used with assert/require, i.e. assert.True(mock.AllMocksCalled()) 1227 func (m *JetStorageMock) AllMocksCalled() bool { 1228 1229 if !m.AddJetsFinished() { 1230 return false 1231 } 1232 1233 if !m.CloneJetTreeFinished() { 1234 return false 1235 } 1236 1237 if !m.DeleteJetTreeFinished() { 1238 return false 1239 } 1240 1241 if !m.FindJetFinished() { 1242 return false 1243 } 1244 1245 if !m.GetJetsFinished() { 1246 return false 1247 } 1248 1249 if !m.SplitJetTreeFinished() { 1250 return false 1251 } 1252 1253 if !m.UpdateJetTreeFinished() { 1254 return false 1255 } 1256 1257 return true 1258 }