Eat several positions in the order book after some time spent

2 files changed, 57 insertions, 0 deletions

diff --git a/tests/test_ccxt_wrapper.py b/tests/test_ccxt_wrapper.py
index c326f0a..44e660e 100644
--- a/tests/test_ccxt_wrapper.py
+++ b/tests/test_ccxt_wrapper.py
@@ -549,3 +549,35 @@ class poloniexETest(unittest.TestCase):
         self.assertTrue(self.s.is_dust_trade(D("0.0000009"), D("1000")))
         self.assertTrue(self.s.is_dust_trade(D("0.000001"), D("10")))
         self.assertFalse(self.s.is_dust_trade(D("0.000001"), D("100")))
+
+    def test_fetch_nth_order_book(self):
+        with mock.patch.object(self.s, "fetch_order_book") as t:
+            t.return_value = {
+                    "asks": [
+                        [1.269e-05, 781.23105917],
+                        [1.271e-05, 108.83577689],
+                        [1.276e-05, 19162.15732141],
+                        [1.277e-05, 34.13657561],
+                        [1.28e-05, 95.14285714],
+                        [1.281e-05, 11.13909862],
+                        [1.282e-05, 43.42379871],
+                        [1.284e-05, 493.67767887],
+                        [1.288e-05, 6179.57843281],
+                        [1.289e-05, 235.16250589]
+                        ],
+                    "bids": [
+                        [1.266e-05, 3496.42283539],
+                        [1.23e-05, 9.02439024],
+                        [1.229e-05, 3244.25987796],
+                        [1.228e-05, 6692.16061185],
+                        [1.207e-05, 9.19635459],
+                        [1.206e-05, 4711.05943978],
+                        [1.194e-05, 84.67400508],
+                        [1.168e-05, 61.75268779],
+                        [1.165e-05, 9.52789699],
+                        [1.157e-05, 16.4900605]
+                        ]
+                    }
+            self.assertAlmostEqual(D("0.00001289"), self.s.fetch_nth_order_book("BTC/HUC", "ask", 10), 8)
+            t.assert_called_once_with("BTC/HUC", limit=10)
+            self.assertAlmostEqual(D("0.00001157"), self.s.fetch_nth_order_book("BTC/HUC", "bid", 10), 8)
diff --git a/tests/test_portfolio.py b/tests/test_portfolio.py
index bc69921..d4e5ab9 100644
--- a/tests/test_portfolio.py
+++ b/tests/test_portfolio.py
@@ -26,6 +26,12 @@ class ComputationTest(WebMockTestCase):
         portfolio.Computation.compute_value("foo", "bid", compute_value="test")
         compute.assert_called_with("foo", "bid")
 
+    def test_eat_several(self):
+        self.m.ccxt.fetch_nth_order_book.return_value = D("0.00001275")
+
+        self.assertEqual(D("0.00001275"), portfolio.Computation.eat_several(self.m)({ "symbol": "BTC/HUC" }, "ask"))
+
+
 @unittest.skipUnless("unit" in limits, "Unit skipped")
 class TradeTest(WebMockTestCase):
 
@@ -506,6 +512,25 @@ class TradeTest(WebMockTestCase):
             trade.orders = []
             self.m.report.log_order.reset_mock()
 
+        with self.subTest(tick=22):
+            trade.update_order(order_mock, 22)
+            order_mock.cancel.assert_called()
+            new_order_mock.run.assert_called()
+            prepare_order.assert_called_with(compute_value=mock.ANY)
+            self.m.report.log_order.assert_called()
+            self.assertEqual(2, self.m.report.log_order.call_count)
+            calls = [
+                    mock.call(order_mock, 22, update="market_adjust_eat",
+                        compute_value=mock.ANY,
+                        new_order=new_order_mock),
+                    mock.call(order_mock, 22, new_order=new_order_mock),
+                    ]
+            self.m.report.log_order.assert_has_calls(calls)
+
+        order_mock.reset_mock()
+        new_order_mock.reset_mock()
+        trade.orders = []
+        self.m.report.log_order.reset_mock()
 
     def test_print_with_order(self):
         value_from = portfolio.Amount("BTC", "0.5")