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