diff --git a/state/protocol/mock/snapshot_execution_subset.go b/state/protocol/mock/snapshot_execution_subset.go
new file mode 100644
index 00000000000..1950f9adace
--- /dev/null
+++ b/state/protocol/mock/snapshot_execution_subset.go
@@ -0,0 +1,87 @@
+// Code generated by mockery v2.43.2. DO NOT EDIT.
+
+package mock
+
+import (
+	flow "github.com/onflow/flow-go/model/flow"
+	mock "github.com/stretchr/testify/mock"
+)
+
+// SnapshotExecutionSubset is an autogenerated mock type for the SnapshotExecutionSubset type
+type SnapshotExecutionSubset struct {
+	mock.Mock
+}
+
+// RandomSource provides a mock function with given fields:
+func (_m *SnapshotExecutionSubset) RandomSource() ([]byte, error) {
+	ret := _m.Called()
+
+	if len(ret) == 0 {
+		panic("no return value specified for RandomSource")
+	}
+
+	var r0 []byte
+	var r1 error
+	if rf, ok := ret.Get(0).(func() ([]byte, error)); ok {
+		return rf()
+	}
+	if rf, ok := ret.Get(0).(func() []byte); ok {
+		r0 = rf()
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).([]byte)
+		}
+	}
+
+	if rf, ok := ret.Get(1).(func() error); ok {
+		r1 = rf()
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// VersionBeacon provides a mock function with given fields:
+func (_m *SnapshotExecutionSubset) VersionBeacon() (*flow.SealedVersionBeacon, error) {
+	ret := _m.Called()
+
+	if len(ret) == 0 {
+		panic("no return value specified for VersionBeacon")
+	}
+
+	var r0 *flow.SealedVersionBeacon
+	var r1 error
+	if rf, ok := ret.Get(0).(func() (*flow.SealedVersionBeacon, error)); ok {
+		return rf()
+	}
+	if rf, ok := ret.Get(0).(func() *flow.SealedVersionBeacon); ok {
+		r0 = rf()
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*flow.SealedVersionBeacon)
+		}
+	}
+
+	if rf, ok := ret.Get(1).(func() error); ok {
+		r1 = rf()
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// NewSnapshotExecutionSubset creates a new instance of SnapshotExecutionSubset. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
+// The first argument is typically a *testing.T value.
+func NewSnapshotExecutionSubset(t interface {
+	mock.TestingT
+	Cleanup(func())
+}) *SnapshotExecutionSubset {
+	mock := &SnapshotExecutionSubset{}
+	mock.Mock.Test(t)
+
+	t.Cleanup(func() { mock.AssertExpectations(t) })
+
+	return mock
+}
diff --git a/state/protocol/mock/snapshot_execution_subset_provider.go b/state/protocol/mock/snapshot_execution_subset_provider.go
new file mode 100644
index 00000000000..445c6daa7f5
--- /dev/null
+++ b/state/protocol/mock/snapshot_execution_subset_provider.go
@@ -0,0 +1,49 @@
+// Code generated by mockery v2.43.2. DO NOT EDIT.
+
+package mock
+
+import (
+	flow "github.com/onflow/flow-go/model/flow"
+	mock "github.com/stretchr/testify/mock"
+
+	protocol "github.com/onflow/flow-go/state/protocol"
+)
+
+// SnapshotExecutionSubsetProvider is an autogenerated mock type for the SnapshotExecutionSubsetProvider type
+type SnapshotExecutionSubsetProvider struct {
+	mock.Mock
+}
+
+// AtBlockID provides a mock function with given fields: blockID
+func (_m *SnapshotExecutionSubsetProvider) AtBlockID(blockID flow.Identifier) protocol.SnapshotExecutionSubset {
+	ret := _m.Called(blockID)
+
+	if len(ret) == 0 {
+		panic("no return value specified for AtBlockID")
+	}
+
+	var r0 protocol.SnapshotExecutionSubset
+	if rf, ok := ret.Get(0).(func(flow.Identifier) protocol.SnapshotExecutionSubset); ok {
+		r0 = rf(blockID)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(protocol.SnapshotExecutionSubset)
+		}
+	}
+
+	return r0
+}
+
+// NewSnapshotExecutionSubsetProvider creates a new instance of SnapshotExecutionSubsetProvider. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
+// The first argument is typically a *testing.T value.
+func NewSnapshotExecutionSubsetProvider(t interface {
+	mock.TestingT
+	Cleanup(func())
+}) *SnapshotExecutionSubsetProvider {
+	mock := &SnapshotExecutionSubsetProvider{}
+	mock.Mock.Test(t)
+
+	t.Cleanup(func() { mock.AssertExpectations(t) })
+
+	return mock
+}