Write tests for Processor classv0.4

Make default repartition less sensitive to BTC
Automatic signature for methods

Fixes https://git.immae.eu/mantisbt/view.php?id=49 https://git.immae.eu/mantisbt/view.php?id=39

1 files changed, 158 insertions, 72 deletions

diff --git a/test.py b/test.py
index 78de76e..a894503 100644
--- a/test.py
+++ b/test.py
@@ -3014,87 +3014,173 @@ class HelperTest(WebMockTestCase):
             mock.call(portfolio.Amount("BTC", "0.95")),
             ])
 
-    def test_process_sell_needed__1_sell(self):
-        helper.process_sell_needed__1_sell(self.m)
+@unittest.skipUnless("unit" in limits, "Unit skipped")
+class ProcessorTest(WebMockTestCase):
+    def test_values(self):
+        processor = helper.Processor(self.m)
 
-        self.m.balances.fetch_balances.assert_has_calls([
-            mock.call(tag="process_sell_needed__1_sell_begin"),
-            mock.call(tag="process_sell_needed__1_sell_end"),
-            ])
-        self.m.prepare_trades.assert_called_with(base_currency="BTC",
-                liquidity="medium")
-        self.m.trades.prepare_orders.assert_called_with(compute_value="average",
-                only="dispose")
-        self.m.trades.run_orders.assert_called()
-        self.m.follow_orders.assert_called()
-        self.m.report.log_stage.assert_has_calls([
-            mock.call("process_sell_needed__1_sell_begin"),
-            mock.call("process_sell_needed__1_sell_end")
-            ])
+        self.assertEqual(self.m, processor.market)
 
-    def test_process_sell_needed__2_buy(self):
-        helper.process_sell_needed__2_buy(self.m)
+    def test_run_action(self):
+        processor = helper.Processor(self.m)
 
-        self.m.balances.fetch_balances.assert_has_calls([
-            mock.call(tag="process_sell_needed__2_buy_begin"),
-            mock.call(tag="process_sell_needed__2_buy_end"),
-            ])
-        self.m.prepare_trades.assert_called_with(base_currency="BTC",
-                liquidity="medium", only="acquire")
-        self.m.trades.prepare_orders.assert_called_with(compute_value="average",
-                only="acquire")
-        self.m.move_balances.assert_called_with()
-        self.m.trades.run_orders.assert_called()
-        self.m.follow_orders.assert_called()
-        self.m.report.log_stage.assert_has_calls([
-            mock.call("process_sell_needed__2_buy_begin"),
-            mock.call("process_sell_needed__2_buy_end")
-            ])
+        with mock.patch.object(processor, "parse_args") as parse_args:
+            method_mock = mock.Mock()
+            parse_args.return_value = [method_mock, { "foo": "bar" }]
 
-    def test_process_sell_all__1_sell(self):
-        helper.process_sell_all__1_all_sell(self.m)
+            processor.run_action("foo", "bar", "baz")
 
-        self.m.balances.fetch_balances.assert_has_calls([
-            mock.call(tag="process_sell_all__1_all_sell_begin"),
-            mock.call(tag="process_sell_all__1_all_sell_end"),
-            ])
-        self.m.prepare_trades.assert_called_with(base_currency="BTC",
-                liquidity="medium", repartition={'BTC': (1, 'long')})
-        self.m.trades.prepare_orders.assert_called_with(compute_value="average")
-        self.m.trades.run_orders.assert_called()
-        self.m.follow_orders.assert_called()
-        self.m.report.log_stage.assert_has_calls([
-            mock.call("process_sell_all__1_all_sell_begin"),
-            mock.call("process_sell_all__1_all_sell_end")
-            ])
+            parse_args.assert_called_with("foo", "bar", "baz")
 
-    @mock.patch("portfolio.Portfolio.wait_for_recent")
-    def test_process_sell_all__2_wait(self, wait):
-        helper.process_sell_all__2_wait(self.m)
+            method_mock.assert_called_with(foo="bar")
 
-        wait.assert_called_once_with(self.m)
-        self.m.report.log_stage.assert_has_calls([
-            mock.call("process_sell_all__2_wait_begin"),
-            mock.call("process_sell_all__2_wait_end")
-            ])
+            processor.run_action("wait_for_recent", "bar", "baz")
 
-    def test_process_sell_all__3_all_buy(self):
-        helper.process_sell_all__3_all_buy(self.m)
+            method_mock.assert_called_with(self.m, foo="bar")
+
+    def test_select_step(self):
+        processor = helper.Processor(self.m)
+
+        scenario = processor.scenarios["sell_all"]
+
+        self.assertEqual(scenario, processor.select_steps(scenario, "all"))
+        self.assertEqual(["all_sell"], list(map(lambda x: x["name"], processor.select_steps(scenario, "before"))))
+        self.assertEqual(["wait", "all_buy"], list(map(lambda x: x["name"], processor.select_steps(scenario, "after"))))
+        self.assertEqual(["wait"], list(map(lambda x: x["name"], processor.select_steps(scenario, 2))))
+        self.assertEqual(["wait"], list(map(lambda x: x["name"], processor.select_steps(scenario, "wait"))))
+
+        with self.assertRaises(TypeError):
+            processor.select_steps(scenario, ["wait"])
+
+    @mock.patch("helper.Processor.process_step")
+    def test_process(self, process_step):
+        processor = helper.Processor(self.m)
+
+        processor.process("sell_all", foo="bar")
+        self.assertEqual(3, process_step.call_count)
+
+        steps = list(map(lambda x: x[1][1]["name"], process_step.mock_calls))
+        scenario_names = list(map(lambda x: x[1][0], process_step.mock_calls))
+        kwargs = list(map(lambda x: x[1][2], process_step.mock_calls))
+        self.assertEqual(["all_sell", "wait", "all_buy"], steps)
+        self.assertEqual(["sell_all", "sell_all", "sell_all"], scenario_names)
+        self.assertEqual([{"foo":"bar"}, {"foo":"bar"}, {"foo":"bar"}], kwargs)
+
+        process_step.reset_mock()
+
+        processor.process("sell_needed", steps=["before", "after"])
+        self.assertEqual(3, process_step.call_count)
+
+    def test_method_arguments(self):
+        ccxt = mock.Mock(spec=market.ccxt.poloniexE)
+        m = market.Market(ccxt)
+
+        processor = helper.Processor(m)
+
+        method, arguments = processor.method_arguments("wait_for_recent")
+        self.assertEqual(portfolio.Portfolio.wait_for_recent, method)
+        self.assertEqual(["delta"], arguments)
+
+        method, arguments = processor.method_arguments("prepare_trades")
+        self.assertEqual(m.prepare_trades, method)
+        self.assertEqual(['base_currency', 'liquidity', 'compute_value', 'repartition', 'only'], arguments)
+
+        method, arguments = processor.method_arguments("prepare_orders")
+        self.assertEqual(m.trades.prepare_orders, method)
+
+        method, arguments = processor.method_arguments("move_balances")
+        self.assertEqual(m.move_balances, method)
+
+        method, arguments = processor.method_arguments("run_orders")
+        self.assertEqual(m.trades.run_orders, method)
+
+        method, arguments = processor.method_arguments("follow_orders")
+        self.assertEqual(m.follow_orders, method)
+
+        method, arguments = processor.method_arguments("close_trades")
+        self.assertEqual(m.trades.close_trades, method)
+
+    def test_process_step(self):
+        processor = helper.Processor(self.m)
+
+        with mock.patch.object(processor, "run_action") as run_action:
+            step = processor.scenarios["sell_needed"][1]
+
+            processor.process_step("foo", step, {"foo":"bar"})
+
+            self.m.report.log_stage.assert_has_calls([
+                mock.call("process_foo__1_sell_begin"),
+                mock.call("process_foo__1_sell_end"),
+                ])
+            self.m.balances.fetch_balances.assert_has_calls([
+                mock.call(tag="process_foo__1_sell_begin"),
+                mock.call(tag="process_foo__1_sell_end"),
+                ])
+
+            self.assertEqual(5, run_action.call_count)
+
+            run_action.assert_has_calls([
+                mock.call('prepare_trades', {}, {'foo': 'bar'}),
+                mock.call('prepare_orders', {'only': 'dispose', 'compute_value': 'average'}, {'foo': 'bar'}),
+                mock.call('run_orders', {}, {'foo': 'bar'}),
+                mock.call('follow_orders', {}, {'foo': 'bar'}),
+                mock.call('close_trades', {}, {'foo': 'bar'}),
+                ])
+
+        self.m.reset_mock()
+        with mock.patch.object(processor, "run_action") as run_action:
+            step = processor.scenarios["sell_needed"][0]
+
+            processor.process_step("foo", step, {"foo":"bar"})
+            self.m.balances.fetch_balances.assert_not_called()
+
+    def test_parse_args(self):
+        processor = helper.Processor(self.m)
+
+        with mock.patch.object(processor, "method_arguments") as method_arguments:
+            method_mock = mock.Mock()
+            method_arguments.return_value = [
+                    method_mock,
+                    ["foo2", "foo"]
+                    ]
+            method, args = processor.parse_args("action", {"foo": "bar", "foo2": "bar"}, {"foo": "bar2", "bla": "bla"})
+
+            self.assertEqual(method_mock, method)
+            self.assertEqual({"foo": "bar2", "foo2": "bar"}, args)
+
+        with mock.patch.object(processor, "method_arguments") as method_arguments:
+            method_mock = mock.Mock()
+            method_arguments.return_value = [
+                    method_mock,
+                    ["repartition"]
+                    ]
+            method, args = processor.parse_args("action", {"repartition": { "base_currency": 1 }}, {})
+
+            self.assertEqual(1, len(args["repartition"]))
+            self.assertIn("BTC", args["repartition"])
+
+        with mock.patch.object(processor, "method_arguments") as method_arguments:
+            method_mock = mock.Mock()
+            method_arguments.return_value = [
+                    method_mock,
+                    ["repartition", "base_currency"]
+                    ]
+            method, args = processor.parse_args("action", {"repartition": { "base_currency": 1 }}, {"base_currency": "USDT"})
+
+            self.assertEqual(1, len(args["repartition"]))
+            self.assertIn("USDT", args["repartition"])
+
+        with mock.patch.object(processor, "method_arguments") as method_arguments:
+            method_mock = mock.Mock()
+            method_arguments.return_value = [
+                    method_mock,
+                    ["repartition", "base_currency"]
+                    ]
+            method, args = processor.parse_args("action", {"repartition": { "ETH": 1 }}, {"base_currency": "USDT"})
+
+            self.assertEqual(1, len(args["repartition"]))
+            self.assertIn("ETH", args["repartition"])
 
-        self.m.balances.fetch_balances.assert_has_calls([
-            mock.call(tag="process_sell_all__3_all_buy_begin"),
-            mock.call(tag="process_sell_all__3_all_buy_end"),
-            ])
-        self.m.prepare_trades.assert_called_with(base_currency="BTC",
-                liquidity="medium")
-        self.m.trades.prepare_orders.assert_called_with(compute_value="average")
-        self.m.move_balances.assert_called_with()
-        self.m.trades.run_orders.assert_called()
-        self.m.follow_orders.assert_called()
-        self.m.report.log_stage.assert_has_calls([
-            mock.call("process_sell_all__3_all_buy_begin"),
-            mock.call("process_sell_all__3_all_buy_end")
-            ])
 
 @unittest.skipUnless("acceptance" in limits, "Acceptance skipped")
 class AcceptanceTest(WebMockTestCase):