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')