Update the test for disseminatePreEncryptedMsg API command

This commit is contained in:
Lee Miller 2024-04-05 00:20:16 +03:00
parent f6bd1546a7
commit 629efb253e
Signed by untrusted user: lee.miller
GPG Key ID: 4F97A5EA88F4AB63

View File

@ -8,9 +8,7 @@ from struct import pack
from six.moves import queue, xmlrpc_client from six.moves import queue, xmlrpc_client
from pybitmessage import protocol from pybitmessage import protocol
from pybitmessage.defaults import ( from pybitmessage.highlevelcrypto import calculateInventoryHash
networkDefaultProofOfWorkNonceTrialsPerByte,
networkDefaultPayloadLengthExtraBytes)
from .partial import TestPartialRun from .partial import TestPartialRun
from .samples import sample_statusbar_msg, sample_object_data from .samples import sample_statusbar_msg, sample_object_data
@ -80,12 +78,14 @@ class TestAPIThread(TestPartialRun):
from inventory import Inventory from inventory import Inventory
proofofwork.init() proofofwork.init()
update_object = pack( self.assertEqual(
unhexlify(self.api.disseminatePreparedObject(
hexlify(sample_object_data).decode())),
calculateInventoryHash(sample_object_data))
update_object = b'\x00' * 8 + pack(
'>Q', int(time.time() + 7200)) + sample_object_data[16:] '>Q', int(time.time() + 7200)) + sample_object_data[16:]
invhash = unhexlify(self.api.disseminatePreEncryptedMsg( invhash = unhexlify(self.api.disseminatePreEncryptedMsg(
hexlify(update_object).decode(), hexlify(update_object).decode()
networkDefaultProofOfWorkNonceTrialsPerByte,
networkDefaultPayloadLengthExtraBytes
)) ))
obj_type, obj_stream, obj_data = Inventory()[invhash][:3] obj_type, obj_stream, obj_data = Inventory()[invhash][:3]
self.assertEqual(obj_type, 42) self.assertEqual(obj_type, 42)