diff --git a/tests/test_core.py b/tests/test_core.py index 94fe91e9..c53bc7e2 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -32,9 +32,7 @@ def test_bigchain_class_default_initialization(config): from bigchaindb.core import Bigchain from bigchaindb.consensus import BaseConsensusRules bigchain = Bigchain() - assert bigchain.host == config['database']['host'] - assert bigchain.port == config['database']['port'] - assert bigchain.dbname == config['database']['name'] + assert bigchain.connection assert bigchain.me == config['keypair']['public'] assert bigchain.me_private == config['keypair']['private'] assert bigchain.nodes_except_me == config['keyring'] @@ -43,16 +41,16 @@ def test_bigchain_class_default_initialization(config): def test_bigchain_class_initialization_with_parameters(config): from bigchaindb.core import Bigchain + from bigchaindb.backend.connection import Connection from bigchaindb.consensus import BaseConsensusRules init_kwargs = { 'public_key': 'white', 'private_key': 'black', 'keyring': ['key_one', 'key_two'], } - bigchain = Bigchain(**init_kwargs) - assert bigchain.host == init_kwargs['host'] - assert bigchain.port == init_kwargs['port'] - assert bigchain.dbname == init_kwargs['dbname'] + connection = Connection() + bigchain = Bigchain(**init_kwargs, connection=connection) + assert bigchain.connection == connection assert bigchain.me == init_kwargs['public_key'] assert bigchain.me_private == init_kwargs['private_key'] assert bigchain.nodes_except_me == init_kwargs['keyring']