Allow dispatch assets to receive liquidity

2 files changed, 3 insertions, 2 deletions

diff --git a/store.py b/store.py
index 841a0fc..abbe5ee 100644
--- 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 8641130..d58b203 100644
--- 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)