Merge pull request #383 from ScorexFoundation/i374

Fix for #377
hyperledger-labs · Nov 4, 2020 · 4405ea4 · 4405ea4
2 parents 257cd96 + a944e63
commit 4405ea4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/main/scala/scorex/core/network/NetworkController.scala b/src/main/scala/scorex/core/network/NetworkController.scala
@@ -13,7 +13,7 @@ import scorex.core.network.NodeViewSynchronizer.ReceivableMessages.{Disconnected
 import scorex.core.network.message.Message.MessageCode
 import scorex.core.network.message.{Message, MessageSpec}
 import scorex.core.network.peer.PeerManager.ReceivableMessages._
-import scorex.core.network.peer.{LocalAddressPeerFeature, PeerInfo, PeerManager, PenaltyType}
+import scorex.core.network.peer.{LocalAddressPeerFeature, PeerInfo, PeerManager, PeersStatus, PenaltyType}
 import scorex.core.settings.NetworkSettings
 import scorex.core.utils.TimeProvider.Time
 import scorex.core.utils.{NetworkUtils, TimeProvider}
@@ -193,7 +193,7 @@ class NetworkController(settings: NetworkSettings,
   //calls from API / application
   private def interfaceCalls: Receive = {
     case GetPeersStatus =>
-      sender() ! PeersStatusResponse(lastIncomingMessageTime, networkTime())
+      sender() ! PeersStatus(lastIncomingMessageTime, networkTime())
 
     case GetConnectedPeers =>
       sender() ! connections.values.filter(_.peerInfo.nonEmpty)

diff --git a/src/test/scala/scorex/network/NetworkControllerSpec.scala b/src/test/scala/scorex/network/NetworkControllerSpec.scala
@@ -301,7 +301,7 @@ class NetworkControllerSpec extends NetworkTests {
     ls should not be ls0
 
     p.send(networkControllerRef, GetPeersStatus)
-    val status = p.expectMsgClass(classOf[PeersStatusResponse])
+    val status = p.expectMsgClass(classOf[PeersStatus])
     status.lastIncomingMessage shouldBe ls
 
     system.terminate()