async def test_on_database_world_selected():
# Setup
cog = DatabaseCommandCog({"guild": 1234}, MagicMock())
area = MagicMock()
area.name = "The Area"
cog._split_worlds[RandovaniaGame.METROID_PRIME_CORRUPTION] = [
tuple(),
SplitWorld(MagicMock(), "The World", [area]),
]
ctx = AsyncMock()
ctx.selected_options = [
f"{RandovaniaGame.METROID_PRIME_CORRUPTION.value}_world_1"
]
# Run
await cog.on_database_world_selected(
ctx, RandovaniaGame.METROID_PRIME_CORRUPTION)
# Assert
ctx.edit_origin.assert_awaited_once_with(
embed=ANY,
components=[
ANY,
{
"type": ComponentType.actionrow,
"components": ANY
},
],
)
async def test_database_command():
# Setup
cog = DatabaseCommandCog({"guild": 1234}, MagicMock())
area = MagicMock()
area.name = "The Area"
cog._split_worlds[RandovaniaGame.METROID_PRIME_CORRUPTION] = [
SplitWorld(MagicMock(), "The World", [area]),
]
ctx = AsyncMock()
# Run
await cog.database_command(ctx,
RandovaniaGame.METROID_PRIME_CORRUPTION.value)
# Assert
ctx.send.assert_awaited_once_with(embed=ANY, components=ANY, hidden=True)
async def test_on_database_area_selected(echoes_game_description, mocker):
# Setup
mock_file: MagicMock = mocker.patch("discord.File")
mock_digraph: MagicMock = mocker.patch("graphviz.Digraph")
dot: MagicMock = mock_digraph.return_value
dot.render.return_value = "bar"
Path("bar").write_bytes(b"1234")
cog = DatabaseCommandCog({"guild": 1234}, MagicMock())
ctx = AsyncMock()
ctx.selected_options = [f"area_1"]
db = echoes_game_description
world = echoes_game_description.world_list.worlds[2]
area = world.areas[0]
split_world = SplitWorld(world, "The World", [MagicMock(), area])
# Run
await cog.on_database_area_selected(ctx,
RandovaniaGame.METROID_PRIME_ECHOES,
split_world, 2)
# Assert
ctx.send.assert_awaited_once_with(
content=
f"**{db.game.long_name}: {db.world_list.area_name(area)}**\nRequested by {ctx.author.display_name}.",
files=[mock_file.return_value],
components=[ANY],
)
mock_digraph.assert_called_once_with(comment=area.name)
dot.node.assert_has_calls(
[call(node.name) for node in area.nodes if not node.is_derived_node])
dot.render.assert_called_once_with(format="png", cleanup=True)
mock_file.assert_called_once_with(ANY, filename=f"{area.name}_graph.png")
v = mock_file.call_args[0][0]
assert isinstance(v, io.BytesIO)
assert v.getvalue() == b"1234"
assert not Path("bar").is_file()
async def test_add_commands():
# Setup
cog = DatabaseCommandCog({"guild": 1234}, MagicMock())
slash = cog.bot.slash
# Run
await cog.add_commands(slash)
# Assert
slash.add_slash_command.assert_called_once_with(
cog.database_command,
name="database-inspect",
description=
"Consult the Randovania's logic database for one specific room.",
guild_ids=None,
options=[ANY],
)
slash.add_component_callback.assert_called_once_with(
cog.on_database_component,
components=ANY,
use_callback_name=False,
)
async def test_on_area_node_selection(echoes_game_description, mocker):
# Setup
mock_embed: MagicMock = mocker.patch("discord.Embed")
cog = DatabaseCommandCog({"guild": 1234}, MagicMock())
world = echoes_game_description.world_list.worlds[2]
area = world.areas[2]
ctx = AsyncMock()
ctx.selected_options = [area.nodes[0].name, area.nodes[2].name]
# Run
await cog.on_area_node_selection(ctx, RandovaniaGame.METROID_PRIME_ECHOES,
area)
# Assert
ctx.edit_origin.assert_awaited_once_with(
embeds=[mock_embed.return_value, mock_embed.return_value])
mock_embed.assert_has_calls([
call(title=area.nodes[0].name, description=ANY),
call(title=area.nodes[2].name, description=ANY),
])