chainsyncer/chainsyncer/unittest/base.py

# standard imports
import os
import logging

# external imports
from hexathon import add_0x

# local imports
from chainsyncer.driver.history import HistorySyncer
from chainsyncer.error import NoBlockForYou

logg = logging.getLogger().getChild(__name__)



class MockConn:

    def do(self, o):
        pass


class MockTx:

    def __init__(self, index, tx_hash):
        self.hash = tx_hash
        self.index = index


    def apply_receipt(self, rcpt):
        self.rcpt = rcpt


class MockBlock:

    def __init__(self, number, txs):
        self.number = number
        self.txs = txs


    def tx(self, i):
        return MockTx(i, self.txs[i])


class TestSyncer(HistorySyncer):


    def __init__(self, backend, chain_interface, tx_counts=[]):
        self.tx_counts = tx_counts
        super(TestSyncer, self).__init__(backend, chain_interface)


    def get(self, conn):
        (pair, fltr) = self.backend.get()
        (target_block, fltr) = self.backend.target()
        block_height = pair[0]

        if block_height == target_block:
            self.running = False
            raise NoBlockForYou()
            return []

        block_txs = []
        if block_height < len(self.tx_counts):
            for i in range(self.tx_counts[block_height]):
                block_txs.append(add_0x(os.urandom(32).hex()))
     
        return MockBlock(block_height, block_txs)