def test_wrong_option_selected_returns_correct_error_template(self, *args):
buff = ['rate', 'breakfast', '10', '5']
rate_context = Helper.get_rate_template_context(buff, self.user_id)
self.assertEqual(rate_context, {'context': {'option_count': 3},
'template': 'invalid_option'})
def test_wrong_meal_selected_returns_correct_error_template(self, *args):
buff = ['rate', 'asdflkj', '2', '5']
rate_context = Helper.get_rate_template_context(buff, self.user_id)
self.assertEqual(rate_context, {'template': 'invalid_meal',
'context': {}})
def test_user_gets_weekend_rate_error_on_weekend(self, *args):
buff = ['rate', 'breakfast', '2', '5']
user_id = 'U0774N56J'
rate_context = Helper.get_rate_template_context(buff, user_id)
self.assertEqual(rate_context, {'template': 'weekend_rate_error',
'context': {}})
def test_invalid_rating_returns_correct_error_template(self, *args):
negative_rate_buff = ['rate', 'breakfast', '2', '-1']
zero_rate_buff = ['rate', 'breakfast', '2', '0']
above_rate_buff = ['rate', 'breakfast', '2', '6']
negative_rate_context = Helper.get_rate_template_context(
negative_rate_buff, self.user_id)
zero_rate_context = Helper.get_rate_template_context(
zero_rate_buff, self.user_id)
above_rate_context = Helper.get_rate_template_context(
above_rate_buff, self.user_id)
self.assertEqual(negative_rate_context, {'context': {},
'template': 'invalid_rating'})
self.assertEqual(zero_rate_context, {'context': {},
'template': 'invalid_rating'})
self.assertEqual(above_rate_context, {'context': {},
'template': 'invalid_rating'})
def test_valid_rate_returns_correct_template(self, *args):
buff = ['rate', 'breakfast', '2', '5']
rate_context = Helper.get_rate_template_context(buff, self.user_id)
self.assertEqual(rate_context, {'template': 'rating_response',
'context': {}})
CustomSQL.query.assert_called_with(
'SELECT id FROM menu_table WHERE meal = (%s) AND day = (%s) AND week = (%s) AND option = (%s)', ('breakfast', 'monday', '1', '2'))
def test_multiple_check_template_rendered_if_rate_count_is_greater_than_zero(self, *args):
buff = ['rate', 'breakfast', '2', '5']
rate_context = Helper.get_rate_template_context(buff, self.user_id)
self.assertEqual(rate_context, {'template': 'multiple_rating',
'context': {}})
