def test_add_then_update(self): cache = TransactionCache(self.store) bytedata_1 = bytes.fromhex(tx_hex_1) tx_hash_1 = bitcoinx.double_sha256(bytedata_1) metadata_1 = TxData(position=11) with SynchronousWriter() as writer: cache.add( [(tx_hash_1, metadata_1, bytedata_1, TxFlags.StateDispatched)], completion_callback=writer.get_callback()) assert writer.succeeded() assert cache.is_cached(tx_hash_1) entry = cache.get_entry(tx_hash_1) assert TxFlags.HasByteData | TxFlags.HasPosition | TxFlags.StateDispatched == entry.flags assert entry.bytedata is not None metadata_2 = TxData(fee=10, height=88) propagate_flags = TxFlags.HasFee | TxFlags.HasHeight with SynchronousWriter() as writer: cache.update([(tx_hash_1, metadata_2, None, propagate_flags | TxFlags.HasPosition)], completion_callback=writer.get_callback()) assert writer.succeeded() entry = cache.get_entry(tx_hash_1) expected_flags = propagate_flags | TxFlags.StateDispatched | TxFlags.HasByteData assert expected_flags == entry.flags, \ f"{TxFlags.to_repr(expected_flags)} != {TxFlags.to_repr(entry.flags)}" assert entry.bytedata is not None
def test_uncleared_bytedata_requirements(self) -> None: cache = TransactionCache(self.store) tx_1 = Transaction.from_hex(tx_hex_1) tx_hash_1 = tx_1.hash() data = TxData(position=11) for state_flag in TRANSACTION_FLAGS: with pytest.raises(wallet_database.InvalidDataError): cache.add([ (tx_hash_1, data, None, state_flag, None) ]) with SynchronousWriter() as writer: cache.add([ (tx_hash_1, data, tx_1, TxFlags.StateSigned, None) ], completion_callback=writer.get_callback()) assert writer.succeeded() # We are applying a clearing of the bytedata, this should be invalid given uncleared. for state_flag in TRANSACTION_FLAGS: with pytest.raises(wallet_database.InvalidDataError): cache.update([ (tx_hash_1, data, None, state_flag | TxFlags.HasByteData) ])
def test_add_then_update(self): cache = TransactionCache(self.store) bytedata_1 = bytes.fromhex(tx_hex_1) tx_hash_1 = bitcoinx.double_sha256(bytedata_1) metadata_1 = TxData(position=11) with SynchronousWriter() as writer: cache.add([(tx_hash_1, metadata_1, bytedata_1, TxFlags.StateDispatched, None)], completion_callback=writer.get_callback()) assert writer.succeeded() assert cache.is_cached(tx_hash_1) entry = cache.get_entry(tx_hash_1) assert TxFlags.HasByteData | TxFlags.HasPosition | TxFlags.StateDispatched == entry.flags assert cache.have_transaction_data_cached(tx_hash_1) # NOTE: We are not updating bytedata, and it should remain the same. The flags we pass # into update are treated specially to achieve this. metadata_2 = TxData(fee=10, height=88) propagate_flags = TxFlags.HasFee | TxFlags.HasHeight with SynchronousWriter() as writer: cache.update([(tx_hash_1, metadata_2, None, propagate_flags | TxFlags.HasPosition)], completion_callback=writer.get_callback()) assert writer.succeeded() # Check the cache to see that the flags are correct and that bytedata is cached. entry = cache.get_entry(tx_hash_1) expected_flags = propagate_flags | TxFlags.StateDispatched | TxFlags.HasByteData assert expected_flags == entry.flags, \ f"{TxFlags.to_repr(expected_flags)} != {TxFlags.to_repr(entry.flags)}" assert cache.have_transaction_data_cached(tx_hash_1) # Check the store to see that the flags are correct and the bytedata is retained. rows = self.store.read(tx_hashes=[tx_hash_1]) assert 1 == len(rows) get_tx_hash, bytedata_get, flags_get, metadata_get = rows[0] assert bytedata_1 == bytedata_get assert flags_get & TxFlags.HasByteData != 0
def test_get_unsynced_hashes(self): cache = TransactionCache(self.store) tx_1 = Transaction.from_hex(tx_hex_1) tx_hash_1 = tx_1.hash() metadata_1 = TxData(height=11) with SynchronousWriter() as writer: cache.add([ (tx_hash_1, metadata_1, None, TxFlags.Unset, None) ], completion_callback=writer.get_callback()) assert writer.succeeded() results = cache.get_unsynced_hashes() assert 1 == len(results) metadata_2 = TxData() with SynchronousWriter() as writer: cache.update([ (tx_hash_1, metadata_2, tx_1, TxFlags.HasByteData) ], completion_callback=writer.get_callback()) assert writer.succeeded() results = cache.get_unsynced_hashes() assert 0 == len(results)