diff options
Diffstat (limited to 'test.py')
-rw-r--r-- | test.py | 13 |
1 files changed, 7 insertions, 6 deletions
@@ -2118,7 +2118,8 @@ class OrderTest(WebMockTestCase): | |||
2118 | self.m.report.log_debug_action.assert_called_once() | 2118 | self.m.report.log_debug_action.assert_called_once() |
2119 | 2119 | ||
2120 | @mock.patch.object(portfolio.Order, "fetch_mouvements") | 2120 | @mock.patch.object(portfolio.Order, "fetch_mouvements") |
2121 | def test_fetch(self, fetch_mouvements): | 2121 | @mock.patch.object(portfolio.Order, "mark_finished_order") |
2122 | def test_fetch(self, mark_finished_order, fetch_mouvements): | ||
2122 | order = portfolio.Order("buy", portfolio.Amount("ETH", 10), | 2123 | order = portfolio.Order("buy", portfolio.Amount("ETH", 10), |
2123 | D("0.1"), "BTC", "long", self.m, "trade") | 2124 | D("0.1"), "BTC", "long", self.m, "trade") |
2124 | order.id = 45 | 2125 | order.id = 45 |
@@ -2128,6 +2129,7 @@ class OrderTest(WebMockTestCase): | |||
2128 | self.m.report.log_debug_action.assert_called_once() | 2129 | self.m.report.log_debug_action.assert_called_once() |
2129 | self.m.report.log_debug_action.reset_mock() | 2130 | self.m.report.log_debug_action.reset_mock() |
2130 | self.m.ccxt.fetch_order.assert_not_called() | 2131 | self.m.ccxt.fetch_order.assert_not_called() |
2132 | mark_finished_order.assert_not_called() | ||
2131 | fetch_mouvements.assert_not_called() | 2133 | fetch_mouvements.assert_not_called() |
2132 | 2134 | ||
2133 | with self.subTest(debug=False): | 2135 | with self.subTest(debug=False): |
@@ -2144,17 +2146,19 @@ class OrderTest(WebMockTestCase): | |||
2144 | self.assertEqual("timestamp", order.timestamp) | 2146 | self.assertEqual("timestamp", order.timestamp) |
2145 | self.assertEqual(1, len(order.results)) | 2147 | self.assertEqual(1, len(order.results)) |
2146 | self.m.report.log_debug_action.assert_not_called() | 2148 | self.m.report.log_debug_action.assert_not_called() |
2149 | mark_finished_order.assert_called_once() | ||
2147 | 2150 | ||
2151 | mark_finished_order.reset_mock() | ||
2148 | with self.subTest(missing_order=True): | 2152 | with self.subTest(missing_order=True): |
2149 | self.m.ccxt.fetch_order.side_effect = [ | 2153 | self.m.ccxt.fetch_order.side_effect = [ |
2150 | portfolio.OrderNotCached, | 2154 | portfolio.OrderNotCached, |
2151 | ] | 2155 | ] |
2152 | order.fetch() | 2156 | order.fetch() |
2153 | self.assertEqual("closed_unknown", order.status) | 2157 | self.assertEqual("closed_unknown", order.status) |
2158 | mark_finished_order.assert_called_once() | ||
2154 | 2159 | ||
2155 | @mock.patch.object(portfolio.Order, "fetch") | 2160 | @mock.patch.object(portfolio.Order, "fetch") |
2156 | @mock.patch.object(portfolio.Order, "mark_finished_order") | 2161 | def test_get_status(self, fetch): |
2157 | def test_get_status(self, mark_finished_order, fetch): | ||
2158 | with self.subTest(debug=True): | 2162 | with self.subTest(debug=True): |
2159 | self.m.debug = True | 2163 | self.m.debug = True |
2160 | order = portfolio.Order("buy", portfolio.Amount("ETH", 10), | 2164 | order = portfolio.Order("buy", portfolio.Amount("ETH", 10), |
@@ -2173,10 +2177,8 @@ class OrderTest(WebMockTestCase): | |||
2173 | return update_status | 2177 | return update_status |
2174 | fetch.side_effect = _fetch(order) | 2178 | fetch.side_effect = _fetch(order) |
2175 | self.assertEqual("open", order.get_status()) | 2179 | self.assertEqual("open", order.get_status()) |
2176 | mark_finished_order.assert_not_called() | ||
2177 | fetch.assert_called_once() | 2180 | fetch.assert_called_once() |
2178 | 2181 | ||
2179 | mark_finished_order.reset_mock() | ||
2180 | fetch.reset_mock() | 2182 | fetch.reset_mock() |
2181 | with self.subTest(debug=False, finished=True): | 2183 | with self.subTest(debug=False, finished=True): |
2182 | self.m.debug = False | 2184 | self.m.debug = False |
@@ -2188,7 +2190,6 @@ class OrderTest(WebMockTestCase): | |||
2188 | return update_status | 2190 | return update_status |
2189 | fetch.side_effect = _fetch(order) | 2191 | fetch.side_effect = _fetch(order) |
2190 | self.assertEqual("closed", order.get_status()) | 2192 | self.assertEqual("closed", order.get_status()) |
2191 | mark_finished_order.assert_called_once() | ||
2192 | fetch.assert_called_once() | 2193 | fetch.assert_called_once() |
2193 | 2194 | ||
2194 | def test_run(self): | 2195 | def test_run(self): |