bigchaindb/tests/pipelines/test_block_creation.py

import time
import random
from unittest.mock import patch

import rethinkdb as r

from bigchaindb.pipelines import block
from multipipes import Pipe, Pipeline


def test_filter_by_assignee(b, user_vk):
    block_maker = block.Block()

    tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
    tx = b.sign_transaction(tx, b.me_private)
    tx['assignee'] = b.me
    tx['assignment_timestamp'] = 111

    # filter_tx has side effects on the `tx` instance by popping 'assignee'
    # and 'assignment_timestamp'
    filtered_tx = block_maker.filter_tx(tx)
    assert filtered_tx == tx
    assert 'assignee' not in filtered_tx
    assert 'assignment_timestamp' not in filtered_tx

    tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
    tx = b.sign_transaction(tx, b.me_private)
    tx['assignee'] = 'nobody'

    assert block_maker.filter_tx(tx) is None


def test_validate_transaction(b, user_vk):
    block_maker = block.Block()

    tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
    tx = b.sign_transaction(tx, b.me_private)
    tx['id'] = 'a' * 64

    assert block_maker.validate_tx(tx) is None

    tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
    tx = b.sign_transaction(tx, b.me_private)

    assert block_maker.validate_tx(tx) == tx


def test_create_block(b, user_vk):
    block_maker = block.Block()

    for i in range(100):
        tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
        tx = b.sign_transaction(tx, b.me_private)
        block_maker.create(tx)

    # force the output triggering a `timeout`
    block_doc = block_maker.create(None, timeout=True)

    assert len(block_doc['block']['transactions']) == 100


def test_write_block(b, user_vk):
    block_maker = block.Block()

    txs = []
    for i in range(100):
        tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
        tx = b.sign_transaction(tx, b.me_private)
        txs.append(tx)

    block_doc = b.create_block(txs)
    block_maker.write(block_doc)

    assert r.table('bigchain').get(block_doc['id']).run(b.conn) == block_doc


def test_duplicate_transaction(b, user_vk):
    block_maker = block.Block()

    txs = []
    for i in range(10):
        tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
        tx = b.sign_transaction(tx, b.me_private)
        txs.append(tx)

    block_doc = b.create_block(txs)
    block_maker.write(block_doc)

    # block is in bigchain
    assert r.table('bigchain').get(block_doc['id']).run(b.conn) == block_doc

    b.write_transaction(txs[0])

    # verify tx is in the backlog
    assert r.table('backlog').get(txs[0]['id']).run(b.conn) is not None

    # try to validate a transaction that's already in the chain; should not
    # work
    assert block_maker.validate_tx(txs[0]) is None

    # duplicate tx should be removed from backlog
    assert r.table('backlog').get(txs[0]['id']).run(b.conn) is None


def test_delete_tx(b, user_vk):
    block_maker = block.Block()

    for i in range(100):
        tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
        tx = b.sign_transaction(tx, b.me_private)
        block_maker.create(tx)
        # make sure the tx appears in the backlog
        b.write_transaction(tx)

    # force the output triggering a `timeout`
    block_doc = block_maker.create(None, timeout=True)

    for tx in block_doc['block']['transactions']:
        returned_tx = r.table('backlog').get(tx['id']).run(b.conn)
        returned_tx.pop('assignee')
        returned_tx.pop('assignment_timestamp')
        assert returned_tx == tx

    returned_block = block_maker.delete_tx(block_doc)

    assert returned_block == block_doc

    for tx in block_doc['block']['transactions']:
        assert r.table('backlog').get(tx['id']).run(b.conn) is None


def test_prefeed(b, user_vk):
    for i in range(100):
        tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
        tx = b.sign_transaction(tx, b.me_private)
        b.write_transaction(tx)

    backlog = block.initial()

    assert len(list(backlog)) == 100


@patch.object(Pipeline, 'start')
def test_start(mock_start):
    # TODO: `block.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 :)
    block.start()
    mock_start.assert_called_with()


def test_full_pipeline(b, user_vk):
    outpipe = Pipe()

    count_assigned_to_me = 0
    for i in range(100):
        tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
        tx = b.sign_transaction(tx, b.me_private)
        assignee = random.choice([b.me, 'aaa', 'bbb', 'ccc'])
        tx['assignee'] = assignee
        tx['assignment_timestamp'] = time.time()
        if assignee == b.me:
            count_assigned_to_me += 1
        r.table('backlog').insert(tx, durability='hard').run(b.conn)

    assert r.table('backlog').count().run(b.conn) == 100

    pipeline = block.create_pipeline()
    pipeline.setup(indata=block.get_changefeed(), outdata=outpipe)
    pipeline.start()

    time.sleep(2)
    pipeline.terminate()

    block_doc = outpipe.get()

    assert len(block_doc['block']['transactions']) == count_assigned_to_me
    assert r.table('bigchain').get(block_doc['id']).run(b.conn) == block_doc
    assert r.table('backlog').count().run(b.conn) == 100 - count_assigned_to_me