from argparse import Namespace from pprint import pprint import pytest @pytest.fixture def mock_run_configure(monkeypatch): from bigchaindb.commands import bigchain monkeypatch.setattr(bigchain, 'run_configure', lambda *args, **kwargs: None) @pytest.fixture def mock_file_config(monkeypatch): from bigchaindb import config_utils monkeypatch.setattr(config_utils, 'file_config', lambda *args: None) @pytest.fixture def mock_db_init_with_existing_db(monkeypatch): from bigchaindb import db from bigchaindb.exceptions import DatabaseAlreadyExists def mockreturn(): raise DatabaseAlreadyExists monkeypatch.setattr(db, 'init', mockreturn) @pytest.fixture def mock_processes_start(monkeypatch): from bigchaindb.processes import Processes monkeypatch.setattr(Processes, 'start', lambda *args: None) @pytest.fixture def mock_rethink_db_drop(monkeypatch): def mockreturn(dbname): class MockDropped(object): def run(self, conn): return return MockDropped() monkeypatch.setattr('rethinkdb.db_drop', mockreturn) def test_bigchain_run_start(mock_run_configure, mock_file_config, mock_processes_start, mock_db_init_with_existing_db): from bigchaindb.commands.bigchain import run_start args = Namespace(config=None, yes=True) run_start(args) # 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 def test_bigchain_show_config(capsys, mock_file_config): from bigchaindb import config from bigchaindb.commands.bigchain import run_show_config args = Namespace(config=None) _, _ = capsys.readouterr() run_show_config(args) output_config, _ = capsys.readouterr() pprint(config) expected_outout_config, _ = capsys.readouterr() assert output_config == expected_outout_config def test_bigchain_run_init_when_db_exists(mock_file_config, mock_db_init_with_existing_db): from bigchaindb.commands.bigchain import run_init args = Namespace(config=None) run_init(args) def test_drop_existing_db(mock_file_config, mock_rethink_db_drop): from bigchaindb.commands.bigchain import run_drop args = Namespace(config=None, yes=True) run_drop(args)