X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=tests%2Ftest_portfolio.py;h=bc69921fe01f58b649492ccbafdcb500eebcb0ce;hb=27bad39d3f2efcec0247b473627088149f462f79;hp=14dc9954f63679c03a12a087c2f69102c3109705;hpb=1178378ab68a426762b9e45d28a5ce97a7727c7d;p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git diff --git a/tests/test_portfolio.py b/tests/test_portfolio.py index 14dc995..bc69921 100644 --- a/tests/test_portfolio.py +++ b/tests/test_portfolio.py @@ -2,6 +2,7 @@ from .helper import * import portfolio import datetime +@unittest.skipUnless("unit" in limits, "Unit skipped") class ComputationTest(WebMockTestCase): def test_compute_value(self): compute = mock.Mock() @@ -25,6 +26,7 @@ class ComputationTest(WebMockTestCase): portfolio.Computation.compute_value("foo", "bid", compute_value="test") compute.assert_called_with("foo", "bid") +@unittest.skipUnless("unit" in limits, "Unit skipped") class TradeTest(WebMockTestCase): def test_values_assertion(self): @@ -140,9 +142,9 @@ class TradeTest(WebMockTestCase): self.assertTrue(trade.is_fullfiled) - order1.filled_amount.assert_called_with(in_base_currency=True) - order2.filled_amount.assert_called_with(in_base_currency=True) - order3.filled_amount.assert_called_with(in_base_currency=True) + order1.filled_amount.assert_called_with(in_base_currency=True, refetch=True) + order2.filled_amount.assert_called_with(in_base_currency=True, refetch=True) + order3.filled_amount.assert_called_with(in_base_currency=True, refetch=True) with self.subTest(inverted=True): value_from = portfolio.Amount("BTC", "0.5") @@ -167,9 +169,9 @@ class TradeTest(WebMockTestCase): self.assertTrue(trade.is_fullfiled) - order1.filled_amount.assert_called_with(in_base_currency=False) - order2.filled_amount.assert_called_with(in_base_currency=False) - order3.filled_amount.assert_called_with(in_base_currency=False) + order1.filled_amount.assert_called_with(in_base_currency=False, refetch=True) + order2.filled_amount.assert_called_with(in_base_currency=False, refetch=True) + order3.filled_amount.assert_called_with(in_base_currency=False, refetch=True) def test_filled_amount(self): @@ -187,16 +189,16 @@ class TradeTest(WebMockTestCase): trade.orders.append(order2) self.assertEqual(portfolio.Amount("ETH", "0.31"), trade.filled_amount()) - order1.filled_amount.assert_called_with(in_base_currency=False) - order2.filled_amount.assert_called_with(in_base_currency=False) + order1.filled_amount.assert_called_with(in_base_currency=False, refetch=False) + order2.filled_amount.assert_called_with(in_base_currency=False, refetch=False) self.assertEqual(portfolio.Amount("ETH", "0.31"), trade.filled_amount(in_base_currency=False)) - order1.filled_amount.assert_called_with(in_base_currency=False) - order2.filled_amount.assert_called_with(in_base_currency=False) + order1.filled_amount.assert_called_with(in_base_currency=False, refetch=False) + order2.filled_amount.assert_called_with(in_base_currency=False, refetch=False) self.assertEqual(portfolio.Amount("ETH", "0.31"), trade.filled_amount(in_base_currency=True)) - order1.filled_amount.assert_called_with(in_base_currency=True) - order2.filled_amount.assert_called_with(in_base_currency=True) + order1.filled_amount.assert_called_with(in_base_currency=True, refetch=False) + order2.filled_amount.assert_called_with(in_base_currency=True, refetch=False) @mock.patch.object(portfolio.Computation, "compute_value") @mock.patch.object(portfolio.Trade, "filled_amount") @@ -609,6 +611,7 @@ class TradeTest(WebMockTestCase): self.assertEqual("ETH", as_json["currency"]) self.assertEqual("BTC", as_json["base_currency"]) +@unittest.skipUnless("unit" in limits, "Unit skipped") class BalanceTest(WebMockTestCase): def test_values(self): balance = portfolio.Balance("BTC", { @@ -684,6 +687,7 @@ class BalanceTest(WebMockTestCase): self.assertEqual(D(0), as_json["margin_available"]) self.assertEqual(D(0), as_json["margin_borrowed"]) +@unittest.skipUnless("unit" in limits, "Unit skipped") class OrderTest(WebMockTestCase): def test_values(self): order = portfolio.Order("buy", portfolio.Amount("ETH", 10), @@ -819,14 +823,20 @@ class OrderTest(WebMockTestCase): order.cancel() self.m.ccxt.cancel_order.assert_not_called() - def test_dust_amount_remaining(self): + def test_mark_dust_amount_remaining(self): order = portfolio.Order("buy", portfolio.Amount("ETH", 10), D("0.1"), "BTC", "long", self.m, "trade") - order.remaining_amount = mock.Mock(return_value=portfolio.Amount("ETH", 1)) - self.assertFalse(order.dust_amount_remaining()) + self.m.ccxt.is_dust_trade.return_value = False + order.mark_dust_amount_remaining_order() + self.assertEqual("pending", order.status) - order.remaining_amount = mock.Mock(return_value=portfolio.Amount("ETH", D("0.0001"))) - self.assertTrue(order.dust_amount_remaining()) + self.m.ccxt.is_dust_trade.return_value = True + order.mark_dust_amount_remaining_order() + self.assertEqual("pending", order.status) + + order.status = "open" + order.mark_dust_amount_remaining_order() + self.assertEqual("closed_dust_remaining", order.status) @mock.patch.object(portfolio.Order, "fetch") @mock.patch.object(portfolio.Order, "filled_amount", return_value=portfolio.Amount("ETH", 1)) @@ -857,6 +867,8 @@ class OrderTest(WebMockTestCase): fetch.assert_not_called() order.status = "open" self.assertEqual(portfolio.Amount("ETH", 5), order.filled_amount(in_base_currency=False)) + fetch.assert_not_called() + self.assertEqual(portfolio.Amount("ETH", 5), order.filled_amount(in_base_currency=False, refetch=True)) fetch.assert_called_once() self.assertEqual(portfolio.Amount("BTC", "0.7"), order.filled_amount(in_base_currency=True)) @@ -961,6 +973,7 @@ class OrderTest(WebMockTestCase): "status": "foo", "datetime": "timestamp" } + self.m.ccxt.is_dust_trade.return_value = False order.fetch() self.m.ccxt.fetch_order.assert_called_once_with(45) @@ -1745,6 +1758,7 @@ class OrderTest(WebMockTestCase): result = order.retrieve_order() self.assertFalse(result) +@unittest.skipUnless("unit" in limits, "Unit skipped") class MouvementTest(WebMockTestCase): def test_values(self): mouvement = portfolio.Mouvement("ETH", "BTC", { @@ -1802,6 +1816,7 @@ class MouvementTest(WebMockTestCase): self.assertEqual("BTC", as_json["base_currency"]) self.assertEqual("ETH", as_json["currency"]) +@unittest.skipUnless("unit" in limits, "Unit skipped") class AmountTest(WebMockTestCase): def test_values(self): amount = portfolio.Amount("BTC", "0.65")