github.com/insolar/vanilla@v0.0.0-20201023172447-248fdf805322/cryptkit/key_store_factory_mock.go (about)

     1  package cryptkit
     2  
     3  // Code generated by http://github.com/gojuno/minimock (dev). DO NOT EDIT.
     4  
     5  import (
     6  	"sync"
     7  	mm_atomic "sync/atomic"
     8  	mm_time "time"
     9  
    10  	"github.com/gojuno/minimock/v3"
    11  )
    12  
    13  // KeyStoreFactoryMock implements KeyStoreFactory
    14  type KeyStoreFactoryMock struct {
    15  	t minimock.Tester
    16  
    17  	funcCreatePublicKeyStore          func(s1 SigningKeyHolder) (p1 PublicKeyStore)
    18  	inspectFuncCreatePublicKeyStore   func(s1 SigningKeyHolder)
    19  	afterCreatePublicKeyStoreCounter  uint64
    20  	beforeCreatePublicKeyStoreCounter uint64
    21  	CreatePublicKeyStoreMock          mKeyStoreFactoryMockCreatePublicKeyStore
    22  }
    23  
    24  // NewKeyStoreFactoryMock returns a mock for KeyStoreFactory
    25  func NewKeyStoreFactoryMock(t minimock.Tester) *KeyStoreFactoryMock {
    26  	m := &KeyStoreFactoryMock{t: t}
    27  	if controller, ok := t.(minimock.MockController); ok {
    28  		controller.RegisterMocker(m)
    29  	}
    30  
    31  	m.CreatePublicKeyStoreMock = mKeyStoreFactoryMockCreatePublicKeyStore{mock: m}
    32  	m.CreatePublicKeyStoreMock.callArgs = []*KeyStoreFactoryMockCreatePublicKeyStoreParams{}
    33  
    34  	return m
    35  }
    36  
    37  type mKeyStoreFactoryMockCreatePublicKeyStore struct {
    38  	mock               *KeyStoreFactoryMock
    39  	defaultExpectation *KeyStoreFactoryMockCreatePublicKeyStoreExpectation
    40  	expectations       []*KeyStoreFactoryMockCreatePublicKeyStoreExpectation
    41  
    42  	callArgs []*KeyStoreFactoryMockCreatePublicKeyStoreParams
    43  	mutex    sync.RWMutex
    44  }
    45  
    46  // KeyStoreFactoryMockCreatePublicKeyStoreExpectation specifies expectation struct of the KeyStoreFactory.CreatePublicKeyStore
    47  type KeyStoreFactoryMockCreatePublicKeyStoreExpectation struct {
    48  	mock    *KeyStoreFactoryMock
    49  	params  *KeyStoreFactoryMockCreatePublicKeyStoreParams
    50  	results *KeyStoreFactoryMockCreatePublicKeyStoreResults
    51  	Counter uint64
    52  }
    53  
    54  // KeyStoreFactoryMockCreatePublicKeyStoreParams contains parameters of the KeyStoreFactory.CreatePublicKeyStore
    55  type KeyStoreFactoryMockCreatePublicKeyStoreParams struct {
    56  	s1 SigningKeyHolder
    57  }
    58  
    59  // KeyStoreFactoryMockCreatePublicKeyStoreResults contains results of the KeyStoreFactory.CreatePublicKeyStore
    60  type KeyStoreFactoryMockCreatePublicKeyStoreResults struct {
    61  	p1 PublicKeyStore
    62  }
    63  
    64  // Expect sets up expected params for KeyStoreFactory.CreatePublicKeyStore
    65  func (mmCreatePublicKeyStore *mKeyStoreFactoryMockCreatePublicKeyStore) Expect(s1 SigningKeyHolder) *mKeyStoreFactoryMockCreatePublicKeyStore {
    66  	if mmCreatePublicKeyStore.mock.funcCreatePublicKeyStore != nil {
    67  		mmCreatePublicKeyStore.mock.t.Fatalf("KeyStoreFactoryMock.CreatePublicKeyStore mock is already set by Set")
    68  	}
    69  
    70  	if mmCreatePublicKeyStore.defaultExpectation == nil {
    71  		mmCreatePublicKeyStore.defaultExpectation = &KeyStoreFactoryMockCreatePublicKeyStoreExpectation{}
    72  	}
    73  
    74  	mmCreatePublicKeyStore.defaultExpectation.params = &KeyStoreFactoryMockCreatePublicKeyStoreParams{s1}
    75  	for _, e := range mmCreatePublicKeyStore.expectations {
    76  		if minimock.Equal(e.params, mmCreatePublicKeyStore.defaultExpectation.params) {
    77  			mmCreatePublicKeyStore.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmCreatePublicKeyStore.defaultExpectation.params)
    78  		}
    79  	}
    80  
    81  	return mmCreatePublicKeyStore
    82  }
    83  
    84  // Inspect accepts an inspector function that has same arguments as the KeyStoreFactory.CreatePublicKeyStore
    85  func (mmCreatePublicKeyStore *mKeyStoreFactoryMockCreatePublicKeyStore) Inspect(f func(s1 SigningKeyHolder)) *mKeyStoreFactoryMockCreatePublicKeyStore {
    86  	if mmCreatePublicKeyStore.mock.inspectFuncCreatePublicKeyStore != nil {
    87  		mmCreatePublicKeyStore.mock.t.Fatalf("Inspect function is already set for KeyStoreFactoryMock.CreatePublicKeyStore")
    88  	}
    89  
    90  	mmCreatePublicKeyStore.mock.inspectFuncCreatePublicKeyStore = f
    91  
    92  	return mmCreatePublicKeyStore
    93  }
    94  
    95  // Return sets up results that will be returned by KeyStoreFactory.CreatePublicKeyStore
    96  func (mmCreatePublicKeyStore *mKeyStoreFactoryMockCreatePublicKeyStore) Return(p1 PublicKeyStore) *KeyStoreFactoryMock {
    97  	if mmCreatePublicKeyStore.mock.funcCreatePublicKeyStore != nil {
    98  		mmCreatePublicKeyStore.mock.t.Fatalf("KeyStoreFactoryMock.CreatePublicKeyStore mock is already set by Set")
    99  	}
   100  
   101  	if mmCreatePublicKeyStore.defaultExpectation == nil {
   102  		mmCreatePublicKeyStore.defaultExpectation = &KeyStoreFactoryMockCreatePublicKeyStoreExpectation{mock: mmCreatePublicKeyStore.mock}
   103  	}
   104  	mmCreatePublicKeyStore.defaultExpectation.results = &KeyStoreFactoryMockCreatePublicKeyStoreResults{p1}
   105  	return mmCreatePublicKeyStore.mock
   106  }
   107  
   108  //Set uses given function f to mock the KeyStoreFactory.CreatePublicKeyStore method
   109  func (mmCreatePublicKeyStore *mKeyStoreFactoryMockCreatePublicKeyStore) Set(f func(s1 SigningKeyHolder) (p1 PublicKeyStore)) *KeyStoreFactoryMock {
   110  	if mmCreatePublicKeyStore.defaultExpectation != nil {
   111  		mmCreatePublicKeyStore.mock.t.Fatalf("Default expectation is already set for the KeyStoreFactory.CreatePublicKeyStore method")
   112  	}
   113  
   114  	if len(mmCreatePublicKeyStore.expectations) > 0 {
   115  		mmCreatePublicKeyStore.mock.t.Fatalf("Some expectations are already set for the KeyStoreFactory.CreatePublicKeyStore method")
   116  	}
   117  
   118  	mmCreatePublicKeyStore.mock.funcCreatePublicKeyStore = f
   119  	return mmCreatePublicKeyStore.mock
   120  }
   121  
   122  // When sets expectation for the KeyStoreFactory.CreatePublicKeyStore which will trigger the result defined by the following
   123  // Then helper
   124  func (mmCreatePublicKeyStore *mKeyStoreFactoryMockCreatePublicKeyStore) When(s1 SigningKeyHolder) *KeyStoreFactoryMockCreatePublicKeyStoreExpectation {
   125  	if mmCreatePublicKeyStore.mock.funcCreatePublicKeyStore != nil {
   126  		mmCreatePublicKeyStore.mock.t.Fatalf("KeyStoreFactoryMock.CreatePublicKeyStore mock is already set by Set")
   127  	}
   128  
   129  	expectation := &KeyStoreFactoryMockCreatePublicKeyStoreExpectation{
   130  		mock:   mmCreatePublicKeyStore.mock,
   131  		params: &KeyStoreFactoryMockCreatePublicKeyStoreParams{s1},
   132  	}
   133  	mmCreatePublicKeyStore.expectations = append(mmCreatePublicKeyStore.expectations, expectation)
   134  	return expectation
   135  }
   136  
   137  // Then sets up KeyStoreFactory.CreatePublicKeyStore return parameters for the expectation previously defined by the When method
   138  func (e *KeyStoreFactoryMockCreatePublicKeyStoreExpectation) Then(p1 PublicKeyStore) *KeyStoreFactoryMock {
   139  	e.results = &KeyStoreFactoryMockCreatePublicKeyStoreResults{p1}
   140  	return e.mock
   141  }
   142  
   143  // CreatePublicKeyStore implements KeyStoreFactory
   144  func (mmCreatePublicKeyStore *KeyStoreFactoryMock) CreatePublicKeyStore(s1 SigningKeyHolder) (p1 PublicKeyStore) {
   145  	mm_atomic.AddUint64(&mmCreatePublicKeyStore.beforeCreatePublicKeyStoreCounter, 1)
   146  	defer mm_atomic.AddUint64(&mmCreatePublicKeyStore.afterCreatePublicKeyStoreCounter, 1)
   147  
   148  	if mmCreatePublicKeyStore.inspectFuncCreatePublicKeyStore != nil {
   149  		mmCreatePublicKeyStore.inspectFuncCreatePublicKeyStore(s1)
   150  	}
   151  
   152  	mm_params := &KeyStoreFactoryMockCreatePublicKeyStoreParams{s1}
   153  
   154  	// Record call args
   155  	mmCreatePublicKeyStore.CreatePublicKeyStoreMock.mutex.Lock()
   156  	mmCreatePublicKeyStore.CreatePublicKeyStoreMock.callArgs = append(mmCreatePublicKeyStore.CreatePublicKeyStoreMock.callArgs, mm_params)
   157  	mmCreatePublicKeyStore.CreatePublicKeyStoreMock.mutex.Unlock()
   158  
   159  	for _, e := range mmCreatePublicKeyStore.CreatePublicKeyStoreMock.expectations {
   160  		if minimock.Equal(e.params, mm_params) {
   161  			mm_atomic.AddUint64(&e.Counter, 1)
   162  			return e.results.p1
   163  		}
   164  	}
   165  
   166  	if mmCreatePublicKeyStore.CreatePublicKeyStoreMock.defaultExpectation != nil {
   167  		mm_atomic.AddUint64(&mmCreatePublicKeyStore.CreatePublicKeyStoreMock.defaultExpectation.Counter, 1)
   168  		mm_want := mmCreatePublicKeyStore.CreatePublicKeyStoreMock.defaultExpectation.params
   169  		mm_got := KeyStoreFactoryMockCreatePublicKeyStoreParams{s1}
   170  		if mm_want != nil && !minimock.Equal(*mm_want, mm_got) {
   171  			mmCreatePublicKeyStore.t.Errorf("KeyStoreFactoryMock.CreatePublicKeyStore got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got))
   172  		}
   173  
   174  		mm_results := mmCreatePublicKeyStore.CreatePublicKeyStoreMock.defaultExpectation.results
   175  		if mm_results == nil {
   176  			mmCreatePublicKeyStore.t.Fatal("No results are set for the KeyStoreFactoryMock.CreatePublicKeyStore")
   177  		}
   178  		return (*mm_results).p1
   179  	}
   180  	if mmCreatePublicKeyStore.funcCreatePublicKeyStore != nil {
   181  		return mmCreatePublicKeyStore.funcCreatePublicKeyStore(s1)
   182  	}
   183  	mmCreatePublicKeyStore.t.Fatalf("Unexpected call to KeyStoreFactoryMock.CreatePublicKeyStore. %v", s1)
   184  	return
   185  }
   186  
   187  // CreatePublicKeyStoreAfterCounter returns a count of finished KeyStoreFactoryMock.CreatePublicKeyStore invocations
   188  func (mmCreatePublicKeyStore *KeyStoreFactoryMock) CreatePublicKeyStoreAfterCounter() uint64 {
   189  	return mm_atomic.LoadUint64(&mmCreatePublicKeyStore.afterCreatePublicKeyStoreCounter)
   190  }
   191  
   192  // CreatePublicKeyStoreBeforeCounter returns a count of KeyStoreFactoryMock.CreatePublicKeyStore invocations
   193  func (mmCreatePublicKeyStore *KeyStoreFactoryMock) CreatePublicKeyStoreBeforeCounter() uint64 {
   194  	return mm_atomic.LoadUint64(&mmCreatePublicKeyStore.beforeCreatePublicKeyStoreCounter)
   195  }
   196  
   197  // Calls returns a list of arguments used in each call to KeyStoreFactoryMock.CreatePublicKeyStore.
   198  // The list is in the same order as the calls were made (i.e. recent calls have a higher index)
   199  func (mmCreatePublicKeyStore *mKeyStoreFactoryMockCreatePublicKeyStore) Calls() []*KeyStoreFactoryMockCreatePublicKeyStoreParams {
   200  	mmCreatePublicKeyStore.mutex.RLock()
   201  
   202  	argCopy := make([]*KeyStoreFactoryMockCreatePublicKeyStoreParams, len(mmCreatePublicKeyStore.callArgs))
   203  	copy(argCopy, mmCreatePublicKeyStore.callArgs)
   204  
   205  	mmCreatePublicKeyStore.mutex.RUnlock()
   206  
   207  	return argCopy
   208  }
   209  
   210  // MinimockCreatePublicKeyStoreDone returns true if the count of the CreatePublicKeyStore invocations corresponds
   211  // the number of defined expectations
   212  func (m *KeyStoreFactoryMock) MinimockCreatePublicKeyStoreDone() bool {
   213  	for _, e := range m.CreatePublicKeyStoreMock.expectations {
   214  		if mm_atomic.LoadUint64(&e.Counter) < 1 {
   215  			return false
   216  		}
   217  	}
   218  
   219  	// if default expectation was set then invocations count should be greater than zero
   220  	if m.CreatePublicKeyStoreMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterCreatePublicKeyStoreCounter) < 1 {
   221  		return false
   222  	}
   223  	// if func was set then invocations count should be greater than zero
   224  	if m.funcCreatePublicKeyStore != nil && mm_atomic.LoadUint64(&m.afterCreatePublicKeyStoreCounter) < 1 {
   225  		return false
   226  	}
   227  	return true
   228  }
   229  
   230  // MinimockCreatePublicKeyStoreInspect logs each unmet expectation
   231  func (m *KeyStoreFactoryMock) MinimockCreatePublicKeyStoreInspect() {
   232  	for _, e := range m.CreatePublicKeyStoreMock.expectations {
   233  		if mm_atomic.LoadUint64(&e.Counter) < 1 {
   234  			m.t.Errorf("Expected call to KeyStoreFactoryMock.CreatePublicKeyStore with params: %#v", *e.params)
   235  		}
   236  	}
   237  
   238  	// if default expectation was set then invocations count should be greater than zero
   239  	if m.CreatePublicKeyStoreMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterCreatePublicKeyStoreCounter) < 1 {
   240  		if m.CreatePublicKeyStoreMock.defaultExpectation.params == nil {
   241  			m.t.Error("Expected call to KeyStoreFactoryMock.CreatePublicKeyStore")
   242  		} else {
   243  			m.t.Errorf("Expected call to KeyStoreFactoryMock.CreatePublicKeyStore with params: %#v", *m.CreatePublicKeyStoreMock.defaultExpectation.params)
   244  		}
   245  	}
   246  	// if func was set then invocations count should be greater than zero
   247  	if m.funcCreatePublicKeyStore != nil && mm_atomic.LoadUint64(&m.afterCreatePublicKeyStoreCounter) < 1 {
   248  		m.t.Error("Expected call to KeyStoreFactoryMock.CreatePublicKeyStore")
   249  	}
   250  }
   251  
   252  // MinimockFinish checks that all mocked methods have been called the expected number of times
   253  func (m *KeyStoreFactoryMock) MinimockFinish() {
   254  	if !m.minimockDone() {
   255  		m.MinimockCreatePublicKeyStoreInspect()
   256  		m.t.FailNow()
   257  	}
   258  }
   259  
   260  // MinimockWait waits for all mocked methods to be called the expected number of times
   261  func (m *KeyStoreFactoryMock) MinimockWait(timeout mm_time.Duration) {
   262  	timeoutCh := mm_time.After(timeout)
   263  	for {
   264  		if m.minimockDone() {
   265  			return
   266  		}
   267  		select {
   268  		case <-timeoutCh:
   269  			m.MinimockFinish()
   270  			return
   271  		case <-mm_time.After(10 * mm_time.Millisecond):
   272  		}
   273  	}
   274  }
   275  
   276  func (m *KeyStoreFactoryMock) minimockDone() bool {
   277  	done := true
   278  	return done &&
   279  		m.MinimockCreatePublicKeyStoreDone()
   280  }