From: Ismaël Bouya <ismael.bouya@normalesup.org>
Date: Mon, 19 Feb 2018 13:06:45 +0000 (+0100)
Subject: Allow dispatch assets to receive liquidity
X-Git-Tag: v0.1~5
X-Git-Url: https://git.immae.eu/?p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git;a=commitdiff_plain;h=7eb9cb36c2ef01097fe8e99d40dd17308abdfe37

Allow dispatch assets to receive liquidity
---

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)