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 }