Authentication (#17)

Author: joostlek

src/async_python_youtube/youtube.py

@@ -35,6 +35,7 @@ class YouTube:
     request_timeout: int = 10
     api_host: str = "youtube.googleapis.com"
     _close_session: bool = False
+    _access_token: str | None = None
 
     async def _request(
         self,
@@ -77,6 +78,9 @@ async def _request(
             "Accept": "application/json, text/plain, */*",
         }
 
+        if self._access_token:
+            headers["Authorization"] = f"Bearer {self._access_token}"
+
         if self.session is None:
             self.session = ClientSession()
             self._close_session = True
@@ -110,6 +114,10 @@ async def _request(
 
         return cast(dict[str, Any], await response.json())
 
+    def authenticate(self, access_token: str) -> None:
+        """Apply authentication to the requests."""
+        self._access_token = access_token
+
     async def get_video(self, video_id: str) -> YouTubeVideo | None:
         """Get a single video."""
         return await first(self.get_videos([video_id]))

tests/test_auth.py

@@ -0,0 +1,37 @@
+"""Tests for the YouTube client."""
+
+import aiohttp
+from aiohttp.web_request import BaseRequest
+from aresponses import Response, ResponsesMockServer
+
+from async_python_youtube.youtube import YouTube
+
+from . import load_fixture
+
+YOUTUBE_URL = "youtube.googleapis.com"
+
+
+async def test_authentication(aresponses: ResponsesMockServer) -> None:
+    """Test request will be sending bearer token after authentication."""
+
+    async def response_handler(req: BaseRequest) -> Response:
+        """Response handler for this test."""
+        assert req.headers.get("Authorization") == "Bearer abc"
+        return aresponses.Response(
+            status=200,
+            headers={"Content-Type": "application/json"},
+            text=load_fixture("video_response_snippet.json"),
+        )
+
+    aresponses.add(
+        YOUTUBE_URL,
+        "/youtube/v3/videos",
+        "GET",
+        response_handler,
+    )
+
+    async with aiohttp.ClientSession() as session:
+        youtube = YouTube(session=session)
+        youtube.authenticate("abc")
+        assert await youtube.get_video(video_id="Ks-_Mh1QhMc")
+        await youtube.close()