Mirroristas/bigchaindb
2
0
Fork 0
mirror of https://github.com/bigchaindb/bigchaindb.git synced 2024-10-13 13:34:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix interactive drop database commands

Browse Source
This commit is contained in:
Brett Sun 2016-12-06 14:22:13 +01:00 committed by Sylvain Bellemare
parent dc31f7f607
commit 597cd778e9
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_commands.py
View File

@ -1,4 +1,3 @@
import builtins
import json import json
from unittest.mock import Mock, patch from unittest.mock import Mock, patch
from argparse import Namespace from argparse import Namespace
@ -216,7 +215,7 @@ def test_bigchain_run_init_when_db_exists(mock_db_init_with_existing_db):
run_init(args) run_init(args)
@patch('bigchaindb.schema.drop_database') @patch('bigchaindb.backend.schema.drop_database')
def test_drop_db_when_assumed_yes(mock_db_drop): def test_drop_db_when_assumed_yes(mock_db_drop):
from bigchaindb.commands.bigchain import run_drop from bigchaindb.commands.bigchain import run_drop
args = Namespace(config=None, yes=True) args = Namespace(config=None, yes=True)
@ -225,21 +224,21 @@ def test_drop_db_when_assumed_yes(mock_db_drop):
assert mock_db_drop.called assert mock_db_drop.called
@patch('bigchaindb.schema.drop_database') @patch('bigchaindb.backend.schema.drop_database')
def test_drop_db_when_interactive_yes(mock_db_drop, monkeypatch): def test_drop_db_when_interactive_yes(mock_db_drop, monkeypatch):
from bigchaindb.commands.bigchain import run_drop from bigchaindb.commands.bigchain import run_drop
args = Namespace(config=None, yes=False) args = Namespace(config=None, yes=False)
monkeypatch.setattr(builtins, 'input', lambda x: 'y') monkeypatch.setattr('bigchaindb.commands.bigchain.input', lambda x: 'y')
run_drop(args) run_drop(args)
assert mock_db_drop.called assert mock_db_drop.called
@patch('bigchaindb.schema.drop_database') @patch('bigchaindb.backend.schema.drop_database')
def test_drop_db_does_not_drop_when_interactive_no(mock_db_drop, monkeypatch): def test_drop_db_does_not_drop_when_interactive_no(mock_db_drop, monkeypatch):
from bigchaindb.commands.bigchain import run_drop from bigchaindb.commands.bigchain import run_drop
args = Namespace(config=None, yes=False) args = Namespace(config=None, yes=False)
monkeypatch.setattr(builtins, 'input', lambda x: 'n') monkeypatch.setattr('bigchaindb.commands.bigchain.input', lambda x: 'n')
run_drop(args) run_drop(args)
assert not mock_db_drop.called assert not mock_db_drop.called