add toke, authorized_clients fixtures and test_get_all_posts

atrox3d · atrox3d · commit e25d44e16979 · 2024-08-28T08:19:20.000+02:00
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -6,5 +6,7 @@
                             new_user, 
                             user_create, 
                             user_create_json, 
-                            user_login_json
+                            user_login_json,
+                            token,
+                            authorized_client
 )
diff --git a/tests/fixtures/user.py b/tests/fixtures/user.py
@@ -1,5 +1,9 @@
+from typing import Generator
+from fastapi.testclient import TestClient
 import pytest
 import schemas, schemas.user
+from app import oauth2
+from .database import client
 
 @pytest.fixture
 def user_create() -> schemas.user.UserCreate:
@@ -35,3 +39,15 @@ def new_user(client, user_create, user_create_json) -> schemas.user.UserDb:
         password=user_create.password
     )
     return userdb
+
+@pytest.fixture
+def token(new_user) -> str:
+    return oauth2.create_access_token({'user_id': new_user.id})
+
+@pytest.fixture
+def authorized_client(client, token) -> Generator[TestClient, None, None]:
+     client.headers = {
+         **client.headers,
+         'Authorization': f'Bearer {token}'
+     }
+     yield client
diff --git a/tests/test_posts.py b/tests/test_posts.py
@@ -0,0 +1,4 @@
+def test_get_all_posts(authorized_client):
+    response = authorized_client.get('/posts')
+    assert response.status_code == 200
+    print(response.json())
