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"]