def test_peek_network_num(self):
network_num = 12345
hash_bytes = helpers.generate_bytearray(SHA256_HASH_LEN)
msg_hash = Sha256Hash(hash_bytes)
block_bytes = helpers.generate_bytearray(1234)
broadcast_msg = BroadcastMessage(message_hash=msg_hash, network_num=network_num, source_id="", blob=block_bytes)
msg_bytes = broadcast_msg.rawbytes()
input_buffer = InputBuffer()
input_buffer.add_bytes(msg_bytes)
peeked_network_num = BroadcastMessage.peek_network_num(input_buffer)
self.assertEqual(network_num, peeked_network_num)
def peek_broadcast_msg_network_num(self, input_buffer):
if self.protocol_version == 1:
return constants.DEFAULT_NETWORK_NUM
return BroadcastMessage.peek_network_num(input_buffer)