From e6015816224f8f405e9b1c9557f22e73b21246e8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Isma=C3=ABl=20Bouya?= Date: Thu, 5 Apr 2018 12:14:49 +0200 Subject: Fix console helper --- test.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index 854e27b..cbce357 100644 --- a/test.py +++ b/test.py @@ -4585,22 +4585,28 @@ class MainTest(WebMockTestCase): def test_get_user_market(self): with mock.patch("main.fetch_markets") as main_fetch_markets,\ + mock.patch("main.parse_args") as main_parse_args,\ mock.patch("main.parse_config") as main_parse_config: with self.subTest(debug=False): - main_parse_config.return_value = ["pg_config", "report_path"] + main_parse_args.return_value = self.market_args() + main_parse_config.return_value = "pg_config" main_fetch_markets.return_value = [(1, {"key": "market_config"}, 3)] m = main.get_user_market("config_path.ini", 1) self.assertIsInstance(m, market.Market) self.assertFalse(m.debug) + main_parse_args.assert_called_once_with(["--config", "config_path.ini"]) + main_parse_args.reset_mock() with self.subTest(debug=True): - main_parse_config.return_value = ["pg_config", "report_path"] + main_parse_args.return_value = self.market_args(debug=True) + main_parse_config.return_value = "pg_config" main_fetch_markets.return_value = [(1, {"key": "market_config"}, 3)] m = main.get_user_market("config_path.ini", 1, debug=True) self.assertIsInstance(m, market.Market) self.assertTrue(m.debug) + main_parse_args.assert_called_once_with(["--config", "config_path.ini", "--debug"]) def test_process(self): with mock.patch("market.Market") as market_mock,\ -- cgit v1.2.3