fix

neo-project · Jul 1, 2019 · 7d6f2cd · igormcoelho · Jul 1, 2019 · igormcoelho
1 parent 1799197
commit 7d6f2cd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/neo.UnitTests/UT_RemoteNode.cs b/neo.UnitTests/UT_RemoteNode.cs
@@ -1,14 +1,11 @@
-using Akka.Actor;
-using Akka.IO;
+using Akka.IO;
 using Akka.TestKit.Xunit2;
 using FluentAssertions;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using Neo.Network.P2P;
 using Neo.Network.P2P.Capabilities;
 using Neo.Network.P2P.Payloads;
 using System;
-using System.Collections.Generic;
-using System.Text;
 
 namespace Neo.UnitTests
 {
@@ -19,6 +16,11 @@ public class UT_RemoteNode : TestKit
 
         private static NeoSystem testBlockchain;
 
+        public UT_RemoteNode()
+            : base($"remote-node-mailbox {{ mailbox-type: \"{typeof(RemoteNodeMailbox).AssemblyQualifiedName}\" }}" +
+                $"protocol-handler-mailbox {{ mailbox-type: \"{typeof(ProtocolHandlerMailbox).AssemblyQualifiedName}\" }}")
+        {
+        }
 
         [ClassInitialize]
         public static void TestSetup(TestContext ctx)

diff --git a/neo/Network/P2P/RemoteNode.cs b/neo/Network/P2P/RemoteNode.cs
@@ -36,7 +36,7 @@ public RemoteNode(NeoSystem system, object connection, IPEndPoint remote, IPEndP
             : base(connection, remote, local)
         {
             this.system = system;
-            this.protocol = system.ActorSystem.ActorOf(ProtocolHandler.Props(system));
+            this.protocol = Context.ActorOf(ProtocolHandler.Props(system));
             LocalNode.Singleton.RemoteNodes.TryAdd(Self, this);
 
             var capabilities = new List<NodeCapability>