Fetch only enabled market

[perso/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Trader.git] / tests / test_main.py

diff --git a/tests/test_main.py b/tests/test_main.py
index 1864c062af75fd1bd7d7c0eabca80127e6a2dc13..298e29e136ca1f4062a32af0e874d6aa331a4eaf 100644 (file)
--- 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)