Replace test occurances of Trade.session with Trade.query.session

This commit is contained in:
Matthias
2021-04-05 07:38:07 +02:00
parent 0407bf755f
commit ea0b47a7f9
6 changed files with 46 additions and 54 deletions

View File

@@ -403,7 +403,7 @@ def test_clean_dry_run_db(default_conf, fee):
exchange='bittrex',
open_order_id='dry_run_buy_12345'
)
Trade.session.add(trade)
Trade.query.session.add(trade)
trade = Trade(
pair='ETC/BTC',
@@ -415,7 +415,7 @@ def test_clean_dry_run_db(default_conf, fee):
exchange='bittrex',
open_order_id='dry_run_sell_12345'
)
Trade.session.add(trade)
Trade.query.session.add(trade)
# Simulate prod entry
trade = Trade(
@@ -428,7 +428,7 @@ def test_clean_dry_run_db(default_conf, fee):
exchange='bittrex',
open_order_id='prod_buy_12345'
)
Trade.session.add(trade)
Trade.query.session.add(trade)
# We have 3 entries: 2 dry_run, 1 prod
assert len(Trade.query.filter(Trade.open_order_id.isnot(None)).all()) == 3
@@ -933,7 +933,7 @@ def test_stoploss_reinitialization(default_conf, fee):
assert trade.stop_loss_pct == -0.05
assert trade.initial_stop_loss == 0.95
assert trade.initial_stop_loss_pct == -0.05
Trade.session.add(trade)
Trade.query.session.add(trade)
# Lower stoploss
Trade.stoploss_reinitialization(0.06)