def test_bicingbot_callback_remove_group_cancel(remove_group_cancel):
remove_group_cancel.return_value = mock.MagicMock()
callback_query = CallbackQuery(1, message_id=111)
bicingbot_callback(chat_id, '{}_unused'.format(REMOVE_CANCEL_CALLBACK),
callback_query)
remove_group_cancel.assert_called_with(chat_id, callback_query)
def test_bicingbot_callback_language(update_language):
update_language.return_value = mock.MagicMock()
callback_query = CallbackQuery(1)
bicingbot_callback(chat_id, '{}_en'.format(LANGUAGE_CALLBACK),
callback_query)
update_language.assert_called_with(chat_id, 'en', callback_query)
def test_bicingbot_callback_remove_group_underscore(remove_group):
remove_group.return_value = mock.MagicMock()
callback_query = CallbackQuery(1, message_id=111)
bicingbot_callback(chat_id, '{}_name_group1'.format(REMOVE_GROUP_CALLBACK),
callback_query)
remove_group.assert_called_with(chat_id, 'name_group1', callback_query)
def test_bicingbot_callback_unknown(remove_group, update_language):
update_language.return_value = mock.MagicMock()
remove_group.return_value = mock.MagicMock()
callback_query = CallbackQuery(1)
bicingbot_callback(chat_id, 'unknown_group1', callback_query)
update_language.assert_not_called()
remove_group.assert_not_called()
def test_update_language_wrong(get_bot, DatabaseConnection):
DatabaseConnection.return_value = mock.MagicMock()
get_bot.return_value = mock.MagicMock()
callback_query = CallbackQuery(1)
update_language(chat_id, 'nolanguage', callback_query)
# Check bot calls
get_bot().answer_callback_query.assert_not_called()
get_bot().send_message.assert_not_called()
DatabaseConnection().add_setting.assert_not_called()
def test_remove_group_cancel(get_bot):
get_bot.return_value = mock.MagicMock()
callback_query = CallbackQuery(1, message_id=111)
remove_group_cancel(chat_id, callback_query)
# Check bot calls
get_bot().answer_callback_query.assert_called_once()
get_bot().edit_message_text.assert_called_with(
chat_id=chat_id,
text=STRINGS['es']['removegroup_canceled'],
message_id=111)
def test_update_language(get_bot, DatabaseConnection):
DatabaseConnection.return_value = mock.MagicMock()
get_bot.return_value = mock.MagicMock()
callback_query = CallbackQuery(1)
update_language(chat_id, 'es', callback_query)
# Check bot calls
get_bot().answer_callback_query.assert_called_once()
get_bot().send_message.assert_called_once()
DatabaseConnection().add_setting.assert_called_with(
chat_id=chat_id, setting=LANGUAGE_SETTING, value='es')
def test_remove_group_not_found(get_bot, DatabaseConnection):
get_bot.return_value = mock.MagicMock()
DatabaseConnection.return_value = mock.MagicMock()
DatabaseConnection().get_groups_names.return_value = ['group1', 'group2']
callback_query = CallbackQuery(1, message_id=111)
remove_group(chat_id, 'notfound', callback_query)
# Check bot calls
get_bot().answer_callback_query.assert_not_called()
get_bot().send_message.assert_called_with(
chat_id=chat_id, text=STRINGS['es']['removegroup_not_found'])
DatabaseConnection().delete_group.assert_not_called()
def test_remove_group(get_bot, DatabaseConnection):
get_bot.return_value = mock.MagicMock()
DatabaseConnection.return_value = mock.MagicMock()
DatabaseConnection().get_groups_names.return_value = ['group1', 'group2']
DatabaseConnection().get_group.return_value = {'stations': [1, 2]}
callback_query = CallbackQuery(1, message_id=111)
remove_group(chat_id, 'group2', callback_query)
# Check bot calls
get_bot().answer_callback_query.assert_called_once()
get_bot().edit_message_text.assert_called_with(
chat_id=chat_id,
text=STRINGS['es']['removegroup_removed'].format('group2', '1, 2'),
message_id=111)
DatabaseConnection().delete_group.assert_called_with(chat_id=chat_id,
name='group2')