- def test_print_orders(self):
- m = market.Market(self.ccxt, self.market_args())
- with mock.patch.object(m.report, "log_stage") as log_stage,\
- mock.patch.object(m.balances, "fetch_balances") as fetch_balances,\
- mock.patch.object(m, "prepare_trades") as prepare_trades,\
- mock.patch.object(m.trades, "prepare_orders") as prepare_orders:
- m.print_orders()
+ report.reset_mock()
+ m = market.Market(self.ccxt, self.market_args(report_redis=False),
+ redis_config="redis_config", user_id=1)
+ with self.subTest(redis_config="present", report_redis=False),\
+ mock.patch.object(m, "report") as report,\
+ mock.patch.object(m, "store_file_report") as file_report,\
+ mock.patch.object(m, "store_redis_report") as redis_report,\
+ mock.patch.object(m, "store_database_report") as db_report,\
+ mock.patch.object(market.datetime, "datetime") as time_mock:
+
+ time_mock.now.return_value = datetime.datetime(2018, 2, 25)
+
+ m.store_report()
+ redis_report.assert_not_called()
+
+ report.reset_mock()
+ m = market.Market(self.ccxt, self.market_args(report_redis=True),
+ user_id=1)
+ with self.subTest(redis_config="absent", report_redis=True),\
+ mock.patch.object(m, "report") as report,\
+ mock.patch.object(m, "store_file_report") as file_report,\
+ mock.patch.object(m, "store_redis_report") as redis_report,\
+ mock.patch.object(m, "store_database_report") as db_report,\
+ mock.patch.object(market.datetime, "datetime") as time_mock:
+
+ time_mock.now.return_value = datetime.datetime(2018, 2, 25)