github.com/braveheart12/insolar-09-08-19@v0.8.7/ledger/storage/jet_storage_mock.go (about)

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