Fix some errors in api responses

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

diff --git a/test.py b/test.py
index 890381541368f5d683d942584152e0dfbd5324f0..f7df9a861ffc46240515f5868d5a508567122313 100644 (file)
--- a/test.py
+++ b/test.py
@@ -1549,12 +1549,12 @@ class OrderTest(WebMockTestCase):
         order = portfolio.Order("buy", portfolio.Amount("ETH", 10),
                 D("0.1"), "BTC", "long", "market", "trade")
         order.mouvements.append(portfolio.Mouvement("ETH", "BTC", {
-            "id": 42, "type": "buy", "fee": "0.0015",
+            "tradeID": 42, "type": "buy", "fee": "0.0015",
             "date": "2017-12-30 12:00:12", "rate": "0.1",
             "amount": "3", "total": "0.3"
             }))
         order.mouvements.append(portfolio.Mouvement("ETH", "BTC", {
-            "id": 43, "type": "buy", "fee": "0.0015",
+            "tradeID": 43, "type": "buy", "fee": "0.0015",
             "date": "2017-12-30 13:00:12", "rate": "0.2",
             "amount": "2", "total": "0.4"
             }))
@@ -1569,12 +1569,12 @@ class OrderTest(WebMockTestCase):
         market = mock.Mock()
         market.privatePostReturnOrderTrades.return_value = [
                 {
-                    "id": 42, "type": "buy", "fee": "0.0015",
+                    "tradeID": 42, "type": "buy", "fee": "0.0015",
                     "date": "2017-12-30 12:00:12", "rate": "0.1",
                     "amount": "3", "total": "0.3"
                     },
                 {
-                    "id": 43, "type": "buy", "fee": "0.0015",
+                    "tradeID": 43, "type": "buy", "fee": "0.0015",
                     "date": "2017-12-30 13:00:12", "rate": "0.2",
                     "amount": "2", "total": "0.4"
                     }
@@ -1591,6 +1591,12 @@ class OrderTest(WebMockTestCase):
         self.assertEqual(42, order.mouvements[0].id)
         self.assertEqual(43, order.mouvements[1].id)
 
+        market.privatePostReturnOrderTrades.side_effect = portfolio.ExchangeError
+        order = portfolio.Order("buy", portfolio.Amount("ETH", 10),
+                D("0.1"), "BTC", "long", market, "trade")
+        order.fetch_mouvements()
+        self.assertEqual(0, len(order.mouvements))
+
     def test_mark_finished_order(self):
         market = mock.Mock()
         order = portfolio.Order("buy", portfolio.Amount("ETH", 10),
@@ -1770,11 +1776,24 @@ class OrderTest(WebMockTestCase):
             self.assertRegex(stdout_mock.getvalue(), "error when running market.create_order")
             self.assertRegex(stdout_mock.getvalue(), "Exception: bouh")
 
+        market.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", market, "trade")
+            market.create_order.side_effect = portfolio.ExchangeNotAvailable
+            order.run()
+            market.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):
     def test_values(self):
         mouvement = portfolio.Mouvement("ETH", "BTC", {
-            "id": 42, "type": "buy", "fee": "0.0015",
+            "tradeID": 42, "type": "buy", "fee": "0.0015",
             "date": "2017-12-30 12:00:12", "rate": "0.1",
             "amount": "10", "total": "1"
             })
@@ -1788,6 +1807,15 @@ class MouvementTest(WebMockTestCase):
         self.assertEqual(portfolio.Amount("ETH", "10"), mouvement.total)
         self.assertEqual(portfolio.Amount("BTC", "1"), mouvement.total_in_base)
 
+        mouvement = portfolio.Mouvement("ETH", "BTC", { "foo": "bar" })
+        self.assertIsNone(mouvement.date)
+        self.assertIsNone(mouvement.id)
+        self.assertIsNone(mouvement.action)
+        self.assertEqual(-1, mouvement.fee_rate)
+        self.assertEqual(0, mouvement.rate)
+        self.assertEqual(portfolio.Amount("ETH", 0), mouvement.total)
+        self.assertEqual(portfolio.Amount("BTC", 0), mouvement.total_in_base)
+
 @unittest.skipUnless("acceptance" in limits, "Acceptance skipped")
 class AcceptanceTest(WebMockTestCase):
     @unittest.expectedFailure
@@ -1939,7 +1967,7 @@ class AcceptanceTest(WebMockTestCase):
         market.fetch_order.return_value = { "status": "closed", "datetime": "2018-01-20 13:40:00" }
         market.privatePostReturnOrderTrades.return_value = [
                 {
-                    "id": 42, "type": "buy", "fee": "0.0015",
+                    "tradeID": 42, "type": "buy", "fee": "0.0015",
                     "date": "2017-12-30 12:00:12", "rate": "0.1",
                     "amount": "10", "total": "1"
                     }