60 lines
1.3 KiB
Python
60 lines
1.3 KiB
Python
# standard imports
|
|
import os
|
|
|
|
# external imports
|
|
from hexathon import add_0x
|
|
|
|
# local imports
|
|
from chainsyncer.driver import HistorySyncer
|
|
from chainsyncer.error import NoBlockForYou
|
|
|
|
|
|
class MockTx:
|
|
|
|
def __init__(self, tx_hash, index):
|
|
self.hash = tx_hash
|
|
self.index = index
|
|
|
|
|
|
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, tx_counts=[]):
|
|
self.tx_counts = tx_counts
|
|
super(TestSyncer, self).__init__(backend)
|
|
|
|
|
|
def get(self, conn):
|
|
if self.backend.block_height == self.backend.target_block:
|
|
self.running = False
|
|
raise NoBlockForYou()
|
|
if self.backend.block_height > len(self.tx_counts):
|
|
return []
|
|
|
|
block_txs = []
|
|
for i in range(self.tx_counts[self.backend.block_height]):
|
|
block_txs.append(add_0x(os.urandom(32).hex()))
|
|
|
|
return MockBlock(self.backend.block_height, block_txs)
|
|
|
|
|
|
def process(self, conn, block):
|
|
i = 0
|
|
for tx in block.txs:
|
|
self.process_single(conn, block, tx)
|
|
i += 1
|
|
self.backend.set(self.backend.block_height + 1, 0)
|
|
|
|
|