github.com/ava-labs/avalanchego@v1.11.11/vms/platformvm/state/mock_state.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: vms/platformvm/state/state.go
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -source=vms/platformvm/state/state.go -destination=vms/platformvm/state/mock_state.go -package=state -exclude_interfaces=Chain -mock_names=MockState=MockState
     7  //
     8  
     9  // Package state is a generated GoMock package.
    10  package state
    11  
    12  import (
    13  	context "context"
    14  	reflect "reflect"
    15  	sync "sync"
    16  	time "time"
    17  
    18  	database "github.com/ava-labs/avalanchego/database"
    19  	ids "github.com/ava-labs/avalanchego/ids"
    20  	validators "github.com/ava-labs/avalanchego/snow/validators"
    21  	iterator "github.com/ava-labs/avalanchego/utils/iterator"
    22  	logging "github.com/ava-labs/avalanchego/utils/logging"
    23  	avax "github.com/ava-labs/avalanchego/vms/components/avax"
    24  	gas "github.com/ava-labs/avalanchego/vms/components/gas"
    25  	block "github.com/ava-labs/avalanchego/vms/platformvm/block"
    26  	fx "github.com/ava-labs/avalanchego/vms/platformvm/fx"
    27  	status "github.com/ava-labs/avalanchego/vms/platformvm/status"
    28  	txs "github.com/ava-labs/avalanchego/vms/platformvm/txs"
    29  	gomock "go.uber.org/mock/gomock"
    30  )
    31  
    32  // MockState is a mock of State interface.
    33  type MockState struct {
    34  	ctrl     *gomock.Controller
    35  	recorder *MockStateMockRecorder
    36  }
    37  
    38  // MockStateMockRecorder is the mock recorder for MockState.
    39  type MockStateMockRecorder struct {
    40  	mock *MockState
    41  }
    42  
    43  // NewMockState creates a new mock instance.
    44  func NewMockState(ctrl *gomock.Controller) *MockState {
    45  	mock := &MockState{ctrl: ctrl}
    46  	mock.recorder = &MockStateMockRecorder{mock}
    47  	return mock
    48  }
    49  
    50  // EXPECT returns an object that allows the caller to indicate expected use.
    51  func (m *MockState) EXPECT() *MockStateMockRecorder {
    52  	return m.recorder
    53  }
    54  
    55  // Abort mocks base method.
    56  func (m *MockState) Abort() {
    57  	m.ctrl.T.Helper()
    58  	m.ctrl.Call(m, "Abort")
    59  }
    60  
    61  // Abort indicates an expected call of Abort.
    62  func (mr *MockStateMockRecorder) Abort() *gomock.Call {
    63  	mr.mock.ctrl.T.Helper()
    64  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Abort", reflect.TypeOf((*MockState)(nil).Abort))
    65  }
    66  
    67  // AddChain mocks base method.
    68  func (m *MockState) AddChain(createChainTx *txs.Tx) {
    69  	m.ctrl.T.Helper()
    70  	m.ctrl.Call(m, "AddChain", createChainTx)
    71  }
    72  
    73  // AddChain indicates an expected call of AddChain.
    74  func (mr *MockStateMockRecorder) AddChain(createChainTx any) *gomock.Call {
    75  	mr.mock.ctrl.T.Helper()
    76  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddChain", reflect.TypeOf((*MockState)(nil).AddChain), createChainTx)
    77  }
    78  
    79  // AddRewardUTXO mocks base method.
    80  func (m *MockState) AddRewardUTXO(txID ids.ID, utxo *avax.UTXO) {
    81  	m.ctrl.T.Helper()
    82  	m.ctrl.Call(m, "AddRewardUTXO", txID, utxo)
    83  }
    84  
    85  // AddRewardUTXO indicates an expected call of AddRewardUTXO.
    86  func (mr *MockStateMockRecorder) AddRewardUTXO(txID, utxo any) *gomock.Call {
    87  	mr.mock.ctrl.T.Helper()
    88  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddRewardUTXO", reflect.TypeOf((*MockState)(nil).AddRewardUTXO), txID, utxo)
    89  }
    90  
    91  // AddStatelessBlock mocks base method.
    92  func (m *MockState) AddStatelessBlock(block block.Block) {
    93  	m.ctrl.T.Helper()
    94  	m.ctrl.Call(m, "AddStatelessBlock", block)
    95  }
    96  
    97  // AddStatelessBlock indicates an expected call of AddStatelessBlock.
    98  func (mr *MockStateMockRecorder) AddStatelessBlock(block any) *gomock.Call {
    99  	mr.mock.ctrl.T.Helper()
   100  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddStatelessBlock", reflect.TypeOf((*MockState)(nil).AddStatelessBlock), block)
   101  }
   102  
   103  // AddSubnet mocks base method.
   104  func (m *MockState) AddSubnet(subnetID ids.ID) {
   105  	m.ctrl.T.Helper()
   106  	m.ctrl.Call(m, "AddSubnet", subnetID)
   107  }
   108  
   109  // AddSubnet indicates an expected call of AddSubnet.
   110  func (mr *MockStateMockRecorder) AddSubnet(subnetID any) *gomock.Call {
   111  	mr.mock.ctrl.T.Helper()
   112  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnet", reflect.TypeOf((*MockState)(nil).AddSubnet), subnetID)
   113  }
   114  
   115  // AddSubnetTransformation mocks base method.
   116  func (m *MockState) AddSubnetTransformation(transformSubnetTx *txs.Tx) {
   117  	m.ctrl.T.Helper()
   118  	m.ctrl.Call(m, "AddSubnetTransformation", transformSubnetTx)
   119  }
   120  
   121  // AddSubnetTransformation indicates an expected call of AddSubnetTransformation.
   122  func (mr *MockStateMockRecorder) AddSubnetTransformation(transformSubnetTx any) *gomock.Call {
   123  	mr.mock.ctrl.T.Helper()
   124  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnetTransformation", reflect.TypeOf((*MockState)(nil).AddSubnetTransformation), transformSubnetTx)
   125  }
   126  
   127  // AddTx mocks base method.
   128  func (m *MockState) AddTx(tx *txs.Tx, status status.Status) {
   129  	m.ctrl.T.Helper()
   130  	m.ctrl.Call(m, "AddTx", tx, status)
   131  }
   132  
   133  // AddTx indicates an expected call of AddTx.
   134  func (mr *MockStateMockRecorder) AddTx(tx, status any) *gomock.Call {
   135  	mr.mock.ctrl.T.Helper()
   136  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddTx", reflect.TypeOf((*MockState)(nil).AddTx), tx, status)
   137  }
   138  
   139  // AddUTXO mocks base method.
   140  func (m *MockState) AddUTXO(utxo *avax.UTXO) {
   141  	m.ctrl.T.Helper()
   142  	m.ctrl.Call(m, "AddUTXO", utxo)
   143  }
   144  
   145  // AddUTXO indicates an expected call of AddUTXO.
   146  func (mr *MockStateMockRecorder) AddUTXO(utxo any) *gomock.Call {
   147  	mr.mock.ctrl.T.Helper()
   148  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddUTXO", reflect.TypeOf((*MockState)(nil).AddUTXO), utxo)
   149  }
   150  
   151  // ApplyValidatorPublicKeyDiffs mocks base method.
   152  func (m *MockState) ApplyValidatorPublicKeyDiffs(ctx context.Context, validators map[ids.NodeID]*validators.GetValidatorOutput, startHeight, endHeight uint64) error {
   153  	m.ctrl.T.Helper()
   154  	ret := m.ctrl.Call(m, "ApplyValidatorPublicKeyDiffs", ctx, validators, startHeight, endHeight)
   155  	ret0, _ := ret[0].(error)
   156  	return ret0
   157  }
   158  
   159  // ApplyValidatorPublicKeyDiffs indicates an expected call of ApplyValidatorPublicKeyDiffs.
   160  func (mr *MockStateMockRecorder) ApplyValidatorPublicKeyDiffs(ctx, validators, startHeight, endHeight any) *gomock.Call {
   161  	mr.mock.ctrl.T.Helper()
   162  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyValidatorPublicKeyDiffs", reflect.TypeOf((*MockState)(nil).ApplyValidatorPublicKeyDiffs), ctx, validators, startHeight, endHeight)
   163  }
   164  
   165  // ApplyValidatorWeightDiffs mocks base method.
   166  func (m *MockState) ApplyValidatorWeightDiffs(ctx context.Context, validators map[ids.NodeID]*validators.GetValidatorOutput, startHeight, endHeight uint64, subnetID ids.ID) error {
   167  	m.ctrl.T.Helper()
   168  	ret := m.ctrl.Call(m, "ApplyValidatorWeightDiffs", ctx, validators, startHeight, endHeight, subnetID)
   169  	ret0, _ := ret[0].(error)
   170  	return ret0
   171  }
   172  
   173  // ApplyValidatorWeightDiffs indicates an expected call of ApplyValidatorWeightDiffs.
   174  func (mr *MockStateMockRecorder) ApplyValidatorWeightDiffs(ctx, validators, startHeight, endHeight, subnetID any) *gomock.Call {
   175  	mr.mock.ctrl.T.Helper()
   176  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyValidatorWeightDiffs", reflect.TypeOf((*MockState)(nil).ApplyValidatorWeightDiffs), ctx, validators, startHeight, endHeight, subnetID)
   177  }
   178  
   179  // Checksum mocks base method.
   180  func (m *MockState) Checksum() ids.ID {
   181  	m.ctrl.T.Helper()
   182  	ret := m.ctrl.Call(m, "Checksum")
   183  	ret0, _ := ret[0].(ids.ID)
   184  	return ret0
   185  }
   186  
   187  // Checksum indicates an expected call of Checksum.
   188  func (mr *MockStateMockRecorder) Checksum() *gomock.Call {
   189  	mr.mock.ctrl.T.Helper()
   190  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Checksum", reflect.TypeOf((*MockState)(nil).Checksum))
   191  }
   192  
   193  // Close mocks base method.
   194  func (m *MockState) Close() error {
   195  	m.ctrl.T.Helper()
   196  	ret := m.ctrl.Call(m, "Close")
   197  	ret0, _ := ret[0].(error)
   198  	return ret0
   199  }
   200  
   201  // Close indicates an expected call of Close.
   202  func (mr *MockStateMockRecorder) Close() *gomock.Call {
   203  	mr.mock.ctrl.T.Helper()
   204  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockState)(nil).Close))
   205  }
   206  
   207  // Commit mocks base method.
   208  func (m *MockState) Commit() error {
   209  	m.ctrl.T.Helper()
   210  	ret := m.ctrl.Call(m, "Commit")
   211  	ret0, _ := ret[0].(error)
   212  	return ret0
   213  }
   214  
   215  // Commit indicates an expected call of Commit.
   216  func (mr *MockStateMockRecorder) Commit() *gomock.Call {
   217  	mr.mock.ctrl.T.Helper()
   218  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Commit", reflect.TypeOf((*MockState)(nil).Commit))
   219  }
   220  
   221  // CommitBatch mocks base method.
   222  func (m *MockState) CommitBatch() (database.Batch, error) {
   223  	m.ctrl.T.Helper()
   224  	ret := m.ctrl.Call(m, "CommitBatch")
   225  	ret0, _ := ret[0].(database.Batch)
   226  	ret1, _ := ret[1].(error)
   227  	return ret0, ret1
   228  }
   229  
   230  // CommitBatch indicates an expected call of CommitBatch.
   231  func (mr *MockStateMockRecorder) CommitBatch() *gomock.Call {
   232  	mr.mock.ctrl.T.Helper()
   233  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommitBatch", reflect.TypeOf((*MockState)(nil).CommitBatch))
   234  }
   235  
   236  // DeleteCurrentDelegator mocks base method.
   237  func (m *MockState) DeleteCurrentDelegator(staker *Staker) {
   238  	m.ctrl.T.Helper()
   239  	m.ctrl.Call(m, "DeleteCurrentDelegator", staker)
   240  }
   241  
   242  // DeleteCurrentDelegator indicates an expected call of DeleteCurrentDelegator.
   243  func (mr *MockStateMockRecorder) DeleteCurrentDelegator(staker any) *gomock.Call {
   244  	mr.mock.ctrl.T.Helper()
   245  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentDelegator", reflect.TypeOf((*MockState)(nil).DeleteCurrentDelegator), staker)
   246  }
   247  
   248  // DeleteCurrentValidator mocks base method.
   249  func (m *MockState) DeleteCurrentValidator(staker *Staker) {
   250  	m.ctrl.T.Helper()
   251  	m.ctrl.Call(m, "DeleteCurrentValidator", staker)
   252  }
   253  
   254  // DeleteCurrentValidator indicates an expected call of DeleteCurrentValidator.
   255  func (mr *MockStateMockRecorder) DeleteCurrentValidator(staker any) *gomock.Call {
   256  	mr.mock.ctrl.T.Helper()
   257  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentValidator", reflect.TypeOf((*MockState)(nil).DeleteCurrentValidator), staker)
   258  }
   259  
   260  // DeletePendingDelegator mocks base method.
   261  func (m *MockState) DeletePendingDelegator(staker *Staker) {
   262  	m.ctrl.T.Helper()
   263  	m.ctrl.Call(m, "DeletePendingDelegator", staker)
   264  }
   265  
   266  // DeletePendingDelegator indicates an expected call of DeletePendingDelegator.
   267  func (mr *MockStateMockRecorder) DeletePendingDelegator(staker any) *gomock.Call {
   268  	mr.mock.ctrl.T.Helper()
   269  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingDelegator", reflect.TypeOf((*MockState)(nil).DeletePendingDelegator), staker)
   270  }
   271  
   272  // DeletePendingValidator mocks base method.
   273  func (m *MockState) DeletePendingValidator(staker *Staker) {
   274  	m.ctrl.T.Helper()
   275  	m.ctrl.Call(m, "DeletePendingValidator", staker)
   276  }
   277  
   278  // DeletePendingValidator indicates an expected call of DeletePendingValidator.
   279  func (mr *MockStateMockRecorder) DeletePendingValidator(staker any) *gomock.Call {
   280  	mr.mock.ctrl.T.Helper()
   281  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingValidator", reflect.TypeOf((*MockState)(nil).DeletePendingValidator), staker)
   282  }
   283  
   284  // DeleteUTXO mocks base method.
   285  func (m *MockState) DeleteUTXO(utxoID ids.ID) {
   286  	m.ctrl.T.Helper()
   287  	m.ctrl.Call(m, "DeleteUTXO", utxoID)
   288  }
   289  
   290  // DeleteUTXO indicates an expected call of DeleteUTXO.
   291  func (mr *MockStateMockRecorder) DeleteUTXO(utxoID any) *gomock.Call {
   292  	mr.mock.ctrl.T.Helper()
   293  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteUTXO", reflect.TypeOf((*MockState)(nil).DeleteUTXO), utxoID)
   294  }
   295  
   296  // GetBlockIDAtHeight mocks base method.
   297  func (m *MockState) GetBlockIDAtHeight(height uint64) (ids.ID, error) {
   298  	m.ctrl.T.Helper()
   299  	ret := m.ctrl.Call(m, "GetBlockIDAtHeight", height)
   300  	ret0, _ := ret[0].(ids.ID)
   301  	ret1, _ := ret[1].(error)
   302  	return ret0, ret1
   303  }
   304  
   305  // GetBlockIDAtHeight indicates an expected call of GetBlockIDAtHeight.
   306  func (mr *MockStateMockRecorder) GetBlockIDAtHeight(height any) *gomock.Call {
   307  	mr.mock.ctrl.T.Helper()
   308  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBlockIDAtHeight", reflect.TypeOf((*MockState)(nil).GetBlockIDAtHeight), height)
   309  }
   310  
   311  // GetChains mocks base method.
   312  func (m *MockState) GetChains(subnetID ids.ID) ([]*txs.Tx, error) {
   313  	m.ctrl.T.Helper()
   314  	ret := m.ctrl.Call(m, "GetChains", subnetID)
   315  	ret0, _ := ret[0].([]*txs.Tx)
   316  	ret1, _ := ret[1].(error)
   317  	return ret0, ret1
   318  }
   319  
   320  // GetChains indicates an expected call of GetChains.
   321  func (mr *MockStateMockRecorder) GetChains(subnetID any) *gomock.Call {
   322  	mr.mock.ctrl.T.Helper()
   323  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetChains", reflect.TypeOf((*MockState)(nil).GetChains), subnetID)
   324  }
   325  
   326  // GetCurrentDelegatorIterator mocks base method.
   327  func (m *MockState) GetCurrentDelegatorIterator(subnetID ids.ID, nodeID ids.NodeID) (iterator.Iterator[*Staker], error) {
   328  	m.ctrl.T.Helper()
   329  	ret := m.ctrl.Call(m, "GetCurrentDelegatorIterator", subnetID, nodeID)
   330  	ret0, _ := ret[0].(iterator.Iterator[*Staker])
   331  	ret1, _ := ret[1].(error)
   332  	return ret0, ret1
   333  }
   334  
   335  // GetCurrentDelegatorIterator indicates an expected call of GetCurrentDelegatorIterator.
   336  func (mr *MockStateMockRecorder) GetCurrentDelegatorIterator(subnetID, nodeID any) *gomock.Call {
   337  	mr.mock.ctrl.T.Helper()
   338  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentDelegatorIterator", reflect.TypeOf((*MockState)(nil).GetCurrentDelegatorIterator), subnetID, nodeID)
   339  }
   340  
   341  // GetCurrentStakerIterator mocks base method.
   342  func (m *MockState) GetCurrentStakerIterator() (iterator.Iterator[*Staker], error) {
   343  	m.ctrl.T.Helper()
   344  	ret := m.ctrl.Call(m, "GetCurrentStakerIterator")
   345  	ret0, _ := ret[0].(iterator.Iterator[*Staker])
   346  	ret1, _ := ret[1].(error)
   347  	return ret0, ret1
   348  }
   349  
   350  // GetCurrentStakerIterator indicates an expected call of GetCurrentStakerIterator.
   351  func (mr *MockStateMockRecorder) GetCurrentStakerIterator() *gomock.Call {
   352  	mr.mock.ctrl.T.Helper()
   353  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentStakerIterator", reflect.TypeOf((*MockState)(nil).GetCurrentStakerIterator))
   354  }
   355  
   356  // GetCurrentSupply mocks base method.
   357  func (m *MockState) GetCurrentSupply(subnetID ids.ID) (uint64, error) {
   358  	m.ctrl.T.Helper()
   359  	ret := m.ctrl.Call(m, "GetCurrentSupply", subnetID)
   360  	ret0, _ := ret[0].(uint64)
   361  	ret1, _ := ret[1].(error)
   362  	return ret0, ret1
   363  }
   364  
   365  // GetCurrentSupply indicates an expected call of GetCurrentSupply.
   366  func (mr *MockStateMockRecorder) GetCurrentSupply(subnetID any) *gomock.Call {
   367  	mr.mock.ctrl.T.Helper()
   368  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentSupply", reflect.TypeOf((*MockState)(nil).GetCurrentSupply), subnetID)
   369  }
   370  
   371  // GetCurrentValidator mocks base method.
   372  func (m *MockState) GetCurrentValidator(subnetID ids.ID, nodeID ids.NodeID) (*Staker, error) {
   373  	m.ctrl.T.Helper()
   374  	ret := m.ctrl.Call(m, "GetCurrentValidator", subnetID, nodeID)
   375  	ret0, _ := ret[0].(*Staker)
   376  	ret1, _ := ret[1].(error)
   377  	return ret0, ret1
   378  }
   379  
   380  // GetCurrentValidator indicates an expected call of GetCurrentValidator.
   381  func (mr *MockStateMockRecorder) GetCurrentValidator(subnetID, nodeID any) *gomock.Call {
   382  	mr.mock.ctrl.T.Helper()
   383  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentValidator", reflect.TypeOf((*MockState)(nil).GetCurrentValidator), subnetID, nodeID)
   384  }
   385  
   386  // GetDelegateeReward mocks base method.
   387  func (m *MockState) GetDelegateeReward(subnetID ids.ID, nodeID ids.NodeID) (uint64, error) {
   388  	m.ctrl.T.Helper()
   389  	ret := m.ctrl.Call(m, "GetDelegateeReward", subnetID, nodeID)
   390  	ret0, _ := ret[0].(uint64)
   391  	ret1, _ := ret[1].(error)
   392  	return ret0, ret1
   393  }
   394  
   395  // GetDelegateeReward indicates an expected call of GetDelegateeReward.
   396  func (mr *MockStateMockRecorder) GetDelegateeReward(subnetID, nodeID any) *gomock.Call {
   397  	mr.mock.ctrl.T.Helper()
   398  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDelegateeReward", reflect.TypeOf((*MockState)(nil).GetDelegateeReward), subnetID, nodeID)
   399  }
   400  
   401  // GetFeeState mocks base method.
   402  func (m *MockState) GetFeeState() gas.State {
   403  	m.ctrl.T.Helper()
   404  	ret := m.ctrl.Call(m, "GetFeeState")
   405  	ret0, _ := ret[0].(gas.State)
   406  	return ret0
   407  }
   408  
   409  // GetFeeState indicates an expected call of GetFeeState.
   410  func (mr *MockStateMockRecorder) GetFeeState() *gomock.Call {
   411  	mr.mock.ctrl.T.Helper()
   412  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFeeState", reflect.TypeOf((*MockState)(nil).GetFeeState))
   413  }
   414  
   415  // GetLastAccepted mocks base method.
   416  func (m *MockState) GetLastAccepted() ids.ID {
   417  	m.ctrl.T.Helper()
   418  	ret := m.ctrl.Call(m, "GetLastAccepted")
   419  	ret0, _ := ret[0].(ids.ID)
   420  	return ret0
   421  }
   422  
   423  // GetLastAccepted indicates an expected call of GetLastAccepted.
   424  func (mr *MockStateMockRecorder) GetLastAccepted() *gomock.Call {
   425  	mr.mock.ctrl.T.Helper()
   426  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLastAccepted", reflect.TypeOf((*MockState)(nil).GetLastAccepted))
   427  }
   428  
   429  // GetPendingDelegatorIterator mocks base method.
   430  func (m *MockState) GetPendingDelegatorIterator(subnetID ids.ID, nodeID ids.NodeID) (iterator.Iterator[*Staker], error) {
   431  	m.ctrl.T.Helper()
   432  	ret := m.ctrl.Call(m, "GetPendingDelegatorIterator", subnetID, nodeID)
   433  	ret0, _ := ret[0].(iterator.Iterator[*Staker])
   434  	ret1, _ := ret[1].(error)
   435  	return ret0, ret1
   436  }
   437  
   438  // GetPendingDelegatorIterator indicates an expected call of GetPendingDelegatorIterator.
   439  func (mr *MockStateMockRecorder) GetPendingDelegatorIterator(subnetID, nodeID any) *gomock.Call {
   440  	mr.mock.ctrl.T.Helper()
   441  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingDelegatorIterator", reflect.TypeOf((*MockState)(nil).GetPendingDelegatorIterator), subnetID, nodeID)
   442  }
   443  
   444  // GetPendingStakerIterator mocks base method.
   445  func (m *MockState) GetPendingStakerIterator() (iterator.Iterator[*Staker], error) {
   446  	m.ctrl.T.Helper()
   447  	ret := m.ctrl.Call(m, "GetPendingStakerIterator")
   448  	ret0, _ := ret[0].(iterator.Iterator[*Staker])
   449  	ret1, _ := ret[1].(error)
   450  	return ret0, ret1
   451  }
   452  
   453  // GetPendingStakerIterator indicates an expected call of GetPendingStakerIterator.
   454  func (mr *MockStateMockRecorder) GetPendingStakerIterator() *gomock.Call {
   455  	mr.mock.ctrl.T.Helper()
   456  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingStakerIterator", reflect.TypeOf((*MockState)(nil).GetPendingStakerIterator))
   457  }
   458  
   459  // GetPendingValidator mocks base method.
   460  func (m *MockState) GetPendingValidator(subnetID ids.ID, nodeID ids.NodeID) (*Staker, error) {
   461  	m.ctrl.T.Helper()
   462  	ret := m.ctrl.Call(m, "GetPendingValidator", subnetID, nodeID)
   463  	ret0, _ := ret[0].(*Staker)
   464  	ret1, _ := ret[1].(error)
   465  	return ret0, ret1
   466  }
   467  
   468  // GetPendingValidator indicates an expected call of GetPendingValidator.
   469  func (mr *MockStateMockRecorder) GetPendingValidator(subnetID, nodeID any) *gomock.Call {
   470  	mr.mock.ctrl.T.Helper()
   471  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingValidator", reflect.TypeOf((*MockState)(nil).GetPendingValidator), subnetID, nodeID)
   472  }
   473  
   474  // GetRewardUTXOs mocks base method.
   475  func (m *MockState) GetRewardUTXOs(txID ids.ID) ([]*avax.UTXO, error) {
   476  	m.ctrl.T.Helper()
   477  	ret := m.ctrl.Call(m, "GetRewardUTXOs", txID)
   478  	ret0, _ := ret[0].([]*avax.UTXO)
   479  	ret1, _ := ret[1].(error)
   480  	return ret0, ret1
   481  }
   482  
   483  // GetRewardUTXOs indicates an expected call of GetRewardUTXOs.
   484  func (mr *MockStateMockRecorder) GetRewardUTXOs(txID any) *gomock.Call {
   485  	mr.mock.ctrl.T.Helper()
   486  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRewardUTXOs", reflect.TypeOf((*MockState)(nil).GetRewardUTXOs), txID)
   487  }
   488  
   489  // GetStartTime mocks base method.
   490  func (m *MockState) GetStartTime(nodeID ids.NodeID, subnetID ids.ID) (time.Time, error) {
   491  	m.ctrl.T.Helper()
   492  	ret := m.ctrl.Call(m, "GetStartTime", nodeID, subnetID)
   493  	ret0, _ := ret[0].(time.Time)
   494  	ret1, _ := ret[1].(error)
   495  	return ret0, ret1
   496  }
   497  
   498  // GetStartTime indicates an expected call of GetStartTime.
   499  func (mr *MockStateMockRecorder) GetStartTime(nodeID, subnetID any) *gomock.Call {
   500  	mr.mock.ctrl.T.Helper()
   501  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStartTime", reflect.TypeOf((*MockState)(nil).GetStartTime), nodeID, subnetID)
   502  }
   503  
   504  // GetStatelessBlock mocks base method.
   505  func (m *MockState) GetStatelessBlock(blockID ids.ID) (block.Block, error) {
   506  	m.ctrl.T.Helper()
   507  	ret := m.ctrl.Call(m, "GetStatelessBlock", blockID)
   508  	ret0, _ := ret[0].(block.Block)
   509  	ret1, _ := ret[1].(error)
   510  	return ret0, ret1
   511  }
   512  
   513  // GetStatelessBlock indicates an expected call of GetStatelessBlock.
   514  func (mr *MockStateMockRecorder) GetStatelessBlock(blockID any) *gomock.Call {
   515  	mr.mock.ctrl.T.Helper()
   516  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStatelessBlock", reflect.TypeOf((*MockState)(nil).GetStatelessBlock), blockID)
   517  }
   518  
   519  // GetSubnetIDs mocks base method.
   520  func (m *MockState) GetSubnetIDs() ([]ids.ID, error) {
   521  	m.ctrl.T.Helper()
   522  	ret := m.ctrl.Call(m, "GetSubnetIDs")
   523  	ret0, _ := ret[0].([]ids.ID)
   524  	ret1, _ := ret[1].(error)
   525  	return ret0, ret1
   526  }
   527  
   528  // GetSubnetIDs indicates an expected call of GetSubnetIDs.
   529  func (mr *MockStateMockRecorder) GetSubnetIDs() *gomock.Call {
   530  	mr.mock.ctrl.T.Helper()
   531  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetIDs", reflect.TypeOf((*MockState)(nil).GetSubnetIDs))
   532  }
   533  
   534  // GetSubnetManager mocks base method.
   535  func (m *MockState) GetSubnetManager(subnetID ids.ID) (ids.ID, []byte, error) {
   536  	m.ctrl.T.Helper()
   537  	ret := m.ctrl.Call(m, "GetSubnetManager", subnetID)
   538  	ret0, _ := ret[0].(ids.ID)
   539  	ret1, _ := ret[1].([]byte)
   540  	ret2, _ := ret[2].(error)
   541  	return ret0, ret1, ret2
   542  }
   543  
   544  // GetSubnetManager indicates an expected call of GetSubnetManager.
   545  func (mr *MockStateMockRecorder) GetSubnetManager(subnetID any) *gomock.Call {
   546  	mr.mock.ctrl.T.Helper()
   547  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetManager", reflect.TypeOf((*MockState)(nil).GetSubnetManager), subnetID)
   548  }
   549  
   550  // GetSubnetOwner mocks base method.
   551  func (m *MockState) GetSubnetOwner(subnetID ids.ID) (fx.Owner, error) {
   552  	m.ctrl.T.Helper()
   553  	ret := m.ctrl.Call(m, "GetSubnetOwner", subnetID)
   554  	ret0, _ := ret[0].(fx.Owner)
   555  	ret1, _ := ret[1].(error)
   556  	return ret0, ret1
   557  }
   558  
   559  // GetSubnetOwner indicates an expected call of GetSubnetOwner.
   560  func (mr *MockStateMockRecorder) GetSubnetOwner(subnetID any) *gomock.Call {
   561  	mr.mock.ctrl.T.Helper()
   562  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetOwner", reflect.TypeOf((*MockState)(nil).GetSubnetOwner), subnetID)
   563  }
   564  
   565  // GetSubnetTransformation mocks base method.
   566  func (m *MockState) GetSubnetTransformation(subnetID ids.ID) (*txs.Tx, error) {
   567  	m.ctrl.T.Helper()
   568  	ret := m.ctrl.Call(m, "GetSubnetTransformation", subnetID)
   569  	ret0, _ := ret[0].(*txs.Tx)
   570  	ret1, _ := ret[1].(error)
   571  	return ret0, ret1
   572  }
   573  
   574  // GetSubnetTransformation indicates an expected call of GetSubnetTransformation.
   575  func (mr *MockStateMockRecorder) GetSubnetTransformation(subnetID any) *gomock.Call {
   576  	mr.mock.ctrl.T.Helper()
   577  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetTransformation", reflect.TypeOf((*MockState)(nil).GetSubnetTransformation), subnetID)
   578  }
   579  
   580  // GetTimestamp mocks base method.
   581  func (m *MockState) GetTimestamp() time.Time {
   582  	m.ctrl.T.Helper()
   583  	ret := m.ctrl.Call(m, "GetTimestamp")
   584  	ret0, _ := ret[0].(time.Time)
   585  	return ret0
   586  }
   587  
   588  // GetTimestamp indicates an expected call of GetTimestamp.
   589  func (mr *MockStateMockRecorder) GetTimestamp() *gomock.Call {
   590  	mr.mock.ctrl.T.Helper()
   591  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTimestamp", reflect.TypeOf((*MockState)(nil).GetTimestamp))
   592  }
   593  
   594  // GetTx mocks base method.
   595  func (m *MockState) GetTx(txID ids.ID) (*txs.Tx, status.Status, error) {
   596  	m.ctrl.T.Helper()
   597  	ret := m.ctrl.Call(m, "GetTx", txID)
   598  	ret0, _ := ret[0].(*txs.Tx)
   599  	ret1, _ := ret[1].(status.Status)
   600  	ret2, _ := ret[2].(error)
   601  	return ret0, ret1, ret2
   602  }
   603  
   604  // GetTx indicates an expected call of GetTx.
   605  func (mr *MockStateMockRecorder) GetTx(txID any) *gomock.Call {
   606  	mr.mock.ctrl.T.Helper()
   607  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTx", reflect.TypeOf((*MockState)(nil).GetTx), txID)
   608  }
   609  
   610  // GetUTXO mocks base method.
   611  func (m *MockState) GetUTXO(utxoID ids.ID) (*avax.UTXO, error) {
   612  	m.ctrl.T.Helper()
   613  	ret := m.ctrl.Call(m, "GetUTXO", utxoID)
   614  	ret0, _ := ret[0].(*avax.UTXO)
   615  	ret1, _ := ret[1].(error)
   616  	return ret0, ret1
   617  }
   618  
   619  // GetUTXO indicates an expected call of GetUTXO.
   620  func (mr *MockStateMockRecorder) GetUTXO(utxoID any) *gomock.Call {
   621  	mr.mock.ctrl.T.Helper()
   622  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUTXO", reflect.TypeOf((*MockState)(nil).GetUTXO), utxoID)
   623  }
   624  
   625  // GetUptime mocks base method.
   626  func (m *MockState) GetUptime(nodeID ids.NodeID, subnetID ids.ID) (time.Duration, time.Time, error) {
   627  	m.ctrl.T.Helper()
   628  	ret := m.ctrl.Call(m, "GetUptime", nodeID, subnetID)
   629  	ret0, _ := ret[0].(time.Duration)
   630  	ret1, _ := ret[1].(time.Time)
   631  	ret2, _ := ret[2].(error)
   632  	return ret0, ret1, ret2
   633  }
   634  
   635  // GetUptime indicates an expected call of GetUptime.
   636  func (mr *MockStateMockRecorder) GetUptime(nodeID, subnetID any) *gomock.Call {
   637  	mr.mock.ctrl.T.Helper()
   638  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUptime", reflect.TypeOf((*MockState)(nil).GetUptime), nodeID, subnetID)
   639  }
   640  
   641  // PutCurrentDelegator mocks base method.
   642  func (m *MockState) PutCurrentDelegator(staker *Staker) {
   643  	m.ctrl.T.Helper()
   644  	m.ctrl.Call(m, "PutCurrentDelegator", staker)
   645  }
   646  
   647  // PutCurrentDelegator indicates an expected call of PutCurrentDelegator.
   648  func (mr *MockStateMockRecorder) PutCurrentDelegator(staker any) *gomock.Call {
   649  	mr.mock.ctrl.T.Helper()
   650  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentDelegator", reflect.TypeOf((*MockState)(nil).PutCurrentDelegator), staker)
   651  }
   652  
   653  // PutCurrentValidator mocks base method.
   654  func (m *MockState) PutCurrentValidator(staker *Staker) error {
   655  	m.ctrl.T.Helper()
   656  	ret := m.ctrl.Call(m, "PutCurrentValidator", staker)
   657  	ret0, _ := ret[0].(error)
   658  	return ret0
   659  }
   660  
   661  // PutCurrentValidator indicates an expected call of PutCurrentValidator.
   662  func (mr *MockStateMockRecorder) PutCurrentValidator(staker any) *gomock.Call {
   663  	mr.mock.ctrl.T.Helper()
   664  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentValidator", reflect.TypeOf((*MockState)(nil).PutCurrentValidator), staker)
   665  }
   666  
   667  // PutPendingDelegator mocks base method.
   668  func (m *MockState) PutPendingDelegator(staker *Staker) {
   669  	m.ctrl.T.Helper()
   670  	m.ctrl.Call(m, "PutPendingDelegator", staker)
   671  }
   672  
   673  // PutPendingDelegator indicates an expected call of PutPendingDelegator.
   674  func (mr *MockStateMockRecorder) PutPendingDelegator(staker any) *gomock.Call {
   675  	mr.mock.ctrl.T.Helper()
   676  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingDelegator", reflect.TypeOf((*MockState)(nil).PutPendingDelegator), staker)
   677  }
   678  
   679  // PutPendingValidator mocks base method.
   680  func (m *MockState) PutPendingValidator(staker *Staker) error {
   681  	m.ctrl.T.Helper()
   682  	ret := m.ctrl.Call(m, "PutPendingValidator", staker)
   683  	ret0, _ := ret[0].(error)
   684  	return ret0
   685  }
   686  
   687  // PutPendingValidator indicates an expected call of PutPendingValidator.
   688  func (mr *MockStateMockRecorder) PutPendingValidator(staker any) *gomock.Call {
   689  	mr.mock.ctrl.T.Helper()
   690  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingValidator", reflect.TypeOf((*MockState)(nil).PutPendingValidator), staker)
   691  }
   692  
   693  // ReindexBlocks mocks base method.
   694  func (m *MockState) ReindexBlocks(lock sync.Locker, log logging.Logger) error {
   695  	m.ctrl.T.Helper()
   696  	ret := m.ctrl.Call(m, "ReindexBlocks", lock, log)
   697  	ret0, _ := ret[0].(error)
   698  	return ret0
   699  }
   700  
   701  // ReindexBlocks indicates an expected call of ReindexBlocks.
   702  func (mr *MockStateMockRecorder) ReindexBlocks(lock, log any) *gomock.Call {
   703  	mr.mock.ctrl.T.Helper()
   704  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReindexBlocks", reflect.TypeOf((*MockState)(nil).ReindexBlocks), lock, log)
   705  }
   706  
   707  // SetCurrentSupply mocks base method.
   708  func (m *MockState) SetCurrentSupply(subnetID ids.ID, cs uint64) {
   709  	m.ctrl.T.Helper()
   710  	m.ctrl.Call(m, "SetCurrentSupply", subnetID, cs)
   711  }
   712  
   713  // SetCurrentSupply indicates an expected call of SetCurrentSupply.
   714  func (mr *MockStateMockRecorder) SetCurrentSupply(subnetID, cs any) *gomock.Call {
   715  	mr.mock.ctrl.T.Helper()
   716  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetCurrentSupply", reflect.TypeOf((*MockState)(nil).SetCurrentSupply), subnetID, cs)
   717  }
   718  
   719  // SetDelegateeReward mocks base method.
   720  func (m *MockState) SetDelegateeReward(subnetID ids.ID, nodeID ids.NodeID, amount uint64) error {
   721  	m.ctrl.T.Helper()
   722  	ret := m.ctrl.Call(m, "SetDelegateeReward", subnetID, nodeID, amount)
   723  	ret0, _ := ret[0].(error)
   724  	return ret0
   725  }
   726  
   727  // SetDelegateeReward indicates an expected call of SetDelegateeReward.
   728  func (mr *MockStateMockRecorder) SetDelegateeReward(subnetID, nodeID, amount any) *gomock.Call {
   729  	mr.mock.ctrl.T.Helper()
   730  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetDelegateeReward", reflect.TypeOf((*MockState)(nil).SetDelegateeReward), subnetID, nodeID, amount)
   731  }
   732  
   733  // SetFeeState mocks base method.
   734  func (m *MockState) SetFeeState(f gas.State) {
   735  	m.ctrl.T.Helper()
   736  	m.ctrl.Call(m, "SetFeeState", f)
   737  }
   738  
   739  // SetFeeState indicates an expected call of SetFeeState.
   740  func (mr *MockStateMockRecorder) SetFeeState(f any) *gomock.Call {
   741  	mr.mock.ctrl.T.Helper()
   742  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetFeeState", reflect.TypeOf((*MockState)(nil).SetFeeState), f)
   743  }
   744  
   745  // SetHeight mocks base method.
   746  func (m *MockState) SetHeight(height uint64) {
   747  	m.ctrl.T.Helper()
   748  	m.ctrl.Call(m, "SetHeight", height)
   749  }
   750  
   751  // SetHeight indicates an expected call of SetHeight.
   752  func (mr *MockStateMockRecorder) SetHeight(height any) *gomock.Call {
   753  	mr.mock.ctrl.T.Helper()
   754  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeight", reflect.TypeOf((*MockState)(nil).SetHeight), height)
   755  }
   756  
   757  // SetLastAccepted mocks base method.
   758  func (m *MockState) SetLastAccepted(blkID ids.ID) {
   759  	m.ctrl.T.Helper()
   760  	m.ctrl.Call(m, "SetLastAccepted", blkID)
   761  }
   762  
   763  // SetLastAccepted indicates an expected call of SetLastAccepted.
   764  func (mr *MockStateMockRecorder) SetLastAccepted(blkID any) *gomock.Call {
   765  	mr.mock.ctrl.T.Helper()
   766  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetLastAccepted", reflect.TypeOf((*MockState)(nil).SetLastAccepted), blkID)
   767  }
   768  
   769  // SetSubnetManager mocks base method.
   770  func (m *MockState) SetSubnetManager(subnetID, chainID ids.ID, addr []byte) {
   771  	m.ctrl.T.Helper()
   772  	m.ctrl.Call(m, "SetSubnetManager", subnetID, chainID, addr)
   773  }
   774  
   775  // SetSubnetManager indicates an expected call of SetSubnetManager.
   776  func (mr *MockStateMockRecorder) SetSubnetManager(subnetID, chainID, addr any) *gomock.Call {
   777  	mr.mock.ctrl.T.Helper()
   778  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetSubnetManager", reflect.TypeOf((*MockState)(nil).SetSubnetManager), subnetID, chainID, addr)
   779  }
   780  
   781  // SetSubnetOwner mocks base method.
   782  func (m *MockState) SetSubnetOwner(subnetID ids.ID, owner fx.Owner) {
   783  	m.ctrl.T.Helper()
   784  	m.ctrl.Call(m, "SetSubnetOwner", subnetID, owner)
   785  }
   786  
   787  // SetSubnetOwner indicates an expected call of SetSubnetOwner.
   788  func (mr *MockStateMockRecorder) SetSubnetOwner(subnetID, owner any) *gomock.Call {
   789  	mr.mock.ctrl.T.Helper()
   790  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetSubnetOwner", reflect.TypeOf((*MockState)(nil).SetSubnetOwner), subnetID, owner)
   791  }
   792  
   793  // SetTimestamp mocks base method.
   794  func (m *MockState) SetTimestamp(tm time.Time) {
   795  	m.ctrl.T.Helper()
   796  	m.ctrl.Call(m, "SetTimestamp", tm)
   797  }
   798  
   799  // SetTimestamp indicates an expected call of SetTimestamp.
   800  func (mr *MockStateMockRecorder) SetTimestamp(tm any) *gomock.Call {
   801  	mr.mock.ctrl.T.Helper()
   802  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTimestamp", reflect.TypeOf((*MockState)(nil).SetTimestamp), tm)
   803  }
   804  
   805  // SetUptime mocks base method.
   806  func (m *MockState) SetUptime(nodeID ids.NodeID, subnetID ids.ID, upDuration time.Duration, lastUpdated time.Time) error {
   807  	m.ctrl.T.Helper()
   808  	ret := m.ctrl.Call(m, "SetUptime", nodeID, subnetID, upDuration, lastUpdated)
   809  	ret0, _ := ret[0].(error)
   810  	return ret0
   811  }
   812  
   813  // SetUptime indicates an expected call of SetUptime.
   814  func (mr *MockStateMockRecorder) SetUptime(nodeID, subnetID, upDuration, lastUpdated any) *gomock.Call {
   815  	mr.mock.ctrl.T.Helper()
   816  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetUptime", reflect.TypeOf((*MockState)(nil).SetUptime), nodeID, subnetID, upDuration, lastUpdated)
   817  }
   818  
   819  // UTXOIDs mocks base method.
   820  func (m *MockState) UTXOIDs(addr []byte, previous ids.ID, limit int) ([]ids.ID, error) {
   821  	m.ctrl.T.Helper()
   822  	ret := m.ctrl.Call(m, "UTXOIDs", addr, previous, limit)
   823  	ret0, _ := ret[0].([]ids.ID)
   824  	ret1, _ := ret[1].(error)
   825  	return ret0, ret1
   826  }
   827  
   828  // UTXOIDs indicates an expected call of UTXOIDs.
   829  func (mr *MockStateMockRecorder) UTXOIDs(addr, previous, limit any) *gomock.Call {
   830  	mr.mock.ctrl.T.Helper()
   831  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UTXOIDs", reflect.TypeOf((*MockState)(nil).UTXOIDs), addr, previous, limit)
   832  }