mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
116 lines
2.8 KiB
Python
116 lines
2.8 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 os
|
|
import copy
|
|
|
|
import pytest
|
|
|
|
|
|
DB_NAME = 'bigchain_test_{}'.format(os.getpid())
|
|
|
|
CONFIG = {
|
|
'database': {
|
|
'name': DB_NAME,
|
|
},
|
|
'keypair': {
|
|
'private': '31Lb1ZGKTyHnmVK3LUMrAUrPNfd4sE2YyBt3UA4A25aA',
|
|
'public': '4XYfCbabAWVUCbjTmRTFEu2sc3dFEdkse4r6X498B1s8',
|
|
}
|
|
}
|
|
|
|
# Test user. inputs will be created for this user. Cryptography Keys
|
|
USER_PRIVATE_KEY = '8eJ8q9ZQpReWyQT5aFCiwtZ5wDZC4eDnCen88p3tQ6ie'
|
|
USER_PUBLIC_KEY = 'JEAkEJqLbbgDRAtMm8YAjGp759Aq2qTn9eaEHUj2XePE'
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
from bigchaindb.backend import connection
|
|
|
|
backends = ', '.join(connection.BACKENDS.keys())
|
|
|
|
parser.addoption('--database-backend', action='store', default='rethinkdb',
|
|
help='Defines the backend to use (available: {})'.format(backends))
|
|
|
|
|
|
# We need this function to avoid loading an existing
|
|
# conf file located in the home of the user running
|
|
# the tests. If it's too aggressive we can change it
|
|
# later.
|
|
@pytest.fixture(scope='function', autouse=True)
|
|
def ignore_local_config_file(monkeypatch):
|
|
def mock_file_config(filename=None):
|
|
raise FileNotFoundError()
|
|
|
|
monkeypatch.setattr('bigchaindb.config_utils.file_config',
|
|
mock_file_config)
|
|
|
|
|
|
@pytest.fixture(scope='function', autouse=True)
|
|
def restore_config(request, node_config):
|
|
from bigchaindb import config_utils
|
|
config_utils.set_config(node_config)
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def node_config(request):
|
|
config = copy.deepcopy(CONFIG)
|
|
config['database']['backend'] = request.config.getoption('--database-backend')
|
|
return config
|
|
|
|
|
|
@pytest.fixture
|
|
def user_sk():
|
|
return USER_PRIVATE_KEY
|
|
|
|
|
|
@pytest.fixture
|
|
def user_pk():
|
|
return USER_PUBLIC_KEY
|
|
|
|
|
|
@pytest.fixture
|
|
def b(request, node_config):
|
|
restore_config(request, node_config)
|
|
from bigchaindb import Bigchain
|
|
return Bigchain()
|
|
|
|
|
|
@pytest.fixture
|
|
def create_tx(b, user_pk):
|
|
from bigchaindb.models import Transaction
|
|
return Transaction.create([b.me], [([user_pk], 1)])
|
|
|
|
|
|
@pytest.fixture
|
|
def signed_create_tx(b, create_tx):
|
|
return create_tx.sign([b.me_private])
|
|
|
|
|
|
@pytest.fixture
|
|
def signed_transfer_tx(signed_create_tx, user_pk, user_sk):
|
|
from bigchaindb.models import Transaction
|
|
inputs = signed_create_tx.to_inputs()
|
|
tx = Transaction.transfer(inputs, [([user_pk], 1)], signed_create_tx.asset)
|
|
return tx.sign([user_sk])
|
|
|
|
|
|
@pytest.fixture
|
|
def structurally_valid_vote():
|
|
return {
|
|
'node_pubkey': 'c' * 44,
|
|
'signature': 'd' * 86,
|
|
'vote': {
|
|
'voting_for_block': 'a' * 64,
|
|
'previous_block': 'b' * 64,
|
|
'is_block_valid': False,
|
|
'invalid_reason': None,
|
|
'timestamp': '1111111111'
|
|
}
|
|
}
|