Isolate ApplicationEngine events

src/Neo/SmartContract/ApplicationEngine.cs

@@ -44,15 +44,24 @@ public partial class ApplicationEngine : ExecutionEngine
         /// </summary>
         public const long TestModeGas = 20_00000000;
 
+        public delegate void DelOnNotify(ApplicationEngine engine, NotifyEventArgs args);
+
         /// <summary>
         /// Triggered when a contract calls System.Runtime.Notify.
         /// </summary>
-        public static event EventHandler<NotifyEventArgs> Notify;
+        public event DelOnNotify Notify;
 
         /// <summary>
         /// Triggered when a contract calls System.Runtime.Log.
         /// </summary>
-        public static event EventHandler<LogEventArgs> Log;
+        public event EventHandler<LogEventArgs> Log;
+
+        public delegate void DelOnApplicationEngine(ApplicationEngine engine);
+
+        /// <summary>
+        /// On Application Engine
+        /// </summary>
+        public static DelOnApplicationEngine OnApplicationEngine;
 
         private static Dictionary<uint, InteropDescriptor> services;
         // Total amount of GAS spent to execute.
@@ -435,8 +444,11 @@ public static ApplicationEngine Create(TriggerType trigger, IVerifiable containe
             // Adjust jump table according persistingBlock
 
             var jumpTable = settings == null || settings.IsHardforkEnabled(Hardfork.HF_Echidna, index) ? DefaultJumpTable : NotEchidnaJumpTable;
-            return Provider?.Create(trigger, container, snapshot, persistingBlock, settings, gas, diagnostic, jumpTable)
+            var engine = Provider?.Create(trigger, container, snapshot, persistingBlock, settings, gas, diagnostic, jumpTable)
                   ?? new ApplicationEngine(trigger, container, snapshot, persistingBlock, settings, gas, diagnostic, jumpTable);
+
+            OnApplicationEngine?.Invoke(engine);
+            return engine;
         }
 
         /// <summary>

src/Plugins/ApplicationLogs/LogReader.cs

@@ -62,10 +62,15 @@ public override void Dispose()
             Blockchain.Committing -= ((ICommittingHandler)this).Blockchain_Committing_Handler;
             Blockchain.Committed -= ((ICommittedHandler)this).Blockchain_Committed_Handler;
             if (Settings.Default.Debug)
-                ApplicationEngine.Log -= ((ILogHandler)this).ApplicationEngine_Log_Handler;
+                ApplicationEngine.OnApplicationEngine -= ConfigureAppEngine;
             GC.SuppressFinalize(this);
         }
 
+        private void ConfigureAppEngine(ApplicationEngine engine)
+        {
+            engine.Log += ((ILogHandler)this).ApplicationEngine_Log_Handler;
+        }
+
         protected override void Configure()
         {
             Settings.Load(GetConfiguration());
@@ -82,7 +87,7 @@ protected override void OnSystemLoaded(NeoSystem system)
             RpcServerPlugin.RegisterMethods(this, Settings.Default.Network);
 
             if (Settings.Default.Debug)
-                ApplicationEngine.Log += ((ILogHandler)this).ApplicationEngine_Log_Handler;
+                ApplicationEngine.OnApplicationEngine += ConfigureAppEngine;
         }
 
         #endregion

tests/Neo.UnitTests/Extensions/NativeContractExtensions.cs

@@ -126,12 +126,34 @@ public static void DeleteContract(this DataCache snapshot, UInt160 hash)
 
         public static StackItem Call(this NativeContract contract, DataCache snapshot, string method, params ContractParameter[] args)
         {
-            return Call(contract, snapshot, null, null, method, args);
+            return Call(contract, snapshot, null, null, method, null, args);
         }
 
-        public static StackItem Call(this NativeContract contract, DataCache snapshot, IVerifiable container, Block persistingBlock, string method, params ContractParameter[] args)
+        public static StackItem Call(this NativeContract contract, DataCache snapshot, string method, ApplicationEngine.DelOnNotify onNotify, params ContractParameter[] args)
+        {
+            return Call(contract, snapshot, null, null, method, onNotify, args);
+        }
+
+        public static StackItem Call(
+           this NativeContract contract, DataCache snapshot, IVerifiable container,
+           Block persistingBlock, string method, params ContractParameter[] args
+           )
+        {
+            return Call(contract, snapshot, container, persistingBlock, method, null, args);
+        }
+
+        public static StackItem Call(
+            this NativeContract contract, DataCache snapshot, IVerifiable container,
+            Block persistingBlock, string method, ApplicationEngine.DelOnNotify onNotify, params ContractParameter[] args
+            )
         {
             using var engine = ApplicationEngine.Create(TriggerType.Application, container, snapshot, persistingBlock, settings: TestProtocolSettings.Default);
+
+            if (onNotify != null)
+            {
+                engine.Notify += onNotify;
+            }
+
             return Call(contract, engine, method, args);
         }
 

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

@@ -58,20 +58,18 @@ public void TestSetAndGet()
             foreach (var role in roles)
             {
                 var snapshot1 = _snapshotCache.CloneCache();
-                UInt160 committeeMultiSigAddr = NativeContract.NEO.GetCommitteeAddress(snapshot1);
-                List<NotifyEventArgs> notifications = new List<NotifyEventArgs>();
-                EventHandler<NotifyEventArgs> ev = (o, e) => notifications.Add(e);
-                ApplicationEngine.Notify += ev;
+                var committeeMultiSigAddr = NativeContract.NEO.GetCommitteeAddress(snapshot1);
+                List<NotifyEventArgs> notifications = [];
+                void Ev(ApplicationEngine o, NotifyEventArgs e) => notifications.Add(e);
                 var ret = NativeContract.RoleManagement.Call(
                     snapshot1,
                     new Nep17NativeContractExtensions.ManualWitness(committeeMultiSigAddr),
                     new Block { Header = new Header() },
-                    "designateAsRole",
+                    "designateAsRole", Ev,
                     new ContractParameter(ContractParameterType.Integer) { Value = new BigInteger((int)role) },
                     new ContractParameter(ContractParameterType.Array) { Value = publicKeys.Select(p => new ContractParameter(ContractParameterType.ByteArray) { Value = p.ToArray() }).ToList() }
                 );
                 snapshot1.Commit();
-                ApplicationEngine.Notify -= ev;
                 Assert.AreEqual(1, notifications.Count);
                 Assert.AreEqual("Designation", notifications[0].EventName);
                 var snapshot2 = _snapshotCache.CloneCache();

tests/Neo.UnitTests/SmartContract/UT_ApplicationEngine.cs

@@ -34,22 +34,22 @@ public void TestNotify()
             var snapshotCache = TestBlockchain.GetTestSnapshotCache();
             using var engine = ApplicationEngine.Create(TriggerType.Application, null, snapshotCache, settings: TestProtocolSettings.Default);
             engine.LoadScript(System.Array.Empty<byte>());
-            ApplicationEngine.Notify += Test_Notify1;
+            engine.Notify += Test_Notify1;
             const string notifyEvent = "TestEvent";
 
             engine.SendNotification(UInt160.Zero, notifyEvent, new Array());
             Assert.AreEqual(notifyEvent, eventName);
 
-            ApplicationEngine.Notify += Test_Notify2;
+            engine.Notify += Test_Notify2;
             engine.SendNotification(UInt160.Zero, notifyEvent, new Array());
             Assert.IsNull(eventName);
 
             eventName = notifyEvent;
-            ApplicationEngine.Notify -= Test_Notify1;
+            engine.Notify -= Test_Notify1;
             engine.SendNotification(UInt160.Zero, notifyEvent, new Array());
             Assert.IsNull(eventName);
 
-            ApplicationEngine.Notify -= Test_Notify2;
+            engine.Notify -= Test_Notify2;
             engine.SendNotification(UInt160.Zero, notifyEvent, new Array());
             Assert.IsNull(eventName);
         }

tests/Neo.UnitTests/SmartContract/UT_InteropService.cs

@@ -368,10 +368,10 @@ public void TestRuntime_Log()
         {
             var engine = GetEngine(true);
             var message = "hello";
-            ApplicationEngine.Log += LogEvent;
+            engine.Log += LogEvent;
             engine.RuntimeLog(Encoding.UTF8.GetBytes(message));
             Assert.AreEqual(new byte[] { 0x01, 0x02, 0x03 }.ToHexString(), ((Transaction)engine.ScriptContainer).Script.Span.ToHexString());
-            ApplicationEngine.Log -= LogEvent;
+            engine.Log -= LogEvent;
         }
 
         [TestMethod]