Auth tests

getdozer · Apr 12, 2023 · 0bbe8b1 · 0bbe8b1
1 parent 88789f3
commit 0bbe8b1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/tests/helper.py b/tests/helper.py
@@ -3,8 +3,10 @@
 
 from time import sleep
 
+from pydozer.auth import AuthClient
 from pydozer.api import ApiClient
 from pydozer.ingest import IngestClient
+
 import pytest
 
 
@@ -32,7 +34,13 @@ def dozer_server():
 
 @pytest.fixture
 def api_client() -> ApiClient:
-    return ApiClient("users", url=DOZER_API_URL)
+    master_token = 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJhdWQiOiJjYWNoZV91c2VyIiwic3ViIjoiYXBpQGRvemVyLmNvbSIsImV4cCI6MTY4MTI4NzcwNDAyNywiYWNjZXNzIjoiQWxsIn0.aD5L6XURIr3hrUp0LzfUQWKHairjK6DDkMlzvnVvzHA'
+
+    client = AuthClient(token=master_token, url=DOZER_API_URL)
+
+    restricted_token = client.get_auth_token()
+
+    return ApiClient("users", url=DOZER_API_URL, token=restricted_token)
 
 
 @pytest.fixture

diff --git a/tests/test_dozer.py b/tests/test_dozer.py
@@ -87,4 +87,5 @@ def test_ingest_query(ingestion_client: IngestClient, api_client: ApiClient):
 
     res = api_client.query({'$limit': 1})
     assert res is not None
-    assert len(res.records) >= 1
+    assert len(res.records) >= 1
+