About 15 margin tests pass

tests/test_persistence.py

@@ -63,6 +63,48 @@ def test_init_dryrun_db(default_conf, tmpdir):
     assert Path(filename).is_file()
 
 
+@pytest.mark.usefixtures("init_persistence")
+def test_is_opening_closing_trade(fee):
+    trade = Trade(
+        id=2,
+        pair='ETH/BTC',
+        stake_amount=0.001,
+        open_rate=0.01,
+        amount=5,
+        is_open=True,
+        open_date=arrow.utcnow().datetime,
+        fee_open=fee.return_value,
+        fee_close=fee.return_value,
+        exchange='binance',
+        is_short=False,
+        leverage=2.0
+    )
+    assert trade.is_opening_trade('buy') == True
+    assert trade.is_opening_trade('sell') == False
+    assert trade.is_closing_trade('buy') == False
+    assert trade.is_closing_trade('sell') == True
+
+    trade = Trade(
+        id=2,
+        pair='ETH/BTC',
+        stake_amount=0.001,
+        open_rate=0.01,
+        amount=5,
+        is_open=True,
+        open_date=arrow.utcnow().datetime,
+        fee_open=fee.return_value,
+        fee_close=fee.return_value,
+        exchange='binance',
+        is_short=True,
+        leverage=2.0
+    )
+
+    assert trade.is_opening_trade('buy') == False
+    assert trade.is_opening_trade('sell') == True
+    assert trade.is_closing_trade('buy') == True
+    assert trade.is_closing_trade('sell') == False
+
+
 @pytest.mark.usefixtures("init_persistence")
 def test_update_with_binance(limit_buy_order, limit_sell_order, fee, caplog):
     """
@@ -196,6 +238,7 @@ def test_calc_open_close_trade_price(limit_buy_order, limit_sell_order, fee):
 
 @pytest.mark.usefixtures("init_persistence")
 def test_trade_close(limit_buy_order, limit_sell_order, fee):
+    # TODO: limit_buy_order and limit_sell_order aren't used, remove them probably
     trade = Trade(
         pair='ETH/BTC',
         stake_amount=0.001,
@@ -1126,6 +1169,42 @@ def test_fee_updated(fee):
     assert not trade.fee_updated('asfd')
 
 
+@pytest.mark.usefixtures("init_persistence")
+def test_update_leverage(fee, ten_minutes_ago):
+    trade = Trade(
+        pair='ETH/BTC',
+        stake_amount=0.001,
+        amount=5,
+        open_rate=0.00001099,
+        open_date=ten_minutes_ago,
+        fee_open=fee.return_value,
+        fee_close=fee.return_value,
+        exchange='binance',
+        is_short=True,
+        interest_rate=0.0005
+    )
+    trade.leverage = 3.0
+    assert trade.borrowed == 15.0
+    assert trade.amount == 15.0
+
+    trade = Trade(
+        pair='ETH/BTC',
+        stake_amount=0.001,
+        amount=5,
+        open_rate=0.00001099,
+        open_date=ten_minutes_ago,
+        fee_open=fee.return_value,
+        fee_close=fee.return_value,
+        exchange='binance',
+        is_short=False,
+        interest_rate=0.0005
+    )
+
+    trade.leverage = 5.0
+    assert trade.borrowed == 20.0
+    assert trade.amount == 25.0
+
+
 @pytest.mark.usefixtures("init_persistence")
 @pytest.mark.parametrize('use_db', [True, False])
 def test_total_open_trades_stakes(fee, use_db):