mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-18 06:34:19 +01:00
d43ab76220
The modules that need to run either bitcoind or the api do now reference to the fixture in the first test that needs it. Since the fixtures are definexd session-wise the rest of the modules will have access to them from that point on.
66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
import pytest
|
|
import logging
|
|
from os import urandom
|
|
from uuid import uuid4
|
|
from hashlib import sha256
|
|
from binascii import unhexlify
|
|
|
|
from pisa.block_processor import BlockProcessor
|
|
|
|
logging.getLogger().disabled = True
|
|
|
|
APPOINTMENT_COUNT = 100
|
|
TEST_SET_SIZE = 200
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def txids():
|
|
return [urandom(32).hex() for _ in range(APPOINTMENT_COUNT)]
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def locator_uuid_map(txids):
|
|
return {sha256(unhexlify(txid)).hexdigest(): uuid4().hex for txid in txids}
|
|
|
|
|
|
@pytest.fixture
|
|
def best_block_hash():
|
|
return BlockProcessor.get_best_block_hash()
|
|
|
|
|
|
def test_get_best_block_hash(run_bitcoind, best_block_hash):
|
|
# As long as bitcoind is running (or mocked in this case) we should always a block hash
|
|
assert best_block_hash is not None and isinstance(best_block_hash, str)
|
|
|
|
|
|
def test_get_block(best_block_hash):
|
|
# Getting a block from a block hash we are aware of should return data
|
|
block = BlockProcessor.get_block(best_block_hash)
|
|
|
|
# Checking that the received block has at least the fields we need
|
|
# FIXME: We could be more strict here, but we'll need to add those restrictions to bitcoind_sim too
|
|
assert isinstance(block, dict)
|
|
assert block.get('hash') == best_block_hash and 'height' in block and 'previousblockhash' in block and 'tx' in block
|
|
|
|
|
|
def test_get_block_count():
|
|
block_count = BlockProcessor.get_block_count()
|
|
assert isinstance(block_count, int) and block_count >= 0
|
|
|
|
|
|
def test_potential_matches(txids, locator_uuid_map):
|
|
potential_matches = BlockProcessor.get_potential_matches(txids, locator_uuid_map)
|
|
|
|
# All the txids must match
|
|
assert locator_uuid_map.keys() == potential_matches.keys()
|
|
|
|
|
|
def test_potential_matches_random_data(locator_uuid_map):
|
|
# The likelihood of finding a potential match with random data should be negligible
|
|
txids = [urandom(32).hex() for _ in range(TEST_SET_SIZE)]
|
|
|
|
potential_matches = BlockProcessor.get_potential_matches(txids, locator_uuid_map)
|
|
|
|
# None of the txids should match
|
|
assert len(potential_matches) == 0
|