mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
e7ffcf5705
* Added replicaset name to bigchaindb config * changed travis replSet to match bigchaindb default * Updated initialize_replica_set It now initializes the replica set with the name provided by the bigchaindb config file. * initialize_replica_set is now called when creating a mongodb connection. This way we are sure that a replica set exists when we return a connection. * Moved the initialize replica set logic to the connection. * update the config documentation
152 lines
5.2 KiB
Python
152 lines
5.2 KiB
Python
from unittest import mock
|
|
|
|
import pytest
|
|
from pymongo import MongoClient
|
|
from pymongo.database import Database
|
|
from pymongo.errors import ConnectionFailure, OperationFailure
|
|
|
|
|
|
pytestmark = pytest.mark.bdb
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_cmd_line_opts():
|
|
return {'argv': ['mongod', '--dbpath=/data', '--replSet=bigchain-rs'],
|
|
'ok': 1.0,
|
|
'parsed': {'replication': {'replSet': 'bigchain-rs'},
|
|
'storage': {'dbPath': '/data'}}}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_opts():
|
|
return {'argv': ['mongod', '--dbpath=/data', '--replSet=bigchain-rs'],
|
|
'ok': 1.0,
|
|
'parsed': {'replication': {'replSetName': 'bigchain-rs'},
|
|
'storage': {'dbPath': '/data'}}}
|
|
|
|
|
|
@pytest.fixture
|
|
def mongodb_connection():
|
|
import bigchaindb
|
|
return MongoClient(host=bigchaindb.config['database']['host'],
|
|
port=bigchaindb.config['database']['port'])
|
|
|
|
|
|
def test_get_connection_returns_the_correct_instance():
|
|
from bigchaindb.backend import connect
|
|
from bigchaindb.backend.connection import Connection
|
|
from bigchaindb.backend.mongodb.connection import MongoDBConnection
|
|
|
|
config = {
|
|
'backend': 'mongodb',
|
|
'host': 'localhost',
|
|
'port': 27017,
|
|
'name': 'test',
|
|
'replicaset': 'bigchain-rs'
|
|
}
|
|
|
|
conn = connect(**config)
|
|
assert isinstance(conn, Connection)
|
|
assert isinstance(conn, MongoDBConnection)
|
|
assert conn.conn._topology_settings.replica_set_name == config['replicaset']
|
|
|
|
|
|
@mock.patch('bigchaindb.backend.mongodb.connection.initialize_replica_set')
|
|
@mock.patch('pymongo.MongoClient.__init__')
|
|
@mock.patch('time.sleep')
|
|
def test_connection_error(mock_sleep, mock_client, mock_init_repl_set):
|
|
from bigchaindb.backend import connect
|
|
|
|
# force the driver to trow ConnectionFailure
|
|
# the mock on time.sleep is to prevent the actual sleep when running
|
|
# the tests
|
|
mock_client.side_effect = ConnectionFailure()
|
|
|
|
with pytest.raises(ConnectionFailure):
|
|
conn = connect()
|
|
conn.db
|
|
|
|
assert mock_client.call_count == 3
|
|
|
|
|
|
def test_check_replica_set_not_enabled(mongodb_connection):
|
|
from bigchaindb.backend.mongodb.connection import _check_replica_set
|
|
from bigchaindb.common.exceptions import ConfigurationError
|
|
|
|
# no replSet option set
|
|
cmd_line_opts = {'argv': ['mongod', '--dbpath=/data'],
|
|
'ok': 1.0,
|
|
'parsed': {'storage': {'dbPath': '/data'}}}
|
|
with mock.patch.object(Database, 'command', return_value=cmd_line_opts):
|
|
with pytest.raises(ConfigurationError):
|
|
_check_replica_set(mongodb_connection)
|
|
|
|
|
|
def test_check_replica_set_command_line(mongodb_connection,
|
|
mock_cmd_line_opts):
|
|
from bigchaindb.backend.mongodb.connection import _check_replica_set
|
|
|
|
# replSet option set through the command line
|
|
with mock.patch.object(Database, 'command',
|
|
return_value=mock_cmd_line_opts):
|
|
assert _check_replica_set(mongodb_connection) is None
|
|
|
|
|
|
def test_check_replica_set_config_file(mongodb_connection, mock_config_opts):
|
|
from bigchaindb.backend.mongodb.connection import _check_replica_set
|
|
|
|
# replSet option set through the config file
|
|
with mock.patch.object(Database, 'command', return_value=mock_config_opts):
|
|
assert _check_replica_set(mongodb_connection) is None
|
|
|
|
|
|
def test_check_replica_set_name_mismatch(mongodb_connection,
|
|
mock_cmd_line_opts):
|
|
from bigchaindb.backend.mongodb.connection import _check_replica_set
|
|
from bigchaindb.common.exceptions import ConfigurationError
|
|
|
|
# change the replica set name so it does not match the bigchaindb config
|
|
mock_cmd_line_opts['parsed']['replication']['replSet'] = 'rs0'
|
|
|
|
with mock.patch.object(Database, 'command',
|
|
return_value=mock_cmd_line_opts):
|
|
with pytest.raises(ConfigurationError):
|
|
_check_replica_set(mongodb_connection)
|
|
|
|
|
|
def test_wait_for_replica_set_initialization(mongodb_connection):
|
|
from bigchaindb.backend.mongodb.connection import _wait_for_replica_set_initialization # noqa
|
|
|
|
with mock.patch.object(Database, 'command') as mock_command:
|
|
mock_command.side_effect = [
|
|
{'log': ['a line']},
|
|
{'log': ['database writes are now permitted']},
|
|
]
|
|
|
|
# check that it returns
|
|
assert _wait_for_replica_set_initialization(mongodb_connection) is None
|
|
|
|
|
|
def test_initialize_replica_set(mock_cmd_line_opts):
|
|
from bigchaindb.backend.mongodb.connection import initialize_replica_set
|
|
|
|
with mock.patch.object(Database, 'command') as mock_command:
|
|
mock_command.side_effect = [
|
|
mock_cmd_line_opts,
|
|
None,
|
|
{'log': ['database writes are now permitted']},
|
|
]
|
|
|
|
# check that it returns
|
|
assert initialize_replica_set() is None
|
|
|
|
# test it raises OperationError if anything wrong
|
|
with mock.patch.object(Database, 'command') as mock_command:
|
|
mock_command.side_effect = [
|
|
mock_cmd_line_opts,
|
|
OperationFailure(None, details={'codeName': ''})
|
|
]
|
|
|
|
with pytest.raises(OperationFailure):
|
|
initialize_replica_set()
|