mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-06-17 18:13:22 +02:00
Test backend.schema.init_database
This commit is contained in:
parent
a902f895fa
commit
b87baaaa1d
|
@ -1,3 +1,6 @@
|
|||
from importlib import import_module
|
||||
from unittest.mock import patch
|
||||
|
||||
from pytest import mark, raises
|
||||
|
||||
|
||||
|
@ -43,3 +46,26 @@ def test_query(query_func_name, args_qty):
|
|||
query_func = getattr(query, query_func_name)
|
||||
with raises(NotImplementedError):
|
||||
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')
|
||||
|
|
Loading…
Reference in New Issue
Block a user