from .helper import *
import main, market
+@unittest.skipUnless("unit" in limits, "Unit skipped")
class MainTest(WebMockTestCase):
def test_make_order(self):
self.m.get_ticker.return_value = {
mock.call.__bool__(),
mock.call("config2", 1, 2, args_mock, "pg_config"),
])
+ with self.subTest(quiet=True):
+ with mock.patch("main.parse_args") as parse_args,\
+ mock.patch("main.parse_config") as parse_config,\
+ mock.patch("main.fetch_markets") as fetch_markets,\
+ mock.patch("store.Portfolio.report") as report,\
+ mock.patch("main.process") as process:
+
+ args_mock = mock.Mock()
+ args_mock.parallel = False
+ args_mock.quiet = True
+ args_mock.user = "user"
+ parse_args.return_value = args_mock
+
+ parse_config.return_value = "pg_config"
+
+ fetch_markets.return_value = [[3, "config1", 1], [1, "config2", 2]]
+
+ main.main(["Foo", "Bar"])
+
+ report.set_verbose.assert_called_once_with(False)
+
+ with self.subTest(quiet=False):
+ with mock.patch("main.parse_args") as parse_args,\
+ mock.patch("main.parse_config") as parse_config,\
+ mock.patch("main.fetch_markets") as fetch_markets,\
+ mock.patch("store.Portfolio.report") as report,\
+ mock.patch("main.process") as process:
+
+ args_mock = mock.Mock()
+ args_mock.parallel = False
+ args_mock.quiet = False
+ args_mock.user = "user"
+ parse_args.return_value = args_mock
+
+ parse_config.return_value = "pg_config"
+
+ fetch_markets.return_value = [[3, "config1", 1], [1, "config2", 2]]
+
+ main.main(["Foo", "Bar"])
+
+ report.set_verbose.assert_called_once_with(True)
+
@mock.patch.object(main.sys, "exit")
@mock.patch("main.os")