mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
7bd6d485bc
* Test Bigchain.get_blocks_status_containing_tx() exception case * Test Bigchaindb.has_previous_vote() * Add missing blank lines (pep 8) * Group imports * Move ImproperVoteError into .exceptions.py * Simplify logic * Simplify formatting * Imrpove the docstrings a bit * Move GenesisBlockAlreadyExistsError into .exceptions.py * Remove unused import * Add missing blank line * Remove extra blank line * Group imports from same module together * Add test for Bigchain.transaction_exists()
109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
"""
|
|
Fixtures and setup / teardown functions
|
|
|
|
Tasks:
|
|
1. setup test database before starting the tests
|
|
2. delete test database after running the tests
|
|
"""
|
|
|
|
import pytest
|
|
import rethinkdb as r
|
|
|
|
from bigchaindb import Bigchain
|
|
from bigchaindb.db import get_conn
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def restore_config(request, node_config):
|
|
from bigchaindb import config_utils
|
|
config_utils.set_config(node_config)
|
|
|
|
|
|
@pytest.fixture(scope='module', autouse=True)
|
|
def setup_database(request, node_config):
|
|
print('Initializing test db')
|
|
db_name = node_config['database']['name']
|
|
get_conn().repl()
|
|
try:
|
|
r.db_create(db_name).run()
|
|
except r.ReqlOpFailedError as e:
|
|
if e.message == 'Database `{}` already exists.'.format(db_name):
|
|
r.db_drop(db_name).run()
|
|
r.db_create(db_name).run()
|
|
else:
|
|
raise
|
|
|
|
print('Finished initializing test db')
|
|
|
|
# setup tables
|
|
r.db(db_name).table_create('bigchain').run()
|
|
r.db(db_name).table_create('backlog').run()
|
|
# create the secondary indexes
|
|
# to order blocks by timestamp
|
|
r.db(db_name).table('bigchain').index_create('block_timestamp', r.row['block']['timestamp']).run()
|
|
# to order blocks by block number
|
|
r.db(db_name).table('bigchain').index_create('block_number', r.row['block']['block_number']).run()
|
|
# to order transactions by timestamp
|
|
r.db(db_name).table('backlog').index_create('transaction_timestamp', r.row['transaction']['timestamp']).run()
|
|
# compound index to read transactions from the backlog per assignee
|
|
r.db(db_name).table('backlog')\
|
|
.index_create('assignee__transaction_timestamp', [r.row['assignee'], r.row['transaction']['timestamp']])\
|
|
.run()
|
|
# order transactions by id
|
|
r.db(db_name).table('bigchain').index_create('transaction_id', r.row['block']['transactions']['id'],
|
|
multi=True).run()
|
|
|
|
r.db(db_name).table('bigchain').index_wait('transaction_id').run()
|
|
|
|
def fin():
|
|
print('Deleting `{}` database'.format(db_name))
|
|
get_conn().repl()
|
|
try:
|
|
r.db_drop(db_name).run()
|
|
except r.ReqlOpFailedError as e:
|
|
if e.message != 'Database `{}` does not exist.'.format(db_name):
|
|
raise
|
|
|
|
print('Finished deleting `{}`'.format(db_name))
|
|
|
|
request.addfinalizer(fin)
|
|
|
|
|
|
@pytest.fixture(scope='function', autouse=True)
|
|
def cleanup_tables(request, node_config):
|
|
db_name = node_config['database']['name']
|
|
|
|
def fin():
|
|
get_conn().repl()
|
|
try:
|
|
r.db(db_name).table('bigchain').delete().run()
|
|
r.db(db_name).table('backlog').delete().run()
|
|
except r.ReqlOpFailedError as e:
|
|
if e.message != 'Database `{}` does not exist.'.format(db_name):
|
|
raise
|
|
|
|
request.addfinalizer(fin)
|
|
|
|
|
|
@pytest.fixture
|
|
def inputs(user_vk, amount=1, b=None):
|
|
from bigchaindb.exceptions import GenesisBlockAlreadyExistsError
|
|
# 1. create the genesis block
|
|
b = b or Bigchain()
|
|
try:
|
|
b.create_genesis_block()
|
|
except GenesisBlockAlreadyExistsError:
|
|
pass
|
|
|
|
# 2. create block with transactions for `USER` to spend
|
|
transactions = []
|
|
for i in range(amount):
|
|
tx = b.create_transaction(b.me, user_vk, None, 'CREATE')
|
|
tx_signed = b.sign_transaction(tx, b.me_private)
|
|
transactions.append(tx_signed)
|
|
b.write_transaction(tx_signed)
|
|
|
|
block = b.create_block(transactions)
|
|
b.write_block(block, durability='hard')
|
|
return block
|