balance bid price between ask and last

test/test_main.py

@@ -4,7 +4,7 @@ from unittest.mock import patch, MagicMock
 from jsonschema import validate
 
 import exchange
-from main import create_trade, handle_trade, close_trade_if_fulfilled, init
+from main import create_trade, handle_trade, close_trade_if_fulfilled, init, get_target_bid
 from misc import CONF_SCHEMA
 from persistence import Trade
 
@@ -20,6 +20,9 @@ class TestMain(unittest.TestCase):
             "720": 0.01,
             "0": 0.02
         },
+        "bid_strategy": {
+            "ask_last_balance": 0.0
+        },
         "bittrex": {
             "enabled": True,
             "key": "key",
@@ -90,6 +93,18 @@ class TestMain(unittest.TestCase):
             self.assertTrue(closed)
             self.assertEqual(trade.is_open, False)
 
+    def test_balance_fully_ask_side(self):
+        with patch.dict('main._CONF', {'bid_strategy': {'ask_last_balance': 0.0}}):
+            self.assertEqual(get_target_bid({'ask': 20, 'last': 10}), 20)
+
+    def test_balance_fully_last_side(self):
+        with patch.dict('main._CONF', {'bid_strategy': {'ask_last_balance': 1.0}}):
+            self.assertEqual(get_target_bid({'ask': 20, 'last': 10}), 10)
+
+    def test_balance_when_last_bigger_than_ask(self):
+        with patch.dict('main._CONF', {'bid_strategy': {'ask_last_balance': 1.0}}):
+            self.assertEqual(get_target_bid({'ask': 5, 'last': 10}), 5)
+
     @classmethod
     def setUpClass(cls):
         validate(cls.conf, CONF_SCHEMA)

test/test_telegram.py

@@ -28,6 +28,9 @@ class TestTelegram(unittest.TestCase):
             "720": 0.01,
             "0": 0.02
         },
+        "bid_strategy": {
+            "ask_last_balance": 0.0
+        },
         "bittrex": {
             "enabled": True,
             "key": "key",