Fix dust amount error

[perso/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Trader.git] / test.py

diff --git a/test.py b/test.py
index 4ed04774645239f42026430a8e0ce7a9935a5dd9..52d737d7528f773261242e0830146415ddb1f487 100644 (file)
--- a/test.py
+++ b/test.py
@@ -571,7 +571,7 @@ class MarketTest(WebMockTestCase):
             ccxt.poloniexE.return_value = self.ccxt
             self.ccxt.session.request.return_value = "response"
 
-            m = market.Market.from_config("config")
+            m = market.Market.from_config({"key": "key", "secred": "secret"})
 
             self.assertEqual(self.ccxt, m.ccxt)
 
@@ -580,7 +580,7 @@ class MarketTest(WebMockTestCase):
             m.report.log_http_request.assert_called_with('GET', 'URL', 'data',
                     'headers', 'response')
 
-        m = market.Market.from_config("config", debug=True)
+        m = market.Market.from_config({"key": "key", "secred": "secret"}, debug=True)
         self.assertEqual(True, m.debug)
 
     def test_get_ticker(self):
@@ -2080,6 +2080,18 @@ class OrderTest(WebMockTestCase):
             self.assertEqual("closed", order.status)
             mark_finished_order.assert_called_once()
 
+        self.m.ccxt.create_order.reset_mock()
+        with self.subTest(dust_amount_exception=True),\
+                mock.patch.object(portfolio.Order, "mark_finished_order") as mark_finished_order:
+            order = portfolio.Order("buy", portfolio.Amount("ETH", 0.001),
+                    D("0.1"), "BTC", "long", self.m, "trade")
+            self.m.ccxt.create_order.side_effect = portfolio.InvalidOrder
+            order.run()
+            self.m.ccxt.create_order.assert_called_once()
+            self.assertEqual(0, len(order.results))
+            self.assertEqual("closed", order.status)
+            mark_finished_order.assert_called_once()
+
 
 @unittest.skipUnless("unit" in limits, "Unit skipped")
 class MouvementTest(WebMockTestCase):