616 lines
24 KiB
Python
616 lines
24 KiB
Python
import json
|
|
from unittest.mock import Mock, patch
|
|
from argparse import Namespace
|
|
import copy
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.tendermint
|
|
def test_make_sure_we_dont_remove_any_command():
|
|
# thanks to: http://stackoverflow.com/a/18161115/597097
|
|
from bigchaindb.commands.bigchaindb import create_parser
|
|
|
|
parser = create_parser()
|
|
|
|
assert parser.parse_args(['configure', 'localmongodb']).command
|
|
assert parser.parse_args(['configure', 'localmongodb']).command
|
|
assert parser.parse_args(['show-config']).command
|
|
assert parser.parse_args(['export-my-pubkey']).command
|
|
assert parser.parse_args(['init']).command
|
|
assert parser.parse_args(['drop']).command
|
|
assert parser.parse_args(['start']).command
|
|
assert parser.parse_args(['set-shards', '1']).command
|
|
assert parser.parse_args(['set-replicas', '1']).command
|
|
assert parser.parse_args(['add-replicas', 'localhost:27017']).command
|
|
assert parser.parse_args(['remove-replicas', 'localhost:27017']).command
|
|
|
|
|
|
@patch('bigchaindb.commands.utils.start')
|
|
def test_main_entrypoint(mock_start):
|
|
from bigchaindb.commands.bigchaindb import main
|
|
main()
|
|
|
|
assert mock_start.called
|
|
|
|
|
|
def test_bigchain_run_start(mock_run_configure,
|
|
mock_processes_start,
|
|
mock_db_init_with_existing_db,
|
|
mocked_setup_logging):
|
|
from bigchaindb import config
|
|
from bigchaindb.commands.bigchaindb import run_start
|
|
args = Namespace(start_rethinkdb=False, allow_temp_keypair=False, config=None, yes=True,
|
|
skip_initialize_database=False)
|
|
run_start(args)
|
|
mocked_setup_logging.assert_called_once_with(user_log_config=config['log'])
|
|
|
|
|
|
@pytest.mark.skipif(reason="BigchainDB doesn't support the automatic creation of a config file anymore")
|
|
def test_bigchain_run_start_assume_yes_create_default_config(
|
|
monkeypatch, mock_processes_start, mock_generate_key_pair,
|
|
mock_db_init_with_existing_db, mocked_setup_logging):
|
|
import bigchaindb
|
|
from bigchaindb.commands.bigchaindb import run_start
|
|
from bigchaindb import config_utils
|
|
|
|
value = {}
|
|
expected_config = copy.deepcopy(bigchaindb._config)
|
|
expected_config['keypair']['public'] = 'pubkey'
|
|
expected_config['keypair']['private'] = 'privkey'
|
|
|
|
def mock_write_config(newconfig, filename=None):
|
|
value['return'] = newconfig
|
|
|
|
monkeypatch.setattr(config_utils, 'write_config', mock_write_config)
|
|
monkeypatch.setattr(config_utils, 'file_config', lambda x: config_utils.set_config(expected_config))
|
|
monkeypatch.setattr('os.path.exists', lambda path: False)
|
|
|
|
args = Namespace(config=None, yes=True)
|
|
run_start(args)
|
|
|
|
mocked_setup_logging.assert_called_once_with()
|
|
assert value['return'] == expected_config
|
|
|
|
|
|
# TODO Please beware, that if debugging, the "-s" switch for pytest will
|
|
# interfere with capsys.
|
|
# See related issue: https://github.com/pytest-dev/pytest/issues/128
|
|
@pytest.mark.usefixtures('ignore_local_config_file')
|
|
def test_bigchain_show_config(capsys):
|
|
from bigchaindb import config
|
|
from bigchaindb.commands.bigchaindb import run_show_config
|
|
|
|
args = Namespace(config=None)
|
|
_, _ = capsys.readouterr()
|
|
run_show_config(args)
|
|
output_config = json.loads(capsys.readouterr()[0])
|
|
del config['CONFIGURED']
|
|
config['keypair']['private'] = 'x' * 45
|
|
assert output_config == config
|
|
|
|
|
|
@pytest.mark.usefixtures('ignore_local_config_file')
|
|
def test_bigchain_export_my_pubkey_when_pubkey_set(capsys, monkeypatch):
|
|
from bigchaindb import config
|
|
from bigchaindb.commands.bigchaindb import run_export_my_pubkey
|
|
|
|
args = Namespace(config=None)
|
|
# so in run_export_my_pubkey(args) below,
|
|
# filename=args.config='dummy' is passed to autoconfigure().
|
|
# We just assume autoconfigure() works and sets
|
|
# config['keypair']['public'] correctly (tested elsewhere).
|
|
# We force-set config['keypair']['public'] using monkeypatch.
|
|
monkeypatch.setitem(config['keypair'], 'public', 'Charlie_Bucket')
|
|
_, _ = capsys.readouterr() # has the effect of clearing capsys
|
|
run_export_my_pubkey(args)
|
|
out, _ = capsys.readouterr()
|
|
lines = out.splitlines()
|
|
assert config['keypair']['public'] in lines
|
|
assert 'Charlie_Bucket' in lines
|
|
|
|
|
|
@pytest.mark.usefixtures('ignore_local_config_file')
|
|
def test_bigchain_export_my_pubkey_when_pubkey_not_set(monkeypatch):
|
|
from bigchaindb import config
|
|
from bigchaindb.commands.bigchaindb import run_export_my_pubkey
|
|
|
|
args = Namespace(config=None)
|
|
monkeypatch.setitem(config['keypair'], 'public', None)
|
|
# assert that run_export_my_pubkey(args) raises SystemExit:
|
|
with pytest.raises(SystemExit) as exc_info:
|
|
run_export_my_pubkey(args)
|
|
# exc_info is an object of class ExceptionInfo
|
|
# https://pytest.org/latest/builtin.html#_pytest._code.ExceptionInfo
|
|
assert exc_info.type == SystemExit
|
|
# exc_info.value is an object of class SystemExit
|
|
# https://docs.python.org/3/library/exceptions.html#SystemExit
|
|
assert exc_info.value.code == \
|
|
"This node's public key wasn't set anywhere so it can't be exported"
|
|
|
|
|
|
def test_bigchain_run_init_when_db_exists(mocker, capsys):
|
|
from bigchaindb.commands.bigchaindb import run_init
|
|
from bigchaindb.common.exceptions import DatabaseAlreadyExists
|
|
init_db_mock = mocker.patch(
|
|
'bigchaindb.commands.bigchaindb.schema.init_database',
|
|
autospec=True,
|
|
spec_set=True,
|
|
)
|
|
init_db_mock.side_effect = DatabaseAlreadyExists
|
|
args = Namespace(config=None)
|
|
run_init(args)
|
|
output_message = capsys.readouterr()[1]
|
|
print(output_message)
|
|
assert output_message == (
|
|
'The database already exists.\n'
|
|
'If you wish to re-initialize it, first drop it.\n'
|
|
)
|
|
|
|
|
|
def test__run_init(mocker):
|
|
from bigchaindb.commands.bigchaindb import _run_init
|
|
bigchain_mock = mocker.patch(
|
|
'bigchaindb.commands.bigchaindb.bigchaindb.Bigchain')
|
|
init_db_mock = mocker.patch(
|
|
'bigchaindb.commands.bigchaindb.schema.init_database',
|
|
autospec=True,
|
|
spec_set=True,
|
|
)
|
|
_run_init()
|
|
bigchain_mock.assert_called_once_with()
|
|
init_db_mock.assert_called_once_with(
|
|
connection=bigchain_mock.return_value.connection)
|
|
bigchain_mock.return_value.create_genesis_block.assert_called_once_with()
|
|
|
|
|
|
@patch('bigchaindb.backend.schema.drop_database')
|
|
def test_drop_db_when_assumed_yes(mock_db_drop):
|
|
from bigchaindb.commands.bigchaindb import run_drop
|
|
args = Namespace(config=None, yes=True)
|
|
|
|
run_drop(args)
|
|
assert mock_db_drop.called
|
|
|
|
|
|
@patch('bigchaindb.backend.schema.drop_database')
|
|
def test_drop_db_when_interactive_yes(mock_db_drop, monkeypatch):
|
|
from bigchaindb.commands.bigchaindb import run_drop
|
|
args = Namespace(config=None, yes=False)
|
|
monkeypatch.setattr('bigchaindb.commands.bigchaindb.input_on_stderr', lambda x: 'y')
|
|
|
|
run_drop(args)
|
|
assert mock_db_drop.called
|
|
|
|
|
|
@patch('bigchaindb.backend.schema.drop_database')
|
|
def test_drop_db_when_db_does_not_exist(mock_db_drop, capsys):
|
|
from bigchaindb import config
|
|
from bigchaindb.commands.bigchaindb import run_drop
|
|
from bigchaindb.common.exceptions import DatabaseDoesNotExist
|
|
args = Namespace(config=None, yes=True)
|
|
mock_db_drop.side_effect = DatabaseDoesNotExist
|
|
|
|
run_drop(args)
|
|
output_message = capsys.readouterr()[1]
|
|
assert output_message == "Cannot drop '{name}'. The database does not exist.\n".format(
|
|
name=config['database']['name'])
|
|
|
|
|
|
@patch('bigchaindb.backend.schema.drop_database')
|
|
def test_drop_db_does_not_drop_when_interactive_no(mock_db_drop, monkeypatch):
|
|
from bigchaindb.commands.bigchaindb import run_drop
|
|
args = Namespace(config=None, yes=False)
|
|
monkeypatch.setattr('bigchaindb.commands.bigchaindb.input_on_stderr', lambda x: 'n')
|
|
|
|
run_drop(args)
|
|
assert not mock_db_drop.called
|
|
|
|
|
|
def test_run_configure_when_config_exists_and_skipping(monkeypatch):
|
|
from bigchaindb.commands.bigchaindb import run_configure
|
|
monkeypatch.setattr('os.path.exists', lambda path: True)
|
|
args = Namespace(config='foo', yes=True)
|
|
return_value = run_configure(args, skip_if_exists=True)
|
|
assert return_value is None
|
|
|
|
|
|
# TODO Beware if you are putting breakpoints in there, and using the '-s'
|
|
# switch with pytest. It will just hang. Seems related to the monkeypatching of
|
|
# input_on_stderr.
|
|
def test_run_configure_when_config_does_not_exist(monkeypatch,
|
|
mock_write_config,
|
|
mock_generate_key_pair,
|
|
mock_bigchaindb_backup_config):
|
|
from bigchaindb.commands.bigchaindb import run_configure
|
|
monkeypatch.setattr('os.path.exists', lambda path: False)
|
|
monkeypatch.setattr('builtins.input', lambda: '\n')
|
|
args = Namespace(config='foo', backend='rethinkdb', yes=True)
|
|
return_value = run_configure(args)
|
|
assert return_value is None
|
|
|
|
|
|
def test_run_configure_when_config_does_exist(monkeypatch,
|
|
mock_write_config,
|
|
mock_generate_key_pair,
|
|
mock_bigchaindb_backup_config):
|
|
value = {}
|
|
|
|
def mock_write_config(newconfig, filename=None):
|
|
value['return'] = newconfig
|
|
|
|
from bigchaindb.commands.bigchaindb import run_configure
|
|
monkeypatch.setattr('os.path.exists', lambda path: True)
|
|
monkeypatch.setattr('builtins.input', lambda: '\n')
|
|
monkeypatch.setattr('bigchaindb.config_utils.write_config', mock_write_config)
|
|
|
|
args = Namespace(config='foo', yes=None)
|
|
run_configure(args)
|
|
assert value == {}
|
|
|
|
|
|
@pytest.mark.parametrize('backend', (
|
|
'rethinkdb',
|
|
'mongodb',
|
|
))
|
|
def test_run_configure_with_backend(backend, monkeypatch, mock_write_config):
|
|
import bigchaindb
|
|
from bigchaindb.commands.bigchaindb import run_configure
|
|
|
|
value = {}
|
|
|
|
def mock_write_config(new_config, filename=None):
|
|
value['return'] = new_config
|
|
|
|
monkeypatch.setattr('os.path.exists', lambda path: False)
|
|
monkeypatch.setattr('builtins.input', lambda: '\n')
|
|
monkeypatch.setattr('bigchaindb.config_utils.write_config',
|
|
mock_write_config)
|
|
|
|
args = Namespace(config='foo', backend=backend, yes=True)
|
|
expected_config = bigchaindb.config
|
|
run_configure(args)
|
|
|
|
# update the expected config with the correct backend and keypair
|
|
backend_conf = getattr(bigchaindb, '_database_' + backend)
|
|
expected_config.update({'database': backend_conf,
|
|
'keypair': value['return']['keypair']})
|
|
|
|
assert value['return'] == expected_config
|
|
|
|
|
|
@patch('bigchaindb.common.crypto.generate_key_pair',
|
|
return_value=('private_key', 'public_key'))
|
|
@pytest.mark.usefixtures('ignore_local_config_file')
|
|
def test_allow_temp_keypair_generates_one_on_the_fly(
|
|
mock_gen_keypair, mock_processes_start,
|
|
mock_db_init_with_existing_db, mocked_setup_logging):
|
|
import bigchaindb
|
|
from bigchaindb.commands.bigchaindb import run_start
|
|
|
|
bigchaindb.config['keypair'] = {'private': None, 'public': None}
|
|
|
|
args = Namespace(allow_temp_keypair=True, start_rethinkdb=False, config=None, yes=True,
|
|
skip_initialize_database=False)
|
|
run_start(args)
|
|
|
|
mocked_setup_logging.assert_called_once_with(
|
|
user_log_config=bigchaindb.config['log'])
|
|
assert bigchaindb.config['keypair']['private'] == 'private_key'
|
|
assert bigchaindb.config['keypair']['public'] == 'public_key'
|
|
|
|
|
|
@patch('bigchaindb.common.crypto.generate_key_pair',
|
|
return_value=('private_key', 'public_key'))
|
|
@pytest.mark.usefixtures('ignore_local_config_file')
|
|
def test_allow_temp_keypair_doesnt_override_if_keypair_found(mock_gen_keypair,
|
|
mock_processes_start,
|
|
mock_db_init_with_existing_db,
|
|
mocked_setup_logging):
|
|
import bigchaindb
|
|
from bigchaindb.commands.bigchaindb import run_start
|
|
|
|
# Preconditions for the test
|
|
original_private_key = bigchaindb.config['keypair']['private']
|
|
original_public_key = bigchaindb.config['keypair']['public']
|
|
|
|
assert isinstance(original_public_key, str)
|
|
assert isinstance(original_private_key, str)
|
|
|
|
args = Namespace(allow_temp_keypair=True, start_rethinkdb=False, config=None, yes=True,
|
|
skip_initialize_database=False)
|
|
run_start(args)
|
|
|
|
mocked_setup_logging.assert_called_once_with(
|
|
user_log_config=bigchaindb.config['log'])
|
|
assert bigchaindb.config['keypair']['private'] == original_private_key
|
|
assert bigchaindb.config['keypair']['public'] == original_public_key
|
|
|
|
|
|
def test_run_start_when_db_already_exists(mocker,
|
|
monkeypatch,
|
|
run_start_args,
|
|
mocked_setup_logging):
|
|
from bigchaindb import config
|
|
from bigchaindb.commands.bigchaindb import run_start
|
|
from bigchaindb.common.exceptions import DatabaseAlreadyExists
|
|
mocked_start = mocker.patch('bigchaindb.processes.start')
|
|
|
|
def mock_run_init():
|
|
raise DatabaseAlreadyExists()
|
|
|
|
monkeypatch.setattr(
|
|
'bigchaindb.commands.bigchaindb._run_init', mock_run_init)
|
|
run_start(run_start_args)
|
|
mocked_setup_logging.assert_called_once_with(user_log_config=config['log'])
|
|
assert mocked_start.called
|
|
|
|
|
|
def test_run_start_when_keypair_not_found(mocker,
|
|
monkeypatch,
|
|
run_start_args,
|
|
mocked_setup_logging):
|
|
from bigchaindb import config
|
|
from bigchaindb.commands.bigchaindb import run_start
|
|
from bigchaindb.commands.messages import CANNOT_START_KEYPAIR_NOT_FOUND
|
|
from bigchaindb.common.exceptions import KeypairNotFoundException
|
|
mocked_start = mocker.patch('bigchaindb.processes.start')
|
|
|
|
def mock_run_init():
|
|
raise KeypairNotFoundException()
|
|
|
|
monkeypatch.setattr(
|
|
'bigchaindb.commands.bigchaindb._run_init', mock_run_init)
|
|
|
|
with pytest.raises(SystemExit) as exc:
|
|
run_start(run_start_args)
|
|
|
|
mocked_setup_logging.assert_called_once_with(user_log_config=config['log'])
|
|
assert len(exc.value.args) == 1
|
|
assert exc.value.args[0] == CANNOT_START_KEYPAIR_NOT_FOUND
|
|
assert not mocked_start.called
|
|
|
|
|
|
def test_run_start_when_start_rethinkdb_fails(mocker,
|
|
monkeypatch,
|
|
run_start_args,
|
|
mocked_setup_logging):
|
|
from bigchaindb import config
|
|
from bigchaindb.commands.bigchaindb import run_start
|
|
from bigchaindb.commands.messages import RETHINKDB_STARTUP_ERROR
|
|
from bigchaindb.common.exceptions import StartupError
|
|
run_start_args.start_rethinkdb = True
|
|
mocked_start = mocker.patch('bigchaindb.processes.start')
|
|
err_msg = 'Error starting rethinkdb.'
|
|
|
|
def mock_start_rethinkdb():
|
|
raise StartupError(err_msg)
|
|
|
|
monkeypatch.setattr(
|
|
'bigchaindb.commands.utils.start_rethinkdb', mock_start_rethinkdb)
|
|
|
|
with pytest.raises(SystemExit) as exc:
|
|
run_start(run_start_args)
|
|
|
|
mocked_setup_logging.assert_called_once_with(user_log_config=config['log'])
|
|
assert len(exc.value.args) == 1
|
|
assert exc.value.args[0] == RETHINKDB_STARTUP_ERROR.format(err_msg)
|
|
assert not mocked_start.called
|
|
|
|
|
|
@patch('argparse.ArgumentParser.parse_args')
|
|
@patch('bigchaindb.commands.utils.base_parser')
|
|
@patch('bigchaindb.commands.utils.start')
|
|
def test_calling_main(start_mock, base_parser_mock, parse_args_mock,
|
|
monkeypatch):
|
|
from bigchaindb.commands.bigchaindb import main
|
|
|
|
argparser_mock = Mock()
|
|
parser = Mock()
|
|
subparsers = Mock()
|
|
subsubparsers = Mock()
|
|
subparsers.add_parser.return_value = subsubparsers
|
|
parser.add_subparsers.return_value = subparsers
|
|
argparser_mock.return_value = parser
|
|
monkeypatch.setattr('argparse.ArgumentParser', argparser_mock)
|
|
main()
|
|
|
|
assert argparser_mock.called is True
|
|
parser.add_subparsers.assert_called_with(title='Commands',
|
|
dest='command')
|
|
subparsers.add_parser.assert_any_call('configure',
|
|
help='Prepare the config file '
|
|
'and create the node keypair')
|
|
subparsers.add_parser.assert_any_call('show-config',
|
|
help='Show the current '
|
|
'configuration')
|
|
subparsers.add_parserassert_any_call('export-my-pubkey',
|
|
help="Export this node's public "
|
|
'key')
|
|
subparsers.add_parser.assert_any_call('init', help='Init the database')
|
|
subparsers.add_parser.assert_any_call('drop', help='Drop the database')
|
|
|
|
subparsers.add_parser.assert_any_call('start', help='Start BigchainDB')
|
|
subsubparsers.add_argument.assert_any_call('--dev-start-rethinkdb',
|
|
dest='start_rethinkdb',
|
|
action='store_true',
|
|
help='Run RethinkDB on start')
|
|
subsubparsers.add_argument.assert_any_call('--dev-allow-temp-keypair',
|
|
dest='allow_temp_keypair',
|
|
action='store_true',
|
|
help='Generate a random keypair on start')
|
|
|
|
subparsers.add_parser.assert_any_call('set-shards',
|
|
help='Configure number of shards')
|
|
subsubparsers.add_argument.assert_any_call('num_shards',
|
|
metavar='num_shards',
|
|
type=int, default=1,
|
|
help='Number of shards')
|
|
|
|
subparsers.add_parser.assert_any_call('set-replicas',
|
|
help='Configure number of replicas')
|
|
subsubparsers.add_argument.assert_any_call('num_replicas',
|
|
metavar='num_replicas',
|
|
type=int, default=1,
|
|
help='Number of replicas (i.e. '
|
|
'the replication factor)')
|
|
|
|
assert start_mock.called is True
|
|
|
|
|
|
@pytest.mark.usefixtures('ignore_local_config_file')
|
|
@patch('bigchaindb.commands.bigchaindb.add_replicas')
|
|
def test_run_add_replicas(mock_add_replicas):
|
|
from bigchaindb.commands.bigchaindb import run_add_replicas
|
|
from bigchaindb.backend.exceptions import OperationError
|
|
|
|
args = Namespace(config=None, replicas=['localhost:27017'])
|
|
|
|
# test add_replicas no raises
|
|
mock_add_replicas.return_value = None
|
|
assert run_add_replicas(args) is None
|
|
assert mock_add_replicas.call_count == 1
|
|
mock_add_replicas.reset_mock()
|
|
|
|
# test add_replicas with `OperationError`
|
|
mock_add_replicas.side_effect = OperationError('err')
|
|
with pytest.raises(SystemExit) as exc:
|
|
run_add_replicas(args)
|
|
assert exc.value.args == ('err',)
|
|
assert mock_add_replicas.call_count == 1
|
|
mock_add_replicas.reset_mock()
|
|
|
|
# test add_replicas with `NotImplementedError`
|
|
mock_add_replicas.side_effect = NotImplementedError('err')
|
|
with pytest.raises(SystemExit) as exc:
|
|
run_add_replicas(args)
|
|
assert exc.value.args == ('err',)
|
|
assert mock_add_replicas.call_count == 1
|
|
mock_add_replicas.reset_mock()
|
|
|
|
|
|
@pytest.mark.usefixtures('ignore_local_config_file')
|
|
@patch('bigchaindb.commands.bigchaindb.remove_replicas')
|
|
def test_run_remove_replicas(mock_remove_replicas):
|
|
from bigchaindb.commands.bigchaindb import run_remove_replicas
|
|
from bigchaindb.backend.exceptions import OperationError
|
|
|
|
args = Namespace(config=None, replicas=['localhost:27017'])
|
|
|
|
# test add_replicas no raises
|
|
mock_remove_replicas.return_value = None
|
|
assert run_remove_replicas(args) is None
|
|
assert mock_remove_replicas.call_count == 1
|
|
mock_remove_replicas.reset_mock()
|
|
|
|
# test add_replicas with `OperationError`
|
|
mock_remove_replicas.side_effect = OperationError('err')
|
|
with pytest.raises(SystemExit) as exc:
|
|
run_remove_replicas(args)
|
|
assert exc.value.args == ('err',)
|
|
assert mock_remove_replicas.call_count == 1
|
|
mock_remove_replicas.reset_mock()
|
|
|
|
# test add_replicas with `NotImplementedError`
|
|
mock_remove_replicas.side_effect = NotImplementedError('err')
|
|
with pytest.raises(SystemExit) as exc:
|
|
run_remove_replicas(args)
|
|
assert exc.value.args == ('err',)
|
|
assert mock_remove_replicas.call_count == 1
|
|
mock_remove_replicas.reset_mock()
|
|
|
|
|
|
@pytest.mark.tendermint
|
|
@pytest.mark.bdb
|
|
def test_recover_db_from_zombie_txn(b, monkeypatch):
|
|
from bigchaindb.commands.bigchaindb import run_recover
|
|
from bigchaindb.models import Transaction
|
|
from bigchaindb.common.crypto import generate_key_pair
|
|
from bigchaindb.tendermint.lib import Block
|
|
from bigchaindb import backend
|
|
|
|
alice = generate_key_pair()
|
|
tx = Transaction.create([alice.public_key],
|
|
[([alice.public_key], 1)],
|
|
asset={'cycle': 'hero'},
|
|
metadata={'name': 'hohenheim'}) \
|
|
.sign([alice.private_key])
|
|
b.store_bulk_transactions([tx])
|
|
block = Block(app_hash='random_app_hash', height=10,
|
|
transactions=[])._asdict()
|
|
b.store_block(block)
|
|
|
|
def mock_get(uri):
|
|
return MockResponse(10)
|
|
monkeypatch.setattr('requests.get', mock_get)
|
|
|
|
run_recover(b)
|
|
|
|
assert list(backend.query.get_metadata(b.connection, [tx.id])) == []
|
|
assert not backend.query.get_asset(b.connection, tx.id)
|
|
assert not b.get_transaction(tx.id)
|
|
|
|
|
|
@pytest.mark.tendermint
|
|
@pytest.mark.bdb
|
|
def test_recover_db_from_zombie_block(b, monkeypatch):
|
|
from bigchaindb.commands.bigchaindb import run_recover
|
|
from bigchaindb.models import Transaction
|
|
from bigchaindb.common.crypto import generate_key_pair
|
|
from bigchaindb.tendermint.lib import Block
|
|
from bigchaindb import backend
|
|
|
|
alice = generate_key_pair()
|
|
tx = Transaction.create([alice.public_key],
|
|
[([alice.public_key], 1)],
|
|
asset={'cycle': 'hero'},
|
|
metadata={'name': 'hohenheim'}) \
|
|
.sign([alice.private_key])
|
|
b.store_bulk_transactions([tx])
|
|
|
|
block9 = Block(app_hash='random_app_hash', height=9,
|
|
transactions=[])._asdict()
|
|
b.store_block(block9)
|
|
block10 = Block(app_hash='random_app_hash', height=10,
|
|
transactions=[tx.id])._asdict()
|
|
b.store_block(block10)
|
|
|
|
def mock_get(uri):
|
|
return MockResponse(9)
|
|
monkeypatch.setattr('requests.get', mock_get)
|
|
|
|
run_recover(b)
|
|
|
|
assert list(backend.query.get_metadata(b.connection, [tx.id])) == []
|
|
assert not backend.query.get_asset(b.connection, tx.id)
|
|
assert not b.get_transaction(tx.id)
|
|
|
|
block = b.get_latest_block()
|
|
assert block['height'] == 9
|
|
|
|
|
|
@patch('bigchaindb.config_utils.autoconfigure')
|
|
@patch('bigchaindb.commands.bigchaindb.run_recover')
|
|
@patch('bigchaindb.tendermint.commands.start')
|
|
def test_recover_db_on_start(mock_autoconfigure,
|
|
mock_run_recover,
|
|
mock_start,
|
|
mocked_setup_logging):
|
|
from bigchaindb.commands.bigchaindb import run_start
|
|
args = Namespace(start_rethinkdb=False, allow_temp_keypair=False, config=None, yes=True,
|
|
skip_initialize_database=False)
|
|
run_start(args)
|
|
|
|
assert mock_run_recover.called
|
|
assert mock_start.called
|
|
|
|
|
|
# Helper
|
|
class MockResponse():
|
|
|
|
def __init__(self, height):
|
|
self.height = height
|
|
|
|
def json(self):
|
|
return {'result': {'latest_block_height': self.height}}
|