def to_mock(self): """Returns an AsyncMock matching the spec for this class""" # we still have to set stuff manually but changing values is nicer mock = AsyncMock(name="Message Mock", spec=discord.Message) # Author section mock.author = MockedMember( name=self.author["name"], member_id=self.author["id"], is_bot=self.author["is_bot"], ).to_mock() # Guild options and 'is_in_guild' are mutually exclusive if not self.is_in_guild: mock.guild = None else: mock.guild = MockedGuild(name=self.guild["name"], guild_id=self.guild["id"]).to_mock() mock.channel = MockedChannel().to_mock() mock.created_at = datetime.datetime.now() mock.id = self.message_id mock.content = self.content mock.clean_content = self.clean_content return mock
def context( dpy_guild: discord.Guild, dpy_channel: discord.TextChannel, dpy_author: discord.User, dpy_message: discord.Message, ) -> discord.Interaction: stub = AsyncMock(spec=commands.Context) stub.guild = dpy_guild stub.channel = dpy_channel stub.channel_id = dpy_channel.id stub.author = dpy_author stub.message = dpy_message return stub
def build_interaction( guild: discord.Guild, channel: discord.TextChannel, author: discord.User, ) -> discord.Interaction: stub = AsyncMock(spec=discord.Interaction) stub.response = AsyncMock() stub.followup = AsyncMock() stub.guild = guild stub.guild_id = guild.id stub.channel = channel stub.channel_id = channel.id stub.user = author stub.original_message = AsyncMock( return_value=build_message(guild, channel, author)) return stub