diff --git a/src/Nethermind/Nethermind.AuRa.Test/Contract/ContractDataStoreTests.cs b/src/Nethermind/Nethermind.AuRa.Test/Contract/ContractDataStoreTests.cs index f88647fade18..a9669604116e 100644 --- a/src/Nethermind/Nethermind.AuRa.Test/Contract/ContractDataStoreTests.cs +++ b/src/Nethermind/Nethermind.AuRa.Test/Contract/ContractDataStoreTests.cs @@ -5,7 +5,6 @@ using System.Collections; using System.Collections.Generic; using System.Linq; -using System.Threading.Tasks; using FluentAssertions; using Nethermind.Abi; using Nethermind.Blockchain; @@ -91,7 +90,7 @@ public void returns_data_from_getAll_on_non_consecutive_receipts_with_incrementa } [Test] - public async Task returns_data_from_receipts_on_non_consecutive_with_not_incremental_changes() + public void returns_data_from_receipts_on_non_consecutive_with_not_incremental_changes() { TestCase
testCase = BuildTestCase
(); testCase.DataContract.IncrementalChanges.Returns(false); @@ -109,9 +108,10 @@ public async Task returns_data_from_receipts_on_non_consecutive_with_not_increme testCase.ContractDataStore.GetItemsFromContractAtBlock(blockHeader); testCase.BlockTree.NewHeadBlock += Raise.EventWith(new BlockEventArgs(secondBlock)); - await Task.Delay(10); // delay for refresh from contract as its async - - testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header).Should().BeEquivalentTo(expected.Cast()); + Assert.That( + () => testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header), + Is.EquivalentTo(expected.Cast()).After(200, 20) + ); } [Test] @@ -127,7 +127,7 @@ public void returns_data_from_getAll_on_non_consecutive_with_not_incremental_cha } [Test] - public async Task returns_data_from_receipts_on_consecutive_with_not_incremental_changes() + public void returns_data_from_receipts_on_consecutive_with_not_incremental_changes() { TestCase
testCase = BuildTestCase
(); testCase.DataContract.IncrementalChanges.Returns(false); @@ -145,13 +145,14 @@ public async Task returns_data_from_receipts_on_consecutive_with_not_incremental testCase.ContractDataStore.GetItemsFromContractAtBlock(blockHeader); testCase.BlockTree.NewHeadBlock += Raise.EventWith(new BlockEventArgs(secondBlock)); - await Task.Delay(10); // delay for refresh from contract as its async - - testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header).Should().BeEquivalentTo(expected.Cast()); + Assert.That( + () => testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header), + Is.EquivalentTo(expected.Cast()).After(200, 20) + ); } [Test] - public async Task returns_data_from_receipts_on_consecutive_with_incremental_changes() + public void returns_data_from_receipts_on_consecutive_with_incremental_changes() { TestCase
testCase = BuildTestCase
(); BlockHeader blockHeader = Build.A.BlockHeader.WithNumber(1).WithHash(TestItem.KeccakA).TestObject; @@ -167,16 +168,14 @@ public async Task returns_data_from_receipts_on_consecutive_with_incremental_cha testCase.ContractDataStore.GetItemsFromContractAtBlock(blockHeader); testCase.BlockTree.NewHeadBlock += Raise.EventWith(new BlockEventArgs(secondBlock)); - await Task.Delay(50); // delay for refresh from contract as its async - Assert.That( () => testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header).ToList(), - Is.EquivalentTo(new ArrayList() { TestItem.AddressA, TestItem.AddressB }).After(1000, 100) + Is.EquivalentTo(new ArrayList() { TestItem.AddressA, TestItem.AddressB }).After(200, 20) ); } [Test] - public async Task returns_unmodified_data_from_empty_receipts_on_consecutive_with_incremental_changes() + public void returns_unmodified_data_from_empty_receipts_on_consecutive_with_incremental_changes() { TestCase
testCase = BuildTestCase
(); BlockHeader blockHeader = Build.A.BlockHeader.WithNumber(1).WithHash(TestItem.KeccakA).TestObject; @@ -192,13 +191,14 @@ public async Task returns_unmodified_data_from_empty_receipts_on_consecutive_wit testCase.ContractDataStore.GetItemsFromContractAtBlock(blockHeader); testCase.BlockTree.NewHeadBlock += Raise.EventWith(new BlockEventArgs(secondBlock)); - await Task.Delay(10); // delay for refresh from contract as its async - - testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header).Should().BeEquivalentTo(TestItem.AddressA, TestItem.AddressC); + Assert.That( + () => testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header), + Is.EquivalentTo(new[] { TestItem.AddressA, TestItem.AddressC }).After(200, 20) + ); } [Test] - public async Task returns_data_from_receipts_on_consecutive_with_incremental_changes_with_identity() + public void returns_data_from_receipts_on_consecutive_with_incremental_changes_with_identity() { TestCase testCase = BuildTestCase( TxPriorityContract.DistinctDestinationMethodComparer.Instance, @@ -227,11 +227,9 @@ public async Task returns_data_from_receipts_on_consecutive_with_incremental_cha testCase.ContractDataStore.GetItemsFromContractAtBlock(blockHeader); testCase.BlockTree.NewHeadBlock += Raise.EventWith(new BlockEventArgs(secondBlock)); - await Task.Delay(10); // delay for refresh from contract as its async - Assert.That( () => testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header).Count(), - Is.EqualTo(3).After(1000, 100) + Is.EqualTo(3).After(200, 20) ); testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header).Should().BeEquivalentTo(new[] diff --git a/src/Nethermind/Nethermind.AuRa.Test/Contract/ContractDataStoreWithLocalDataTests.cs b/src/Nethermind/Nethermind.AuRa.Test/Contract/ContractDataStoreWithLocalDataTests.cs index 735267cdbb63..9d1f68f70926 100644 --- a/src/Nethermind/Nethermind.AuRa.Test/Contract/ContractDataStoreWithLocalDataTests.cs +++ b/src/Nethermind/Nethermind.AuRa.Test/Contract/ContractDataStoreWithLocalDataTests.cs @@ -86,7 +86,7 @@ public void combines_contract_and_local_data_correctly() Assert.That( () => testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header), - Is.EquivalentTo(expected.Cast()).After(1000, 100) + Is.EquivalentTo(expected.Cast()).After(200, 20) ); localDataSource.Data.Returns(new[] { TestItem.AddressC, TestItem.AddressD }); @@ -95,7 +95,7 @@ public void combines_contract_and_local_data_correctly() Assert.That( () => testCase.ContractDataStore.GetItemsFromContractAtBlock(secondBlock.Header), - Is.EquivalentTo(expected.Cast()).After(1000, 100) + Is.EquivalentTo(expected.Cast()).After(200, 20) ); }