github.com/braveheart12/insolar-09-08-19@v0.8.7/ledger/storage/pulse_tracker_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 "PulseTracker" 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 16 testify_assert "github.com/stretchr/testify/assert" 17 ) 18 19 //PulseTrackerMock implements github.com/insolar/insolar/ledger/storage.PulseTracker 20 type PulseTrackerMock struct { 21 t minimock.Tester 22 23 AddPulseFunc func(p context.Context, p1 core.Pulse) (r error) 24 AddPulseCounter uint64 25 AddPulsePreCounter uint64 26 AddPulseMock mPulseTrackerMockAddPulse 27 28 DeletePulseFunc func(p context.Context, p1 core.PulseNumber) (r error) 29 DeletePulseCounter uint64 30 DeletePulsePreCounter uint64 31 DeletePulseMock mPulseTrackerMockDeletePulse 32 33 GetLatestPulseFunc func(p context.Context) (r *Pulse, r1 error) 34 GetLatestPulseCounter uint64 35 GetLatestPulsePreCounter uint64 36 GetLatestPulseMock mPulseTrackerMockGetLatestPulse 37 38 GetNthPrevPulseFunc func(p context.Context, p1 uint, p2 core.PulseNumber) (r *Pulse, r1 error) 39 GetNthPrevPulseCounter uint64 40 GetNthPrevPulsePreCounter uint64 41 GetNthPrevPulseMock mPulseTrackerMockGetNthPrevPulse 42 43 GetPreviousPulseFunc func(p context.Context, p1 core.PulseNumber) (r *Pulse, r1 error) 44 GetPreviousPulseCounter uint64 45 GetPreviousPulsePreCounter uint64 46 GetPreviousPulseMock mPulseTrackerMockGetPreviousPulse 47 48 GetPulseFunc func(p context.Context, p1 core.PulseNumber) (r *Pulse, r1 error) 49 GetPulseCounter uint64 50 GetPulsePreCounter uint64 51 GetPulseMock mPulseTrackerMockGetPulse 52 } 53 54 //NewPulseTrackerMock returns a mock for github.com/insolar/insolar/ledger/storage.PulseTracker 55 func NewPulseTrackerMock(t minimock.Tester) *PulseTrackerMock { 56 m := &PulseTrackerMock{t: t} 57 58 if controller, ok := t.(minimock.MockController); ok { 59 controller.RegisterMocker(m) 60 } 61 62 m.AddPulseMock = mPulseTrackerMockAddPulse{mock: m} 63 m.DeletePulseMock = mPulseTrackerMockDeletePulse{mock: m} 64 m.GetLatestPulseMock = mPulseTrackerMockGetLatestPulse{mock: m} 65 m.GetNthPrevPulseMock = mPulseTrackerMockGetNthPrevPulse{mock: m} 66 m.GetPreviousPulseMock = mPulseTrackerMockGetPreviousPulse{mock: m} 67 m.GetPulseMock = mPulseTrackerMockGetPulse{mock: m} 68 69 return m 70 } 71 72 type mPulseTrackerMockAddPulse struct { 73 mock *PulseTrackerMock 74 mainExpectation *PulseTrackerMockAddPulseExpectation 75 expectationSeries []*PulseTrackerMockAddPulseExpectation 76 } 77 78 type PulseTrackerMockAddPulseExpectation struct { 79 input *PulseTrackerMockAddPulseInput 80 result *PulseTrackerMockAddPulseResult 81 } 82 83 type PulseTrackerMockAddPulseInput struct { 84 p context.Context 85 p1 core.Pulse 86 } 87 88 type PulseTrackerMockAddPulseResult struct { 89 r error 90 } 91 92 //Expect specifies that invocation of PulseTracker.AddPulse is expected from 1 to Infinity times 93 func (m *mPulseTrackerMockAddPulse) Expect(p context.Context, p1 core.Pulse) *mPulseTrackerMockAddPulse { 94 m.mock.AddPulseFunc = nil 95 m.expectationSeries = nil 96 97 if m.mainExpectation == nil { 98 m.mainExpectation = &PulseTrackerMockAddPulseExpectation{} 99 } 100 m.mainExpectation.input = &PulseTrackerMockAddPulseInput{p, p1} 101 return m 102 } 103 104 //Return specifies results of invocation of PulseTracker.AddPulse 105 func (m *mPulseTrackerMockAddPulse) Return(r error) *PulseTrackerMock { 106 m.mock.AddPulseFunc = nil 107 m.expectationSeries = nil 108 109 if m.mainExpectation == nil { 110 m.mainExpectation = &PulseTrackerMockAddPulseExpectation{} 111 } 112 m.mainExpectation.result = &PulseTrackerMockAddPulseResult{r} 113 return m.mock 114 } 115 116 //ExpectOnce specifies that invocation of PulseTracker.AddPulse is expected once 117 func (m *mPulseTrackerMockAddPulse) ExpectOnce(p context.Context, p1 core.Pulse) *PulseTrackerMockAddPulseExpectation { 118 m.mock.AddPulseFunc = nil 119 m.mainExpectation = nil 120 121 expectation := &PulseTrackerMockAddPulseExpectation{} 122 expectation.input = &PulseTrackerMockAddPulseInput{p, p1} 123 m.expectationSeries = append(m.expectationSeries, expectation) 124 return expectation 125 } 126 127 func (e *PulseTrackerMockAddPulseExpectation) Return(r error) { 128 e.result = &PulseTrackerMockAddPulseResult{r} 129 } 130 131 //Set uses given function f as a mock of PulseTracker.AddPulse method 132 func (m *mPulseTrackerMockAddPulse) Set(f func(p context.Context, p1 core.Pulse) (r error)) *PulseTrackerMock { 133 m.mainExpectation = nil 134 m.expectationSeries = nil 135 136 m.mock.AddPulseFunc = f 137 return m.mock 138 } 139 140 //AddPulse implements github.com/insolar/insolar/ledger/storage.PulseTracker interface 141 func (m *PulseTrackerMock) AddPulse(p context.Context, p1 core.Pulse) (r error) { 142 counter := atomic.AddUint64(&m.AddPulsePreCounter, 1) 143 defer atomic.AddUint64(&m.AddPulseCounter, 1) 144 145 if len(m.AddPulseMock.expectationSeries) > 0 { 146 if counter > uint64(len(m.AddPulseMock.expectationSeries)) { 147 m.t.Fatalf("Unexpected call to PulseTrackerMock.AddPulse. %v %v", p, p1) 148 return 149 } 150 151 input := m.AddPulseMock.expectationSeries[counter-1].input 152 testify_assert.Equal(m.t, *input, PulseTrackerMockAddPulseInput{p, p1}, "PulseTracker.AddPulse got unexpected parameters") 153 154 result := m.AddPulseMock.expectationSeries[counter-1].result 155 if result == nil { 156 m.t.Fatal("No results are set for the PulseTrackerMock.AddPulse") 157 return 158 } 159 160 r = result.r 161 162 return 163 } 164 165 if m.AddPulseMock.mainExpectation != nil { 166 167 input := m.AddPulseMock.mainExpectation.input 168 if input != nil { 169 testify_assert.Equal(m.t, *input, PulseTrackerMockAddPulseInput{p, p1}, "PulseTracker.AddPulse got unexpected parameters") 170 } 171 172 result := m.AddPulseMock.mainExpectation.result 173 if result == nil { 174 m.t.Fatal("No results are set for the PulseTrackerMock.AddPulse") 175 } 176 177 r = result.r 178 179 return 180 } 181 182 if m.AddPulseFunc == nil { 183 m.t.Fatalf("Unexpected call to PulseTrackerMock.AddPulse. %v %v", p, p1) 184 return 185 } 186 187 return m.AddPulseFunc(p, p1) 188 } 189 190 //AddPulseMinimockCounter returns a count of PulseTrackerMock.AddPulseFunc invocations 191 func (m *PulseTrackerMock) AddPulseMinimockCounter() uint64 { 192 return atomic.LoadUint64(&m.AddPulseCounter) 193 } 194 195 //AddPulseMinimockPreCounter returns the value of PulseTrackerMock.AddPulse invocations 196 func (m *PulseTrackerMock) AddPulseMinimockPreCounter() uint64 { 197 return atomic.LoadUint64(&m.AddPulsePreCounter) 198 } 199 200 //AddPulseFinished returns true if mock invocations count is ok 201 func (m *PulseTrackerMock) AddPulseFinished() bool { 202 // if expectation series were set then invocations count should be equal to expectations count 203 if len(m.AddPulseMock.expectationSeries) > 0 { 204 return atomic.LoadUint64(&m.AddPulseCounter) == uint64(len(m.AddPulseMock.expectationSeries)) 205 } 206 207 // if main expectation was set then invocations count should be greater than zero 208 if m.AddPulseMock.mainExpectation != nil { 209 return atomic.LoadUint64(&m.AddPulseCounter) > 0 210 } 211 212 // if func was set then invocations count should be greater than zero 213 if m.AddPulseFunc != nil { 214 return atomic.LoadUint64(&m.AddPulseCounter) > 0 215 } 216 217 return true 218 } 219 220 type mPulseTrackerMockDeletePulse struct { 221 mock *PulseTrackerMock 222 mainExpectation *PulseTrackerMockDeletePulseExpectation 223 expectationSeries []*PulseTrackerMockDeletePulseExpectation 224 } 225 226 type PulseTrackerMockDeletePulseExpectation struct { 227 input *PulseTrackerMockDeletePulseInput 228 result *PulseTrackerMockDeletePulseResult 229 } 230 231 type PulseTrackerMockDeletePulseInput struct { 232 p context.Context 233 p1 core.PulseNumber 234 } 235 236 type PulseTrackerMockDeletePulseResult struct { 237 r error 238 } 239 240 //Expect specifies that invocation of PulseTracker.DeletePulse is expected from 1 to Infinity times 241 func (m *mPulseTrackerMockDeletePulse) Expect(p context.Context, p1 core.PulseNumber) *mPulseTrackerMockDeletePulse { 242 m.mock.DeletePulseFunc = nil 243 m.expectationSeries = nil 244 245 if m.mainExpectation == nil { 246 m.mainExpectation = &PulseTrackerMockDeletePulseExpectation{} 247 } 248 m.mainExpectation.input = &PulseTrackerMockDeletePulseInput{p, p1} 249 return m 250 } 251 252 //Return specifies results of invocation of PulseTracker.DeletePulse 253 func (m *mPulseTrackerMockDeletePulse) Return(r error) *PulseTrackerMock { 254 m.mock.DeletePulseFunc = nil 255 m.expectationSeries = nil 256 257 if m.mainExpectation == nil { 258 m.mainExpectation = &PulseTrackerMockDeletePulseExpectation{} 259 } 260 m.mainExpectation.result = &PulseTrackerMockDeletePulseResult{r} 261 return m.mock 262 } 263 264 //ExpectOnce specifies that invocation of PulseTracker.DeletePulse is expected once 265 func (m *mPulseTrackerMockDeletePulse) ExpectOnce(p context.Context, p1 core.PulseNumber) *PulseTrackerMockDeletePulseExpectation { 266 m.mock.DeletePulseFunc = nil 267 m.mainExpectation = nil 268 269 expectation := &PulseTrackerMockDeletePulseExpectation{} 270 expectation.input = &PulseTrackerMockDeletePulseInput{p, p1} 271 m.expectationSeries = append(m.expectationSeries, expectation) 272 return expectation 273 } 274 275 func (e *PulseTrackerMockDeletePulseExpectation) Return(r error) { 276 e.result = &PulseTrackerMockDeletePulseResult{r} 277 } 278 279 //Set uses given function f as a mock of PulseTracker.DeletePulse method 280 func (m *mPulseTrackerMockDeletePulse) Set(f func(p context.Context, p1 core.PulseNumber) (r error)) *PulseTrackerMock { 281 m.mainExpectation = nil 282 m.expectationSeries = nil 283 284 m.mock.DeletePulseFunc = f 285 return m.mock 286 } 287 288 //DeletePulse implements github.com/insolar/insolar/ledger/storage.PulseTracker interface 289 func (m *PulseTrackerMock) DeletePulse(p context.Context, p1 core.PulseNumber) (r error) { 290 counter := atomic.AddUint64(&m.DeletePulsePreCounter, 1) 291 defer atomic.AddUint64(&m.DeletePulseCounter, 1) 292 293 if len(m.DeletePulseMock.expectationSeries) > 0 { 294 if counter > uint64(len(m.DeletePulseMock.expectationSeries)) { 295 m.t.Fatalf("Unexpected call to PulseTrackerMock.DeletePulse. %v %v", p, p1) 296 return 297 } 298 299 input := m.DeletePulseMock.expectationSeries[counter-1].input 300 testify_assert.Equal(m.t, *input, PulseTrackerMockDeletePulseInput{p, p1}, "PulseTracker.DeletePulse got unexpected parameters") 301 302 result := m.DeletePulseMock.expectationSeries[counter-1].result 303 if result == nil { 304 m.t.Fatal("No results are set for the PulseTrackerMock.DeletePulse") 305 return 306 } 307 308 r = result.r 309 310 return 311 } 312 313 if m.DeletePulseMock.mainExpectation != nil { 314 315 input := m.DeletePulseMock.mainExpectation.input 316 if input != nil { 317 testify_assert.Equal(m.t, *input, PulseTrackerMockDeletePulseInput{p, p1}, "PulseTracker.DeletePulse got unexpected parameters") 318 } 319 320 result := m.DeletePulseMock.mainExpectation.result 321 if result == nil { 322 m.t.Fatal("No results are set for the PulseTrackerMock.DeletePulse") 323 } 324 325 r = result.r 326 327 return 328 } 329 330 if m.DeletePulseFunc == nil { 331 m.t.Fatalf("Unexpected call to PulseTrackerMock.DeletePulse. %v %v", p, p1) 332 return 333 } 334 335 return m.DeletePulseFunc(p, p1) 336 } 337 338 //DeletePulseMinimockCounter returns a count of PulseTrackerMock.DeletePulseFunc invocations 339 func (m *PulseTrackerMock) DeletePulseMinimockCounter() uint64 { 340 return atomic.LoadUint64(&m.DeletePulseCounter) 341 } 342 343 //DeletePulseMinimockPreCounter returns the value of PulseTrackerMock.DeletePulse invocations 344 func (m *PulseTrackerMock) DeletePulseMinimockPreCounter() uint64 { 345 return atomic.LoadUint64(&m.DeletePulsePreCounter) 346 } 347 348 //DeletePulseFinished returns true if mock invocations count is ok 349 func (m *PulseTrackerMock) DeletePulseFinished() bool { 350 // if expectation series were set then invocations count should be equal to expectations count 351 if len(m.DeletePulseMock.expectationSeries) > 0 { 352 return atomic.LoadUint64(&m.DeletePulseCounter) == uint64(len(m.DeletePulseMock.expectationSeries)) 353 } 354 355 // if main expectation was set then invocations count should be greater than zero 356 if m.DeletePulseMock.mainExpectation != nil { 357 return atomic.LoadUint64(&m.DeletePulseCounter) > 0 358 } 359 360 // if func was set then invocations count should be greater than zero 361 if m.DeletePulseFunc != nil { 362 return atomic.LoadUint64(&m.DeletePulseCounter) > 0 363 } 364 365 return true 366 } 367 368 type mPulseTrackerMockGetLatestPulse struct { 369 mock *PulseTrackerMock 370 mainExpectation *PulseTrackerMockGetLatestPulseExpectation 371 expectationSeries []*PulseTrackerMockGetLatestPulseExpectation 372 } 373 374 type PulseTrackerMockGetLatestPulseExpectation struct { 375 input *PulseTrackerMockGetLatestPulseInput 376 result *PulseTrackerMockGetLatestPulseResult 377 } 378 379 type PulseTrackerMockGetLatestPulseInput struct { 380 p context.Context 381 } 382 383 type PulseTrackerMockGetLatestPulseResult struct { 384 r *Pulse 385 r1 error 386 } 387 388 //Expect specifies that invocation of PulseTracker.GetLatestPulse is expected from 1 to Infinity times 389 func (m *mPulseTrackerMockGetLatestPulse) Expect(p context.Context) *mPulseTrackerMockGetLatestPulse { 390 m.mock.GetLatestPulseFunc = nil 391 m.expectationSeries = nil 392 393 if m.mainExpectation == nil { 394 m.mainExpectation = &PulseTrackerMockGetLatestPulseExpectation{} 395 } 396 m.mainExpectation.input = &PulseTrackerMockGetLatestPulseInput{p} 397 return m 398 } 399 400 //Return specifies results of invocation of PulseTracker.GetLatestPulse 401 func (m *mPulseTrackerMockGetLatestPulse) Return(r *Pulse, r1 error) *PulseTrackerMock { 402 m.mock.GetLatestPulseFunc = nil 403 m.expectationSeries = nil 404 405 if m.mainExpectation == nil { 406 m.mainExpectation = &PulseTrackerMockGetLatestPulseExpectation{} 407 } 408 m.mainExpectation.result = &PulseTrackerMockGetLatestPulseResult{r, r1} 409 return m.mock 410 } 411 412 //ExpectOnce specifies that invocation of PulseTracker.GetLatestPulse is expected once 413 func (m *mPulseTrackerMockGetLatestPulse) ExpectOnce(p context.Context) *PulseTrackerMockGetLatestPulseExpectation { 414 m.mock.GetLatestPulseFunc = nil 415 m.mainExpectation = nil 416 417 expectation := &PulseTrackerMockGetLatestPulseExpectation{} 418 expectation.input = &PulseTrackerMockGetLatestPulseInput{p} 419 m.expectationSeries = append(m.expectationSeries, expectation) 420 return expectation 421 } 422 423 func (e *PulseTrackerMockGetLatestPulseExpectation) Return(r *Pulse, r1 error) { 424 e.result = &PulseTrackerMockGetLatestPulseResult{r, r1} 425 } 426 427 //Set uses given function f as a mock of PulseTracker.GetLatestPulse method 428 func (m *mPulseTrackerMockGetLatestPulse) Set(f func(p context.Context) (r *Pulse, r1 error)) *PulseTrackerMock { 429 m.mainExpectation = nil 430 m.expectationSeries = nil 431 432 m.mock.GetLatestPulseFunc = f 433 return m.mock 434 } 435 436 //GetLatestPulse implements github.com/insolar/insolar/ledger/storage.PulseTracker interface 437 func (m *PulseTrackerMock) GetLatestPulse(p context.Context) (r *Pulse, r1 error) { 438 counter := atomic.AddUint64(&m.GetLatestPulsePreCounter, 1) 439 defer atomic.AddUint64(&m.GetLatestPulseCounter, 1) 440 441 if len(m.GetLatestPulseMock.expectationSeries) > 0 { 442 if counter > uint64(len(m.GetLatestPulseMock.expectationSeries)) { 443 m.t.Fatalf("Unexpected call to PulseTrackerMock.GetLatestPulse. %v", p) 444 return 445 } 446 447 input := m.GetLatestPulseMock.expectationSeries[counter-1].input 448 testify_assert.Equal(m.t, *input, PulseTrackerMockGetLatestPulseInput{p}, "PulseTracker.GetLatestPulse got unexpected parameters") 449 450 result := m.GetLatestPulseMock.expectationSeries[counter-1].result 451 if result == nil { 452 m.t.Fatal("No results are set for the PulseTrackerMock.GetLatestPulse") 453 return 454 } 455 456 r = result.r 457 r1 = result.r1 458 459 return 460 } 461 462 if m.GetLatestPulseMock.mainExpectation != nil { 463 464 input := m.GetLatestPulseMock.mainExpectation.input 465 if input != nil { 466 testify_assert.Equal(m.t, *input, PulseTrackerMockGetLatestPulseInput{p}, "PulseTracker.GetLatestPulse got unexpected parameters") 467 } 468 469 result := m.GetLatestPulseMock.mainExpectation.result 470 if result == nil { 471 m.t.Fatal("No results are set for the PulseTrackerMock.GetLatestPulse") 472 } 473 474 r = result.r 475 r1 = result.r1 476 477 return 478 } 479 480 if m.GetLatestPulseFunc == nil { 481 m.t.Fatalf("Unexpected call to PulseTrackerMock.GetLatestPulse. %v", p) 482 return 483 } 484 485 return m.GetLatestPulseFunc(p) 486 } 487 488 //GetLatestPulseMinimockCounter returns a count of PulseTrackerMock.GetLatestPulseFunc invocations 489 func (m *PulseTrackerMock) GetLatestPulseMinimockCounter() uint64 { 490 return atomic.LoadUint64(&m.GetLatestPulseCounter) 491 } 492 493 //GetLatestPulseMinimockPreCounter returns the value of PulseTrackerMock.GetLatestPulse invocations 494 func (m *PulseTrackerMock) GetLatestPulseMinimockPreCounter() uint64 { 495 return atomic.LoadUint64(&m.GetLatestPulsePreCounter) 496 } 497 498 //GetLatestPulseFinished returns true if mock invocations count is ok 499 func (m *PulseTrackerMock) GetLatestPulseFinished() bool { 500 // if expectation series were set then invocations count should be equal to expectations count 501 if len(m.GetLatestPulseMock.expectationSeries) > 0 { 502 return atomic.LoadUint64(&m.GetLatestPulseCounter) == uint64(len(m.GetLatestPulseMock.expectationSeries)) 503 } 504 505 // if main expectation was set then invocations count should be greater than zero 506 if m.GetLatestPulseMock.mainExpectation != nil { 507 return atomic.LoadUint64(&m.GetLatestPulseCounter) > 0 508 } 509 510 // if func was set then invocations count should be greater than zero 511 if m.GetLatestPulseFunc != nil { 512 return atomic.LoadUint64(&m.GetLatestPulseCounter) > 0 513 } 514 515 return true 516 } 517 518 type mPulseTrackerMockGetNthPrevPulse struct { 519 mock *PulseTrackerMock 520 mainExpectation *PulseTrackerMockGetNthPrevPulseExpectation 521 expectationSeries []*PulseTrackerMockGetNthPrevPulseExpectation 522 } 523 524 type PulseTrackerMockGetNthPrevPulseExpectation struct { 525 input *PulseTrackerMockGetNthPrevPulseInput 526 result *PulseTrackerMockGetNthPrevPulseResult 527 } 528 529 type PulseTrackerMockGetNthPrevPulseInput struct { 530 p context.Context 531 p1 uint 532 p2 core.PulseNumber 533 } 534 535 type PulseTrackerMockGetNthPrevPulseResult struct { 536 r *Pulse 537 r1 error 538 } 539 540 //Expect specifies that invocation of PulseTracker.GetNthPrevPulse is expected from 1 to Infinity times 541 func (m *mPulseTrackerMockGetNthPrevPulse) Expect(p context.Context, p1 uint, p2 core.PulseNumber) *mPulseTrackerMockGetNthPrevPulse { 542 m.mock.GetNthPrevPulseFunc = nil 543 m.expectationSeries = nil 544 545 if m.mainExpectation == nil { 546 m.mainExpectation = &PulseTrackerMockGetNthPrevPulseExpectation{} 547 } 548 m.mainExpectation.input = &PulseTrackerMockGetNthPrevPulseInput{p, p1, p2} 549 return m 550 } 551 552 //Return specifies results of invocation of PulseTracker.GetNthPrevPulse 553 func (m *mPulseTrackerMockGetNthPrevPulse) Return(r *Pulse, r1 error) *PulseTrackerMock { 554 m.mock.GetNthPrevPulseFunc = nil 555 m.expectationSeries = nil 556 557 if m.mainExpectation == nil { 558 m.mainExpectation = &PulseTrackerMockGetNthPrevPulseExpectation{} 559 } 560 m.mainExpectation.result = &PulseTrackerMockGetNthPrevPulseResult{r, r1} 561 return m.mock 562 } 563 564 //ExpectOnce specifies that invocation of PulseTracker.GetNthPrevPulse is expected once 565 func (m *mPulseTrackerMockGetNthPrevPulse) ExpectOnce(p context.Context, p1 uint, p2 core.PulseNumber) *PulseTrackerMockGetNthPrevPulseExpectation { 566 m.mock.GetNthPrevPulseFunc = nil 567 m.mainExpectation = nil 568 569 expectation := &PulseTrackerMockGetNthPrevPulseExpectation{} 570 expectation.input = &PulseTrackerMockGetNthPrevPulseInput{p, p1, p2} 571 m.expectationSeries = append(m.expectationSeries, expectation) 572 return expectation 573 } 574 575 func (e *PulseTrackerMockGetNthPrevPulseExpectation) Return(r *Pulse, r1 error) { 576 e.result = &PulseTrackerMockGetNthPrevPulseResult{r, r1} 577 } 578 579 //Set uses given function f as a mock of PulseTracker.GetNthPrevPulse method 580 func (m *mPulseTrackerMockGetNthPrevPulse) Set(f func(p context.Context, p1 uint, p2 core.PulseNumber) (r *Pulse, r1 error)) *PulseTrackerMock { 581 m.mainExpectation = nil 582 m.expectationSeries = nil 583 584 m.mock.GetNthPrevPulseFunc = f 585 return m.mock 586 } 587 588 //GetNthPrevPulse implements github.com/insolar/insolar/ledger/storage.PulseTracker interface 589 func (m *PulseTrackerMock) GetNthPrevPulse(p context.Context, p1 uint, p2 core.PulseNumber) (r *Pulse, r1 error) { 590 counter := atomic.AddUint64(&m.GetNthPrevPulsePreCounter, 1) 591 defer atomic.AddUint64(&m.GetNthPrevPulseCounter, 1) 592 593 if len(m.GetNthPrevPulseMock.expectationSeries) > 0 { 594 if counter > uint64(len(m.GetNthPrevPulseMock.expectationSeries)) { 595 m.t.Fatalf("Unexpected call to PulseTrackerMock.GetNthPrevPulse. %v %v %v", p, p1, p2) 596 return 597 } 598 599 input := m.GetNthPrevPulseMock.expectationSeries[counter-1].input 600 testify_assert.Equal(m.t, *input, PulseTrackerMockGetNthPrevPulseInput{p, p1, p2}, "PulseTracker.GetNthPrevPulse got unexpected parameters") 601 602 result := m.GetNthPrevPulseMock.expectationSeries[counter-1].result 603 if result == nil { 604 m.t.Fatal("No results are set for the PulseTrackerMock.GetNthPrevPulse") 605 return 606 } 607 608 r = result.r 609 r1 = result.r1 610 611 return 612 } 613 614 if m.GetNthPrevPulseMock.mainExpectation != nil { 615 616 input := m.GetNthPrevPulseMock.mainExpectation.input 617 if input != nil { 618 testify_assert.Equal(m.t, *input, PulseTrackerMockGetNthPrevPulseInput{p, p1, p2}, "PulseTracker.GetNthPrevPulse got unexpected parameters") 619 } 620 621 result := m.GetNthPrevPulseMock.mainExpectation.result 622 if result == nil { 623 m.t.Fatal("No results are set for the PulseTrackerMock.GetNthPrevPulse") 624 } 625 626 r = result.r 627 r1 = result.r1 628 629 return 630 } 631 632 if m.GetNthPrevPulseFunc == nil { 633 m.t.Fatalf("Unexpected call to PulseTrackerMock.GetNthPrevPulse. %v %v %v", p, p1, p2) 634 return 635 } 636 637 return m.GetNthPrevPulseFunc(p, p1, p2) 638 } 639 640 //GetNthPrevPulseMinimockCounter returns a count of PulseTrackerMock.GetNthPrevPulseFunc invocations 641 func (m *PulseTrackerMock) GetNthPrevPulseMinimockCounter() uint64 { 642 return atomic.LoadUint64(&m.GetNthPrevPulseCounter) 643 } 644 645 //GetNthPrevPulseMinimockPreCounter returns the value of PulseTrackerMock.GetNthPrevPulse invocations 646 func (m *PulseTrackerMock) GetNthPrevPulseMinimockPreCounter() uint64 { 647 return atomic.LoadUint64(&m.GetNthPrevPulsePreCounter) 648 } 649 650 //GetNthPrevPulseFinished returns true if mock invocations count is ok 651 func (m *PulseTrackerMock) GetNthPrevPulseFinished() bool { 652 // if expectation series were set then invocations count should be equal to expectations count 653 if len(m.GetNthPrevPulseMock.expectationSeries) > 0 { 654 return atomic.LoadUint64(&m.GetNthPrevPulseCounter) == uint64(len(m.GetNthPrevPulseMock.expectationSeries)) 655 } 656 657 // if main expectation was set then invocations count should be greater than zero 658 if m.GetNthPrevPulseMock.mainExpectation != nil { 659 return atomic.LoadUint64(&m.GetNthPrevPulseCounter) > 0 660 } 661 662 // if func was set then invocations count should be greater than zero 663 if m.GetNthPrevPulseFunc != nil { 664 return atomic.LoadUint64(&m.GetNthPrevPulseCounter) > 0 665 } 666 667 return true 668 } 669 670 type mPulseTrackerMockGetPreviousPulse struct { 671 mock *PulseTrackerMock 672 mainExpectation *PulseTrackerMockGetPreviousPulseExpectation 673 expectationSeries []*PulseTrackerMockGetPreviousPulseExpectation 674 } 675 676 type PulseTrackerMockGetPreviousPulseExpectation struct { 677 input *PulseTrackerMockGetPreviousPulseInput 678 result *PulseTrackerMockGetPreviousPulseResult 679 } 680 681 type PulseTrackerMockGetPreviousPulseInput struct { 682 p context.Context 683 p1 core.PulseNumber 684 } 685 686 type PulseTrackerMockGetPreviousPulseResult struct { 687 r *Pulse 688 r1 error 689 } 690 691 //Expect specifies that invocation of PulseTracker.GetPreviousPulse is expected from 1 to Infinity times 692 func (m *mPulseTrackerMockGetPreviousPulse) Expect(p context.Context, p1 core.PulseNumber) *mPulseTrackerMockGetPreviousPulse { 693 m.mock.GetPreviousPulseFunc = nil 694 m.expectationSeries = nil 695 696 if m.mainExpectation == nil { 697 m.mainExpectation = &PulseTrackerMockGetPreviousPulseExpectation{} 698 } 699 m.mainExpectation.input = &PulseTrackerMockGetPreviousPulseInput{p, p1} 700 return m 701 } 702 703 //Return specifies results of invocation of PulseTracker.GetPreviousPulse 704 func (m *mPulseTrackerMockGetPreviousPulse) Return(r *Pulse, r1 error) *PulseTrackerMock { 705 m.mock.GetPreviousPulseFunc = nil 706 m.expectationSeries = nil 707 708 if m.mainExpectation == nil { 709 m.mainExpectation = &PulseTrackerMockGetPreviousPulseExpectation{} 710 } 711 m.mainExpectation.result = &PulseTrackerMockGetPreviousPulseResult{r, r1} 712 return m.mock 713 } 714 715 //ExpectOnce specifies that invocation of PulseTracker.GetPreviousPulse is expected once 716 func (m *mPulseTrackerMockGetPreviousPulse) ExpectOnce(p context.Context, p1 core.PulseNumber) *PulseTrackerMockGetPreviousPulseExpectation { 717 m.mock.GetPreviousPulseFunc = nil 718 m.mainExpectation = nil 719 720 expectation := &PulseTrackerMockGetPreviousPulseExpectation{} 721 expectation.input = &PulseTrackerMockGetPreviousPulseInput{p, p1} 722 m.expectationSeries = append(m.expectationSeries, expectation) 723 return expectation 724 } 725 726 func (e *PulseTrackerMockGetPreviousPulseExpectation) Return(r *Pulse, r1 error) { 727 e.result = &PulseTrackerMockGetPreviousPulseResult{r, r1} 728 } 729 730 //Set uses given function f as a mock of PulseTracker.GetPreviousPulse method 731 func (m *mPulseTrackerMockGetPreviousPulse) Set(f func(p context.Context, p1 core.PulseNumber) (r *Pulse, r1 error)) *PulseTrackerMock { 732 m.mainExpectation = nil 733 m.expectationSeries = nil 734 735 m.mock.GetPreviousPulseFunc = f 736 return m.mock 737 } 738 739 //GetPreviousPulse implements github.com/insolar/insolar/ledger/storage.PulseTracker interface 740 func (m *PulseTrackerMock) GetPreviousPulse(p context.Context, p1 core.PulseNumber) (r *Pulse, r1 error) { 741 counter := atomic.AddUint64(&m.GetPreviousPulsePreCounter, 1) 742 defer atomic.AddUint64(&m.GetPreviousPulseCounter, 1) 743 744 if len(m.GetPreviousPulseMock.expectationSeries) > 0 { 745 if counter > uint64(len(m.GetPreviousPulseMock.expectationSeries)) { 746 m.t.Fatalf("Unexpected call to PulseTrackerMock.GetPreviousPulse. %v %v", p, p1) 747 return 748 } 749 750 input := m.GetPreviousPulseMock.expectationSeries[counter-1].input 751 testify_assert.Equal(m.t, *input, PulseTrackerMockGetPreviousPulseInput{p, p1}, "PulseTracker.GetPreviousPulse got unexpected parameters") 752 753 result := m.GetPreviousPulseMock.expectationSeries[counter-1].result 754 if result == nil { 755 m.t.Fatal("No results are set for the PulseTrackerMock.GetPreviousPulse") 756 return 757 } 758 759 r = result.r 760 r1 = result.r1 761 762 return 763 } 764 765 if m.GetPreviousPulseMock.mainExpectation != nil { 766 767 input := m.GetPreviousPulseMock.mainExpectation.input 768 if input != nil { 769 testify_assert.Equal(m.t, *input, PulseTrackerMockGetPreviousPulseInput{p, p1}, "PulseTracker.GetPreviousPulse got unexpected parameters") 770 } 771 772 result := m.GetPreviousPulseMock.mainExpectation.result 773 if result == nil { 774 m.t.Fatal("No results are set for the PulseTrackerMock.GetPreviousPulse") 775 } 776 777 r = result.r 778 r1 = result.r1 779 780 return 781 } 782 783 if m.GetPreviousPulseFunc == nil { 784 m.t.Fatalf("Unexpected call to PulseTrackerMock.GetPreviousPulse. %v %v", p, p1) 785 return 786 } 787 788 return m.GetPreviousPulseFunc(p, p1) 789 } 790 791 //GetPreviousPulseMinimockCounter returns a count of PulseTrackerMock.GetPreviousPulseFunc invocations 792 func (m *PulseTrackerMock) GetPreviousPulseMinimockCounter() uint64 { 793 return atomic.LoadUint64(&m.GetPreviousPulseCounter) 794 } 795 796 //GetPreviousPulseMinimockPreCounter returns the value of PulseTrackerMock.GetPreviousPulse invocations 797 func (m *PulseTrackerMock) GetPreviousPulseMinimockPreCounter() uint64 { 798 return atomic.LoadUint64(&m.GetPreviousPulsePreCounter) 799 } 800 801 //GetPreviousPulseFinished returns true if mock invocations count is ok 802 func (m *PulseTrackerMock) GetPreviousPulseFinished() bool { 803 // if expectation series were set then invocations count should be equal to expectations count 804 if len(m.GetPreviousPulseMock.expectationSeries) > 0 { 805 return atomic.LoadUint64(&m.GetPreviousPulseCounter) == uint64(len(m.GetPreviousPulseMock.expectationSeries)) 806 } 807 808 // if main expectation was set then invocations count should be greater than zero 809 if m.GetPreviousPulseMock.mainExpectation != nil { 810 return atomic.LoadUint64(&m.GetPreviousPulseCounter) > 0 811 } 812 813 // if func was set then invocations count should be greater than zero 814 if m.GetPreviousPulseFunc != nil { 815 return atomic.LoadUint64(&m.GetPreviousPulseCounter) > 0 816 } 817 818 return true 819 } 820 821 type mPulseTrackerMockGetPulse struct { 822 mock *PulseTrackerMock 823 mainExpectation *PulseTrackerMockGetPulseExpectation 824 expectationSeries []*PulseTrackerMockGetPulseExpectation 825 } 826 827 type PulseTrackerMockGetPulseExpectation struct { 828 input *PulseTrackerMockGetPulseInput 829 result *PulseTrackerMockGetPulseResult 830 } 831 832 type PulseTrackerMockGetPulseInput struct { 833 p context.Context 834 p1 core.PulseNumber 835 } 836 837 type PulseTrackerMockGetPulseResult struct { 838 r *Pulse 839 r1 error 840 } 841 842 //Expect specifies that invocation of PulseTracker.GetPulse is expected from 1 to Infinity times 843 func (m *mPulseTrackerMockGetPulse) Expect(p context.Context, p1 core.PulseNumber) *mPulseTrackerMockGetPulse { 844 m.mock.GetPulseFunc = nil 845 m.expectationSeries = nil 846 847 if m.mainExpectation == nil { 848 m.mainExpectation = &PulseTrackerMockGetPulseExpectation{} 849 } 850 m.mainExpectation.input = &PulseTrackerMockGetPulseInput{p, p1} 851 return m 852 } 853 854 //Return specifies results of invocation of PulseTracker.GetPulse 855 func (m *mPulseTrackerMockGetPulse) Return(r *Pulse, r1 error) *PulseTrackerMock { 856 m.mock.GetPulseFunc = nil 857 m.expectationSeries = nil 858 859 if m.mainExpectation == nil { 860 m.mainExpectation = &PulseTrackerMockGetPulseExpectation{} 861 } 862 m.mainExpectation.result = &PulseTrackerMockGetPulseResult{r, r1} 863 return m.mock 864 } 865 866 //ExpectOnce specifies that invocation of PulseTracker.GetPulse is expected once 867 func (m *mPulseTrackerMockGetPulse) ExpectOnce(p context.Context, p1 core.PulseNumber) *PulseTrackerMockGetPulseExpectation { 868 m.mock.GetPulseFunc = nil 869 m.mainExpectation = nil 870 871 expectation := &PulseTrackerMockGetPulseExpectation{} 872 expectation.input = &PulseTrackerMockGetPulseInput{p, p1} 873 m.expectationSeries = append(m.expectationSeries, expectation) 874 return expectation 875 } 876 877 func (e *PulseTrackerMockGetPulseExpectation) Return(r *Pulse, r1 error) { 878 e.result = &PulseTrackerMockGetPulseResult{r, r1} 879 } 880 881 //Set uses given function f as a mock of PulseTracker.GetPulse method 882 func (m *mPulseTrackerMockGetPulse) Set(f func(p context.Context, p1 core.PulseNumber) (r *Pulse, r1 error)) *PulseTrackerMock { 883 m.mainExpectation = nil 884 m.expectationSeries = nil 885 886 m.mock.GetPulseFunc = f 887 return m.mock 888 } 889 890 //GetPulse implements github.com/insolar/insolar/ledger/storage.PulseTracker interface 891 func (m *PulseTrackerMock) GetPulse(p context.Context, p1 core.PulseNumber) (r *Pulse, r1 error) { 892 counter := atomic.AddUint64(&m.GetPulsePreCounter, 1) 893 defer atomic.AddUint64(&m.GetPulseCounter, 1) 894 895 if len(m.GetPulseMock.expectationSeries) > 0 { 896 if counter > uint64(len(m.GetPulseMock.expectationSeries)) { 897 m.t.Fatalf("Unexpected call to PulseTrackerMock.GetPulse. %v %v", p, p1) 898 return 899 } 900 901 input := m.GetPulseMock.expectationSeries[counter-1].input 902 testify_assert.Equal(m.t, *input, PulseTrackerMockGetPulseInput{p, p1}, "PulseTracker.GetPulse got unexpected parameters") 903 904 result := m.GetPulseMock.expectationSeries[counter-1].result 905 if result == nil { 906 m.t.Fatal("No results are set for the PulseTrackerMock.GetPulse") 907 return 908 } 909 910 r = result.r 911 r1 = result.r1 912 913 return 914 } 915 916 if m.GetPulseMock.mainExpectation != nil { 917 918 input := m.GetPulseMock.mainExpectation.input 919 if input != nil { 920 testify_assert.Equal(m.t, *input, PulseTrackerMockGetPulseInput{p, p1}, "PulseTracker.GetPulse got unexpected parameters") 921 } 922 923 result := m.GetPulseMock.mainExpectation.result 924 if result == nil { 925 m.t.Fatal("No results are set for the PulseTrackerMock.GetPulse") 926 } 927 928 r = result.r 929 r1 = result.r1 930 931 return 932 } 933 934 if m.GetPulseFunc == nil { 935 m.t.Fatalf("Unexpected call to PulseTrackerMock.GetPulse. %v %v", p, p1) 936 return 937 } 938 939 return m.GetPulseFunc(p, p1) 940 } 941 942 //GetPulseMinimockCounter returns a count of PulseTrackerMock.GetPulseFunc invocations 943 func (m *PulseTrackerMock) GetPulseMinimockCounter() uint64 { 944 return atomic.LoadUint64(&m.GetPulseCounter) 945 } 946 947 //GetPulseMinimockPreCounter returns the value of PulseTrackerMock.GetPulse invocations 948 func (m *PulseTrackerMock) GetPulseMinimockPreCounter() uint64 { 949 return atomic.LoadUint64(&m.GetPulsePreCounter) 950 } 951 952 //GetPulseFinished returns true if mock invocations count is ok 953 func (m *PulseTrackerMock) GetPulseFinished() bool { 954 // if expectation series were set then invocations count should be equal to expectations count 955 if len(m.GetPulseMock.expectationSeries) > 0 { 956 return atomic.LoadUint64(&m.GetPulseCounter) == uint64(len(m.GetPulseMock.expectationSeries)) 957 } 958 959 // if main expectation was set then invocations count should be greater than zero 960 if m.GetPulseMock.mainExpectation != nil { 961 return atomic.LoadUint64(&m.GetPulseCounter) > 0 962 } 963 964 // if func was set then invocations count should be greater than zero 965 if m.GetPulseFunc != nil { 966 return atomic.LoadUint64(&m.GetPulseCounter) > 0 967 } 968 969 return true 970 } 971 972 //ValidateCallCounters checks that all mocked methods of the interface have been called at least once 973 //Deprecated: please use MinimockFinish method or use Finish method of minimock.Controller 974 func (m *PulseTrackerMock) ValidateCallCounters() { 975 976 if !m.AddPulseFinished() { 977 m.t.Fatal("Expected call to PulseTrackerMock.AddPulse") 978 } 979 980 if !m.DeletePulseFinished() { 981 m.t.Fatal("Expected call to PulseTrackerMock.DeletePulse") 982 } 983 984 if !m.GetLatestPulseFinished() { 985 m.t.Fatal("Expected call to PulseTrackerMock.GetLatestPulse") 986 } 987 988 if !m.GetNthPrevPulseFinished() { 989 m.t.Fatal("Expected call to PulseTrackerMock.GetNthPrevPulse") 990 } 991 992 if !m.GetPreviousPulseFinished() { 993 m.t.Fatal("Expected call to PulseTrackerMock.GetPreviousPulse") 994 } 995 996 if !m.GetPulseFinished() { 997 m.t.Fatal("Expected call to PulseTrackerMock.GetPulse") 998 } 999 1000 } 1001 1002 //CheckMocksCalled checks that all mocked methods of the interface have been called at least once 1003 //Deprecated: please use MinimockFinish method or use Finish method of minimock.Controller 1004 func (m *PulseTrackerMock) CheckMocksCalled() { 1005 m.Finish() 1006 } 1007 1008 //Finish checks that all mocked methods of the interface have been called at least once 1009 //Deprecated: please use MinimockFinish or use Finish method of minimock.Controller 1010 func (m *PulseTrackerMock) Finish() { 1011 m.MinimockFinish() 1012 } 1013 1014 //MinimockFinish checks that all mocked methods of the interface have been called at least once 1015 func (m *PulseTrackerMock) MinimockFinish() { 1016 1017 if !m.AddPulseFinished() { 1018 m.t.Fatal("Expected call to PulseTrackerMock.AddPulse") 1019 } 1020 1021 if !m.DeletePulseFinished() { 1022 m.t.Fatal("Expected call to PulseTrackerMock.DeletePulse") 1023 } 1024 1025 if !m.GetLatestPulseFinished() { 1026 m.t.Fatal("Expected call to PulseTrackerMock.GetLatestPulse") 1027 } 1028 1029 if !m.GetNthPrevPulseFinished() { 1030 m.t.Fatal("Expected call to PulseTrackerMock.GetNthPrevPulse") 1031 } 1032 1033 if !m.GetPreviousPulseFinished() { 1034 m.t.Fatal("Expected call to PulseTrackerMock.GetPreviousPulse") 1035 } 1036 1037 if !m.GetPulseFinished() { 1038 m.t.Fatal("Expected call to PulseTrackerMock.GetPulse") 1039 } 1040 1041 } 1042 1043 //Wait waits for all mocked methods to be called at least once 1044 //Deprecated: please use MinimockWait or use Wait method of minimock.Controller 1045 func (m *PulseTrackerMock) Wait(timeout time.Duration) { 1046 m.MinimockWait(timeout) 1047 } 1048 1049 //MinimockWait waits for all mocked methods to be called at least once 1050 //this method is called by minimock.Controller 1051 func (m *PulseTrackerMock) MinimockWait(timeout time.Duration) { 1052 timeoutCh := time.After(timeout) 1053 for { 1054 ok := true 1055 ok = ok && m.AddPulseFinished() 1056 ok = ok && m.DeletePulseFinished() 1057 ok = ok && m.GetLatestPulseFinished() 1058 ok = ok && m.GetNthPrevPulseFinished() 1059 ok = ok && m.GetPreviousPulseFinished() 1060 ok = ok && m.GetPulseFinished() 1061 1062 if ok { 1063 return 1064 } 1065 1066 select { 1067 case <-timeoutCh: 1068 1069 if !m.AddPulseFinished() { 1070 m.t.Error("Expected call to PulseTrackerMock.AddPulse") 1071 } 1072 1073 if !m.DeletePulseFinished() { 1074 m.t.Error("Expected call to PulseTrackerMock.DeletePulse") 1075 } 1076 1077 if !m.GetLatestPulseFinished() { 1078 m.t.Error("Expected call to PulseTrackerMock.GetLatestPulse") 1079 } 1080 1081 if !m.GetNthPrevPulseFinished() { 1082 m.t.Error("Expected call to PulseTrackerMock.GetNthPrevPulse") 1083 } 1084 1085 if !m.GetPreviousPulseFinished() { 1086 m.t.Error("Expected call to PulseTrackerMock.GetPreviousPulse") 1087 } 1088 1089 if !m.GetPulseFinished() { 1090 m.t.Error("Expected call to PulseTrackerMock.GetPulse") 1091 } 1092 1093 m.t.Fatalf("Some mocks were not called on time: %s", timeout) 1094 return 1095 default: 1096 time.Sleep(time.Millisecond) 1097 } 1098 } 1099 } 1100 1101 //AllMocksCalled returns true if all mocked methods were called before the execution of AllMocksCalled, 1102 //it can be used with assert/require, i.e. assert.True(mock.AllMocksCalled()) 1103 func (m *PulseTrackerMock) AllMocksCalled() bool { 1104 1105 if !m.AddPulseFinished() { 1106 return false 1107 } 1108 1109 if !m.DeletePulseFinished() { 1110 return false 1111 } 1112 1113 if !m.GetLatestPulseFinished() { 1114 return false 1115 } 1116 1117 if !m.GetNthPrevPulseFinished() { 1118 return false 1119 } 1120 1121 if !m.GetPreviousPulseFinished() { 1122 return false 1123 } 1124 1125 if !m.GetPulseFinished() { 1126 return false 1127 } 1128 1129 return true 1130 }