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

Test backend.schema.init_database

Browse Source
This commit is contained in:
Sylvain Bellemare 2016-12-13 11:47:50 +01:00
parent a902f895fa
commit b87baaaa1d
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tests/backend/test_generics.py
View File

@ -1,3 +1,6 @@
from importlib import import_module
from unittest.mock import patch
from pytest import mark, raises from pytest import mark, raises
@ -43,3 +46,26 @@ def test_query(query_func_name, args_qty):
query_func = getattr(query, query_func_name) query_func = getattr(query, query_func_name)
with raises(NotImplementedError): with raises(NotImplementedError):
query_func(None, *range(args_qty)) query_func(None, *range(args_qty))
@mark.parametrize('db,conn_cls', (
('mongodb', 'MongoDBConnection'),
('rethinkdb', 'RethinkDBConnection'),
))
@patch('bigchaindb.backend.schema.create_indexes',
autospec=True, return_value=None)
@patch('bigchaindb.backend.schema.create_tables',
autospec=True, return_value=None)
@patch('bigchaindb.backend.schema.create_database',
autospec=True, return_value=None)
def test_init_database(mock_create_database, mock_create_tables,
mock_create_indexes, db, conn_cls):
from bigchaindb.backend.schema import init_database
conn = getattr(
import_module('bigchaindb.backend.{}.connection'.format(db)),
conn_cls,
)('host', 'port', 'dbname')
init_database(connection=conn, dbname='mickeymouse')
mock_create_database.assert_called_once_with(conn, 'mickeymouse')
mock_create_tables.assert_called_once_with(conn, 'mickeymouse')
mock_create_indexes.assert_called_once_with(conn, 'mickeymouse')