async def test_add_emoji_role_pair(mocker): """Adds a emoji/role pair to the role message creation.""" cog, mock_bot, reaction_for_role_message = init_mocks() cog.delete_setup_messages = mocker.AsyncMock() cog.send_reply.return_value = tosurnament_mock.MessageMock(tosurnament_mock.SETUP_MESSAGE_MOCK.id + 1) mock_channel = tosurnament_mock.ChannelMock(tosurnament_mock.SETUP_CHANNEL_MOCK.id + 1) mock_ctx = tosurnament_mock.CtxMock(mock_bot, channel=mock_channel) mock_preview_message = tosurnament_mock.MessageMock(tosurnament_mock.PREVIEW_MESSAGE_MOCK.id + 1) mock_ctx.send.return_value = mock_preview_message await cog.add_emoji_role_pair(cog, mock_ctx, "šļø", tosurnament_mock.VERIFIED_ROLE_MOCK) cog.delete_setup_messages.assert_called_once_with(reaction_for_role_message) cog.send_reply.assert_called_once_with(mocker.ANY, "success", "šļø | Verified\n") mock_ctx.send.assert_called_once_with("Some text") mock_bot.session.update.assert_called_once_with( tosurnament_mock.Matcher( ReactionForRoleMessage( guild_id=tosurnament_mock.DEFAULT_GUILD_MOCK.id, author_id=tosurnament_mock.DEFAULT_USER_MOCK.id, setup_channel_id=tosurnament_mock.SETUP_CHANNEL_MOCK.id + 1, setup_message_id=tosurnament_mock.SETUP_MESSAGE_MOCK.id + 1, preview_message_id=tosurnament_mock.PREVIEW_MESSAGE_MOCK.id + 1, channel_id=tosurnament_mock.DEFAULT_CHANNEL_MOCK.id, text="Some text", emojis="šļø", roles=str(tosurnament_mock.VERIFIED_ROLE_MOCK.id), ) ) ) assert mock_preview_message.reactions == ["šļø"]
async def test_set_post_result_channel(mocker): """Sets the post result channel.""" cog, mock_bot, bracket = init_mocks() channel = tosurnament_mock.ChannelMock(324769) assert bracket.post_result_channel_id != channel.id await cog.set_post_result_channel(cog, tosurnament_mock.CtxMock(mock_bot), channel=channel) mock_bot.session.update.assert_called_once_with( tosurnament_mock.Matcher(Bracket(tournament_id=1, post_result_channel_id=channel.id, name=BRACKET1_NAME)) )
async def test_reaction_on_end_tournament_message(mocker): """Sends a message to react on in order to end the tournament.""" mock_bot = tosurnament_mock.BotMock() cog = tosurnament_mock.mock_cog(guild_owner.get_class(mock_bot)) mock_bot.session.add_stub( Tournament(id=1, guild_id=tosurnament_mock.GUILD_ID)) mock_bot.session.add_stub( Bracket(tournament_id=1, schedules_spreadsheet_id=1, players_spreadsheet_id=1)) mock_bot.session.add_stub(SchedulesSpreadsheet(id=1)) mock_bot.session.add_stub(PlayersSpreadsheet(id=1)) mock_bot.session.add_stub(Bracket(tournament_id=1)) mock_bot.session.add_stub( Bracket(tournament_id=1, schedules_spreadsheet_id=42, players_spreadsheet_id=42)) mock_bot.session.add_stub(SchedulesSpreadsheet(id=42)) mock_bot.session.add_stub(PlayersSpreadsheet(id=42)) mock_bot.session.add_stub( Bracket(tournament_id=1, schedules_spreadsheet_id=43)) mock_bot.session.add_stub(RescheduleMessage(tournament_id=1)) mock_bot.session.add_stub(RescheduleMessage(tournament_id=1)) mock_bot.session.add_stub(StaffRescheduleMessage(tournament_id=1)) mock_bot.session.add_stub(StaffRescheduleMessage(tournament_id=1)) mock_bot.session.add_stub(EndTournamentMessage(message_id=MESSAGE_ID)) mock_bot.session.add_stub(Tournament(id=2)) mock_bot.session.add_stub( Bracket(tournament_id=2, schedules_spreadsheet_id=2, players_spreadsheet_id=2)) mock_bot.session.add_stub(SchedulesSpreadsheet(id=2)) mock_bot.session.add_stub(PlayersSpreadsheet(id=2)) mock_bot.session.add_stub(RescheduleMessage(tournament_id=2)) await cog.reaction_on_end_tournament_message( MESSAGE_ID, tosurnament_mock.EmojiMock("ā "), tosurnament_mock.GuildMock(tosurnament_mock.GUILD_ID), tosurnament_mock.ChannelMock(), tosurnament_mock.UserMock(user_id=tosurnament_mock.GuildMock.OWNER_ID), ) cog.send_reply.assert_called_with(mocker.ANY, mocker.ANY, "success") assert len(mock_bot.session.tables[Tournament.__tablename__]) == 1 assert len(mock_bot.session.tables[Bracket.__tablename__]) == 1 assert len( mock_bot.session.tables[SchedulesSpreadsheet.__tablename__]) == 1 assert len(mock_bot.session.tables[PlayersSpreadsheet.__tablename__]) == 1 assert len(mock_bot.session.tables[RescheduleMessage.__tablename__]) == 1 assert len( mock_bot.session.tables[StaffRescheduleMessage.__tablename__]) == 0 assert len( mock_bot.session.tables[EndTournamentMessage.__tablename__]) == 0
async def test_reaction_on_end_tournament_message_invalid_emoji(): """Sends a message to react on in order to end the tournament.""" mock_bot = tosurnament_mock.BotMock() cog = tosurnament_mock.mock_cog(guild_owner.get_class(mock_bot)) await cog.reaction_on_end_tournament_message( MESSAGE_ID, tosurnament_mock.EmojiMock("=)"), tosurnament_mock.GuildMock(tosurnament_mock.GUILD_ID), tosurnament_mock.ChannelMock(), tosurnament_mock.UserMock(user_id=tosurnament_mock.GuildMock.OWNER_ID), ) assert mock_bot.session.query.call_count == 0
async def test_reaction_on_end_tournament_message_no_tournament(): """Sends a message to react on in order to end the tournament.""" mock_bot = tosurnament_mock.BotMock() cog = tosurnament_mock.mock_cog(guild_owner.get_class(mock_bot)) mock_bot.session.add_stub(EndTournamentMessage(message_id=MESSAGE_ID)) await cog.reaction_on_end_tournament_message( MESSAGE_ID, tosurnament_mock.EmojiMock("ā "), tosurnament_mock.GuildMock(tosurnament_mock.GUILD_ID), tosurnament_mock.ChannelMock(), tosurnament_mock.UserMock(user_id=tosurnament_mock.GuildMock.OWNER_ID), ) assert mock_bot.session.delete.call_count == 1
async def test_reaction_on_end_tournament_message_refuse(mocker): """Sends a message to react on in order to end the tournament.""" mock_bot = tosurnament_mock.BotMock() cog = tosurnament_mock.mock_cog(guild_owner.get_class(mock_bot)) mock_bot.session.add_stub(Tournament(guild_id=tosurnament_mock.GUILD_ID)) mock_bot.session.add_stub(Bracket()) mock_bot.session.add_stub(EndTournamentMessage(message_id=MESSAGE_ID)) await cog.reaction_on_end_tournament_message( MESSAGE_ID, tosurnament_mock.EmojiMock("ā"), tosurnament_mock.GuildMock(tosurnament_mock.GUILD_ID), tosurnament_mock.ChannelMock(), tosurnament_mock.UserMock(user_id=tosurnament_mock.GuildMock.OWNER_ID), ) cog.send_reply.assert_called_with(mocker.ANY, mocker.ANY, "refused")
async def test_remove_match_to_ignore(mocker): """Removes a match to ignore.""" MATCH_ID_1 = "T1-1" MATCH_ID_2 = "t1-2" MATCH_ID_3 = "T1-3" mock_bot = tosurnament_mock.BotMock() mock_bot.session.add_stub( Tournament( id=1, guild_id=tosurnament_mock.GUILD_ID, matches_to_ignore=MATCH_ID_1 + "\n" + MATCH_ID_3, current_bracket_id=1, staff_channel_id=tosurnament_mock.ChannelMock.STAFF_CHANNEL_ID, )) mock_bot.session.add_stub( Bracket(id=1, tournament_id=1, schedules_spreadsheet_id=1)) mock_bot.session.add_stub( SchedulesSpreadsheet(id=1, spreadsheet_id="single", sheet_name="Tier 1")) cog = tosurnament_mock.mock_cog(tournament.get_class(mock_bot)) await cog.remove_match_to_ignore(cog, tosurnament_mock.CtxMock(mock_bot), MATCH_ID_2, MATCH_ID_3) assert mock_bot.session.tables[ Tournament.__tablename__][0].matches_to_ignore == MATCH_ID_1 expected_replies = [ mocker.call(mocker.ANY, mocker.ANY, "success", MATCH_ID_1), mocker.call( tosurnament_mock.ChannelMock( tosurnament_mock.ChannelMock.STAFF_CHANNEL_ID), mocker.ANY, "to_not_ignore", mocker.ANY, MATCH_ID_3, ), ] assert cog.send_reply.call_args_list == expected_replies
async def test_add_emoji_role_pair_with_already_existing_emoji_role_pairs(mocker): """Adds a emoji/role pair to the role message creation, and it already contains emojis and roles.""" cog, mock_bot, reaction_for_role_message = init_mocks() reaction_for_role_message.emojis = "ā \nā" reaction_for_role_message.roles = str(tosurnament_mock.PLAYER_ROLE_MOCK.id) + "\n12349089" cog.delete_setup_messages = mocker.AsyncMock() cog.send_reply.return_value = tosurnament_mock.MessageMock(tosurnament_mock.SETUP_MESSAGE_MOCK.id + 1) mock_channel = tosurnament_mock.ChannelMock(tosurnament_mock.SETUP_CHANNEL_MOCK.id + 1) mock_command = tosurnament_mock.CommandMock(cog.qualified_name, "add_emoji_role_pair") mock_ctx = tosurnament_mock.CtxMock(mock_bot, channel=mock_channel, command=mock_command) mock_preview_message = tosurnament_mock.MessageMock(tosurnament_mock.PREVIEW_MESSAGE_MOCK.id + 1) mock_ctx.send.return_value = mock_preview_message await cog.add_emoji_role_pair(cog, mock_ctx, "šļø", tosurnament_mock.VERIFIED_ROLE_MOCK) cog.delete_setup_messages.assert_called_once_with(reaction_for_role_message) expected_success_output = "ā | " + tosurnament_mock.PLAYER_ROLE_MOCK.name + "\n" expected_success_output += "ā | Unknown role\n" expected_success_output += "šļø | Verified\n" cog.send_reply.assert_called_once_with(mocker.ANY, "success", expected_success_output) mock_ctx.send.assert_called_once_with("Some text") expected_roles = str(tosurnament_mock.PLAYER_ROLE_MOCK.id) + "\n" expected_roles += "12349089\n" expected_roles += str(tosurnament_mock.VERIFIED_ROLE_MOCK.id) mock_bot.session.update.assert_called_once_with( tosurnament_mock.Matcher( ReactionForRoleMessage( guild_id=tosurnament_mock.DEFAULT_GUILD_MOCK.id, author_id=tosurnament_mock.DEFAULT_USER_MOCK.id, setup_channel_id=tosurnament_mock.SETUP_CHANNEL_MOCK.id + 1, setup_message_id=tosurnament_mock.SETUP_MESSAGE_MOCK.id + 1, preview_message_id=tosurnament_mock.PREVIEW_MESSAGE_MOCK.id + 1, channel_id=tosurnament_mock.DEFAULT_CHANNEL_MOCK.id, text="Some text", emojis="ā \nā\nšļø", roles=expected_roles, ) ) ) assert mock_preview_message.reactions == ["ā ", "ā", "šļø"]