github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/state/protocol/protocol_state/epochs/mock/state_machine.go (about)

     1  // Code generated by mockery v2.21.4. DO NOT EDIT.
     2  
     3  package mock
     4  
     5  import (
     6  	flow "github.com/onflow/flow-go/model/flow"
     7  	mock "github.com/stretchr/testify/mock"
     8  )
     9  
    10  // StateMachine is an autogenerated mock type for the StateMachine type
    11  type StateMachine struct {
    12  	mock.Mock
    13  }
    14  
    15  // Build provides a mock function with given fields:
    16  func (_m *StateMachine) Build() (*flow.ProtocolStateEntry, flow.Identifier, bool) {
    17  	ret := _m.Called()
    18  
    19  	var r0 *flow.ProtocolStateEntry
    20  	var r1 flow.Identifier
    21  	var r2 bool
    22  	if rf, ok := ret.Get(0).(func() (*flow.ProtocolStateEntry, flow.Identifier, bool)); ok {
    23  		return rf()
    24  	}
    25  	if rf, ok := ret.Get(0).(func() *flow.ProtocolStateEntry); ok {
    26  		r0 = rf()
    27  	} else {
    28  		if ret.Get(0) != nil {
    29  			r0 = ret.Get(0).(*flow.ProtocolStateEntry)
    30  		}
    31  	}
    32  
    33  	if rf, ok := ret.Get(1).(func() flow.Identifier); ok {
    34  		r1 = rf()
    35  	} else {
    36  		if ret.Get(1) != nil {
    37  			r1 = ret.Get(1).(flow.Identifier)
    38  		}
    39  	}
    40  
    41  	if rf, ok := ret.Get(2).(func() bool); ok {
    42  		r2 = rf()
    43  	} else {
    44  		r2 = ret.Get(2).(bool)
    45  	}
    46  
    47  	return r0, r1, r2
    48  }
    49  
    50  // EjectIdentity provides a mock function with given fields: nodeID
    51  func (_m *StateMachine) EjectIdentity(nodeID flow.Identifier) error {
    52  	ret := _m.Called(nodeID)
    53  
    54  	var r0 error
    55  	if rf, ok := ret.Get(0).(func(flow.Identifier) error); ok {
    56  		r0 = rf(nodeID)
    57  	} else {
    58  		r0 = ret.Error(0)
    59  	}
    60  
    61  	return r0
    62  }
    63  
    64  // ParentState provides a mock function with given fields:
    65  func (_m *StateMachine) ParentState() *flow.RichProtocolStateEntry {
    66  	ret := _m.Called()
    67  
    68  	var r0 *flow.RichProtocolStateEntry
    69  	if rf, ok := ret.Get(0).(func() *flow.RichProtocolStateEntry); ok {
    70  		r0 = rf()
    71  	} else {
    72  		if ret.Get(0) != nil {
    73  			r0 = ret.Get(0).(*flow.RichProtocolStateEntry)
    74  		}
    75  	}
    76  
    77  	return r0
    78  }
    79  
    80  // ProcessEpochCommit provides a mock function with given fields: epochCommit
    81  func (_m *StateMachine) ProcessEpochCommit(epochCommit *flow.EpochCommit) (bool, error) {
    82  	ret := _m.Called(epochCommit)
    83  
    84  	var r0 bool
    85  	var r1 error
    86  	if rf, ok := ret.Get(0).(func(*flow.EpochCommit) (bool, error)); ok {
    87  		return rf(epochCommit)
    88  	}
    89  	if rf, ok := ret.Get(0).(func(*flow.EpochCommit) bool); ok {
    90  		r0 = rf(epochCommit)
    91  	} else {
    92  		r0 = ret.Get(0).(bool)
    93  	}
    94  
    95  	if rf, ok := ret.Get(1).(func(*flow.EpochCommit) error); ok {
    96  		r1 = rf(epochCommit)
    97  	} else {
    98  		r1 = ret.Error(1)
    99  	}
   100  
   101  	return r0, r1
   102  }
   103  
   104  // ProcessEpochSetup provides a mock function with given fields: epochSetup
   105  func (_m *StateMachine) ProcessEpochSetup(epochSetup *flow.EpochSetup) (bool, error) {
   106  	ret := _m.Called(epochSetup)
   107  
   108  	var r0 bool
   109  	var r1 error
   110  	if rf, ok := ret.Get(0).(func(*flow.EpochSetup) (bool, error)); ok {
   111  		return rf(epochSetup)
   112  	}
   113  	if rf, ok := ret.Get(0).(func(*flow.EpochSetup) bool); ok {
   114  		r0 = rf(epochSetup)
   115  	} else {
   116  		r0 = ret.Get(0).(bool)
   117  	}
   118  
   119  	if rf, ok := ret.Get(1).(func(*flow.EpochSetup) error); ok {
   120  		r1 = rf(epochSetup)
   121  	} else {
   122  		r1 = ret.Error(1)
   123  	}
   124  
   125  	return r0, r1
   126  }
   127  
   128  // TransitionToNextEpoch provides a mock function with given fields:
   129  func (_m *StateMachine) TransitionToNextEpoch() error {
   130  	ret := _m.Called()
   131  
   132  	var r0 error
   133  	if rf, ok := ret.Get(0).(func() error); ok {
   134  		r0 = rf()
   135  	} else {
   136  		r0 = ret.Error(0)
   137  	}
   138  
   139  	return r0
   140  }
   141  
   142  // View provides a mock function with given fields:
   143  func (_m *StateMachine) View() uint64 {
   144  	ret := _m.Called()
   145  
   146  	var r0 uint64
   147  	if rf, ok := ret.Get(0).(func() uint64); ok {
   148  		r0 = rf()
   149  	} else {
   150  		r0 = ret.Get(0).(uint64)
   151  	}
   152  
   153  	return r0
   154  }
   155  
   156  type mockConstructorTestingTNewStateMachine interface {
   157  	mock.TestingT
   158  	Cleanup(func())
   159  }
   160  
   161  // NewStateMachine creates a new instance of StateMachine. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
   162  func NewStateMachine(t mockConstructorTestingTNewStateMachine) *StateMachine {
   163  	mock := &StateMachine{}
   164  	mock.Mock.Test(t)
   165  
   166  	t.Cleanup(func() { mock.AssertExpectations(t) })
   167  
   168  	return mock
   169  }