mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
125 lines
4.2 KiB
Python
125 lines
4.2 KiB
Python
import rethinkdb as r
|
|
from bigchaindb import Bigchain
|
|
from bigchaindb.pipelines import stale
|
|
from multipipes import Pipe, Pipeline
|
|
from unittest.mock import patch
|
|
from bigchaindb import config_utils
|
|
import time
|
|
import os
|
|
|
|
|
|
def test_get_stale(b, user_vk):
|
|
from bigchaindb.models import Transaction
|
|
tx = Transaction.create([b.me], [user_vk])
|
|
tx = tx.sign([b.me_private])
|
|
b.write_transaction(tx, durability='hard')
|
|
|
|
stm = stale.StaleTransactionMonitor(timeout=0.001,
|
|
backlog_reassign_delay=0.001)
|
|
tx_stale = stm.check_transactions()
|
|
|
|
for _tx in tx_stale:
|
|
_tx.pop('assignee')
|
|
_tx.pop('assignment_timestamp')
|
|
assert tx.to_dict() == _tx
|
|
|
|
|
|
def test_reassign_transactions(b, user_vk):
|
|
from bigchaindb.models import Transaction
|
|
# test with single node
|
|
tx = Transaction.create([b.me], [user_vk])
|
|
tx = tx.sign([b.me_private])
|
|
b.write_transaction(tx, durability='hard')
|
|
|
|
stm = stale.StaleTransactionMonitor(timeout=0.001,
|
|
backlog_reassign_delay=0.001)
|
|
stm.reassign_transactions(tx.to_dict())
|
|
|
|
# test with federation
|
|
tx = Transaction.create([b.me], [user_vk])
|
|
tx = tx.sign([b.me_private])
|
|
b.write_transaction(tx, durability='hard')
|
|
|
|
stm = stale.StaleTransactionMonitor(timeout=0.001,
|
|
backlog_reassign_delay=0.001)
|
|
stm.bigchain.nodes_except_me = ['aaa', 'bbb', 'ccc']
|
|
tx = list(b.connection.run(r.table('backlog')))[0]
|
|
stm.reassign_transactions(tx)
|
|
|
|
reassigned_tx = b.connection.run(r.table('backlog').get(tx['id']))
|
|
assert reassigned_tx['assignment_timestamp'] > tx['assignment_timestamp']
|
|
assert reassigned_tx['assignee'] != tx['assignee']
|
|
|
|
# test with node not in federation
|
|
tx = Transaction.create([b.me], [user_vk])
|
|
tx = tx.sign([b.me_private]).to_dict()
|
|
tx.update({'assignee': 'lol'})
|
|
tx.update({'assignment_timestamp': time.time()})
|
|
b.connection.run(r.table('backlog').insert(tx, durability='hard'))
|
|
|
|
tx = list(b.connection.run(r.table('backlog')))[0]
|
|
stm.reassign_transactions(tx)
|
|
assert b.connection.run(r.table('backlog').get(tx['id']))['assignee'] != 'lol'
|
|
|
|
|
|
def test_full_pipeline(monkeypatch, user_vk):
|
|
from bigchaindb.models import Transaction
|
|
CONFIG = {
|
|
'database': {
|
|
'name': 'bigchain_test_{}'.format(os.getpid())
|
|
},
|
|
'keypair': {
|
|
'private': '31Lb1ZGKTyHnmVK3LUMrAUrPNfd4sE2YyBt3UA4A25aA',
|
|
'public': '4XYfCbabAWVUCbjTmRTFEu2sc3dFEdkse4r6X498B1s8'
|
|
},
|
|
'keyring': ['aaa', 'bbb'],
|
|
'backlog_reassign_delay': 0.01
|
|
}
|
|
config_utils.set_config(CONFIG)
|
|
b = Bigchain()
|
|
outpipe = Pipe()
|
|
|
|
original_txs = {}
|
|
original_txc = []
|
|
|
|
monkeypatch.setattr('time.time', lambda: 1)
|
|
|
|
for i in range(100):
|
|
tx = Transaction.create([b.me], [user_vk])
|
|
tx = tx.sign([b.me_private])
|
|
original_txc.append(tx.to_dict())
|
|
|
|
b.write_transaction(tx)
|
|
original_txs[tx.id] = b.connection.run(r.table('backlog').get(tx.id))
|
|
|
|
assert b.connection.run(r.table('backlog').count()) == 100
|
|
|
|
monkeypatch.undo()
|
|
|
|
pipeline = stale.create_pipeline(backlog_reassign_delay=1,
|
|
timeout=1)
|
|
pipeline.setup(outdata=outpipe)
|
|
pipeline.start()
|
|
|
|
# to terminate
|
|
for _ in range(100):
|
|
outpipe.get()
|
|
|
|
pipeline.terminate()
|
|
|
|
assert b.connection.run(r.table('backlog').count()) == 100
|
|
reassigned_txs = list(b.connection.run(r.table('backlog')))
|
|
|
|
# check that every assignment timestamp has increased, and every tx has a new assignee
|
|
for reassigned_tx in reassigned_txs:
|
|
assert reassigned_tx['assignment_timestamp'] > original_txs[reassigned_tx['id']]['assignment_timestamp']
|
|
assert reassigned_tx['assignee'] != original_txs[reassigned_tx['id']]['assignee']
|
|
|
|
@patch.object(Pipeline, 'start')
|
|
def test_start(mock_start):
|
|
# TODO: `sta,e.start` is just a wrapper around `block.create_pipeline`,
|
|
# that is tested by `test_full_pipeline`.
|
|
# If anyone has better ideas on how to test this, please do a PR :)
|
|
stale.start()
|
|
mock_start.assert_called_with()
|