- cursor_mock.execute.assert_called_once_with("SELECT id,config,user_id FROM market_configs WHERE status='enabled'")
+ cursor_mock.execute.assert_called_once_with("SELECT id,config,user_id,portfolio_profile FROM market_configs_augmented WHERE status='enabled'")
- self.assertEqual(["row_1", "row_2"], rows)
+ self.assertEqual([
+ (1, 'cfg', 1, {'liquidity': 'high'}),
+ (2, 'cfg2', 3, {'liquidity': 'medium'})
+ ], rows)
cursor_mock.execute.reset_mock()
with self.subTest(user=1):
rows = list(main.fetch_markets(1))
cursor_mock.execute.reset_mock()
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 status='enabled' AND user_id = %s", [1])
+ cursor_mock.execute.assert_called_once_with("SELECT id,config,user_id,portfolio_profile FROM market_configs_augmented WHERE status='enabled' AND user_id = %s", [1])