def _build_get_blocks_message_for_block_confirmation(self, hashes: List[Sha256Hash]) -> AbstractMessage:
return GetBlocksBtcMessage(
version=self.version,
magic=self.magic,
hashes=hashes,
hash_stop=NULL_BTC_BLOCK_HASH
)
def test_parse_message_success_all_types(self):
# TODO: pull these numbers into constants, along with all the BTC messages
self.create_message_successfully(self.VERSION_BTC_MESSAGE,
VersionBtcMessage)
self.create_message_successfully(VerAckBtcMessage(self.MAGIC),
VerAckBtcMessage)
self.create_message_successfully(PingBtcMessage(self.MAGIC),
PingBtcMessage)
self.create_message_successfully(PongBtcMessage(self.MAGIC, 123),
PongBtcMessage)
self.create_message_successfully(GetAddrBtcMessage(self.MAGIC),
GetAddrBtcMessage)
self.create_message_successfully(
AddrBtcMessage(self.MAGIC,
[(int(time.time()), "127.0.0.1", 8000)]),
AddrBtcMessage)
inv_vector = [(1, self.HASH), (2, self.HASH)]
self.create_message_successfully(InvBtcMessage(self.MAGIC, inv_vector),
InvBtcMessage)
self.create_message_successfully(
GetDataBtcMessage(self.MAGIC, inv_vector), GetDataBtcMessage)
self.create_message_successfully(
NotFoundBtcMessage(self.MAGIC, inv_vector), NotFoundBtcMessage)
hashes = [self.HASH, self.HASH]
self.create_message_successfully(
GetHeadersBtcMessage(self.MAGIC, self.VERSION, hashes, self.HASH),
GetHeadersBtcMessage)
self.create_message_successfully(
GetBlocksBtcMessage(self.MAGIC, self.VERSION, hashes, self.HASH),
GetBlocksBtcMessage)
self.create_message_successfully(
TxBtcMessage(self.MAGIC, self.VERSION, [], [], 0), TxBtcMessage)
txs = [TxIn(buf=bytearray(10), length=10, off=0).rawbytes()] * 5
self.create_message_successfully(
BlockBtcMessage(self.MAGIC, self.VERSION, self.HASH, self.HASH, 0,
0, 0, txs), BlockBtcMessage)
self.create_message_successfully(
HeadersBtcMessage(self.MAGIC,
[helpers.generate_bytearray(81)] * 2),
HeadersBtcMessage)
self.create_message_successfully(
RejectBtcMessage(self.MAGIC, b"a message",
RejectBtcMessage.REJECT_MALFORMED, b"test break",
helpers.generate_bytearray(10)), RejectBtcMessage)
self.create_message_successfully(SendHeadersBtcMessage(self.MAGIC),
SendHeadersBtcMessage)
self.create_message_successfully(
FeeFilterBtcMessage(self.MAGIC, fee_rate=100), FeeFilterBtcMessage)
self.create_message_successfully(
BtcMessage(self.MAGIC, b'xversion', 0, bytearray(30)),
XversionBtcMessage)
def test_message_preview_success_all_gateway_types(self):
self.get_message_preview_successfully(
GatewayHelloMessage(123, 1, "127.0.0.1", 40000, 1),
GatewayMessageType.HELLO,
GatewayHelloMessage.PAYLOAD_LENGTH,
)
self.get_message_preview_successfully(
BlockReceivedMessage(self.HASH),
GatewayMessageType.BLOCK_RECEIVED,
BlockReceivedMessage.PAYLOAD_LENGTH,
)
self.get_message_preview_successfully(
BlockPropagationRequestMessage(self.BLOCK),
GatewayMessageType.BLOCK_PROPAGATION_REQUEST,
len(self.BLOCK) + constants.CONTROL_FLAGS_LEN,
)
self.get_message_preview_successfully(
BlockHoldingMessage(self.HASH, network_num=123),
BloxrouteMessageType.BLOCK_HOLDING,
BlockHoldingMessage.PAYLOAD_LENGTH,
)
self.get_message_preview_successfully(
ConfirmedTxMessage(self.HASH, self.TX_VAL),
GatewayMessageType.CONFIRMED_TX,
ConfirmedTxMessage.PAYLOAD_LENGTH + len(self.TX_VAL),
)
self.get_message_preview_successfully(
RequestTxStreamMessage(),
GatewayMessageType.REQUEST_TX_STREAM,
constants.CONTROL_FLAGS_LEN,
)
hash_val = BtcObjectHash(buf=crypto.double_sha256(b"123"),
length=crypto.SHA256_HASH_LEN)
blockchain_message = GetBlocksBtcMessage(12345, 23456, [hash_val],
hash_val).rawbytes()
self.get_message_preview_successfully(
BlockchainSyncRequestMessage(GetBlocksBtcMessage.MESSAGE_TYPE,
blockchain_message),
BlockchainSyncRequestMessage.MESSAGE_TYPE,
MSG_TYPE_LEN + len(blockchain_message),
)
self.get_message_preview_successfully(
BlockchainSyncResponseMessage(GetBlocksBtcMessage.MESSAGE_TYPE,
blockchain_message),
BlockchainSyncResponseMessage.MESSAGE_TYPE,
MSG_TYPE_LEN + len(blockchain_message),
)
def test_create_message_success_all_gateway_types(self):
hello_message = self.create_message_successfully(
GatewayHelloMessage(123, 1, "127.0.0.1", 40001, 1),
GatewayHelloMessage)
self.assertEqual(123, hello_message.protocol_version())
self.assertEqual(1, hello_message.network_num())
self.assertEqual("127.0.0.1", hello_message.ip())
self.assertEqual(40001, hello_message.port())
self.assertEqual(1, hello_message.ordering())
block_recv_message = self.create_message_successfully(
BlockReceivedMessage(self.HASH), BlockReceivedMessage)
self.assertEqual(self.HASH, block_recv_message.block_hash())
block_holding_message = self.create_message_successfully(
BlockHoldingMessage(self.HASH, network_num=123),
BlockHoldingMessage)
self.assertEqual(self.HASH, block_holding_message.block_hash())
block_propagation_request_message = self.create_message_successfully(
BlockPropagationRequestMessage(self.BLOCK),
BlockPropagationRequestMessage)
self.assertEqual(self.BLOCK, block_propagation_request_message.blob())
hash_val = BtcObjectHash(buf=crypto.double_sha256(b"123"),
length=crypto.SHA256_HASH_LEN)
blockchain_message_in = GetBlocksBtcMessage(12345, 23456, [hash_val],
hash_val).rawbytes()
sync_request_message = self.create_message_successfully(
BlockchainSyncRequestMessage(GetBlocksBtcMessage.MESSAGE_TYPE,
blockchain_message_in),
BlockchainSyncRequestMessage,
)
blockchain_message_out = GetBlocksBtcMessage(
buf=sync_request_message.payload())
self.assertEqual(12345, blockchain_message_out.magic())
self.assertEqual(23456, blockchain_message_out.version())
self.assertEqual(1, blockchain_message_out.hash_count())
self.assertEqual(hash_val, blockchain_message_out.hash_stop())
self.create_message_successfully(
BlockchainSyncResponseMessage(GetBlocksBtcMessage.MESSAGE_TYPE,
blockchain_message_in),
BlockchainSyncResponseMessage,
)
blockchain_message_out = GetBlocksBtcMessage(
buf=sync_request_message.payload())
self.assertEqual(12345, blockchain_message_out.magic())
self.assertEqual(23456, blockchain_message_out.version())
self.assertEqual(1, blockchain_message_out.hash_count())
self.assertEqual(hash_val, blockchain_message_out.hash_stop())
confirmed_tx = self.create_message_successfully(
ConfirmedTxMessage(self.HASH, self.TX_VAL), ConfirmedTxMessage)
self.assertEqual(self.HASH, confirmed_tx.tx_hash())
self.assertEqual(self.TX_VAL, confirmed_tx.tx_val())
confirmed_tx_no_content = self.create_message_successfully(
ConfirmedTxMessage(self.HASH), ConfirmedTxMessage)
self.assertEqual(self.HASH, confirmed_tx_no_content.tx_hash())
self.assertEqual(TxMessage.EMPTY_TX_VAL,
confirmed_tx_no_content.tx_val())
self.create_message_successfully(RequestTxStreamMessage(),
RequestTxStreamMessage)