From 044df880e68d513d5e3baf2d451e656631e9194d Mon Sep 17 00:00:00 2001 From: Matthias Date: Thu, 13 Aug 2020 08:28:46 +0200 Subject: [PATCH] Move persistence into it's own submodule --- freqtrade/persistence/__init__.py | 3 +++ freqtrade/{persistence.py => persistence/models.py} | 0 tests/test_persistence.py | 12 ++++++------ 3 files changed, 9 insertions(+), 6 deletions(-) create mode 100644 freqtrade/persistence/__init__.py rename freqtrade/{persistence.py => persistence/models.py} (100%) diff --git a/freqtrade/persistence/__init__.py b/freqtrade/persistence/__init__.py new file mode 100644 index 000000000..0973fab3f --- /dev/null +++ b/freqtrade/persistence/__init__.py @@ -0,0 +1,3 @@ +# flake8: noqa: F401 + +from freqtrade.persistence.models import Trade, clean_dry_run_db, cleanup, init diff --git a/freqtrade/persistence.py b/freqtrade/persistence/models.py similarity index 100% rename from freqtrade/persistence.py rename to freqtrade/persistence/models.py diff --git a/tests/test_persistence.py b/tests/test_persistence.py index 65c83e05b..dbb62e636 100644 --- a/tests/test_persistence.py +++ b/tests/test_persistence.py @@ -22,7 +22,7 @@ def test_init_create_session(default_conf): def test_init_custom_db_url(default_conf, mocker): # Update path to a value other than default, but still in-memory default_conf.update({'db_url': 'sqlite:///tmp/freqtrade2_test.sqlite'}) - create_engine_mock = mocker.patch('freqtrade.persistence.create_engine', MagicMock()) + create_engine_mock = mocker.patch('freqtrade.persistence.models.create_engine', MagicMock()) init(default_conf['db_url'], default_conf['dry_run']) assert create_engine_mock.call_count == 1 @@ -40,7 +40,7 @@ def test_init_prod_db(default_conf, mocker): default_conf.update({'dry_run': False}) default_conf.update({'db_url': constants.DEFAULT_DB_PROD_URL}) - create_engine_mock = mocker.patch('freqtrade.persistence.create_engine', MagicMock()) + create_engine_mock = mocker.patch('freqtrade.persistence.models.create_engine', MagicMock()) init(default_conf['db_url'], default_conf['dry_run']) assert create_engine_mock.call_count == 1 @@ -51,7 +51,7 @@ def test_init_dryrun_db(default_conf, mocker): default_conf.update({'dry_run': True}) default_conf.update({'db_url': constants.DEFAULT_DB_DRYRUN_URL}) - create_engine_mock = mocker.patch('freqtrade.persistence.create_engine', MagicMock()) + create_engine_mock = mocker.patch('freqtrade.persistence.models.create_engine', MagicMock()) init(default_conf['db_url'], default_conf['dry_run']) assert create_engine_mock.call_count == 1 @@ -440,7 +440,7 @@ def test_migrate_old(mocker, default_conf, fee): amount=amount ) engine = create_engine('sqlite://') - mocker.patch('freqtrade.persistence.create_engine', lambda *args, **kwargs: engine) + mocker.patch('freqtrade.persistence.models.create_engine', lambda *args, **kwargs: engine) # Create table using the old format engine.execute(create_table_old) @@ -524,7 +524,7 @@ def test_migrate_new(mocker, default_conf, fee, caplog): amount=amount ) engine = create_engine('sqlite://') - mocker.patch('freqtrade.persistence.create_engine', lambda *args, **kwargs: engine) + mocker.patch('freqtrade.persistence.models.create_engine', lambda *args, **kwargs: engine) # Create table using the old format engine.execute(create_table_old) @@ -601,7 +601,7 @@ def test_migrate_mid_state(mocker, default_conf, fee, caplog): amount=amount ) engine = create_engine('sqlite://') - mocker.patch('freqtrade.persistence.create_engine', lambda *args, **kwargs: engine) + mocker.patch('freqtrade.persistence.models.create_engine', lambda *args, **kwargs: engine) # Create table using the old format engine.execute(create_table_old)