From 20449d1f018a665874bafc399e6169220384c916 Mon Sep 17 00:00:00 2001 From: Joost Lekkerkerker Date: Thu, 8 Jun 2023 16:08:18 +0200 Subject: [PATCH] Catch exception when user has no lastfm friends (#94235) --- homeassistant/components/lastfm/config_flow.py | 6 +++--- tests/components/lastfm/__init__.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/lastfm/config_flow.py b/homeassistant/components/lastfm/config_flow.py index 3f00443147d..54406a6e03b 100644 --- a/homeassistant/components/lastfm/config_flow.py +++ b/homeassistant/components/lastfm/config_flow.py @@ -3,7 +3,7 @@ from __future__ import annotations from typing import Any -from pylast import LastFMNetwork, User, WSError +from pylast import LastFMNetwork, PyLastError, User, WSError import voluptuous as vol from homeassistant.config_entries import ( @@ -135,7 +135,7 @@ class LastFmConfigFlowHandler(ConfigFlow, domain=DOMAIN): SelectOptionDict(value=friend.name, label=friend.get_name(True)) for friend in friends_response ] - except WSError: + except PyLastError: friends = [] return self.async_show_form( step_id="friends", @@ -207,7 +207,7 @@ class LastFmOptionsFlowHandler(OptionsFlowWithConfigEntry): SelectOptionDict(value=friend.name, label=friend.get_name(True)) for friend in friends_response ] - except WSError: + except PyLastError: friends = [] else: friends = [] diff --git a/tests/components/lastfm/__init__.py b/tests/components/lastfm/__init__.py index 568983f400d..7ee8665e28a 100644 --- a/tests/components/lastfm/__init__.py +++ b/tests/components/lastfm/__init__.py @@ -1,7 +1,7 @@ """The tests for lastfm.""" from unittest.mock import patch -from pylast import Track, WSError +from pylast import PyLastError, Track from homeassistant.components.lastfm.const import CONF_MAIN_USER, CONF_USERS from homeassistant.const import CONF_API_KEY @@ -65,7 +65,7 @@ class MockUser: def get_friends(self): """Get mock friends.""" if self._has_friends is False: - raise WSError("network", "status", "Page not found") + raise PyLastError("network", "status", "Page not found") return [MockUser(None, None, True, USERNAME_2)]