aboutsummaryrefslogtreecommitdiff
path: root/tests/test_main.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_main.py')
-rw-r--r--tests/test_main.py42
1 files changed, 42 insertions, 0 deletions
diff --git a/tests/test_main.py b/tests/test_main.py
index 06fc84e..cee89ce 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -205,6 +205,48 @@ class MainTest(WebMockTestCase):
205 mock.call.__bool__(), 205 mock.call.__bool__(),
206 mock.call("config2", 1, 2, args_mock, "pg_config"), 206 mock.call("config2", 1, 2, args_mock, "pg_config"),
207 ]) 207 ])
208 with self.subTest(quiet=True):
209 with mock.patch("main.parse_args") as parse_args,\
210 mock.patch("main.parse_config") as parse_config,\
211 mock.patch("main.fetch_markets") as fetch_markets,\
212 mock.patch("store.Portfolio.report") as report,\
213 mock.patch("main.process") as process:
214
215 args_mock = mock.Mock()
216 args_mock.parallel = False
217 args_mock.quiet = True
218 args_mock.user = "user"
219 parse_args.return_value = args_mock
220
221 parse_config.return_value = "pg_config"
222
223 fetch_markets.return_value = [[3, "config1", 1], [1, "config2", 2]]
224
225 main.main(["Foo", "Bar"])
226
227 report.set_verbose.assert_called_once_with(False)
228
229 with self.subTest(quiet=False):
230 with mock.patch("main.parse_args") as parse_args,\
231 mock.patch("main.parse_config") as parse_config,\
232 mock.patch("main.fetch_markets") as fetch_markets,\
233 mock.patch("store.Portfolio.report") as report,\
234 mock.patch("main.process") as process:
235
236 args_mock = mock.Mock()
237 args_mock.parallel = False
238 args_mock.quiet = False
239 args_mock.user = "user"
240 parse_args.return_value = args_mock
241
242 parse_config.return_value = "pg_config"
243
244 fetch_markets.return_value = [[3, "config1", 1], [1, "config2", 2]]
245
246 main.main(["Foo", "Bar"])
247
248 report.set_verbose.assert_called_once_with(True)
249
208 250
209 @mock.patch.object(main.sys, "exit") 251 @mock.patch.object(main.sys, "exit")
210 @mock.patch("main.os") 252 @mock.patch("main.os")