Allow dispatch assets to receive liquidity

diff --git a/store.py b/store.py
index 841a0fc9e4f581ad468a2c3e8664507b003b4232..abbe5eef556eb88455aec50f12d07b041c0f5c67 100644 (file)
--- a/store.py
+++ b/store.py
@@ -27,9 +27,9 @@ class BalanceStore:
                 cls.all[currency] = portfolio.Balance(currency, balance)
 
     @classmethod
-    def dispatch_assets(cls, amount, repartition=None):
+    def dispatch_assets(cls, amount, liquidity="medium", repartition=None):
         if repartition is None:
-            repartition = portfolio.Portfolio.repartition()
+            repartition = portfolio.Portfolio.repartition(liquidity=liquidity)
         sum_ratio = sum([v[0] for k, v in repartition.items()])
         amounts = {}
         for currency, (ptt, trade_type) in repartition.items():

diff --git a/test.py b/test.py
index 86411301e39c0af52bfebe7b806dc73ebfa788fb..d58b2035af902c63609daf0857a2aaea2b42827f 100644 (file)
--- a/test.py
+++ b/test.py
@@ -1247,6 +1247,7 @@ class BalanceStoreTest(WebMockTestCase):
                 }
 
         amounts = portfolio.BalanceStore.dispatch_assets(portfolio.Amount("BTC", "11.1"))
+        repartition.assert_called_with(liquidity="medium")
         self.assertIn("XEM", portfolio.BalanceStore.currencies())
         self.assertEqual(D("2.6"), amounts["BTC"].value)
         self.assertEqual(D("7.5"), amounts["XEM"].value)