X-Git-Url: https://git.immae.eu/?p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git;a=blobdiff_plain;f=tests%2Ftest_main.py;h=298e29e136ca1f4062a32af0e874d6aa331a4eaf;hp=1864c062af75fd1bd7d7c0eabca80127e6a2dc13;hb=e45c64a5ace2637a9ff290a42cde89e2403ae53d;hpb=5321200c05b3b693581ec4238c74eb02e0b715d7 diff --git a/tests/test_main.py b/tests/test_main.py index 1864c06..298e29e 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -325,7 +325,7 @@ class MainTest(WebMockTestCase): with self.subTest(user=None): rows = list(main.fetch_markets(None)) - cursor_mock.execute.assert_called_once_with("SELECT id,config,user_id FROM market_configs") + cursor_mock.execute.assert_called_once_with("SELECT id,config,user_id FROM market_configs WHERE status='enabled'") self.assertEqual(["row_1", "row_2"], rows) @@ -333,7 +333,7 @@ class MainTest(WebMockTestCase): with self.subTest(user=1): rows = list(main.fetch_markets(1)) - cursor_mock.execute.assert_called_once_with("SELECT id,config,user_id FROM market_configs WHERE user_id = %s", 1) + cursor_mock.execute.assert_called_once_with("SELECT id,config,user_id FROM market_configs WHERE status='enabled' AND user_id = %s", 1) self.assertEqual(["row_1", "row_2"], rows)