force flush in create_trade and execute_sell (fixes #128)

freqtrade/tests/test_main.py

@@ -115,9 +115,9 @@ def test_create_trade(default_conf, ticker, limit_buy_order, mocker):
     whitelist = copy.deepcopy(default_conf['exchange']['pair_whitelist'])
 
     init(default_conf, create_engine('sqlite://'))
-    trade = create_trade(15.0)
-    Trade.session.add(trade)
-    Trade.session.flush()
+    create_trade(15.0)
+
+    trade = Trade.query.first()
     assert trade is not None
     assert trade.stake_amount == 15.0
     assert trade.is_open
@@ -176,13 +176,14 @@ def test_handle_trade(default_conf, limit_buy_order, limit_sell_order, mocker):
                           buy=MagicMock(return_value='mocked_limit_buy'),
                           sell=MagicMock(return_value='mocked_limit_sell'))
     init(default_conf, create_engine('sqlite://'))
-    trade = create_trade(15.0)
-    trade.update(limit_buy_order)
-    Trade.session.add(trade)
-    Trade.session.flush()
-    trade = Trade.query.filter(Trade.is_open.is_(True)).first()
+    create_trade(15.0)
+
+    trade = Trade.query.first()
     assert trade
 
+    trade.update(limit_buy_order)
+    assert trade.is_open is True
+
     handle_trade(trade)
     assert trade.open_order_id == 'mocked_limit_sell'
 
@@ -205,15 +206,14 @@ def test_close_trade(default_conf, ticker, limit_buy_order, limit_sell_order, mo
 
     # Create trade and sell it
     init(default_conf, create_engine('sqlite://'))
-    trade = create_trade(15.0)
-    Trade.session.add(trade)
-    trade.update(limit_buy_order)
-    trade = Trade.query.filter(Trade.is_open.is_(True)).first()
+    create_trade(15.0)
+
+    trade = Trade.query.first()
     assert trade
 
+    trade.update(limit_buy_order)
     trade.update(limit_sell_order)
-    trade = Trade.query.filter(Trade.is_open.is_(False)).first()
-    assert trade
+    assert trade.is_open is False
 
     with pytest.raises(ValueError, match=r'.*closed trade.*'):
         handle_trade(trade)

freqtrade/tests/test_rpc_telegram.py

@@ -101,11 +101,7 @@ def test_status_handle(default_conf, update, ticker, mocker):
     msg_mock.reset_mock()
 
     # Create some test data
-    trade = create_trade(15.0)
-    assert trade
-    Trade.session.add(trade)
-    Trade.session.flush()
-
+    create_trade(15.0)
     # Trigger status while we have a fulfilled order for the open trade
     _status(bot=MagicMock(), update=update)
 
@@ -141,10 +137,7 @@ def test_status_table_handle(default_conf, update, ticker, mocker):
     msg_mock.reset_mock()
 
     # Create some test data
-    trade = create_trade(15.0)
-    assert trade
-    Trade.session.add(trade)
-    Trade.session.flush()
+    create_trade(15.0)
 
     _status_table(bot=MagicMock(), update=update)
 
@@ -177,8 +170,8 @@ def test_profit_handle(default_conf, update, ticker, limit_buy_order, limit_sell
     msg_mock.reset_mock()
 
     # Create some test data
-    trade = create_trade(15.0)
-    assert trade
+    create_trade(15.0)
+    trade = Trade.query.first()
 
     # Simulate fulfilled LIMIT_BUY order for trade
     trade.update(limit_buy_order)
@@ -193,8 +186,6 @@ def test_profit_handle(default_conf, update, ticker, limit_buy_order, limit_sell
 
     trade.close_date = datetime.utcnow()
     trade.is_open = False
-    Trade.session.add(trade)
-    Trade.session.flush()
 
     _profit(bot=MagicMock(), update=update)
     assert msg_mock.call_count == 1
@@ -216,11 +207,10 @@ def test_forcesell_handle(default_conf, update, ticker, mocker):
     init(default_conf, create_engine('sqlite://'))
 
     # Create some test data
-    trade = create_trade(15.0)
-    assert trade
+    create_trade(15.0)
 
-    Trade.session.add(trade)
-    Trade.session.flush()
+    trade = Trade.query.first()
+    assert trade
 
     update.message.text = '/forcesell 1'
     _forcesell(bot=MagicMock(), update=update)
@@ -245,8 +235,7 @@ def test_forcesell_all_handle(default_conf, update, ticker, mocker):
 
     # Create some test data
     for _ in range(4):
-        Trade.session.add(create_trade(15.0))
-    Trade.session.flush()
+        create_trade(15.0)
     rpc_mock.reset_mock()
 
     update.message.text = '/forcesell all'
@@ -309,7 +298,8 @@ def test_performance_handle(
     init(default_conf, create_engine('sqlite://'))
 
     # Create some test data
-    trade = create_trade(15.0)
+    create_trade(15.0)
+    trade = Trade.query.first()
     assert trade
 
     # Simulate fulfilled LIMIT_BUY order for trade
@@ -320,8 +310,6 @@ def test_performance_handle(
 
     trade.close_date = datetime.utcnow()
     trade.is_open = False
-    Trade.session.add(trade)
-    Trade.session.flush()
 
     _performance(bot=MagicMock(), update=update)
     assert msg_mock.call_count == 1
@@ -351,9 +339,7 @@ def test_count_handle(default_conf, update, ticker, mocker):
     update_state(State.RUNNING)
 
     # Create some test data
-    Trade.session.add(create_trade(15.0))
-    Trade.session.flush()
-
+    create_trade(15.0)
     msg_mock.reset_mock()
     _count(bot=MagicMock(), update=update)