async def test_end_tournament(self):
"""Sends a message to react on in order to end the tournament."""
bot_mock = tosurnament_mock.BotMock()
bot_mock.session.add_stub(Tournament())
cog = tosurnament_mock.mock_cog(guild_owner.get_class(bot_mock))
await cog.end_tournament(cog, tosurnament_mock.CtxMock(bot_mock))
cog.send_reply.assert_called_with(mock.ANY, mock.ANY, "are_you_sure")
assert bot_mock.session.add.call_count == 1
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_end_tournament(mocker):
"""Sends a message to react on in order to end the tournament."""
mock_bot = tosurnament_mock.BotMock()
mock_bot.session.add_stub(Tournament(guild_id=tosurnament_mock.GUILD_ID))
cog = tosurnament_mock.mock_cog(guild_owner.get_class(mock_bot))
await cog.end_tournament(cog, tosurnament_mock.CtxMock(mock_bot))
cog.send_reply.assert_called_with(mocker.ANY, mocker.ANY, "are_you_sure")
mock_bot.session.add.assert_called_once_with(
tosurnament_mock.Matcher(EndTournamentMessage()))
async def test_create_tournament_already_created(self):
"""Creates a tournament but one has already been created."""
bot_mock = tosurnament_mock.BotMock()
bot_mock.session.add_stub(Tournament())
cog = tosurnament_mock.mock_cog(guild_owner.get_class(bot_mock))
with self.assertRaises(tosurnament.TournamentAlreadyCreated):
await cog.create_tournament(
cog,
tosurnament_mock.CtxMock(bot_mock),
TOURNAMENT_ACRONYM,
TOURNAMENT_NAME,
)
async def test_create_tournament_already_created():
"""Creates a tournament but one has already been created."""
mock_bot = tosurnament_mock.BotMock()
mock_bot.session.add_stub(Tournament(guild_id=tosurnament_mock.GUILD_ID))
cog = tosurnament_mock.mock_cog(guild_owner.get_class(mock_bot))
with pytest.raises(tosurnament.TournamentAlreadyCreated):
await cog.create_tournament(
cog,
tosurnament_mock.CtxMock(mock_bot),
TOURNAMENT_ACRONYM,
TOURNAMENT_NAME,
)
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_create_tournament(self):
"""Creates a tournament."""
bot_mock = tosurnament_mock.BotMock()
cog = tosurnament_mock.mock_cog(guild_owner.get_class(bot_mock))
await cog.create_tournament(cog, tosurnament_mock.CtxMock(bot_mock),
TOURNAMENT_ACRONYM, TOURNAMENT_NAME)
assert bot_mock.session.update.call_count == 1
tournament_matcher = tosurnament_mock.Matcher(
Tournament(acronym=TOURNAMENT_ACRONYM,
name=TOURNAMENT_NAME,
current_bracket_id=1), )
bracket_matcher = tosurnament_mock.Matcher(
Bracket(tournament_id=1, name=TOURNAMENT_NAME))
expected = [mock.call(tournament_matcher), mock.call(bracket_matcher)]
assert bot_mock.session.add.call_args_list == expected
cog.send_reply.assert_called_with(mock.ANY, mock.ANY, "success",
TOURNAMENT_ACRONYM, TOURNAMENT_NAME,
TOURNAMENT_NAME)
async def test_create_tournament_with_bracket_name(mocker):
"""Creates a tournament with a specified bracket name."""
mock_bot = tosurnament_mock.BotMock()
cog = tosurnament_mock.mock_cog(guild_owner.get_class(mock_bot))
await cog.create_tournament(
cog,
tosurnament_mock.CtxMock(mock_bot),
TOURNAMENT_ACRONYM,
TOURNAMENT_NAME,
BRACKET_NAME,
)
assert mock_bot.session.update.call_count == 1
tournament_matcher = tosurnament_mock.Matcher(
Tournament(acronym=TOURNAMENT_ACRONYM,
name=TOURNAMENT_NAME,
current_bracket_id=1), )
bracket_matcher = tosurnament_mock.Matcher(
Bracket(tournament_id=1, name=BRACKET_NAME))
expected = [mocker.call(tournament_matcher), mocker.call(bracket_matcher)]
assert mock_bot.session.add.call_args_list == expected
cog.send_reply.assert_called_with(mocker.ANY, mocker.ANY, "success",
TOURNAMENT_ACRONYM, TOURNAMENT_NAME,
BRACKET_NAME)