github.com/braveheart12/insolar-09-08-19@v0.8.7/pulsar/pulsartestutils/pulsar_storage_mock.go (about)

     1  /*
     2   *    Copyright 2019 Insolar Technologies
     3   *
     4   *    Licensed under the Apache License, Version 2.0 (the "License");
     5   *    you may not use this file except in compliance with the License.
     6   *    You may obtain a copy of the License at
     7   *
     8   *        http://www.apache.org/licenses/LICENSE-2.0
     9   *
    10   *    Unless required by applicable law or agreed to in writing, software
    11   *    distributed under the License is distributed on an "AS IS" BASIS,
    12   *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13   *    See the License for the specific language governing permissions and
    14   *    limitations under the License.
    15   */
    16  
    17  package pulsartestutils
    18  
    19  /*
    20  DO NOT EDIT!
    21  This code was generated automatically using github.com/gojuno/minimock v1.9
    22  The original interface "PulsarStorage" can be found in github.com/insolar/insolar/pulsar/storage
    23  */
    24  import (
    25  	"sync/atomic"
    26  	"time"
    27  
    28  	"github.com/gojuno/minimock"
    29  	"github.com/insolar/insolar/core"
    30  
    31  	testify_assert "github.com/stretchr/testify/assert"
    32  )
    33  
    34  //PulsarStorageMock implements github.com/insolar/insolar/pulsar/storage.PulsarStorage
    35  type PulsarStorageMock struct {
    36  	t minimock.Tester
    37  
    38  	CloseFunc       func() (r error)
    39  	CloseCounter    uint64
    40  	ClosePreCounter uint64
    41  	CloseMock       mPulsarStorageMockClose
    42  
    43  	GetLastPulseFunc       func() (r *core.Pulse, r1 error)
    44  	GetLastPulseCounter    uint64
    45  	GetLastPulsePreCounter uint64
    46  	GetLastPulseMock       mPulsarStorageMockGetLastPulse
    47  
    48  	SavePulseFunc       func(p *core.Pulse) (r error)
    49  	SavePulseCounter    uint64
    50  	SavePulsePreCounter uint64
    51  	SavePulseMock       mPulsarStorageMockSavePulse
    52  
    53  	SetLastPulseFunc       func(p *core.Pulse) (r error)
    54  	SetLastPulseCounter    uint64
    55  	SetLastPulsePreCounter uint64
    56  	SetLastPulseMock       mPulsarStorageMockSetLastPulse
    57  }
    58  
    59  //NewPulsarStorageMock returns a mock for github.com/insolar/insolar/pulsar/storage.PulsarStorage
    60  func NewPulsarStorageMock(t minimock.Tester) *PulsarStorageMock {
    61  	m := &PulsarStorageMock{t: t}
    62  
    63  	if controller, ok := t.(minimock.MockController); ok {
    64  		controller.RegisterMocker(m)
    65  	}
    66  
    67  	m.CloseMock = mPulsarStorageMockClose{mock: m}
    68  	m.GetLastPulseMock = mPulsarStorageMockGetLastPulse{mock: m}
    69  	m.SavePulseMock = mPulsarStorageMockSavePulse{mock: m}
    70  	m.SetLastPulseMock = mPulsarStorageMockSetLastPulse{mock: m}
    71  
    72  	return m
    73  }
    74  
    75  type mPulsarStorageMockClose struct {
    76  	mock *PulsarStorageMock
    77  }
    78  
    79  //Return sets up a mock for PulsarStorage.Close to return Return's arguments
    80  func (m *mPulsarStorageMockClose) Return(r error) *PulsarStorageMock {
    81  	m.mock.CloseFunc = func() error {
    82  		return r
    83  	}
    84  	return m.mock
    85  }
    86  
    87  //Set uses given function f as a mock of PulsarStorage.Close method
    88  func (m *mPulsarStorageMockClose) Set(f func() (r error)) *PulsarStorageMock {
    89  	m.mock.CloseFunc = f
    90  
    91  	return m.mock
    92  }
    93  
    94  //Close implements github.com/insolar/insolar/pulsar/storage.PulsarStorage interface
    95  func (m *PulsarStorageMock) Close() (r error) {
    96  	atomic.AddUint64(&m.ClosePreCounter, 1)
    97  	defer atomic.AddUint64(&m.CloseCounter, 1)
    98  
    99  	if m.CloseFunc == nil {
   100  		m.t.Fatal("Unexpected call to PulsarStorageMock.Close")
   101  		return
   102  	}
   103  
   104  	return m.CloseFunc()
   105  }
   106  
   107  //CloseMinimockCounter returns a count of PulsarStorageMock.CloseFunc invocations
   108  func (m *PulsarStorageMock) CloseMinimockCounter() uint64 {
   109  	return atomic.LoadUint64(&m.CloseCounter)
   110  }
   111  
   112  //CloseMinimockPreCounter returns the value of PulsarStorageMock.Close invocations
   113  func (m *PulsarStorageMock) CloseMinimockPreCounter() uint64 {
   114  	return atomic.LoadUint64(&m.ClosePreCounter)
   115  }
   116  
   117  type mPulsarStorageMockGetLastPulse struct {
   118  	mock *PulsarStorageMock
   119  }
   120  
   121  //Return sets up a mock for PulsarStorage.GetLastPulse to return Return's arguments
   122  func (m *mPulsarStorageMockGetLastPulse) Return(r *core.Pulse, r1 error) *PulsarStorageMock {
   123  	m.mock.GetLastPulseFunc = func() (*core.Pulse, error) {
   124  		return r, r1
   125  	}
   126  	return m.mock
   127  }
   128  
   129  //Set uses given function f as a mock of PulsarStorage.GetLastPulse method
   130  func (m *mPulsarStorageMockGetLastPulse) Set(f func() (r *core.Pulse, r1 error)) *PulsarStorageMock {
   131  	m.mock.GetLastPulseFunc = f
   132  
   133  	return m.mock
   134  }
   135  
   136  //GetLastPulse implements github.com/insolar/insolar/pulsar/storage.PulsarStorage interface
   137  func (m *PulsarStorageMock) GetLastPulse() (r *core.Pulse, r1 error) {
   138  	atomic.AddUint64(&m.GetLastPulsePreCounter, 1)
   139  	defer atomic.AddUint64(&m.GetLastPulseCounter, 1)
   140  
   141  	if m.GetLastPulseFunc == nil {
   142  		m.t.Fatal("Unexpected call to PulsarStorageMock.GetLastPulse")
   143  		return
   144  	}
   145  
   146  	return m.GetLastPulseFunc()
   147  }
   148  
   149  //GetLastPulseMinimockCounter returns a count of PulsarStorageMock.GetLastPulseFunc invocations
   150  func (m *PulsarStorageMock) GetLastPulseMinimockCounter() uint64 {
   151  	return atomic.LoadUint64(&m.GetLastPulseCounter)
   152  }
   153  
   154  //GetLastPulseMinimockPreCounter returns the value of PulsarStorageMock.GetLastPulse invocations
   155  func (m *PulsarStorageMock) GetLastPulseMinimockPreCounter() uint64 {
   156  	return atomic.LoadUint64(&m.GetLastPulsePreCounter)
   157  }
   158  
   159  type mPulsarStorageMockSavePulse struct {
   160  	mock             *PulsarStorageMock
   161  	mockExpectations *PulsarStorageMockSavePulseParams
   162  }
   163  
   164  //PulsarStorageMockSavePulseParams represents input parameters of the PulsarStorage.SavePulse
   165  type PulsarStorageMockSavePulseParams struct {
   166  	p *core.Pulse
   167  }
   168  
   169  //Expect sets up expected params for the PulsarStorage.SavePulse
   170  func (m *mPulsarStorageMockSavePulse) Expect(p *core.Pulse) *mPulsarStorageMockSavePulse {
   171  	m.mockExpectations = &PulsarStorageMockSavePulseParams{p}
   172  	return m
   173  }
   174  
   175  //Return sets up a mock for PulsarStorage.SavePulse to return Return's arguments
   176  func (m *mPulsarStorageMockSavePulse) Return(r error) *PulsarStorageMock {
   177  	m.mock.SavePulseFunc = func(p *core.Pulse) error {
   178  		return r
   179  	}
   180  	return m.mock
   181  }
   182  
   183  //Set uses given function f as a mock of PulsarStorage.SavePulse method
   184  func (m *mPulsarStorageMockSavePulse) Set(f func(p *core.Pulse) (r error)) *PulsarStorageMock {
   185  	m.mock.SavePulseFunc = f
   186  	m.mockExpectations = nil
   187  	return m.mock
   188  }
   189  
   190  //SavePulse implements github.com/insolar/insolar/pulsar/storage.PulsarStorage interface
   191  func (m *PulsarStorageMock) SavePulse(p *core.Pulse) (r error) {
   192  	atomic.AddUint64(&m.SavePulsePreCounter, 1)
   193  	defer atomic.AddUint64(&m.SavePulseCounter, 1)
   194  
   195  	if m.SavePulseMock.mockExpectations != nil {
   196  		testify_assert.Equal(m.t, *m.SavePulseMock.mockExpectations, PulsarStorageMockSavePulseParams{p},
   197  			"PulsarStorage.SavePulse got unexpected parameters")
   198  
   199  		if m.SavePulseFunc == nil {
   200  
   201  			m.t.Fatal("No results are set for the PulsarStorageMock.SavePulse")
   202  
   203  			return
   204  		}
   205  	}
   206  
   207  	if m.SavePulseFunc == nil {
   208  		m.t.Fatal("Unexpected call to PulsarStorageMock.SavePulse")
   209  		return
   210  	}
   211  
   212  	return m.SavePulseFunc(p)
   213  }
   214  
   215  //SavePulseMinimockCounter returns a count of PulsarStorageMock.SavePulseFunc invocations
   216  func (m *PulsarStorageMock) SavePulseMinimockCounter() uint64 {
   217  	return atomic.LoadUint64(&m.SavePulseCounter)
   218  }
   219  
   220  //SavePulseMinimockPreCounter returns the value of PulsarStorageMock.SavePulse invocations
   221  func (m *PulsarStorageMock) SavePulseMinimockPreCounter() uint64 {
   222  	return atomic.LoadUint64(&m.SavePulsePreCounter)
   223  }
   224  
   225  type mPulsarStorageMockSetLastPulse struct {
   226  	mock             *PulsarStorageMock
   227  	mockExpectations *PulsarStorageMockSetLastPulseParams
   228  }
   229  
   230  //PulsarStorageMockSetLastPulseParams represents input parameters of the PulsarStorage.SetLastPulse
   231  type PulsarStorageMockSetLastPulseParams struct {
   232  	p *core.Pulse
   233  }
   234  
   235  //Expect sets up expected params for the PulsarStorage.SetLastPulse
   236  func (m *mPulsarStorageMockSetLastPulse) Expect(p *core.Pulse) *mPulsarStorageMockSetLastPulse {
   237  	m.mockExpectations = &PulsarStorageMockSetLastPulseParams{p}
   238  	return m
   239  }
   240  
   241  //Return sets up a mock for PulsarStorage.SetLastPulse to return Return's arguments
   242  func (m *mPulsarStorageMockSetLastPulse) Return(r error) *PulsarStorageMock {
   243  	m.mock.SetLastPulseFunc = func(p *core.Pulse) error {
   244  		return r
   245  	}
   246  	return m.mock
   247  }
   248  
   249  //Set uses given function f as a mock of PulsarStorage.SetLastPulse method
   250  func (m *mPulsarStorageMockSetLastPulse) Set(f func(p *core.Pulse) (r error)) *PulsarStorageMock {
   251  	m.mock.SetLastPulseFunc = f
   252  	m.mockExpectations = nil
   253  	return m.mock
   254  }
   255  
   256  //SetLastPulse implements github.com/insolar/insolar/pulsar/storage.PulsarStorage interface
   257  func (m *PulsarStorageMock) SetLastPulse(p *core.Pulse) (r error) {
   258  	atomic.AddUint64(&m.SetLastPulsePreCounter, 1)
   259  	defer atomic.AddUint64(&m.SetLastPulseCounter, 1)
   260  
   261  	if m.SetLastPulseMock.mockExpectations != nil {
   262  		testify_assert.Equal(m.t, *m.SetLastPulseMock.mockExpectations, PulsarStorageMockSetLastPulseParams{p},
   263  			"PulsarStorage.SetLastPulse got unexpected parameters")
   264  
   265  		if m.SetLastPulseFunc == nil {
   266  
   267  			m.t.Fatal("No results are set for the PulsarStorageMock.SetLastPulse")
   268  
   269  			return
   270  		}
   271  	}
   272  
   273  	if m.SetLastPulseFunc == nil {
   274  		m.t.Fatal("Unexpected call to PulsarStorageMock.SetLastPulse")
   275  		return
   276  	}
   277  
   278  	return m.SetLastPulseFunc(p)
   279  }
   280  
   281  //SetLastPulseMinimockCounter returns a count of PulsarStorageMock.SetLastPulseFunc invocations
   282  func (m *PulsarStorageMock) SetLastPulseMinimockCounter() uint64 {
   283  	return atomic.LoadUint64(&m.SetLastPulseCounter)
   284  }
   285  
   286  //SetLastPulseMinimockPreCounter returns the value of PulsarStorageMock.SetLastPulse invocations
   287  func (m *PulsarStorageMock) SetLastPulseMinimockPreCounter() uint64 {
   288  	return atomic.LoadUint64(&m.SetLastPulsePreCounter)
   289  }
   290  
   291  //ValidateCallCounters checks that all mocked methods of the interface have been called at least once
   292  //Deprecated: please use MinimockFinish method or use Finish method of minimock.Controller
   293  func (m *PulsarStorageMock) ValidateCallCounters() {
   294  
   295  	if m.CloseFunc != nil && atomic.LoadUint64(&m.CloseCounter) == 0 {
   296  		m.t.Fatal("Expected call to PulsarStorageMock.Close")
   297  	}
   298  
   299  	if m.GetLastPulseFunc != nil && atomic.LoadUint64(&m.GetLastPulseCounter) == 0 {
   300  		m.t.Fatal("Expected call to PulsarStorageMock.GetLastPulse")
   301  	}
   302  
   303  	if m.SavePulseFunc != nil && atomic.LoadUint64(&m.SavePulseCounter) == 0 {
   304  		m.t.Fatal("Expected call to PulsarStorageMock.SavePulse")
   305  	}
   306  
   307  	if m.SetLastPulseFunc != nil && atomic.LoadUint64(&m.SetLastPulseCounter) == 0 {
   308  		m.t.Fatal("Expected call to PulsarStorageMock.SetLastPulse")
   309  	}
   310  
   311  }
   312  
   313  //CheckMocksCalled checks that all mocked methods of the interface have been called at least once
   314  //Deprecated: please use MinimockFinish method or use Finish method of minimock.Controller
   315  //noinspection GoDeprecation
   316  func (m *PulsarStorageMock) CheckMocksCalled() {
   317  	m.Finish()
   318  }
   319  
   320  //Finish checks that all mocked methods of the interface have been called at least once
   321  //Deprecated: please use MinimockFinish or use Finish method of minimock.Controller
   322  func (m *PulsarStorageMock) Finish() {
   323  	m.MinimockFinish()
   324  }
   325  
   326  //MinimockFinish checks that all mocked methods of the interface have been called at least once
   327  func (m *PulsarStorageMock) MinimockFinish() {
   328  
   329  	if m.CloseFunc != nil && atomic.LoadUint64(&m.CloseCounter) == 0 {
   330  		m.t.Fatal("Expected call to PulsarStorageMock.Close")
   331  	}
   332  
   333  	if m.GetLastPulseFunc != nil && atomic.LoadUint64(&m.GetLastPulseCounter) == 0 {
   334  		m.t.Fatal("Expected call to PulsarStorageMock.GetLastPulse")
   335  	}
   336  
   337  	if m.SavePulseFunc != nil && atomic.LoadUint64(&m.SavePulseCounter) == 0 {
   338  		m.t.Fatal("Expected call to PulsarStorageMock.SavePulse")
   339  	}
   340  
   341  	if m.SetLastPulseFunc != nil && atomic.LoadUint64(&m.SetLastPulseCounter) == 0 {
   342  		m.t.Fatal("Expected call to PulsarStorageMock.SetLastPulse")
   343  	}
   344  
   345  }
   346  
   347  //Wait waits for all mocked methods to be called at least once
   348  //Deprecated: please use MinimockWait or use Wait method of minimock.Controller
   349  func (m *PulsarStorageMock) Wait(timeout time.Duration) {
   350  	m.MinimockWait(timeout)
   351  }
   352  
   353  //MinimockWait waits for all mocked methods to be called at least once
   354  //this method is called by minimock.Controller
   355  func (m *PulsarStorageMock) MinimockWait(timeout time.Duration) {
   356  	timeoutCh := time.After(timeout)
   357  	for {
   358  		ok := true
   359  		ok = ok && (m.CloseFunc == nil || atomic.LoadUint64(&m.CloseCounter) > 0)
   360  		ok = ok && (m.GetLastPulseFunc == nil || atomic.LoadUint64(&m.GetLastPulseCounter) > 0)
   361  		ok = ok && (m.SavePulseFunc == nil || atomic.LoadUint64(&m.SavePulseCounter) > 0)
   362  		ok = ok && (m.SetLastPulseFunc == nil || atomic.LoadUint64(&m.SetLastPulseCounter) > 0)
   363  
   364  		if ok {
   365  			return
   366  		}
   367  
   368  		select {
   369  		case <-timeoutCh:
   370  
   371  			if m.CloseFunc != nil && atomic.LoadUint64(&m.CloseCounter) == 0 {
   372  				m.t.Error("Expected call to PulsarStorageMock.Close")
   373  			}
   374  
   375  			if m.GetLastPulseFunc != nil && atomic.LoadUint64(&m.GetLastPulseCounter) == 0 {
   376  				m.t.Error("Expected call to PulsarStorageMock.GetLastPulse")
   377  			}
   378  
   379  			if m.SavePulseFunc != nil && atomic.LoadUint64(&m.SavePulseCounter) == 0 {
   380  				m.t.Error("Expected call to PulsarStorageMock.SavePulse")
   381  			}
   382  
   383  			if m.SetLastPulseFunc != nil && atomic.LoadUint64(&m.SetLastPulseCounter) == 0 {
   384  				m.t.Error("Expected call to PulsarStorageMock.SetLastPulse")
   385  			}
   386  
   387  			m.t.Fatalf("Some mocks were not called on time: %s", timeout)
   388  			return
   389  		default:
   390  			time.Sleep(time.Millisecond)
   391  		}
   392  	}
   393  }
   394  
   395  //AllMocksCalled returns true if all mocked methods were called before the execution of AllMocksCalled,
   396  //it can be used with assert/require, i.e. require.True(mock.AllMocksCalled())
   397  func (m *PulsarStorageMock) AllMocksCalled() bool {
   398  
   399  	if m.CloseFunc != nil && atomic.LoadUint64(&m.CloseCounter) == 0 {
   400  		return false
   401  	}
   402  
   403  	if m.GetLastPulseFunc != nil && atomic.LoadUint64(&m.GetLastPulseCounter) == 0 {
   404  		return false
   405  	}
   406  
   407  	if m.SavePulseFunc != nil && atomic.LoadUint64(&m.SavePulseCounter) == 0 {
   408  		return false
   409  	}
   410  
   411  	if m.SetLastPulseFunc != nil && atomic.LoadUint64(&m.SetLastPulseCounter) == 0 {
   412  		return false
   413  	}
   414  
   415  	return true
   416  }