Skip to content

Commit 25a1a99

Browse files
[ETCM-355] Remove MessageDecoders.fromBytesUnsafe
1 parent d41ca8e commit 25a1a99

File tree

5 files changed

+7
-12
lines changed

5 files changed

+7
-12
lines changed

src/main/scala/io/iohk/ethereum/network/p2p/Message.scala

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,6 @@ trait MessageDecoder extends Logger { self =>
2727

2828
def fromBytes(`type`: Int, payload: Array[Byte]): Either[DecodingError, Message]
2929

30-
def fromBytesUnsafe(`type`: Int, payload: Array[Byte]): Message = self.fromBytes(`type`, payload) match {
31-
case Left(err) => throw err
32-
case Right(res) => res
33-
}
34-
3530
def orElse(otherMessageDecoder: MessageDecoder): MessageDecoder = new MessageDecoder {
3631
override def fromBytes(`type`: Int, payload: Array[Byte]): Either[DecodingError, Message] =
3732
self.fromBytes(`type`, payload).leftFlatMap { err =>

src/test/scala/io/iohk/ethereum/network/p2p/MessageDecodersSpec.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ class MessageDecodersSpec extends AnyFlatSpec with Matchers with SecureRandomBui
4646
)
4747
)
4848
)
49-
NetworkMessageDecoder.fromBytesUnsafe(WireProtocol.Hello.code, helloBytes) shouldBe hello
49+
NetworkMessageDecoder.fromBytes(WireProtocol.Hello.code, helloBytes) shouldBe Right(hello)
5050
}
5151

5252
it should "decode NewBlockHashes message for all supported versions of protocol" in {

src/test/scala/io/iohk/ethereum/network/p2p/messages/MessagesSerializationSpec.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ class MessagesSerializationSpec extends AnyWordSpec with ScalaCheckPropertyCheck
176176
//scalastyle:on
177177

178178
def verify[T](msg: T, encode: T => Array[Byte], code: Int, version: Capability): Unit =
179-
messageDecoder(version).fromBytesUnsafe(code, encode(msg)) shouldEqual msg
179+
messageDecoder(version).fromBytes(code, encode(msg)) shouldEqual Right(msg)
180180

181181
private def messageDecoder(version: Capability) =
182182
NetworkMessageDecoder.orElse(EthereumMessageDecoder.ethMessageDecoder(version))

src/test/scala/io/iohk/ethereum/network/p2p/messages/NodeDataSpec.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ class NodeDataSpec extends AnyFlatSpec with Matchers {
9090
it should "be decoded properly" in {
9191
val result = EthereumMessageDecoder
9292
.ethMessageDecoder(ProtocolVersions.ETH63)
93-
.fromBytesUnsafe(Codes.NodeDataCode, encode(encodedNodeData))
93+
.fromBytes(Codes.NodeDataCode, encode(encodedNodeData)).getOrElse(fail("Should have decoded NodeData"))
9494

9595
result match {
9696
case m: NodeData =>
@@ -106,7 +106,7 @@ class NodeDataSpec extends AnyFlatSpec with Matchers {
106106
it should "be decoded previously encoded value" in {
107107
EthereumMessageDecoder
108108
.ethMessageDecoder(ProtocolVersions.ETH63)
109-
.fromBytesUnsafe(Codes.NodeDataCode, nodeData.toBytes) shouldBe nodeData
109+
.fromBytes(Codes.NodeDataCode, nodeData.toBytes) shouldBe Right(nodeData)
110110
}
111111

112112
it should "decode branch node with values in leafs that looks like RLP list" in {

src/test/scala/io/iohk/ethereum/network/p2p/messages/ReceiptsSpec.scala

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,15 +57,15 @@ class ReceiptsSpec extends AnyFlatSpec with Matchers {
5757
it should "decode receipts" in {
5858
EthereumMessageDecoder
5959
.ethMessageDecoder(ProtocolVersions.ETH63)
60-
.fromBytesUnsafe(
60+
.fromBytes(
6161
Codes.ReceiptsCode,
6262
encode(encodedReceipts)
63-
) shouldBe receipts
63+
) shouldBe Right(receipts)
6464
}
6565

6666
it should "decode encoded receipts" in {
6767
EthereumMessageDecoder
6868
.ethMessageDecoder(ProtocolVersions.ETH63)
69-
.fromBytesUnsafe(Codes.ReceiptsCode, receipts.toBytes) shouldBe receipts
69+
.fromBytes(Codes.ReceiptsCode, receipts.toBytes) shouldBe Right(receipts)
7070
}
7171
}

0 commit comments

Comments
 (0)