Improve error handling (#49)

Author: joostlek

src/youtubeaio/youtube.py

@@ -6,7 +6,7 @@
 from typing import Any, TypeVar
 
 import async_timeout
-from aiohttp import ClientResponse, ClientSession
+from aiohttp import ClientError, ClientResponse, ClientSession
 
 from youtubeaio.helper import (
     build_url,
@@ -16,6 +16,7 @@
 from youtubeaio.types import (
     AuthScope,
     MissingScopeError,
+    UnauthorizedError,
     YouTubeAPIError,
     YouTubeBackendError,
     YouTubeResourceNotFoundError,
@@ -76,8 +77,13 @@ async def _check_request_return(self, response: ClientResponse) -> ClientRespons
             )
         if response.status == 404:
             raise YouTubeResourceNotFoundError
+        if response.status == 401:
+            raise UnauthorizedError
         if 400 <= response.status < 500:
-            raise YouTubeAPIError
+            try:
+                response.raise_for_status()
+            except ClientError as exc:
+                raise YouTubeAPIError from exc
         return response
 
     async def _api_get_request(

tests/test_youtube.py

@@ -7,6 +7,7 @@
 from aresponses import Response, ResponsesMockServer
 
 from youtubeaio.types import (
+    UnauthorizedError,
     YouTubeAPIError,
     YouTubeBackendError,
     YouTubeResourceNotFoundError,
@@ -164,3 +165,25 @@ async def test_bad_request(
         with pytest.raises(YouTubeAPIError):
             await youtube.get_video(video_id="Ks-_Mh1QhMc")
         await youtube.close()
+
+
+async def test_unauthorized(
+    aresponses: ResponsesMockServer,
+) -> None:
+    """Test handling being unauthorized."""
+    aresponses.add(
+        YOUTUBE_URL,
+        "/youtube/v3/videos",
+        "GET",
+        aresponses.Response(
+            status=401,
+            headers={"Content-Type": "application/json"},
+            text='{"message":"Something went wrong"}',
+        ),
+    )
+
+    async with aiohttp.ClientSession() as session:
+        youtube = YouTube(session=session)
+        with pytest.raises(UnauthorizedError):
+            await youtube.get_video(video_id="Ks-_Mh1QhMc")
+        await youtube.close()