def test_get_transaction_after_metadata(self):
# Getting an entry for a settled transaction should update from metadata-only to full.
bytedata_set = bytes.fromhex(tx_hex_1)
tx_hash = bitcoinx.double_sha256(bytedata_set)
metadata_set = TxData(height=1, fee=2, position=None, date_added=1, date_updated=1)
with SynchronousWriter() as writer:
self.store.create([ (tx_hash, metadata_set, bytedata_set, TxFlags.StateSettled, None) ],
completion_callback=writer.get_callback())
assert writer.succeeded()
cache = TransactionCache(self.store)
metadata_get = cache.get_metadata(tx_hash)
assert metadata_get is not None
# Initial priming of cache will be only metadata.
cached_entry_1 = cache._cache[tx_hash]
assert not cache.have_transaction_data_cached(tx_hash)
# Entry request will hit the database.
entry = cache.get_entry(tx_hash)
assert cache.have_transaction_data_cached(tx_hash)
cached_entry_2 = cache._cache[tx_hash]
assert entry.metadata == cached_entry_2.metadata
assert entry.flags == cached_entry_2.flags
def test_get_metadata(self):
# Full entry caching for non-settled transactions, otherwise only metadata.
bytedata_set_1 = bytes.fromhex(tx_hex_1)
tx_hash_1 = bitcoinx.double_sha256(bytedata_set_1)
metadata_set_1 = TxData(height=None, fee=2, position=None, date_added=1, date_updated=1)
bytedata_set_2 = bytes.fromhex(tx_hex_2)
tx_hash_2 = bitcoinx.double_sha256(bytedata_set_2)
metadata_set_2 = TxData(height=1, fee=2, position=10, date_added=1, date_updated=1)
with SynchronousWriter() as writer:
self.store.create([ (tx_hash_1, metadata_set_1, bytedata_set_1, TxFlags.Unset, None),
(tx_hash_2, metadata_set_2, bytedata_set_2, TxFlags.StateSettled, None), ],
completion_callback=writer.get_callback())
assert writer.succeeded()
cache = TransactionCache(self.store)
metadata_get = cache.get_metadata(tx_hash_1)
assert metadata_set_1.height == metadata_get.height
assert metadata_set_1.fee == metadata_get.fee
assert metadata_set_1.position == metadata_get.position
metadata_get = cache.get_metadata(tx_hash_2)
assert metadata_set_2.height == metadata_get.height
assert metadata_set_2.fee == metadata_get.fee
assert metadata_set_2.position == metadata_get.position
entry = cache._cache[tx_hash_1]
assert cache.have_transaction_data_cached(tx_hash_1)
entry = cache._cache[tx_hash_2]
assert not cache.have_transaction_data_cached(tx_hash_2)
def test_add_transaction_update(self):
cache = TransactionCache(self.store)
tx = Transaction.from_hex(tx_hex_1)
data = [
tx.hash(),
TxData(height=1295924,
position=4,
fee=None,
date_added=1,
date_updated=1), None, TxFlags.Unset, None
]
with SynchronousWriter() as writer:
cache.add([data], completion_callback=writer.get_callback())
assert writer.succeeded()
entry = cache.get_entry(tx.hash())
assert entry is not None
assert TxFlags.Unset == entry.flags & TxFlags.STATE_MASK
with SynchronousWriter() as writer:
cache.add_transaction(tx,
TxFlags.StateCleared,
completion_callback=writer.get_callback())
assert writer.succeeded()
tx_hash = tx.hash()
entry = cache.get_entry(tx_hash)
assert entry is not None
assert cache.have_transaction_data_cached(tx_hash)
assert TxFlags.StateCleared == entry.flags & TxFlags.StateCleared
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_add_transaction(self):
cache = TransactionCache(self.store)
tx = Transaction.from_hex(tx_hex_1)
tx_hash = tx.hash()
with SynchronousWriter() as writer:
cache.add_transaction(tx_hash, tx, completion_callback=writer.get_callback())
assert writer.succeeded()
assert cache.is_cached(tx_hash)
entry = cache.get_entry(tx_hash)
assert TxFlags.HasByteData == entry.flags & TxFlags.HasByteData
assert cache.have_transaction_data_cached(tx_hash)