diff --git a/tests/conftest.py b/tests/conftest.py index f4a097f..01bd5c1 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -145,12 +145,12 @@ def _setup_database(_configure_planetmint): # TODO Here is located setup databa @pytest.fixture def _bdb(_setup_database, _configure_planetmint): - from planetmint.backend import Connection + from planetmint.backend.connection import connect from planetmint.transactions.common.memoize import to_dict, from_dict from planetmint.models import Transaction from .utils import flush_db from planetmint.config import Config - conn = Connection() + conn = connect() yield dbname = Config().get()['database']['name'] flush_db(conn, dbname)