Fix ut

benchmarks/Neo.Benchmarks/Benchmarks.cs

@@ -10,6 +10,7 @@
 // modifications are permitted.
 
 using Neo.Network.P2P.Payloads;
+using Neo.Persistence;
 using Neo.SmartContract;
 using Neo.VM;
 using System.Diagnostics;
@@ -19,7 +20,7 @@ namespace Neo;
 static class Benchmarks
 {
     private static readonly ProtocolSettings protocol = ProtocolSettings.Default;
-    private static readonly NeoSystem system = new(protocol);
+    private static readonly NeoSystem system = new(protocol, new MemoryStore());
 
     public static void NeoIssue2725()
     {

src/Neo.CLI/CLI/MainService.cs

@@ -17,6 +17,7 @@
 using Neo.Ledger;
 using Neo.Network.P2P;
 using Neo.Network.P2P.Payloads;
+using Neo.Persistence;
 using Neo.Plugins;
 using Neo.SmartContract;
 using Neo.SmartContract.Manifest;
@@ -376,7 +377,8 @@ public async void Start(CommandLineOptions options)
             ProtocolSettings protocol = ProtocolSettings.Load("config.json");
             CustomProtocolSettings(options, protocol);
             CustomApplicationSettings(options, Settings.Default);
-            NeoSystem = new NeoSystem(protocol, Settings.Default.Storage.Engine, string.Format(Settings.Default.Storage.Path, protocol.Network.ToString("X8")));
+            var store = StoreFactory.GetStore(Settings.Default.Storage.Engine, string.Format(Settings.Default.Storage.Path, protocol.Network.ToString("X8")));
+            NeoSystem = new NeoSystem(protocol, store);
             NeoSystem.AddService(this);
 
             LocalNode = NeoSystem.LocalNode.Ask<LocalNode>(new LocalNode.GetInstance()).Result;

src/Neo/NeoSystem.cs

@@ -97,7 +97,6 @@ public class NeoSystem : IDisposable
         internal RelayCache RelayCache { get; } = new(100);
 
         private ImmutableList<object> services = ImmutableList<object>.Empty;
-        private readonly string storage_engine;
         private readonly IStore store;
         private ChannelsConfig start_message = null;
         private int suspend = 0;
@@ -114,14 +113,12 @@ static NeoSystem()
         /// Initializes a new instance of the <see cref="NeoSystem"/> class.
         /// </summary>
         /// <param name="settings">The protocol settings of the <see cref="NeoSystem"/>.</param>
-        /// <param name="storageEngine">The storage engine used to create the <see cref="IStore"/> objects. If this parameter is <see langword="null"/>, a default in-memory storage engine will be used.</param>
-        /// <param name="storagePath">The path of the storage. If <paramref name="storageEngine"/> is the default in-memory storage engine, this parameter is ignored.</param>
-        public NeoSystem(ProtocolSettings settings, string storageEngine = null, string storagePath = null)
+        /// <param name="storage">The <see cref="IStore"/> to use.</param>
+        public NeoSystem(ProtocolSettings settings, IStore storage)
         {
             this.Settings = settings;
             this.GenesisBlock = CreateGenesisBlock(settings);
-            this.storage_engine = storageEngine ?? nameof(MemoryStore);
-            this.store = LoadStore(storagePath);
+            this.store = storage;
             this.MemPool = new MemoryPool(this);
             this.Blockchain = ActorSystem.ActorOf(Ledger.Blockchain.Props(this));
             this.LocalNode = ActorSystem.ActorOf(Network.P2P.LocalNode.Props(this));
@@ -211,16 +208,6 @@ public void EnsureStopped(IActorRef actor)
             inbox.Receive(TimeSpan.FromMinutes(5));
         }
 
-        /// <summary>
-        /// Loads an <see cref="IStore"/> at the specified path.
-        /// </summary>
-        /// <param name="path">The path of the storage.</param>
-        /// <returns>The loaded <see cref="IStore"/>.</returns>
-        public IStore LoadStore(string path)
-        {
-            return StoreFactory.GetStore(storage_engine, path);
-        }
-
         /// <summary>
         /// Resumes the startup process of <see cref="LocalNode"/>.
         /// </summary>

src/Neo/Persistence/MemoryStore.cs

@@ -65,5 +65,10 @@ public bool Contains(byte[] key)
         {
             return innerData.ContainsKey(key);
         }
+
+        internal void Clean()
+        {
+            innerData.Clear();
+        }
     }
 }

src/Neo/Persistence/StoreFactory.cs

@@ -33,6 +33,12 @@ public static void RegisterProvider(IStoreProvider provider)
         providers.Add(provider.Name, provider);
     }
 
+    /// <summary>
+    /// Get store from name
+    /// </summary>
+    /// <param name="storageEngine">The storage engine used to create the <see cref="IStore"/> objects. If this parameter is <see langword="null"/>, a default in-memory storage engine will be used.</param>
+    /// <param name="path">The path of the storage. If <paramref name="storageEngine"/> is the default in-memory storage engine, this parameter is ignored.</param>
+    /// <returns>The storage engine.</returns>
     public static IStore GetStore(string storageEngine, string path)
     {
         return providers[storageEngine].GetStore(path);

tests/Neo.UnitTests/Ledger/UT_Blockchain.cs

@@ -48,6 +48,12 @@ public void Initialize()
             system.MemPool.TryAdd(txSample, TestBlockchain.GetTestSnapshot());
         }
 
+        [TestCleanup]
+        public void Clean()
+        {
+            TestBlockchain.ResetStore();
+        }
+
         [TestMethod]
         public void TestValidTransaction()
         {
@@ -113,7 +119,7 @@ private static Transaction CreateValidTx(DataCache snapshot, NEP6Wallet wallet,
         [TestMethod]
         public void TestMaliciousOnChainConflict()
         {
-            var snapshot = TestBlockchain.TheNeoSystem.GetSnapshot();
+            using var snapshot = TestBlockchain.TheNeoSystem.GetSnapshot();
             var walletA = TestUtils.GenerateTestWallet("123");
             var accA = walletA.CreateAccount();
             var walletB = TestUtils.GenerateTestWallet("456");

tests/Neo.UnitTests/Ledger/UT_MemoryPool.cs

@@ -57,7 +57,7 @@ public void TestSetup()
             TimeProvider.ResetToDefault();
 
             // Create a MemoryPool with capacity of 100
-            _unit = new MemoryPool(new NeoSystem(TestProtocolSettings.Default with { MemoryPoolMaxTransactions = 100 }));
+            _unit = new MemoryPool(new NeoSystem(TestProtocolSettings.Default with { MemoryPoolMaxTransactions = 100 }, new MemoryStore()));
 
             // Verify capacity equals the amount specified
             _unit.Capacity.Should().Be(100);
@@ -648,7 +648,8 @@ public void TestGetVerifiedTransactions()
         [TestMethod]
         public void TestReVerifyTopUnverifiedTransactionsIfNeeded()
         {
-            _unit = new MemoryPool(new NeoSystem(TestProtocolSettings.Default with { MemoryPoolMaxTransactions = 600 }));
+            var store = new MemoryStore();
+            _unit = new MemoryPool(new NeoSystem(TestProtocolSettings.Default with { MemoryPoolMaxTransactions = 600 }, store));
 
             AddTransaction(CreateTransaction(100000001));
             AddTransaction(CreateTransaction(100000001));

tests/Neo.UnitTests/Network/P2P/Payloads/UT_Transaction.cs

@@ -38,6 +38,12 @@ public void TestSetup()
             uut = new Transaction();
         }
 
+        [TestCleanup]
+        public void Clean()
+        {
+            TestBlockchain.ResetStore();
+        }
+
         [TestMethod]
         public void Script_Get()
         {

tests/Neo.UnitTests/SmartContract/Native/UT_RoleManagement.cs

@@ -36,6 +36,12 @@ public void TestSetup()
             _snapshot = TestBlockchain.GetTestSnapshot();
         }
 
+        [TestCleanup]
+        public void Clean()
+        {
+            TestBlockchain.ResetStore();
+        }
+
         [TestMethod]
         public void TestSetAndGet()
         {

tests/Neo.UnitTests/TestBlockchain.cs

@@ -9,6 +9,8 @@
 // Redistribution and use in source and binary forms with or without
 // modifications are permitted.
 
+using Akka.Actor;
+using Neo.Ledger;
 using Neo.Persistence;
 using System;
 
@@ -18,11 +20,18 @@ public static class TestBlockchain
     {
         public static readonly NeoSystem TheNeoSystem;
         public static readonly UInt160[] DefaultExtensibleWitnessWhiteList;
+        private static readonly MemoryStore Store = new();
 
         static TestBlockchain()
         {
             Console.WriteLine("initialize NeoSystem");
-            TheNeoSystem = new NeoSystem(TestProtocolSettings.Default, null, null);
+            TheNeoSystem = new NeoSystem(TestProtocolSettings.Default, Store);
+        }
+
+        internal static void ResetStore()
+        {
+            Store.Clean();
+            TheNeoSystem.Blockchain.Ask(new Blockchain.Initialize()).Wait();
         }
 
         internal static DataCache GetTestSnapshot()

tests/Neo.UnitTests/Wallets/UT_Wallet.cs

@@ -221,13 +221,11 @@ public void TestGetAvailable()
             var key = NativeContract.GAS.CreateStorageKey(20, account.ScriptHash);
             var entry = snapshot.GetAndChange(key, () => new StorageItem(new AccountState()));
             entry.GetInteroperable<AccountState>().Balance = 10000 * NativeContract.GAS.Factor;
-            snapshot.Commit();
 
             wallet.GetAvailable(snapshot, NativeContract.GAS.Hash).Should().Be(new BigDecimal(new BigInteger(1000000000000M), 8));
 
             entry = snapshot.GetAndChange(key, () => new StorageItem(new AccountState()));
             entry.GetInteroperable<AccountState>().Balance = 0;
-            snapshot.Commit();
         }
 
         [TestMethod]
@@ -243,14 +241,12 @@ public void TestGetBalance()
             var key = NativeContract.GAS.CreateStorageKey(20, account.ScriptHash);
             var entry = snapshot.GetAndChange(key, () => new StorageItem(new AccountState()));
             entry.GetInteroperable<AccountState>().Balance = 10000 * NativeContract.GAS.Factor;
-            snapshot.Commit();
 
             wallet.GetBalance(snapshot, UInt160.Zero, new UInt160[] { account.ScriptHash }).Should().Be(new BigDecimal(BigInteger.Zero, 0));
             wallet.GetBalance(snapshot, NativeContract.GAS.Hash, new UInt160[] { account.ScriptHash }).Should().Be(new BigDecimal(new BigInteger(1000000000000M), 8));
 
             entry = snapshot.GetAndChange(key, () => new StorageItem(new AccountState()));
             entry.GetInteroperable<AccountState>().Balance = 0;
-            snapshot.Commit();
         }
 
         [TestMethod]
@@ -345,8 +341,6 @@ public void TestMakeTransaction1()
             var entry2 = snapshot.GetAndChange(key, () => new StorageItem(new NeoToken.NeoAccountState()));
             entry2.GetInteroperable<NeoToken.NeoAccountState>().Balance = 10000 * NativeContract.NEO.Factor;
 
-            snapshot.Commit();
-
             var tx = wallet.MakeTransaction(snapshot, new TransferOutput[]
             {
                 new TransferOutput()
@@ -374,7 +368,6 @@ public void TestMakeTransaction1()
             entry2 = snapshot.GetAndChange(key, () => new StorageItem(new AccountState()));
             entry1.GetInteroperable<AccountState>().Balance = 0;
             entry2.GetInteroperable<NeoToken.NeoAccountState>().Balance = 0;
-            snapshot.Commit();
         }
 
         [TestMethod]
@@ -393,7 +386,6 @@ public void TestMakeTransaction2()
             var key = NativeContract.GAS.CreateStorageKey(20, account.ScriptHash);
             var entry = snapshot.GetAndChange(key, () => new StorageItem(new AccountState()));
             entry.GetInteroperable<AccountState>().Balance = 1000000 * NativeContract.GAS.Factor;
-            snapshot.Commit();
 
             var tx = wallet.MakeTransaction(snapshot, Array.Empty<byte>(), account.ScriptHash, new[]{ new Signer()
             {
@@ -408,7 +400,6 @@ public void TestMakeTransaction2()
 
             entry = snapshot.GetAndChange(key, () => new StorageItem(new AccountState()));
             entry.GetInteroperable<AccountState>().Balance = 0;
-            snapshot.Commit();
         }
 
         [TestMethod]