36 lines
781 B
Python
36 lines
781 B
Python
import sys
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
sys.path.append("..")
|
|
|
|
from src.main import app, get_db
|
|
from tests.config.database import override_get_db
|
|
|
|
app.dependency_overrides[get_db] = override_get_db
|
|
|
|
client = TestClient(app)
|
|
|
|
username = "user1"
|
|
password = "password"
|
|
avatar_index = 1
|
|
|
|
|
|
def get_headers(token=None):
|
|
if token:
|
|
return {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
|
|
else:
|
|
return {"Content-Type": "application/json"}
|
|
|
|
|
|
async def register_user():
|
|
response = client.post(
|
|
"/register",
|
|
headers=get_headers(),
|
|
json={"username": username, "password": password, "avatar_index": avatar_index},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
|
|
return response.json()["access_token"]
|