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)