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

initialize replica set for tests

Browse Source
This commit is contained in:
Rodolphe Marques 2017-01-09 13:32:17 +01:00
parent 7f05974f0f
commit 8926178e52
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/conftest.py
View File

@ -136,11 +136,18 @@ def _configure_bigchaindb(request):
def _setup_database(_configure_bigchaindb): def _setup_database(_configure_bigchaindb):
from bigchaindb import config from bigchaindb import config
from bigchaindb.backend import connect, schema from bigchaindb.backend import connect, schema
from bigchaindb.backend.mongodb.schema import initialize_replica_set
from bigchaindb.common.exceptions import DatabaseDoesNotExist from bigchaindb.common.exceptions import DatabaseDoesNotExist
print('Initializing test db') print('Initializing test db')
dbname = config['database']['name'] dbname = config['database']['name']
conn = connect() conn = connect()
# if we are setting up mongodb for the first time we need to make sure
# that the replica set is initialized before doing any operation in the
# database
if config['database']['backend'] == 'mongodb':
initialize_replica_set(conn)
try: try:
schema.drop_database(conn, dbname) schema.drop_database(conn, dbname)
except DatabaseDoesNotExist: except DatabaseDoesNotExist: