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

2021-04-05 07:38:07 +02:00
parent 0407bf755f
commit ea0b47a7f9
6 changed files with 46 additions and 54 deletions
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -197,7 +197,7 @@ def create_mock_trades(fee, use_db: bool = True):
    """
    def add_trade(trade):
        if use_db:
-            Trade.session.add(trade)
+            Trade.query.session.add(trade)
        else:
            LocalTrade.add_bt_trade(trade)