diff options
Diffstat (limited to 'tests/test_main.py')
-rw-r--r-- | tests/test_main.py | 42 |
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") |