@@ -53,7 +53,7 @@ public abstract class AbstractStoreTest {
53
53
protected final ForkChoiceStrategy dummyForkChoiceStrategy = mock (ForkChoiceStrategy .class );
54
54
55
55
protected void processChainWithLimitedCache (
56
- BiConsumer <UpdatableStore , SignedBlockAndState > chainProcessor ) {
56
+ final BiConsumer <UpdatableStore , SignedBlockAndState > chainProcessor ) {
57
57
final int cacheSize = 10 ;
58
58
final int cacheMultiplier = 3 ;
59
59
@@ -84,7 +84,7 @@ protected void processChainWithLimitedCache(
84
84
}
85
85
86
86
protected void processCheckpointsWithLimitedCache (
87
- BiConsumer <UpdatableStore , CheckpointState > chainProcessor ) {
87
+ final BiConsumer <UpdatableStore , CheckpointState > chainProcessor ) {
88
88
final int cacheSize = 3 ;
89
89
final int epochsToProcess = cacheSize * 3 ;
90
90
@@ -122,7 +122,7 @@ protected void processCheckpointsWithLimitedCache(
122
122
}
123
123
124
124
protected void processChainHeadWithMockForkChoiceStrategy (
125
- BiConsumer <UpdatableStore , SignedBlockAndState > chainProcessor ) {
125
+ final BiConsumer <UpdatableStore , SignedBlockAndState > chainProcessor ) {
126
126
final StoreConfig pruningOptions = StoreConfig .builder ().build ();
127
127
128
128
final UpdatableStore store = createGenesisStoreWithMockForkChoiceStrategy (pruningOptions );
@@ -150,15 +150,54 @@ protected UpdatableStore createGenesisStore() {
150
150
return createGenesisStore (defaultStoreConfig );
151
151
}
152
152
153
+ protected UpdatableStore createGenesisStore (
154
+ final EarliestBlobSidecarSlotProvider earliestBlobSidecarSlotProvider ) {
155
+ return createStoreBuilder (defaultStoreConfig , chainBuilder , earliestBlobSidecarSlotProvider )
156
+ .build ();
157
+ }
158
+
159
+ protected UpdatableStore createGenesisStore (final StoreConfig pruningOptions ) {
160
+ return createStoreBuilder (pruningOptions ).build ();
161
+ }
162
+
163
+ protected StoreBuilder createStoreBuilder (
164
+ final StoreConfig pruningOptions ,
165
+ final ChainBuilder chainBuilder ,
166
+ final EarliestBlobSidecarSlotProvider earliestBlobSidecarSlotProvider ) {
167
+ final SignedBlockAndState genesis = chainBuilder .generateGenesis ();
168
+ final Checkpoint genesisCheckpoint = chainBuilder .getCurrentCheckpointForEpoch (0 );
169
+ return createStoreBuilder (
170
+ pruningOptions , genesis , genesisCheckpoint , earliestBlobSidecarSlotProvider );
171
+ }
172
+
153
173
protected StoreBuilder createStoreBuilder (final StoreConfig pruningOptions ) {
154
174
final SignedBlockAndState genesis = chainBuilder .generateGenesis ();
155
175
final Checkpoint genesisCheckpoint = chainBuilder .getCurrentCheckpointForEpoch (0 );
176
+ return createStoreBuilder (pruningOptions , genesis , genesisCheckpoint );
177
+ }
178
+
179
+ private StoreBuilder createStoreBuilder (
180
+ final StoreConfig pruningOptions ,
181
+ final SignedBlockAndState genesis ,
182
+ final Checkpoint genesisCheckpoint ) {
183
+ return createStoreBuilder (
184
+ pruningOptions ,
185
+ genesis ,
186
+ genesisCheckpoint ,
187
+ earliestBlobSidecarSlotProviderFromChainBuilder ());
188
+ }
189
+
190
+ private StoreBuilder createStoreBuilder (
191
+ final StoreConfig pruningOptions ,
192
+ final SignedBlockAndState genesis ,
193
+ final Checkpoint genesisCheckpoint ,
194
+ final EarliestBlobSidecarSlotProvider earliestBlobSidecarSlotProvider ) {
156
195
return StoreBuilder .create ()
157
196
.asyncRunner (SYNC_RUNNER )
158
197
.metricsSystem (new StubMetricsSystem ())
159
198
.specProvider (spec )
160
199
.blockProvider (blockProviderFromChainBuilder ())
161
- .earliestBlobSidecarSlotProvider (earliestBlobSidecarSlotProviderFromChainBuilder () )
200
+ .earliestBlobSidecarSlotProvider (earliestBlobSidecarSlotProvider )
162
201
.stateProvider (StateAndBlockSummaryProvider .NOOP )
163
202
.anchor (Optional .empty ())
164
203
.genesisTime (genesis .getState ().getGenesisTime ())
@@ -181,10 +220,6 @@ protected StoreBuilder createStoreBuilder(final StoreConfig pruningOptions) {
181
220
.votes (emptyMap ());
182
221
}
183
222
184
- protected UpdatableStore createGenesisStore (final StoreConfig pruningOptions ) {
185
- return createStoreBuilder (pruningOptions ).build ();
186
- }
187
-
188
223
protected UpdatableStore createGenesisStoreWithMockForkChoiceStrategy (
189
224
final StoreConfig pruningOptions ) {
190
225
return createStoreBuilder (pruningOptions ).forkChoiceStrategy (dummyForkChoiceStrategy ).build ();
0 commit comments