def test_fire_registered_event_cancel(self): bot = Bot("app_name", "version") bot.guild_logs_file = self.LOG_PATH on_guild_remove = AsyncMock() on_guild_remove.__name__ = "on_guild_remove" on_guild_remove.return_value = False bot.register_event(on_guild_remove) guild = AsyncMock() self._await(bot.on_guild_remove(guild)) on_guild_remove.assert_awaited_once_with(guild) self.assertFalse(path.exists(self.LOG_PATH))
def test_fire_registered_event(self): bot = Bot("app_name", "version") bot.guild_logs_file = self.LOG_PATH on_guild_join = AsyncMock() on_guild_join.__name__ = "on_guild_join" on_guild_join.return_value = True bot.register_event(on_guild_join) guild = AsyncMock() self._await(bot.on_guild_join(guild)) on_guild_join.assert_awaited_once_with(guild) self.assertTrue(path.exists(self.LOG_PATH))
def test_fire_registered_event_cancel(self): bot = Bot("app_name", "version") on_message = AsyncMock() on_message.__name__ = "on_message" on_message.return_value = False bot.register_event(on_message) watcher_callback = AsyncMock() bot.register_watcher(watcher_callback) message = AsyncMock() message.content = "hello there" self._await(bot.on_message(message)) on_message.assert_awaited_once_with(message) watcher_callback.assert_not_awaited()
def test_fire_registered_event_cancel(self, client_mock): bot = Bot("app_name", "version") bot.guild_logs_file = None on_ready = AsyncMock() on_ready.__name__ = "on_ready" on_ready.return_value = False bot.register_event(on_ready) ex = Exception("test") client_mock.change_presence.side_effect = ex try: with patch("discord.Game") as game_mock: game_mock.return_value = "activity" self._await(bot.on_ready()) except Exception as error: self.assertEqual(ex, error) on_ready.assert_awaited_once() client_mock.change_presence.assert_not_called()
def mock_schedule_background_tasks(): mock = AsyncMock(return_value=Success()) mock.__name__ = "schedule_background_tasks" return mock
def stop_async(api: Api): stop = AsyncMock() stop.__name__ = "stop_async" api.on_stop(stop) return stop
def start_async(api: Api): start = AsyncMock() start.__name__ = "start_async" api.on_start(start) return start