Skip to content

Commit 197836e

Browse files
authored
Reduce use of IWorldStateManager in test (#9292)
* Reduce use of worldstatemanager * Fix build * Fix build
1 parent d806ac1 commit 197836e

32 files changed

+92
-137
lines changed

src/Nethermind/Nethermind.AuRa.Test/AuraBlockProcessorTests.cs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,8 +170,7 @@ static BlockHeader Process(BranchProcessor auRaBlockProcessor, BlockHeader paren
170170

171171
private (BranchProcessor Processor, IWorldState StateProvider) CreateProcessor(ITxFilter? txFilter = null, ContractRewriter? contractRewriter = null)
172172
{
173-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest();
174-
IWorldState stateProvider = worldStateManager.GlobalWorldState;
173+
IWorldState stateProvider = TestWorldStateFactory.CreateForTest();
175174
ITransactionProcessor transactionProcessor = Substitute.For<ITransactionProcessor>();
176175
AuRaBlockProcessor processor = new AuRaBlockProcessor(
177176
HoleskySpecProvider.Instance,

src/Nethermind/Nethermind.Blockchain.Test/BlockProcessorTests.cs

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,7 @@ public class BlockProcessorTests
4343
[Test, MaxTime(Timeout.MaxTestTime)]
4444
public void Prepared_block_contains_author_field()
4545
{
46-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest();
47-
IWorldState stateProvider = worldStateManager.GlobalWorldState;
46+
IWorldState stateProvider = TestWorldStateFactory.CreateForTest();
4847
ITransactionProcessor transactionProcessor = Substitute.For<ITransactionProcessor>();
4948
BlockProcessor processor = new BlockProcessor(HoleskySpecProvider.Instance,
5049
TestBlockValidator.AlwaysValid,
@@ -78,8 +77,7 @@ public void Prepared_block_contains_author_field()
7877
[Test, MaxTime(Timeout.MaxTestTime)]
7978
public void Recovers_state_on_cancel()
8079
{
81-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest();
82-
IWorldState stateProvider = worldStateManager.GlobalWorldState;
80+
IWorldState stateProvider = TestWorldStateFactory.CreateForTest();
8381
ITransactionProcessor transactionProcessor = Substitute.For<ITransactionProcessor>();
8482
BlockProcessor processor = new BlockProcessor(
8583
HoleskySpecProvider.Instance,

src/Nethermind/Nethermind.Blockchain.Test/BlockhashProviderTests.cs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,7 @@ public class BlockhashProviderTests
2323
{
2424
private static (IWorldState, Hash256) CreateWorldState()
2525
{
26-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest();
27-
IWorldState worldState = worldStateManager.GlobalWorldState;
26+
IWorldState worldState = TestWorldStateFactory.CreateForTest();
2827
using var _ = worldState.BeginScope(IWorldState.PreGenesis);
2928
worldState.CreateAccount(Eip2935Constants.BlockHashHistoryAddress, 0, 1);
3029
worldState.Commit(Frontier.Instance);

src/Nethermind/Nethermind.Blockchain.Test/FullPruning/CopyTreeVisitorTests.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
using Nethermind.Core.Test.Builders;
1313
using Nethermind.Db;
1414
using Nethermind.Db.FullPruning;
15+
using Nethermind.Evm.State;
1516
using Nethermind.Logging;
1617
using Nethermind.State;
1718
using Nethermind.Trie;
@@ -90,8 +91,7 @@ private IPruningContext CopyDb(IPruningContext pruningContext, CancellationToken
9091
dbProvider.CodeDb.Returns(new MemDb());
9192

9293
// Use TestWorldStateFactory.CreateForTest() with the custom DbProvider
93-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest(dbProvider, logManager);
94-
IStateReader stateReader = worldStateManager.GlobalStateReader;
94+
(IWorldState worldState, IStateReader stateReader) = TestWorldStateFactory.CreateForTestWithStateReader(dbProvider, logManager);
9595

9696
if (_keyScheme == INodeStorage.KeyScheme.Hash)
9797
{

src/Nethermind/Nethermind.Blockchain.Test/GenesisBuilderTests.cs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,7 @@ private Block GetGenesisBlock(string chainspecPath)
5555
{
5656
string path = Path.Combine(TestContext.CurrentContext.WorkDirectory, chainspecPath);
5757
ChainSpec chainSpec = LoadChainSpec(path);
58-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest();
59-
IWorldState stateProvider = worldStateManager.GlobalWorldState;
58+
IWorldState stateProvider = TestWorldStateFactory.CreateForTest();
6059
ISpecProvider specProvider = Substitute.For<ISpecProvider>();
6160
specProvider.GetSpec(Arg.Any<ForkActivation>()).Returns(Berlin.Instance);
6261
ITransactionProcessor transactionProcessor = Substitute.For<ITransactionProcessor>();

src/Nethermind/Nethermind.Blockchain.Test/ReorgTests.cs

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,7 @@ public void Setup()
4545
{
4646
ISpecProvider specProvider = MainnetSpecProvider.Instance;
4747
IDbProvider memDbProvider = TestMemDbProvider.Init();
48-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest(memDbProvider, LimboLogs.Instance);
49-
IWorldState stateProvider = worldStateManager.GlobalWorldState;
48+
(IWorldState stateProvider, IStateReader stateReader) = TestWorldStateFactory.CreateForTestWithStateReader(memDbProvider, LimboLogs.Instance);
5049

5150
IReleaseSpec finalSpec = specProvider.GetFinalSpec();
5251

@@ -70,7 +69,6 @@ public void Setup()
7069
_genesis = Build.A.BlockHeader.WithStateRoot(stateProvider.StateRoot).TestObject;
7170
}
7271

73-
IStateReader stateReader = worldStateManager.GlobalStateReader;
7472
EthereumEcdsa ecdsa = new(1);
7573
ITransactionComparerProvider transactionComparerProvider =
7674
new TransactionComparerProvider(specProvider, _blockTree);
@@ -84,7 +82,7 @@ public void Setup()
8482
ecdsa,
8583
new BlobTxStorage(),
8684
new ChainHeadInfoProvider(
87-
new ChainHeadSpecProvider(specProvider, _blockTree), _blockTree, worldStateManager.GlobalStateReader),
85+
new ChainHeadSpecProvider(specProvider, _blockTree), _blockTree, stateReader),
8886
new TxPoolConfig(),
8987
new TxValidator(specProvider.ChainId),
9088
LimboLogs.Instance,

src/Nethermind/Nethermind.Blockchain.Test/TransactionProcessorEip7702Tests.cs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,7 @@ internal class TransactionProcessorEip7702Tests
4040
public void Setup()
4141
{
4242
_specProvider = new TestSpecProvider(Prague.Instance);
43-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest();
44-
_stateProvider = worldStateManager.GlobalWorldState;
43+
_stateProvider = TestWorldStateFactory.CreateForTest();
4544
_worldStateCloser = _stateProvider.BeginScope(IWorldState.PreGenesis);
4645
EthereumCodeInfoRepository codeInfoRepository = new(_stateProvider);
4746
VirtualMachine virtualMachine = new(new TestBlockhashProvider(_specProvider), _specProvider, LimboLogs.Instance);

src/Nethermind/Nethermind.Blockchain.Test/TransactionProcessorTests.cs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,7 @@ public TransactionProcessorTests(bool eip155Enabled)
5757
[SetUp]
5858
public void Setup()
5959
{
60-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest();
61-
_stateProvider = worldStateManager.GlobalWorldState;
60+
_stateProvider = TestWorldStateFactory.CreateForTest();
6261
_stateCloser = _stateProvider.BeginScope(IWorldState.PreGenesis);
6362
_stateProvider.CreateAccount(TestItem.AddressA, AccountBalance);
6463
_stateProvider.Commit(_specProvider.GenesisSpec);

src/Nethermind/Nethermind.Blockchain.Test/TransactionSelectorTests.cs

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -475,9 +475,7 @@ void AddTxs(int txCount, int blobsPerTx, int account, List<Transaction> txs, ref
475475
[TestCaseSource(nameof(BlobTransactionOrderingTestCases))]
476476
public void Proper_transactions_selected(ProperTransactionsSelectedTestCase testCase)
477477
{
478-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest();
479-
IWorldState stateProvider = worldStateManager.GlobalWorldState;
480-
IStateReader _ = worldStateManager.GlobalStateReader;
478+
IWorldState stateProvider = TestWorldStateFactory.CreateForTest();
481479
ISpecProvider specProvider = Substitute.For<ISpecProvider>();
482480

483481
Hash256 SetAccountStates(IEnumerable<Address> missingAddresses)

src/Nethermind/Nethermind.Blockchain.Test/TransactionsExecutorTests.cs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -263,8 +263,7 @@ public static IEnumerable EIP3860TestCases
263263
[TestCaseSource(nameof(EIP3860TestCases))]
264264
public void Proper_transactions_selected(TransactionSelectorTests.ProperTransactionsSelectedTestCase testCase)
265265
{
266-
IWorldStateManager worldStateManager = TestWorldStateFactory.CreateForTest();
267-
IWorldState stateProvider = worldStateManager.GlobalWorldState;
266+
IWorldState stateProvider = TestWorldStateFactory.CreateForTest();
268267
using var _ = stateProvider.BeginScope(IWorldState.PreGenesis);
269268
ISpecProvider specProvider = Substitute.For<ISpecProvider>();
270269

0 commit comments

Comments
 (0)