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  }