From 9ad213c3166e73b185281f37d64d39be8d865c5b Mon Sep 17 00:00:00 2001
From: Dhruba Basu <7675102+dhrubabasu@users.noreply.github.com>
Date: Mon, 27 Nov 2023 11:45:03 -0800
Subject: [PATCH] `vms/platformvm`: Consolidate `state` pkg mocks (#2370)

---
 scripts/mocks.mockgen.txt             |    5 +-
 vms/platformvm/state/mock_chain.go    |  513 ------------
 vms/platformvm/state/mock_diff.go     |  527 -------------
 vms/platformvm/state/mock_state.go    | 1040 ++++++++++++++++++++++++-
 vms/platformvm/state/mock_versions.go |   53 --
 5 files changed, 1040 insertions(+), 1098 deletions(-)
 delete mode 100644 vms/platformvm/state/mock_chain.go
 delete mode 100644 vms/platformvm/state/mock_diff.go
 delete mode 100644 vms/platformvm/state/mock_versions.go

diff --git a/scripts/mocks.mockgen.txt b/scripts/mocks.mockgen.txt
index 4b7afa01849f..ba55a16a4b71 100644
--- a/scripts/mocks.mockgen.txt
+++ b/scripts/mocks.mockgen.txt
@@ -33,11 +33,8 @@ github.com/ava-labs/avalanchego/vms/components/avax=TransferableIn=vms/component
 github.com/ava-labs/avalanchego/vms/components/verify=Verifiable=vms/components/verify/mock_verifiable.go
 github.com/ava-labs/avalanchego/vms/platformvm/block/executor=Manager=vms/platformvm/block/executor/mock_manager.go
 github.com/ava-labs/avalanchego/vms/platformvm/block=Block=vms/platformvm/block/mock_block.go
-github.com/ava-labs/avalanchego/vms/platformvm/state=Chain=vms/platformvm/state/mock_chain.go
-github.com/ava-labs/avalanchego/vms/platformvm/state=Diff=vms/platformvm/state/mock_diff.go
+github.com/ava-labs/avalanchego/vms/platformvm/state=Chain,Diff,State,Versions=vms/platformvm/state/mock_state.go
 github.com/ava-labs/avalanchego/vms/platformvm/state=StakerIterator=vms/platformvm/state/mock_staker_iterator.go
-github.com/ava-labs/avalanchego/vms/platformvm/state=State=vms/platformvm/state/mock_state.go
-github.com/ava-labs/avalanchego/vms/platformvm/state=Versions=vms/platformvm/state/mock_versions.go
 github.com/ava-labs/avalanchego/vms/platformvm/txs/builder=Builder=vms/platformvm/txs/builder/mock_builder.go
 github.com/ava-labs/avalanchego/vms/platformvm/txs/mempool=Mempool=vms/platformvm/txs/mempool/mock_mempool.go
 github.com/ava-labs/avalanchego/vms/platformvm/utxo=Verifier=vms/platformvm/utxo/mock_verifier.go
diff --git a/vms/platformvm/state/mock_chain.go b/vms/platformvm/state/mock_chain.go
deleted file mode 100644
index c82ceb3af831..000000000000
--- a/vms/platformvm/state/mock_chain.go
+++ /dev/null
@@ -1,513 +0,0 @@
-// Copyright (C) 2019-2023, Ava Labs, Inc. All rights reserved.
-// See the file LICENSE for licensing terms.
-
-// Code generated by MockGen. DO NOT EDIT.
-// Source: github.com/ava-labs/avalanchego/vms/platformvm/state (interfaces: Chain)
-
-// Package state is a generated GoMock package.
-package state
-
-import (
-	reflect "reflect"
-	time "time"
-
-	ids "github.com/ava-labs/avalanchego/ids"
-	avax "github.com/ava-labs/avalanchego/vms/components/avax"
-	fx "github.com/ava-labs/avalanchego/vms/platformvm/fx"
-	status "github.com/ava-labs/avalanchego/vms/platformvm/status"
-	txs "github.com/ava-labs/avalanchego/vms/platformvm/txs"
-	gomock "go.uber.org/mock/gomock"
-)
-
-// MockChain is a mock of Chain interface.
-type MockChain struct {
-	ctrl     *gomock.Controller
-	recorder *MockChainMockRecorder
-}
-
-// MockChainMockRecorder is the mock recorder for MockChain.
-type MockChainMockRecorder struct {
-	mock *MockChain
-}
-
-// NewMockChain creates a new mock instance.
-func NewMockChain(ctrl *gomock.Controller) *MockChain {
-	mock := &MockChain{ctrl: ctrl}
-	mock.recorder = &MockChainMockRecorder{mock}
-	return mock
-}
-
-// EXPECT returns an object that allows the caller to indicate expected use.
-func (m *MockChain) EXPECT() *MockChainMockRecorder {
-	return m.recorder
-}
-
-// AddChain mocks base method.
-func (m *MockChain) AddChain(arg0 *txs.Tx) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddChain", arg0)
-}
-
-// AddChain indicates an expected call of AddChain.
-func (mr *MockChainMockRecorder) AddChain(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddChain", reflect.TypeOf((*MockChain)(nil).AddChain), arg0)
-}
-
-// AddRewardUTXO mocks base method.
-func (m *MockChain) AddRewardUTXO(arg0 ids.ID, arg1 *avax.UTXO) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddRewardUTXO", arg0, arg1)
-}
-
-// AddRewardUTXO indicates an expected call of AddRewardUTXO.
-func (mr *MockChainMockRecorder) AddRewardUTXO(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddRewardUTXO", reflect.TypeOf((*MockChain)(nil).AddRewardUTXO), arg0, arg1)
-}
-
-// AddSubnet mocks base method.
-func (m *MockChain) AddSubnet(arg0 *txs.Tx) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddSubnet", arg0)
-}
-
-// AddSubnet indicates an expected call of AddSubnet.
-func (mr *MockChainMockRecorder) AddSubnet(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnet", reflect.TypeOf((*MockChain)(nil).AddSubnet), arg0)
-}
-
-// AddSubnetTransformation mocks base method.
-func (m *MockChain) AddSubnetTransformation(arg0 *txs.Tx) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddSubnetTransformation", arg0)
-}
-
-// AddSubnetTransformation indicates an expected call of AddSubnetTransformation.
-func (mr *MockChainMockRecorder) AddSubnetTransformation(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnetTransformation", reflect.TypeOf((*MockChain)(nil).AddSubnetTransformation), arg0)
-}
-
-// AddTx mocks base method.
-func (m *MockChain) AddTx(arg0 *txs.Tx, arg1 status.Status) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddTx", arg0, arg1)
-}
-
-// AddTx indicates an expected call of AddTx.
-func (mr *MockChainMockRecorder) AddTx(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddTx", reflect.TypeOf((*MockChain)(nil).AddTx), arg0, arg1)
-}
-
-// AddUTXO mocks base method.
-func (m *MockChain) AddUTXO(arg0 *avax.UTXO) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddUTXO", arg0)
-}
-
-// AddUTXO indicates an expected call of AddUTXO.
-func (mr *MockChainMockRecorder) AddUTXO(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddUTXO", reflect.TypeOf((*MockChain)(nil).AddUTXO), arg0)
-}
-
-// DeleteCurrentDelegator mocks base method.
-func (m *MockChain) DeleteCurrentDelegator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeleteCurrentDelegator", arg0)
-}
-
-// DeleteCurrentDelegator indicates an expected call of DeleteCurrentDelegator.
-func (mr *MockChainMockRecorder) DeleteCurrentDelegator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentDelegator", reflect.TypeOf((*MockChain)(nil).DeleteCurrentDelegator), arg0)
-}
-
-// DeleteCurrentValidator mocks base method.
-func (m *MockChain) DeleteCurrentValidator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeleteCurrentValidator", arg0)
-}
-
-// DeleteCurrentValidator indicates an expected call of DeleteCurrentValidator.
-func (mr *MockChainMockRecorder) DeleteCurrentValidator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentValidator", reflect.TypeOf((*MockChain)(nil).DeleteCurrentValidator), arg0)
-}
-
-// DeletePendingDelegator mocks base method.
-func (m *MockChain) DeletePendingDelegator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeletePendingDelegator", arg0)
-}
-
-// DeletePendingDelegator indicates an expected call of DeletePendingDelegator.
-func (mr *MockChainMockRecorder) DeletePendingDelegator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingDelegator", reflect.TypeOf((*MockChain)(nil).DeletePendingDelegator), arg0)
-}
-
-// DeletePendingValidator mocks base method.
-func (m *MockChain) DeletePendingValidator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeletePendingValidator", arg0)
-}
-
-// DeletePendingValidator indicates an expected call of DeletePendingValidator.
-func (mr *MockChainMockRecorder) DeletePendingValidator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingValidator", reflect.TypeOf((*MockChain)(nil).DeletePendingValidator), arg0)
-}
-
-// DeleteUTXO mocks base method.
-func (m *MockChain) DeleteUTXO(arg0 ids.ID) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeleteUTXO", arg0)
-}
-
-// DeleteUTXO indicates an expected call of DeleteUTXO.
-func (mr *MockChainMockRecorder) DeleteUTXO(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteUTXO", reflect.TypeOf((*MockChain)(nil).DeleteUTXO), arg0)
-}
-
-// GetChains mocks base method.
-func (m *MockChain) GetChains(arg0 ids.ID) ([]*txs.Tx, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetChains", arg0)
-	ret0, _ := ret[0].([]*txs.Tx)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetChains indicates an expected call of GetChains.
-func (mr *MockChainMockRecorder) GetChains(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetChains", reflect.TypeOf((*MockChain)(nil).GetChains), arg0)
-}
-
-// GetCurrentDelegatorIterator mocks base method.
-func (m *MockChain) GetCurrentDelegatorIterator(arg0 ids.ID, arg1 ids.NodeID) (StakerIterator, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetCurrentDelegatorIterator", arg0, arg1)
-	ret0, _ := ret[0].(StakerIterator)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetCurrentDelegatorIterator indicates an expected call of GetCurrentDelegatorIterator.
-func (mr *MockChainMockRecorder) GetCurrentDelegatorIterator(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentDelegatorIterator", reflect.TypeOf((*MockChain)(nil).GetCurrentDelegatorIterator), arg0, arg1)
-}
-
-// GetCurrentStakerIterator mocks base method.
-func (m *MockChain) GetCurrentStakerIterator() (StakerIterator, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetCurrentStakerIterator")
-	ret0, _ := ret[0].(StakerIterator)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetCurrentStakerIterator indicates an expected call of GetCurrentStakerIterator.
-func (mr *MockChainMockRecorder) GetCurrentStakerIterator() *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentStakerIterator", reflect.TypeOf((*MockChain)(nil).GetCurrentStakerIterator))
-}
-
-// GetCurrentSupply mocks base method.
-func (m *MockChain) GetCurrentSupply(arg0 ids.ID) (uint64, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetCurrentSupply", arg0)
-	ret0, _ := ret[0].(uint64)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetCurrentSupply indicates an expected call of GetCurrentSupply.
-func (mr *MockChainMockRecorder) GetCurrentSupply(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentSupply", reflect.TypeOf((*MockChain)(nil).GetCurrentSupply), arg0)
-}
-
-// GetCurrentValidator mocks base method.
-func (m *MockChain) GetCurrentValidator(arg0 ids.ID, arg1 ids.NodeID) (*Staker, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetCurrentValidator", arg0, arg1)
-	ret0, _ := ret[0].(*Staker)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetCurrentValidator indicates an expected call of GetCurrentValidator.
-func (mr *MockChainMockRecorder) GetCurrentValidator(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentValidator", reflect.TypeOf((*MockChain)(nil).GetCurrentValidator), arg0, arg1)
-}
-
-// GetDelegateeReward mocks base method.
-func (m *MockChain) GetDelegateeReward(arg0 ids.ID, arg1 ids.NodeID) (uint64, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetDelegateeReward", arg0, arg1)
-	ret0, _ := ret[0].(uint64)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetDelegateeReward indicates an expected call of GetDelegateeReward.
-func (mr *MockChainMockRecorder) GetDelegateeReward(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDelegateeReward", reflect.TypeOf((*MockChain)(nil).GetDelegateeReward), arg0, arg1)
-}
-
-// GetPendingDelegatorIterator mocks base method.
-func (m *MockChain) GetPendingDelegatorIterator(arg0 ids.ID, arg1 ids.NodeID) (StakerIterator, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetPendingDelegatorIterator", arg0, arg1)
-	ret0, _ := ret[0].(StakerIterator)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetPendingDelegatorIterator indicates an expected call of GetPendingDelegatorIterator.
-func (mr *MockChainMockRecorder) GetPendingDelegatorIterator(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingDelegatorIterator", reflect.TypeOf((*MockChain)(nil).GetPendingDelegatorIterator), arg0, arg1)
-}
-
-// GetPendingStakerIterator mocks base method.
-func (m *MockChain) GetPendingStakerIterator() (StakerIterator, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetPendingStakerIterator")
-	ret0, _ := ret[0].(StakerIterator)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetPendingStakerIterator indicates an expected call of GetPendingStakerIterator.
-func (mr *MockChainMockRecorder) GetPendingStakerIterator() *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingStakerIterator", reflect.TypeOf((*MockChain)(nil).GetPendingStakerIterator))
-}
-
-// GetPendingValidator mocks base method.
-func (m *MockChain) GetPendingValidator(arg0 ids.ID, arg1 ids.NodeID) (*Staker, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetPendingValidator", arg0, arg1)
-	ret0, _ := ret[0].(*Staker)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetPendingValidator indicates an expected call of GetPendingValidator.
-func (mr *MockChainMockRecorder) GetPendingValidator(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingValidator", reflect.TypeOf((*MockChain)(nil).GetPendingValidator), arg0, arg1)
-}
-
-// GetRewardUTXOs mocks base method.
-func (m *MockChain) GetRewardUTXOs(arg0 ids.ID) ([]*avax.UTXO, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetRewardUTXOs", arg0)
-	ret0, _ := ret[0].([]*avax.UTXO)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetRewardUTXOs indicates an expected call of GetRewardUTXOs.
-func (mr *MockChainMockRecorder) GetRewardUTXOs(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRewardUTXOs", reflect.TypeOf((*MockChain)(nil).GetRewardUTXOs), arg0)
-}
-
-// GetSubnetOwner mocks base method.
-func (m *MockChain) GetSubnetOwner(arg0 ids.ID) (fx.Owner, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetSubnetOwner", arg0)
-	ret0, _ := ret[0].(fx.Owner)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetSubnetOwner indicates an expected call of GetSubnetOwner.
-func (mr *MockChainMockRecorder) GetSubnetOwner(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetOwner", reflect.TypeOf((*MockChain)(nil).GetSubnetOwner), arg0)
-}
-
-// GetSubnetTransformation mocks base method.
-func (m *MockChain) GetSubnetTransformation(arg0 ids.ID) (*txs.Tx, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetSubnetTransformation", arg0)
-	ret0, _ := ret[0].(*txs.Tx)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetSubnetTransformation indicates an expected call of GetSubnetTransformation.
-func (mr *MockChainMockRecorder) GetSubnetTransformation(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetTransformation", reflect.TypeOf((*MockChain)(nil).GetSubnetTransformation), arg0)
-}
-
-// GetSubnets mocks base method.
-func (m *MockChain) GetSubnets() ([]*txs.Tx, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetSubnets")
-	ret0, _ := ret[0].([]*txs.Tx)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetSubnets indicates an expected call of GetSubnets.
-func (mr *MockChainMockRecorder) GetSubnets() *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnets", reflect.TypeOf((*MockChain)(nil).GetSubnets))
-}
-
-// GetTimestamp mocks base method.
-func (m *MockChain) GetTimestamp() time.Time {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetTimestamp")
-	ret0, _ := ret[0].(time.Time)
-	return ret0
-}
-
-// GetTimestamp indicates an expected call of GetTimestamp.
-func (mr *MockChainMockRecorder) GetTimestamp() *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTimestamp", reflect.TypeOf((*MockChain)(nil).GetTimestamp))
-}
-
-// GetTx mocks base method.
-func (m *MockChain) GetTx(arg0 ids.ID) (*txs.Tx, status.Status, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetTx", arg0)
-	ret0, _ := ret[0].(*txs.Tx)
-	ret1, _ := ret[1].(status.Status)
-	ret2, _ := ret[2].(error)
-	return ret0, ret1, ret2
-}
-
-// GetTx indicates an expected call of GetTx.
-func (mr *MockChainMockRecorder) GetTx(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTx", reflect.TypeOf((*MockChain)(nil).GetTx), arg0)
-}
-
-// GetUTXO mocks base method.
-func (m *MockChain) GetUTXO(arg0 ids.ID) (*avax.UTXO, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetUTXO", arg0)
-	ret0, _ := ret[0].(*avax.UTXO)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetUTXO indicates an expected call of GetUTXO.
-func (mr *MockChainMockRecorder) GetUTXO(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUTXO", reflect.TypeOf((*MockChain)(nil).GetUTXO), arg0)
-}
-
-// PutCurrentDelegator mocks base method.
-func (m *MockChain) PutCurrentDelegator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "PutCurrentDelegator", arg0)
-}
-
-// PutCurrentDelegator indicates an expected call of PutCurrentDelegator.
-func (mr *MockChainMockRecorder) PutCurrentDelegator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentDelegator", reflect.TypeOf((*MockChain)(nil).PutCurrentDelegator), arg0)
-}
-
-// PutCurrentValidator mocks base method.
-func (m *MockChain) PutCurrentValidator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "PutCurrentValidator", arg0)
-}
-
-// PutCurrentValidator indicates an expected call of PutCurrentValidator.
-func (mr *MockChainMockRecorder) PutCurrentValidator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentValidator", reflect.TypeOf((*MockChain)(nil).PutCurrentValidator), arg0)
-}
-
-// PutPendingDelegator mocks base method.
-func (m *MockChain) PutPendingDelegator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "PutPendingDelegator", arg0)
-}
-
-// PutPendingDelegator indicates an expected call of PutPendingDelegator.
-func (mr *MockChainMockRecorder) PutPendingDelegator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingDelegator", reflect.TypeOf((*MockChain)(nil).PutPendingDelegator), arg0)
-}
-
-// PutPendingValidator mocks base method.
-func (m *MockChain) PutPendingValidator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "PutPendingValidator", arg0)
-}
-
-// PutPendingValidator indicates an expected call of PutPendingValidator.
-func (mr *MockChainMockRecorder) PutPendingValidator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingValidator", reflect.TypeOf((*MockChain)(nil).PutPendingValidator), arg0)
-}
-
-// SetCurrentSupply mocks base method.
-func (m *MockChain) SetCurrentSupply(arg0 ids.ID, arg1 uint64) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "SetCurrentSupply", arg0, arg1)
-}
-
-// SetCurrentSupply indicates an expected call of SetCurrentSupply.
-func (mr *MockChainMockRecorder) SetCurrentSupply(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetCurrentSupply", reflect.TypeOf((*MockChain)(nil).SetCurrentSupply), arg0, arg1)
-}
-
-// SetDelegateeReward mocks base method.
-func (m *MockChain) SetDelegateeReward(arg0 ids.ID, arg1 ids.NodeID, arg2 uint64) error {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "SetDelegateeReward", arg0, arg1, arg2)
-	ret0, _ := ret[0].(error)
-	return ret0
-}
-
-// SetDelegateeReward indicates an expected call of SetDelegateeReward.
-func (mr *MockChainMockRecorder) SetDelegateeReward(arg0, arg1, arg2 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetDelegateeReward", reflect.TypeOf((*MockChain)(nil).SetDelegateeReward), arg0, arg1, arg2)
-}
-
-// SetSubnetOwner mocks base method.
-func (m *MockChain) SetSubnetOwner(arg0 ids.ID, arg1 fx.Owner) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "SetSubnetOwner", arg0, arg1)
-}
-
-// SetSubnetOwner indicates an expected call of SetSubnetOwner.
-func (mr *MockChainMockRecorder) SetSubnetOwner(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetSubnetOwner", reflect.TypeOf((*MockChain)(nil).SetSubnetOwner), arg0, arg1)
-}
-
-// SetTimestamp mocks base method.
-func (m *MockChain) SetTimestamp(arg0 time.Time) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "SetTimestamp", arg0)
-}
-
-// SetTimestamp indicates an expected call of SetTimestamp.
-func (mr *MockChainMockRecorder) SetTimestamp(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTimestamp", reflect.TypeOf((*MockChain)(nil).SetTimestamp), arg0)
-}
diff --git a/vms/platformvm/state/mock_diff.go b/vms/platformvm/state/mock_diff.go
deleted file mode 100644
index abb380ca6e5f..000000000000
--- a/vms/platformvm/state/mock_diff.go
+++ /dev/null
@@ -1,527 +0,0 @@
-// Copyright (C) 2019-2023, Ava Labs, Inc. All rights reserved.
-// See the file LICENSE for licensing terms.
-
-// Code generated by MockGen. DO NOT EDIT.
-// Source: github.com/ava-labs/avalanchego/vms/platformvm/state (interfaces: Diff)
-
-// Package state is a generated GoMock package.
-package state
-
-import (
-	reflect "reflect"
-	time "time"
-
-	ids "github.com/ava-labs/avalanchego/ids"
-	avax "github.com/ava-labs/avalanchego/vms/components/avax"
-	fx "github.com/ava-labs/avalanchego/vms/platformvm/fx"
-	status "github.com/ava-labs/avalanchego/vms/platformvm/status"
-	txs "github.com/ava-labs/avalanchego/vms/platformvm/txs"
-	gomock "go.uber.org/mock/gomock"
-)
-
-// MockDiff is a mock of Diff interface.
-type MockDiff struct {
-	ctrl     *gomock.Controller
-	recorder *MockDiffMockRecorder
-}
-
-// MockDiffMockRecorder is the mock recorder for MockDiff.
-type MockDiffMockRecorder struct {
-	mock *MockDiff
-}
-
-// NewMockDiff creates a new mock instance.
-func NewMockDiff(ctrl *gomock.Controller) *MockDiff {
-	mock := &MockDiff{ctrl: ctrl}
-	mock.recorder = &MockDiffMockRecorder{mock}
-	return mock
-}
-
-// EXPECT returns an object that allows the caller to indicate expected use.
-func (m *MockDiff) EXPECT() *MockDiffMockRecorder {
-	return m.recorder
-}
-
-// AddChain mocks base method.
-func (m *MockDiff) AddChain(arg0 *txs.Tx) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddChain", arg0)
-}
-
-// AddChain indicates an expected call of AddChain.
-func (mr *MockDiffMockRecorder) AddChain(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddChain", reflect.TypeOf((*MockDiff)(nil).AddChain), arg0)
-}
-
-// AddRewardUTXO mocks base method.
-func (m *MockDiff) AddRewardUTXO(arg0 ids.ID, arg1 *avax.UTXO) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddRewardUTXO", arg0, arg1)
-}
-
-// AddRewardUTXO indicates an expected call of AddRewardUTXO.
-func (mr *MockDiffMockRecorder) AddRewardUTXO(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddRewardUTXO", reflect.TypeOf((*MockDiff)(nil).AddRewardUTXO), arg0, arg1)
-}
-
-// AddSubnet mocks base method.
-func (m *MockDiff) AddSubnet(arg0 *txs.Tx) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddSubnet", arg0)
-}
-
-// AddSubnet indicates an expected call of AddSubnet.
-func (mr *MockDiffMockRecorder) AddSubnet(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnet", reflect.TypeOf((*MockDiff)(nil).AddSubnet), arg0)
-}
-
-// AddSubnetTransformation mocks base method.
-func (m *MockDiff) AddSubnetTransformation(arg0 *txs.Tx) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddSubnetTransformation", arg0)
-}
-
-// AddSubnetTransformation indicates an expected call of AddSubnetTransformation.
-func (mr *MockDiffMockRecorder) AddSubnetTransformation(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnetTransformation", reflect.TypeOf((*MockDiff)(nil).AddSubnetTransformation), arg0)
-}
-
-// AddTx mocks base method.
-func (m *MockDiff) AddTx(arg0 *txs.Tx, arg1 status.Status) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddTx", arg0, arg1)
-}
-
-// AddTx indicates an expected call of AddTx.
-func (mr *MockDiffMockRecorder) AddTx(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddTx", reflect.TypeOf((*MockDiff)(nil).AddTx), arg0, arg1)
-}
-
-// AddUTXO mocks base method.
-func (m *MockDiff) AddUTXO(arg0 *avax.UTXO) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "AddUTXO", arg0)
-}
-
-// AddUTXO indicates an expected call of AddUTXO.
-func (mr *MockDiffMockRecorder) AddUTXO(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddUTXO", reflect.TypeOf((*MockDiff)(nil).AddUTXO), arg0)
-}
-
-// Apply mocks base method.
-func (m *MockDiff) Apply(arg0 Chain) error {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "Apply", arg0)
-	ret0, _ := ret[0].(error)
-	return ret0
-}
-
-// Apply indicates an expected call of Apply.
-func (mr *MockDiffMockRecorder) Apply(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Apply", reflect.TypeOf((*MockDiff)(nil).Apply), arg0)
-}
-
-// DeleteCurrentDelegator mocks base method.
-func (m *MockDiff) DeleteCurrentDelegator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeleteCurrentDelegator", arg0)
-}
-
-// DeleteCurrentDelegator indicates an expected call of DeleteCurrentDelegator.
-func (mr *MockDiffMockRecorder) DeleteCurrentDelegator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentDelegator", reflect.TypeOf((*MockDiff)(nil).DeleteCurrentDelegator), arg0)
-}
-
-// DeleteCurrentValidator mocks base method.
-func (m *MockDiff) DeleteCurrentValidator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeleteCurrentValidator", arg0)
-}
-
-// DeleteCurrentValidator indicates an expected call of DeleteCurrentValidator.
-func (mr *MockDiffMockRecorder) DeleteCurrentValidator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentValidator", reflect.TypeOf((*MockDiff)(nil).DeleteCurrentValidator), arg0)
-}
-
-// DeletePendingDelegator mocks base method.
-func (m *MockDiff) DeletePendingDelegator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeletePendingDelegator", arg0)
-}
-
-// DeletePendingDelegator indicates an expected call of DeletePendingDelegator.
-func (mr *MockDiffMockRecorder) DeletePendingDelegator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingDelegator", reflect.TypeOf((*MockDiff)(nil).DeletePendingDelegator), arg0)
-}
-
-// DeletePendingValidator mocks base method.
-func (m *MockDiff) DeletePendingValidator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeletePendingValidator", arg0)
-}
-
-// DeletePendingValidator indicates an expected call of DeletePendingValidator.
-func (mr *MockDiffMockRecorder) DeletePendingValidator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingValidator", reflect.TypeOf((*MockDiff)(nil).DeletePendingValidator), arg0)
-}
-
-// DeleteUTXO mocks base method.
-func (m *MockDiff) DeleteUTXO(arg0 ids.ID) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "DeleteUTXO", arg0)
-}
-
-// DeleteUTXO indicates an expected call of DeleteUTXO.
-func (mr *MockDiffMockRecorder) DeleteUTXO(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteUTXO", reflect.TypeOf((*MockDiff)(nil).DeleteUTXO), arg0)
-}
-
-// GetChains mocks base method.
-func (m *MockDiff) GetChains(arg0 ids.ID) ([]*txs.Tx, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetChains", arg0)
-	ret0, _ := ret[0].([]*txs.Tx)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetChains indicates an expected call of GetChains.
-func (mr *MockDiffMockRecorder) GetChains(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetChains", reflect.TypeOf((*MockDiff)(nil).GetChains), arg0)
-}
-
-// GetCurrentDelegatorIterator mocks base method.
-func (m *MockDiff) GetCurrentDelegatorIterator(arg0 ids.ID, arg1 ids.NodeID) (StakerIterator, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetCurrentDelegatorIterator", arg0, arg1)
-	ret0, _ := ret[0].(StakerIterator)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetCurrentDelegatorIterator indicates an expected call of GetCurrentDelegatorIterator.
-func (mr *MockDiffMockRecorder) GetCurrentDelegatorIterator(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentDelegatorIterator", reflect.TypeOf((*MockDiff)(nil).GetCurrentDelegatorIterator), arg0, arg1)
-}
-
-// GetCurrentStakerIterator mocks base method.
-func (m *MockDiff) GetCurrentStakerIterator() (StakerIterator, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetCurrentStakerIterator")
-	ret0, _ := ret[0].(StakerIterator)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetCurrentStakerIterator indicates an expected call of GetCurrentStakerIterator.
-func (mr *MockDiffMockRecorder) GetCurrentStakerIterator() *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentStakerIterator", reflect.TypeOf((*MockDiff)(nil).GetCurrentStakerIterator))
-}
-
-// GetCurrentSupply mocks base method.
-func (m *MockDiff) GetCurrentSupply(arg0 ids.ID) (uint64, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetCurrentSupply", arg0)
-	ret0, _ := ret[0].(uint64)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetCurrentSupply indicates an expected call of GetCurrentSupply.
-func (mr *MockDiffMockRecorder) GetCurrentSupply(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentSupply", reflect.TypeOf((*MockDiff)(nil).GetCurrentSupply), arg0)
-}
-
-// GetCurrentValidator mocks base method.
-func (m *MockDiff) GetCurrentValidator(arg0 ids.ID, arg1 ids.NodeID) (*Staker, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetCurrentValidator", arg0, arg1)
-	ret0, _ := ret[0].(*Staker)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetCurrentValidator indicates an expected call of GetCurrentValidator.
-func (mr *MockDiffMockRecorder) GetCurrentValidator(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentValidator", reflect.TypeOf((*MockDiff)(nil).GetCurrentValidator), arg0, arg1)
-}
-
-// GetDelegateeReward mocks base method.
-func (m *MockDiff) GetDelegateeReward(arg0 ids.ID, arg1 ids.NodeID) (uint64, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetDelegateeReward", arg0, arg1)
-	ret0, _ := ret[0].(uint64)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetDelegateeReward indicates an expected call of GetDelegateeReward.
-func (mr *MockDiffMockRecorder) GetDelegateeReward(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDelegateeReward", reflect.TypeOf((*MockDiff)(nil).GetDelegateeReward), arg0, arg1)
-}
-
-// GetPendingDelegatorIterator mocks base method.
-func (m *MockDiff) GetPendingDelegatorIterator(arg0 ids.ID, arg1 ids.NodeID) (StakerIterator, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetPendingDelegatorIterator", arg0, arg1)
-	ret0, _ := ret[0].(StakerIterator)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetPendingDelegatorIterator indicates an expected call of GetPendingDelegatorIterator.
-func (mr *MockDiffMockRecorder) GetPendingDelegatorIterator(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingDelegatorIterator", reflect.TypeOf((*MockDiff)(nil).GetPendingDelegatorIterator), arg0, arg1)
-}
-
-// GetPendingStakerIterator mocks base method.
-func (m *MockDiff) GetPendingStakerIterator() (StakerIterator, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetPendingStakerIterator")
-	ret0, _ := ret[0].(StakerIterator)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetPendingStakerIterator indicates an expected call of GetPendingStakerIterator.
-func (mr *MockDiffMockRecorder) GetPendingStakerIterator() *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingStakerIterator", reflect.TypeOf((*MockDiff)(nil).GetPendingStakerIterator))
-}
-
-// GetPendingValidator mocks base method.
-func (m *MockDiff) GetPendingValidator(arg0 ids.ID, arg1 ids.NodeID) (*Staker, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetPendingValidator", arg0, arg1)
-	ret0, _ := ret[0].(*Staker)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetPendingValidator indicates an expected call of GetPendingValidator.
-func (mr *MockDiffMockRecorder) GetPendingValidator(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingValidator", reflect.TypeOf((*MockDiff)(nil).GetPendingValidator), arg0, arg1)
-}
-
-// GetRewardUTXOs mocks base method.
-func (m *MockDiff) GetRewardUTXOs(arg0 ids.ID) ([]*avax.UTXO, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetRewardUTXOs", arg0)
-	ret0, _ := ret[0].([]*avax.UTXO)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetRewardUTXOs indicates an expected call of GetRewardUTXOs.
-func (mr *MockDiffMockRecorder) GetRewardUTXOs(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRewardUTXOs", reflect.TypeOf((*MockDiff)(nil).GetRewardUTXOs), arg0)
-}
-
-// GetSubnetOwner mocks base method.
-func (m *MockDiff) GetSubnetOwner(arg0 ids.ID) (fx.Owner, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetSubnetOwner", arg0)
-	ret0, _ := ret[0].(fx.Owner)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetSubnetOwner indicates an expected call of GetSubnetOwner.
-func (mr *MockDiffMockRecorder) GetSubnetOwner(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetOwner", reflect.TypeOf((*MockDiff)(nil).GetSubnetOwner), arg0)
-}
-
-// GetSubnetTransformation mocks base method.
-func (m *MockDiff) GetSubnetTransformation(arg0 ids.ID) (*txs.Tx, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetSubnetTransformation", arg0)
-	ret0, _ := ret[0].(*txs.Tx)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetSubnetTransformation indicates an expected call of GetSubnetTransformation.
-func (mr *MockDiffMockRecorder) GetSubnetTransformation(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetTransformation", reflect.TypeOf((*MockDiff)(nil).GetSubnetTransformation), arg0)
-}
-
-// GetSubnets mocks base method.
-func (m *MockDiff) GetSubnets() ([]*txs.Tx, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetSubnets")
-	ret0, _ := ret[0].([]*txs.Tx)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetSubnets indicates an expected call of GetSubnets.
-func (mr *MockDiffMockRecorder) GetSubnets() *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnets", reflect.TypeOf((*MockDiff)(nil).GetSubnets))
-}
-
-// GetTimestamp mocks base method.
-func (m *MockDiff) GetTimestamp() time.Time {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetTimestamp")
-	ret0, _ := ret[0].(time.Time)
-	return ret0
-}
-
-// GetTimestamp indicates an expected call of GetTimestamp.
-func (mr *MockDiffMockRecorder) GetTimestamp() *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTimestamp", reflect.TypeOf((*MockDiff)(nil).GetTimestamp))
-}
-
-// GetTx mocks base method.
-func (m *MockDiff) GetTx(arg0 ids.ID) (*txs.Tx, status.Status, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetTx", arg0)
-	ret0, _ := ret[0].(*txs.Tx)
-	ret1, _ := ret[1].(status.Status)
-	ret2, _ := ret[2].(error)
-	return ret0, ret1, ret2
-}
-
-// GetTx indicates an expected call of GetTx.
-func (mr *MockDiffMockRecorder) GetTx(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTx", reflect.TypeOf((*MockDiff)(nil).GetTx), arg0)
-}
-
-// GetUTXO mocks base method.
-func (m *MockDiff) GetUTXO(arg0 ids.ID) (*avax.UTXO, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetUTXO", arg0)
-	ret0, _ := ret[0].(*avax.UTXO)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// GetUTXO indicates an expected call of GetUTXO.
-func (mr *MockDiffMockRecorder) GetUTXO(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUTXO", reflect.TypeOf((*MockDiff)(nil).GetUTXO), arg0)
-}
-
-// PutCurrentDelegator mocks base method.
-func (m *MockDiff) PutCurrentDelegator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "PutCurrentDelegator", arg0)
-}
-
-// PutCurrentDelegator indicates an expected call of PutCurrentDelegator.
-func (mr *MockDiffMockRecorder) PutCurrentDelegator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentDelegator", reflect.TypeOf((*MockDiff)(nil).PutCurrentDelegator), arg0)
-}
-
-// PutCurrentValidator mocks base method.
-func (m *MockDiff) PutCurrentValidator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "PutCurrentValidator", arg0)
-}
-
-// PutCurrentValidator indicates an expected call of PutCurrentValidator.
-func (mr *MockDiffMockRecorder) PutCurrentValidator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentValidator", reflect.TypeOf((*MockDiff)(nil).PutCurrentValidator), arg0)
-}
-
-// PutPendingDelegator mocks base method.
-func (m *MockDiff) PutPendingDelegator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "PutPendingDelegator", arg0)
-}
-
-// PutPendingDelegator indicates an expected call of PutPendingDelegator.
-func (mr *MockDiffMockRecorder) PutPendingDelegator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingDelegator", reflect.TypeOf((*MockDiff)(nil).PutPendingDelegator), arg0)
-}
-
-// PutPendingValidator mocks base method.
-func (m *MockDiff) PutPendingValidator(arg0 *Staker) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "PutPendingValidator", arg0)
-}
-
-// PutPendingValidator indicates an expected call of PutPendingValidator.
-func (mr *MockDiffMockRecorder) PutPendingValidator(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingValidator", reflect.TypeOf((*MockDiff)(nil).PutPendingValidator), arg0)
-}
-
-// SetCurrentSupply mocks base method.
-func (m *MockDiff) SetCurrentSupply(arg0 ids.ID, arg1 uint64) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "SetCurrentSupply", arg0, arg1)
-}
-
-// SetCurrentSupply indicates an expected call of SetCurrentSupply.
-func (mr *MockDiffMockRecorder) SetCurrentSupply(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetCurrentSupply", reflect.TypeOf((*MockDiff)(nil).SetCurrentSupply), arg0, arg1)
-}
-
-// SetDelegateeReward mocks base method.
-func (m *MockDiff) SetDelegateeReward(arg0 ids.ID, arg1 ids.NodeID, arg2 uint64) error {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "SetDelegateeReward", arg0, arg1, arg2)
-	ret0, _ := ret[0].(error)
-	return ret0
-}
-
-// SetDelegateeReward indicates an expected call of SetDelegateeReward.
-func (mr *MockDiffMockRecorder) SetDelegateeReward(arg0, arg1, arg2 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetDelegateeReward", reflect.TypeOf((*MockDiff)(nil).SetDelegateeReward), arg0, arg1, arg2)
-}
-
-// SetSubnetOwner mocks base method.
-func (m *MockDiff) SetSubnetOwner(arg0 ids.ID, arg1 fx.Owner) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "SetSubnetOwner", arg0, arg1)
-}
-
-// SetSubnetOwner indicates an expected call of SetSubnetOwner.
-func (mr *MockDiffMockRecorder) SetSubnetOwner(arg0, arg1 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetSubnetOwner", reflect.TypeOf((*MockDiff)(nil).SetSubnetOwner), arg0, arg1)
-}
-
-// SetTimestamp mocks base method.
-func (m *MockDiff) SetTimestamp(arg0 time.Time) {
-	m.ctrl.T.Helper()
-	m.ctrl.Call(m, "SetTimestamp", arg0)
-}
-
-// SetTimestamp indicates an expected call of SetTimestamp.
-func (mr *MockDiffMockRecorder) SetTimestamp(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTimestamp", reflect.TypeOf((*MockDiff)(nil).SetTimestamp), arg0)
-}
diff --git a/vms/platformvm/state/mock_state.go b/vms/platformvm/state/mock_state.go
index 41ce946a12e0..465b3fda2bac 100644
--- a/vms/platformvm/state/mock_state.go
+++ b/vms/platformvm/state/mock_state.go
@@ -2,7 +2,7 @@
 // See the file LICENSE for licensing terms.
 
 // Code generated by MockGen. DO NOT EDIT.
-// Source: github.com/ava-labs/avalanchego/vms/platformvm/state (interfaces: State)
+// Source: github.com/ava-labs/avalanchego/vms/platformvm/state (interfaces: Chain,Diff,State,Versions)
 
 // Package state is a generated GoMock package.
 package state
@@ -25,6 +25,1006 @@ import (
 	gomock "go.uber.org/mock/gomock"
 )
 
+// MockChain is a mock of Chain interface.
+type MockChain struct {
+	ctrl     *gomock.Controller
+	recorder *MockChainMockRecorder
+}
+
+// MockChainMockRecorder is the mock recorder for MockChain.
+type MockChainMockRecorder struct {
+	mock *MockChain
+}
+
+// NewMockChain creates a new mock instance.
+func NewMockChain(ctrl *gomock.Controller) *MockChain {
+	mock := &MockChain{ctrl: ctrl}
+	mock.recorder = &MockChainMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockChain) EXPECT() *MockChainMockRecorder {
+	return m.recorder
+}
+
+// AddChain mocks base method.
+func (m *MockChain) AddChain(arg0 *txs.Tx) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddChain", arg0)
+}
+
+// AddChain indicates an expected call of AddChain.
+func (mr *MockChainMockRecorder) AddChain(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddChain", reflect.TypeOf((*MockChain)(nil).AddChain), arg0)
+}
+
+// AddRewardUTXO mocks base method.
+func (m *MockChain) AddRewardUTXO(arg0 ids.ID, arg1 *avax.UTXO) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddRewardUTXO", arg0, arg1)
+}
+
+// AddRewardUTXO indicates an expected call of AddRewardUTXO.
+func (mr *MockChainMockRecorder) AddRewardUTXO(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddRewardUTXO", reflect.TypeOf((*MockChain)(nil).AddRewardUTXO), arg0, arg1)
+}
+
+// AddSubnet mocks base method.
+func (m *MockChain) AddSubnet(arg0 *txs.Tx) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddSubnet", arg0)
+}
+
+// AddSubnet indicates an expected call of AddSubnet.
+func (mr *MockChainMockRecorder) AddSubnet(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnet", reflect.TypeOf((*MockChain)(nil).AddSubnet), arg0)
+}
+
+// AddSubnetTransformation mocks base method.
+func (m *MockChain) AddSubnetTransformation(arg0 *txs.Tx) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddSubnetTransformation", arg0)
+}
+
+// AddSubnetTransformation indicates an expected call of AddSubnetTransformation.
+func (mr *MockChainMockRecorder) AddSubnetTransformation(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnetTransformation", reflect.TypeOf((*MockChain)(nil).AddSubnetTransformation), arg0)
+}
+
+// AddTx mocks base method.
+func (m *MockChain) AddTx(arg0 *txs.Tx, arg1 status.Status) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddTx", arg0, arg1)
+}
+
+// AddTx indicates an expected call of AddTx.
+func (mr *MockChainMockRecorder) AddTx(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddTx", reflect.TypeOf((*MockChain)(nil).AddTx), arg0, arg1)
+}
+
+// AddUTXO mocks base method.
+func (m *MockChain) AddUTXO(arg0 *avax.UTXO) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddUTXO", arg0)
+}
+
+// AddUTXO indicates an expected call of AddUTXO.
+func (mr *MockChainMockRecorder) AddUTXO(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddUTXO", reflect.TypeOf((*MockChain)(nil).AddUTXO), arg0)
+}
+
+// DeleteCurrentDelegator mocks base method.
+func (m *MockChain) DeleteCurrentDelegator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeleteCurrentDelegator", arg0)
+}
+
+// DeleteCurrentDelegator indicates an expected call of DeleteCurrentDelegator.
+func (mr *MockChainMockRecorder) DeleteCurrentDelegator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentDelegator", reflect.TypeOf((*MockChain)(nil).DeleteCurrentDelegator), arg0)
+}
+
+// DeleteCurrentValidator mocks base method.
+func (m *MockChain) DeleteCurrentValidator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeleteCurrentValidator", arg0)
+}
+
+// DeleteCurrentValidator indicates an expected call of DeleteCurrentValidator.
+func (mr *MockChainMockRecorder) DeleteCurrentValidator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentValidator", reflect.TypeOf((*MockChain)(nil).DeleteCurrentValidator), arg0)
+}
+
+// DeletePendingDelegator mocks base method.
+func (m *MockChain) DeletePendingDelegator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeletePendingDelegator", arg0)
+}
+
+// DeletePendingDelegator indicates an expected call of DeletePendingDelegator.
+func (mr *MockChainMockRecorder) DeletePendingDelegator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingDelegator", reflect.TypeOf((*MockChain)(nil).DeletePendingDelegator), arg0)
+}
+
+// DeletePendingValidator mocks base method.
+func (m *MockChain) DeletePendingValidator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeletePendingValidator", arg0)
+}
+
+// DeletePendingValidator indicates an expected call of DeletePendingValidator.
+func (mr *MockChainMockRecorder) DeletePendingValidator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingValidator", reflect.TypeOf((*MockChain)(nil).DeletePendingValidator), arg0)
+}
+
+// DeleteUTXO mocks base method.
+func (m *MockChain) DeleteUTXO(arg0 ids.ID) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeleteUTXO", arg0)
+}
+
+// DeleteUTXO indicates an expected call of DeleteUTXO.
+func (mr *MockChainMockRecorder) DeleteUTXO(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteUTXO", reflect.TypeOf((*MockChain)(nil).DeleteUTXO), arg0)
+}
+
+// GetChains mocks base method.
+func (m *MockChain) GetChains(arg0 ids.ID) ([]*txs.Tx, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetChains", arg0)
+	ret0, _ := ret[0].([]*txs.Tx)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetChains indicates an expected call of GetChains.
+func (mr *MockChainMockRecorder) GetChains(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetChains", reflect.TypeOf((*MockChain)(nil).GetChains), arg0)
+}
+
+// GetCurrentDelegatorIterator mocks base method.
+func (m *MockChain) GetCurrentDelegatorIterator(arg0 ids.ID, arg1 ids.NodeID) (StakerIterator, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetCurrentDelegatorIterator", arg0, arg1)
+	ret0, _ := ret[0].(StakerIterator)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetCurrentDelegatorIterator indicates an expected call of GetCurrentDelegatorIterator.
+func (mr *MockChainMockRecorder) GetCurrentDelegatorIterator(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentDelegatorIterator", reflect.TypeOf((*MockChain)(nil).GetCurrentDelegatorIterator), arg0, arg1)
+}
+
+// GetCurrentStakerIterator mocks base method.
+func (m *MockChain) GetCurrentStakerIterator() (StakerIterator, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetCurrentStakerIterator")
+	ret0, _ := ret[0].(StakerIterator)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetCurrentStakerIterator indicates an expected call of GetCurrentStakerIterator.
+func (mr *MockChainMockRecorder) GetCurrentStakerIterator() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentStakerIterator", reflect.TypeOf((*MockChain)(nil).GetCurrentStakerIterator))
+}
+
+// GetCurrentSupply mocks base method.
+func (m *MockChain) GetCurrentSupply(arg0 ids.ID) (uint64, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetCurrentSupply", arg0)
+	ret0, _ := ret[0].(uint64)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetCurrentSupply indicates an expected call of GetCurrentSupply.
+func (mr *MockChainMockRecorder) GetCurrentSupply(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentSupply", reflect.TypeOf((*MockChain)(nil).GetCurrentSupply), arg0)
+}
+
+// GetCurrentValidator mocks base method.
+func (m *MockChain) GetCurrentValidator(arg0 ids.ID, arg1 ids.NodeID) (*Staker, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetCurrentValidator", arg0, arg1)
+	ret0, _ := ret[0].(*Staker)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetCurrentValidator indicates an expected call of GetCurrentValidator.
+func (mr *MockChainMockRecorder) GetCurrentValidator(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentValidator", reflect.TypeOf((*MockChain)(nil).GetCurrentValidator), arg0, arg1)
+}
+
+// GetDelegateeReward mocks base method.
+func (m *MockChain) GetDelegateeReward(arg0 ids.ID, arg1 ids.NodeID) (uint64, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetDelegateeReward", arg0, arg1)
+	ret0, _ := ret[0].(uint64)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetDelegateeReward indicates an expected call of GetDelegateeReward.
+func (mr *MockChainMockRecorder) GetDelegateeReward(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDelegateeReward", reflect.TypeOf((*MockChain)(nil).GetDelegateeReward), arg0, arg1)
+}
+
+// GetPendingDelegatorIterator mocks base method.
+func (m *MockChain) GetPendingDelegatorIterator(arg0 ids.ID, arg1 ids.NodeID) (StakerIterator, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetPendingDelegatorIterator", arg0, arg1)
+	ret0, _ := ret[0].(StakerIterator)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetPendingDelegatorIterator indicates an expected call of GetPendingDelegatorIterator.
+func (mr *MockChainMockRecorder) GetPendingDelegatorIterator(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingDelegatorIterator", reflect.TypeOf((*MockChain)(nil).GetPendingDelegatorIterator), arg0, arg1)
+}
+
+// GetPendingStakerIterator mocks base method.
+func (m *MockChain) GetPendingStakerIterator() (StakerIterator, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetPendingStakerIterator")
+	ret0, _ := ret[0].(StakerIterator)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetPendingStakerIterator indicates an expected call of GetPendingStakerIterator.
+func (mr *MockChainMockRecorder) GetPendingStakerIterator() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingStakerIterator", reflect.TypeOf((*MockChain)(nil).GetPendingStakerIterator))
+}
+
+// GetPendingValidator mocks base method.
+func (m *MockChain) GetPendingValidator(arg0 ids.ID, arg1 ids.NodeID) (*Staker, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetPendingValidator", arg0, arg1)
+	ret0, _ := ret[0].(*Staker)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetPendingValidator indicates an expected call of GetPendingValidator.
+func (mr *MockChainMockRecorder) GetPendingValidator(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingValidator", reflect.TypeOf((*MockChain)(nil).GetPendingValidator), arg0, arg1)
+}
+
+// GetRewardUTXOs mocks base method.
+func (m *MockChain) GetRewardUTXOs(arg0 ids.ID) ([]*avax.UTXO, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetRewardUTXOs", arg0)
+	ret0, _ := ret[0].([]*avax.UTXO)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetRewardUTXOs indicates an expected call of GetRewardUTXOs.
+func (mr *MockChainMockRecorder) GetRewardUTXOs(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRewardUTXOs", reflect.TypeOf((*MockChain)(nil).GetRewardUTXOs), arg0)
+}
+
+// GetSubnetOwner mocks base method.
+func (m *MockChain) GetSubnetOwner(arg0 ids.ID) (fx.Owner, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetSubnetOwner", arg0)
+	ret0, _ := ret[0].(fx.Owner)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetSubnetOwner indicates an expected call of GetSubnetOwner.
+func (mr *MockChainMockRecorder) GetSubnetOwner(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetOwner", reflect.TypeOf((*MockChain)(nil).GetSubnetOwner), arg0)
+}
+
+// GetSubnetTransformation mocks base method.
+func (m *MockChain) GetSubnetTransformation(arg0 ids.ID) (*txs.Tx, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetSubnetTransformation", arg0)
+	ret0, _ := ret[0].(*txs.Tx)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetSubnetTransformation indicates an expected call of GetSubnetTransformation.
+func (mr *MockChainMockRecorder) GetSubnetTransformation(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetTransformation", reflect.TypeOf((*MockChain)(nil).GetSubnetTransformation), arg0)
+}
+
+// GetSubnets mocks base method.
+func (m *MockChain) GetSubnets() ([]*txs.Tx, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetSubnets")
+	ret0, _ := ret[0].([]*txs.Tx)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetSubnets indicates an expected call of GetSubnets.
+func (mr *MockChainMockRecorder) GetSubnets() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnets", reflect.TypeOf((*MockChain)(nil).GetSubnets))
+}
+
+// GetTimestamp mocks base method.
+func (m *MockChain) GetTimestamp() time.Time {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetTimestamp")
+	ret0, _ := ret[0].(time.Time)
+	return ret0
+}
+
+// GetTimestamp indicates an expected call of GetTimestamp.
+func (mr *MockChainMockRecorder) GetTimestamp() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTimestamp", reflect.TypeOf((*MockChain)(nil).GetTimestamp))
+}
+
+// GetTx mocks base method.
+func (m *MockChain) GetTx(arg0 ids.ID) (*txs.Tx, status.Status, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetTx", arg0)
+	ret0, _ := ret[0].(*txs.Tx)
+	ret1, _ := ret[1].(status.Status)
+	ret2, _ := ret[2].(error)
+	return ret0, ret1, ret2
+}
+
+// GetTx indicates an expected call of GetTx.
+func (mr *MockChainMockRecorder) GetTx(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTx", reflect.TypeOf((*MockChain)(nil).GetTx), arg0)
+}
+
+// GetUTXO mocks base method.
+func (m *MockChain) GetUTXO(arg0 ids.ID) (*avax.UTXO, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetUTXO", arg0)
+	ret0, _ := ret[0].(*avax.UTXO)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetUTXO indicates an expected call of GetUTXO.
+func (mr *MockChainMockRecorder) GetUTXO(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUTXO", reflect.TypeOf((*MockChain)(nil).GetUTXO), arg0)
+}
+
+// PutCurrentDelegator mocks base method.
+func (m *MockChain) PutCurrentDelegator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "PutCurrentDelegator", arg0)
+}
+
+// PutCurrentDelegator indicates an expected call of PutCurrentDelegator.
+func (mr *MockChainMockRecorder) PutCurrentDelegator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentDelegator", reflect.TypeOf((*MockChain)(nil).PutCurrentDelegator), arg0)
+}
+
+// PutCurrentValidator mocks base method.
+func (m *MockChain) PutCurrentValidator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "PutCurrentValidator", arg0)
+}
+
+// PutCurrentValidator indicates an expected call of PutCurrentValidator.
+func (mr *MockChainMockRecorder) PutCurrentValidator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentValidator", reflect.TypeOf((*MockChain)(nil).PutCurrentValidator), arg0)
+}
+
+// PutPendingDelegator mocks base method.
+func (m *MockChain) PutPendingDelegator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "PutPendingDelegator", arg0)
+}
+
+// PutPendingDelegator indicates an expected call of PutPendingDelegator.
+func (mr *MockChainMockRecorder) PutPendingDelegator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingDelegator", reflect.TypeOf((*MockChain)(nil).PutPendingDelegator), arg0)
+}
+
+// PutPendingValidator mocks base method.
+func (m *MockChain) PutPendingValidator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "PutPendingValidator", arg0)
+}
+
+// PutPendingValidator indicates an expected call of PutPendingValidator.
+func (mr *MockChainMockRecorder) PutPendingValidator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingValidator", reflect.TypeOf((*MockChain)(nil).PutPendingValidator), arg0)
+}
+
+// SetCurrentSupply mocks base method.
+func (m *MockChain) SetCurrentSupply(arg0 ids.ID, arg1 uint64) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "SetCurrentSupply", arg0, arg1)
+}
+
+// SetCurrentSupply indicates an expected call of SetCurrentSupply.
+func (mr *MockChainMockRecorder) SetCurrentSupply(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetCurrentSupply", reflect.TypeOf((*MockChain)(nil).SetCurrentSupply), arg0, arg1)
+}
+
+// SetDelegateeReward mocks base method.
+func (m *MockChain) SetDelegateeReward(arg0 ids.ID, arg1 ids.NodeID, arg2 uint64) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "SetDelegateeReward", arg0, arg1, arg2)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// SetDelegateeReward indicates an expected call of SetDelegateeReward.
+func (mr *MockChainMockRecorder) SetDelegateeReward(arg0, arg1, arg2 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetDelegateeReward", reflect.TypeOf((*MockChain)(nil).SetDelegateeReward), arg0, arg1, arg2)
+}
+
+// SetSubnetOwner mocks base method.
+func (m *MockChain) SetSubnetOwner(arg0 ids.ID, arg1 fx.Owner) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "SetSubnetOwner", arg0, arg1)
+}
+
+// SetSubnetOwner indicates an expected call of SetSubnetOwner.
+func (mr *MockChainMockRecorder) SetSubnetOwner(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetSubnetOwner", reflect.TypeOf((*MockChain)(nil).SetSubnetOwner), arg0, arg1)
+}
+
+// SetTimestamp mocks base method.
+func (m *MockChain) SetTimestamp(arg0 time.Time) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "SetTimestamp", arg0)
+}
+
+// SetTimestamp indicates an expected call of SetTimestamp.
+func (mr *MockChainMockRecorder) SetTimestamp(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTimestamp", reflect.TypeOf((*MockChain)(nil).SetTimestamp), arg0)
+}
+
+// MockDiff is a mock of Diff interface.
+type MockDiff struct {
+	ctrl     *gomock.Controller
+	recorder *MockDiffMockRecorder
+}
+
+// MockDiffMockRecorder is the mock recorder for MockDiff.
+type MockDiffMockRecorder struct {
+	mock *MockDiff
+}
+
+// NewMockDiff creates a new mock instance.
+func NewMockDiff(ctrl *gomock.Controller) *MockDiff {
+	mock := &MockDiff{ctrl: ctrl}
+	mock.recorder = &MockDiffMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockDiff) EXPECT() *MockDiffMockRecorder {
+	return m.recorder
+}
+
+// AddChain mocks base method.
+func (m *MockDiff) AddChain(arg0 *txs.Tx) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddChain", arg0)
+}
+
+// AddChain indicates an expected call of AddChain.
+func (mr *MockDiffMockRecorder) AddChain(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddChain", reflect.TypeOf((*MockDiff)(nil).AddChain), arg0)
+}
+
+// AddRewardUTXO mocks base method.
+func (m *MockDiff) AddRewardUTXO(arg0 ids.ID, arg1 *avax.UTXO) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddRewardUTXO", arg0, arg1)
+}
+
+// AddRewardUTXO indicates an expected call of AddRewardUTXO.
+func (mr *MockDiffMockRecorder) AddRewardUTXO(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddRewardUTXO", reflect.TypeOf((*MockDiff)(nil).AddRewardUTXO), arg0, arg1)
+}
+
+// AddSubnet mocks base method.
+func (m *MockDiff) AddSubnet(arg0 *txs.Tx) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddSubnet", arg0)
+}
+
+// AddSubnet indicates an expected call of AddSubnet.
+func (mr *MockDiffMockRecorder) AddSubnet(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnet", reflect.TypeOf((*MockDiff)(nil).AddSubnet), arg0)
+}
+
+// AddSubnetTransformation mocks base method.
+func (m *MockDiff) AddSubnetTransformation(arg0 *txs.Tx) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddSubnetTransformation", arg0)
+}
+
+// AddSubnetTransformation indicates an expected call of AddSubnetTransformation.
+func (mr *MockDiffMockRecorder) AddSubnetTransformation(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSubnetTransformation", reflect.TypeOf((*MockDiff)(nil).AddSubnetTransformation), arg0)
+}
+
+// AddTx mocks base method.
+func (m *MockDiff) AddTx(arg0 *txs.Tx, arg1 status.Status) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddTx", arg0, arg1)
+}
+
+// AddTx indicates an expected call of AddTx.
+func (mr *MockDiffMockRecorder) AddTx(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddTx", reflect.TypeOf((*MockDiff)(nil).AddTx), arg0, arg1)
+}
+
+// AddUTXO mocks base method.
+func (m *MockDiff) AddUTXO(arg0 *avax.UTXO) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "AddUTXO", arg0)
+}
+
+// AddUTXO indicates an expected call of AddUTXO.
+func (mr *MockDiffMockRecorder) AddUTXO(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddUTXO", reflect.TypeOf((*MockDiff)(nil).AddUTXO), arg0)
+}
+
+// Apply mocks base method.
+func (m *MockDiff) Apply(arg0 Chain) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Apply", arg0)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Apply indicates an expected call of Apply.
+func (mr *MockDiffMockRecorder) Apply(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Apply", reflect.TypeOf((*MockDiff)(nil).Apply), arg0)
+}
+
+// DeleteCurrentDelegator mocks base method.
+func (m *MockDiff) DeleteCurrentDelegator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeleteCurrentDelegator", arg0)
+}
+
+// DeleteCurrentDelegator indicates an expected call of DeleteCurrentDelegator.
+func (mr *MockDiffMockRecorder) DeleteCurrentDelegator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentDelegator", reflect.TypeOf((*MockDiff)(nil).DeleteCurrentDelegator), arg0)
+}
+
+// DeleteCurrentValidator mocks base method.
+func (m *MockDiff) DeleteCurrentValidator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeleteCurrentValidator", arg0)
+}
+
+// DeleteCurrentValidator indicates an expected call of DeleteCurrentValidator.
+func (mr *MockDiffMockRecorder) DeleteCurrentValidator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCurrentValidator", reflect.TypeOf((*MockDiff)(nil).DeleteCurrentValidator), arg0)
+}
+
+// DeletePendingDelegator mocks base method.
+func (m *MockDiff) DeletePendingDelegator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeletePendingDelegator", arg0)
+}
+
+// DeletePendingDelegator indicates an expected call of DeletePendingDelegator.
+func (mr *MockDiffMockRecorder) DeletePendingDelegator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingDelegator", reflect.TypeOf((*MockDiff)(nil).DeletePendingDelegator), arg0)
+}
+
+// DeletePendingValidator mocks base method.
+func (m *MockDiff) DeletePendingValidator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeletePendingValidator", arg0)
+}
+
+// DeletePendingValidator indicates an expected call of DeletePendingValidator.
+func (mr *MockDiffMockRecorder) DeletePendingValidator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePendingValidator", reflect.TypeOf((*MockDiff)(nil).DeletePendingValidator), arg0)
+}
+
+// DeleteUTXO mocks base method.
+func (m *MockDiff) DeleteUTXO(arg0 ids.ID) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "DeleteUTXO", arg0)
+}
+
+// DeleteUTXO indicates an expected call of DeleteUTXO.
+func (mr *MockDiffMockRecorder) DeleteUTXO(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteUTXO", reflect.TypeOf((*MockDiff)(nil).DeleteUTXO), arg0)
+}
+
+// GetChains mocks base method.
+func (m *MockDiff) GetChains(arg0 ids.ID) ([]*txs.Tx, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetChains", arg0)
+	ret0, _ := ret[0].([]*txs.Tx)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetChains indicates an expected call of GetChains.
+func (mr *MockDiffMockRecorder) GetChains(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetChains", reflect.TypeOf((*MockDiff)(nil).GetChains), arg0)
+}
+
+// GetCurrentDelegatorIterator mocks base method.
+func (m *MockDiff) GetCurrentDelegatorIterator(arg0 ids.ID, arg1 ids.NodeID) (StakerIterator, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetCurrentDelegatorIterator", arg0, arg1)
+	ret0, _ := ret[0].(StakerIterator)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetCurrentDelegatorIterator indicates an expected call of GetCurrentDelegatorIterator.
+func (mr *MockDiffMockRecorder) GetCurrentDelegatorIterator(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentDelegatorIterator", reflect.TypeOf((*MockDiff)(nil).GetCurrentDelegatorIterator), arg0, arg1)
+}
+
+// GetCurrentStakerIterator mocks base method.
+func (m *MockDiff) GetCurrentStakerIterator() (StakerIterator, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetCurrentStakerIterator")
+	ret0, _ := ret[0].(StakerIterator)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetCurrentStakerIterator indicates an expected call of GetCurrentStakerIterator.
+func (mr *MockDiffMockRecorder) GetCurrentStakerIterator() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentStakerIterator", reflect.TypeOf((*MockDiff)(nil).GetCurrentStakerIterator))
+}
+
+// GetCurrentSupply mocks base method.
+func (m *MockDiff) GetCurrentSupply(arg0 ids.ID) (uint64, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetCurrentSupply", arg0)
+	ret0, _ := ret[0].(uint64)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetCurrentSupply indicates an expected call of GetCurrentSupply.
+func (mr *MockDiffMockRecorder) GetCurrentSupply(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentSupply", reflect.TypeOf((*MockDiff)(nil).GetCurrentSupply), arg0)
+}
+
+// GetCurrentValidator mocks base method.
+func (m *MockDiff) GetCurrentValidator(arg0 ids.ID, arg1 ids.NodeID) (*Staker, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetCurrentValidator", arg0, arg1)
+	ret0, _ := ret[0].(*Staker)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetCurrentValidator indicates an expected call of GetCurrentValidator.
+func (mr *MockDiffMockRecorder) GetCurrentValidator(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCurrentValidator", reflect.TypeOf((*MockDiff)(nil).GetCurrentValidator), arg0, arg1)
+}
+
+// GetDelegateeReward mocks base method.
+func (m *MockDiff) GetDelegateeReward(arg0 ids.ID, arg1 ids.NodeID) (uint64, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetDelegateeReward", arg0, arg1)
+	ret0, _ := ret[0].(uint64)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetDelegateeReward indicates an expected call of GetDelegateeReward.
+func (mr *MockDiffMockRecorder) GetDelegateeReward(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDelegateeReward", reflect.TypeOf((*MockDiff)(nil).GetDelegateeReward), arg0, arg1)
+}
+
+// GetPendingDelegatorIterator mocks base method.
+func (m *MockDiff) GetPendingDelegatorIterator(arg0 ids.ID, arg1 ids.NodeID) (StakerIterator, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetPendingDelegatorIterator", arg0, arg1)
+	ret0, _ := ret[0].(StakerIterator)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetPendingDelegatorIterator indicates an expected call of GetPendingDelegatorIterator.
+func (mr *MockDiffMockRecorder) GetPendingDelegatorIterator(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingDelegatorIterator", reflect.TypeOf((*MockDiff)(nil).GetPendingDelegatorIterator), arg0, arg1)
+}
+
+// GetPendingStakerIterator mocks base method.
+func (m *MockDiff) GetPendingStakerIterator() (StakerIterator, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetPendingStakerIterator")
+	ret0, _ := ret[0].(StakerIterator)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetPendingStakerIterator indicates an expected call of GetPendingStakerIterator.
+func (mr *MockDiffMockRecorder) GetPendingStakerIterator() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingStakerIterator", reflect.TypeOf((*MockDiff)(nil).GetPendingStakerIterator))
+}
+
+// GetPendingValidator mocks base method.
+func (m *MockDiff) GetPendingValidator(arg0 ids.ID, arg1 ids.NodeID) (*Staker, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetPendingValidator", arg0, arg1)
+	ret0, _ := ret[0].(*Staker)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetPendingValidator indicates an expected call of GetPendingValidator.
+func (mr *MockDiffMockRecorder) GetPendingValidator(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPendingValidator", reflect.TypeOf((*MockDiff)(nil).GetPendingValidator), arg0, arg1)
+}
+
+// GetRewardUTXOs mocks base method.
+func (m *MockDiff) GetRewardUTXOs(arg0 ids.ID) ([]*avax.UTXO, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetRewardUTXOs", arg0)
+	ret0, _ := ret[0].([]*avax.UTXO)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetRewardUTXOs indicates an expected call of GetRewardUTXOs.
+func (mr *MockDiffMockRecorder) GetRewardUTXOs(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRewardUTXOs", reflect.TypeOf((*MockDiff)(nil).GetRewardUTXOs), arg0)
+}
+
+// GetSubnetOwner mocks base method.
+func (m *MockDiff) GetSubnetOwner(arg0 ids.ID) (fx.Owner, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetSubnetOwner", arg0)
+	ret0, _ := ret[0].(fx.Owner)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetSubnetOwner indicates an expected call of GetSubnetOwner.
+func (mr *MockDiffMockRecorder) GetSubnetOwner(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetOwner", reflect.TypeOf((*MockDiff)(nil).GetSubnetOwner), arg0)
+}
+
+// GetSubnetTransformation mocks base method.
+func (m *MockDiff) GetSubnetTransformation(arg0 ids.ID) (*txs.Tx, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetSubnetTransformation", arg0)
+	ret0, _ := ret[0].(*txs.Tx)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetSubnetTransformation indicates an expected call of GetSubnetTransformation.
+func (mr *MockDiffMockRecorder) GetSubnetTransformation(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnetTransformation", reflect.TypeOf((*MockDiff)(nil).GetSubnetTransformation), arg0)
+}
+
+// GetSubnets mocks base method.
+func (m *MockDiff) GetSubnets() ([]*txs.Tx, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetSubnets")
+	ret0, _ := ret[0].([]*txs.Tx)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetSubnets indicates an expected call of GetSubnets.
+func (mr *MockDiffMockRecorder) GetSubnets() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSubnets", reflect.TypeOf((*MockDiff)(nil).GetSubnets))
+}
+
+// GetTimestamp mocks base method.
+func (m *MockDiff) GetTimestamp() time.Time {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetTimestamp")
+	ret0, _ := ret[0].(time.Time)
+	return ret0
+}
+
+// GetTimestamp indicates an expected call of GetTimestamp.
+func (mr *MockDiffMockRecorder) GetTimestamp() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTimestamp", reflect.TypeOf((*MockDiff)(nil).GetTimestamp))
+}
+
+// GetTx mocks base method.
+func (m *MockDiff) GetTx(arg0 ids.ID) (*txs.Tx, status.Status, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetTx", arg0)
+	ret0, _ := ret[0].(*txs.Tx)
+	ret1, _ := ret[1].(status.Status)
+	ret2, _ := ret[2].(error)
+	return ret0, ret1, ret2
+}
+
+// GetTx indicates an expected call of GetTx.
+func (mr *MockDiffMockRecorder) GetTx(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTx", reflect.TypeOf((*MockDiff)(nil).GetTx), arg0)
+}
+
+// GetUTXO mocks base method.
+func (m *MockDiff) GetUTXO(arg0 ids.ID) (*avax.UTXO, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetUTXO", arg0)
+	ret0, _ := ret[0].(*avax.UTXO)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetUTXO indicates an expected call of GetUTXO.
+func (mr *MockDiffMockRecorder) GetUTXO(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUTXO", reflect.TypeOf((*MockDiff)(nil).GetUTXO), arg0)
+}
+
+// PutCurrentDelegator mocks base method.
+func (m *MockDiff) PutCurrentDelegator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "PutCurrentDelegator", arg0)
+}
+
+// PutCurrentDelegator indicates an expected call of PutCurrentDelegator.
+func (mr *MockDiffMockRecorder) PutCurrentDelegator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentDelegator", reflect.TypeOf((*MockDiff)(nil).PutCurrentDelegator), arg0)
+}
+
+// PutCurrentValidator mocks base method.
+func (m *MockDiff) PutCurrentValidator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "PutCurrentValidator", arg0)
+}
+
+// PutCurrentValidator indicates an expected call of PutCurrentValidator.
+func (mr *MockDiffMockRecorder) PutCurrentValidator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutCurrentValidator", reflect.TypeOf((*MockDiff)(nil).PutCurrentValidator), arg0)
+}
+
+// PutPendingDelegator mocks base method.
+func (m *MockDiff) PutPendingDelegator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "PutPendingDelegator", arg0)
+}
+
+// PutPendingDelegator indicates an expected call of PutPendingDelegator.
+func (mr *MockDiffMockRecorder) PutPendingDelegator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingDelegator", reflect.TypeOf((*MockDiff)(nil).PutPendingDelegator), arg0)
+}
+
+// PutPendingValidator mocks base method.
+func (m *MockDiff) PutPendingValidator(arg0 *Staker) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "PutPendingValidator", arg0)
+}
+
+// PutPendingValidator indicates an expected call of PutPendingValidator.
+func (mr *MockDiffMockRecorder) PutPendingValidator(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutPendingValidator", reflect.TypeOf((*MockDiff)(nil).PutPendingValidator), arg0)
+}
+
+// SetCurrentSupply mocks base method.
+func (m *MockDiff) SetCurrentSupply(arg0 ids.ID, arg1 uint64) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "SetCurrentSupply", arg0, arg1)
+}
+
+// SetCurrentSupply indicates an expected call of SetCurrentSupply.
+func (mr *MockDiffMockRecorder) SetCurrentSupply(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetCurrentSupply", reflect.TypeOf((*MockDiff)(nil).SetCurrentSupply), arg0, arg1)
+}
+
+// SetDelegateeReward mocks base method.
+func (m *MockDiff) SetDelegateeReward(arg0 ids.ID, arg1 ids.NodeID, arg2 uint64) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "SetDelegateeReward", arg0, arg1, arg2)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// SetDelegateeReward indicates an expected call of SetDelegateeReward.
+func (mr *MockDiffMockRecorder) SetDelegateeReward(arg0, arg1, arg2 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetDelegateeReward", reflect.TypeOf((*MockDiff)(nil).SetDelegateeReward), arg0, arg1, arg2)
+}
+
+// SetSubnetOwner mocks base method.
+func (m *MockDiff) SetSubnetOwner(arg0 ids.ID, arg1 fx.Owner) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "SetSubnetOwner", arg0, arg1)
+}
+
+// SetSubnetOwner indicates an expected call of SetSubnetOwner.
+func (mr *MockDiffMockRecorder) SetSubnetOwner(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetSubnetOwner", reflect.TypeOf((*MockDiff)(nil).SetSubnetOwner), arg0, arg1)
+}
+
+// SetTimestamp mocks base method.
+func (m *MockDiff) SetTimestamp(arg0 time.Time) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "SetTimestamp", arg0)
+}
+
+// SetTimestamp indicates an expected call of SetTimestamp.
+func (mr *MockDiffMockRecorder) SetTimestamp(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTimestamp", reflect.TypeOf((*MockDiff)(nil).SetTimestamp), arg0)
+}
+
 // MockState is a mock of State interface.
 type MockState struct {
 	ctrl     *gomock.Controller
@@ -783,3 +1783,41 @@ func (mr *MockStateMockRecorder) UTXOIDs(arg0, arg1, arg2 interface{}) *gomock.C
 	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UTXOIDs", reflect.TypeOf((*MockState)(nil).UTXOIDs), arg0, arg1, arg2)
 }
+
+// MockVersions is a mock of Versions interface.
+type MockVersions struct {
+	ctrl     *gomock.Controller
+	recorder *MockVersionsMockRecorder
+}
+
+// MockVersionsMockRecorder is the mock recorder for MockVersions.
+type MockVersionsMockRecorder struct {
+	mock *MockVersions
+}
+
+// NewMockVersions creates a new mock instance.
+func NewMockVersions(ctrl *gomock.Controller) *MockVersions {
+	mock := &MockVersions{ctrl: ctrl}
+	mock.recorder = &MockVersionsMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockVersions) EXPECT() *MockVersionsMockRecorder {
+	return m.recorder
+}
+
+// GetState mocks base method.
+func (m *MockVersions) GetState(arg0 ids.ID) (Chain, bool) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetState", arg0)
+	ret0, _ := ret[0].(Chain)
+	ret1, _ := ret[1].(bool)
+	return ret0, ret1
+}
+
+// GetState indicates an expected call of GetState.
+func (mr *MockVersionsMockRecorder) GetState(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetState", reflect.TypeOf((*MockVersions)(nil).GetState), arg0)
+}
diff --git a/vms/platformvm/state/mock_versions.go b/vms/platformvm/state/mock_versions.go
deleted file mode 100644
index 3f8a20550b22..000000000000
--- a/vms/platformvm/state/mock_versions.go
+++ /dev/null
@@ -1,53 +0,0 @@
-// Copyright (C) 2019-2023, Ava Labs, Inc. All rights reserved.
-// See the file LICENSE for licensing terms.
-
-// Code generated by MockGen. DO NOT EDIT.
-// Source: github.com/ava-labs/avalanchego/vms/platformvm/state (interfaces: Versions)
-
-// Package state is a generated GoMock package.
-package state
-
-import (
-	reflect "reflect"
-
-	ids "github.com/ava-labs/avalanchego/ids"
-	gomock "go.uber.org/mock/gomock"
-)
-
-// MockVersions is a mock of Versions interface.
-type MockVersions struct {
-	ctrl     *gomock.Controller
-	recorder *MockVersionsMockRecorder
-}
-
-// MockVersionsMockRecorder is the mock recorder for MockVersions.
-type MockVersionsMockRecorder struct {
-	mock *MockVersions
-}
-
-// NewMockVersions creates a new mock instance.
-func NewMockVersions(ctrl *gomock.Controller) *MockVersions {
-	mock := &MockVersions{ctrl: ctrl}
-	mock.recorder = &MockVersionsMockRecorder{mock}
-	return mock
-}
-
-// EXPECT returns an object that allows the caller to indicate expected use.
-func (m *MockVersions) EXPECT() *MockVersionsMockRecorder {
-	return m.recorder
-}
-
-// GetState mocks base method.
-func (m *MockVersions) GetState(arg0 ids.ID) (Chain, bool) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetState", arg0)
-	ret0, _ := ret[0].(Chain)
-	ret1, _ := ret[1].(bool)
-	return ret0, ret1
-}
-
-// GetState indicates an expected call of GetState.
-func (mr *MockVersionsMockRecorder) GetState(arg0 interface{}) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetState", reflect.TypeOf((*MockVersions)(nil).GetState), arg0)
-}