Add tests

tests/sources/helpers/rest_client/test_client.py

@@ -1,6 +1,7 @@
 import os
 import pytest
 from typing import Any, cast
+from requests.auth import AuthBase
 from dlt.common.typing import TSecretStrValue
 from dlt.sources.helpers.requests import Response, Request
 from dlt.sources.helpers.rest_client import RESTClient
@@ -57,7 +58,6 @@ def test_page_context(self, rest_client: RESTClient) -> None:
         for page in rest_client.paginate(
             "/posts",
             paginator=JSONResponsePaginator(next_url_path="next_page"),
-            auth=AuthConfigBase(),
         ):
             # response that produced data
             assert isinstance(page.response, Response)
@@ -183,3 +183,45 @@ def test_oauth_jwt_auth_success(self, rest_client: RESTClient):
         )
 
         assert_pagination(list(pages_iter))
+
+    def test_custom_auth_success(self, rest_client: RESTClient):
+        class CustomAuthConfigBase(AuthConfigBase):
+            def __init__(self, token: str):
+                self.token = token
+
+            def __call__(self, request: Request) -> Request:
+                request.headers["Authorization"] = f"Bearer {self.token}"
+                return request
+
+        class CustomAuthAuthBase(AuthBase):
+            def __init__(self, token: str):
+                self.token = token
+
+            def __call__(self, request: Request) -> Request:
+                request.headers["Authorization"] = f"Bearer {self.token}"
+                return request
+
+
+        auth_list = [
+            CustomAuthConfigBase("test-token"),
+            CustomAuthAuthBase("test-token"),
+        ]
+
+        for auth in auth_list:
+            response = rest_client.get(
+                "/protected/posts/bearer-token",
+                auth=auth,
+            )
+
+            assert response.status_code == 200
+            assert response.json()["data"][0] == {"id": 0, "title": "Post 0"}
+
+            pages_iter = rest_client.paginate(
+                "/protected/posts/bearer-token",
+                auth=auth,
+            )
+
+            pages_list = list(pages_iter)
+            assert_pagination(pages_list)
+
+            assert pages_list[0].response.request.headers["Authorization"] == "Bearer test-token"