@@ -5,7 +5,7 @@ from datetime import datetime, timezone
|
||||
import pytest
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
from devplacepy.database import get_table, internal_gateway_key, purge, refresh_snapshot
|
||||
from devplacepy.database import get_table, purge, refresh_snapshot
|
||||
from devplacepy.main import app
|
||||
from devplacepy.services.manager import service_manager
|
||||
from devplacepy.utils import generate_uid
|
||||
@@ -55,4 +55,32 @@ def client():
|
||||
|
||||
@pytest.fixture
|
||||
def auth(client):
|
||||
return {"X-API-KEY": internal_gateway_key()}
|
||||
from devplacepy.database import get_primary_admin_uid, invalidate_admins_cache
|
||||
from devplacepy.utils import clear_user_cache
|
||||
|
||||
users = get_table("users")
|
||||
uid = get_primary_admin_uid()
|
||||
if uid is None:
|
||||
uid = generate_uid()
|
||||
users.insert(
|
||||
{
|
||||
"uid": uid,
|
||||
"username": f"dbapi_admin_{uid[:8]}",
|
||||
"email": f"{uid[:8]}@dbapi.test",
|
||||
"role": "Admin",
|
||||
"api_key": f"dbapikey_{uid}",
|
||||
"xp": 0,
|
||||
"level": 1,
|
||||
"created_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
)
|
||||
invalidate_admins_cache()
|
||||
refresh_snapshot()
|
||||
admin = users.find_one(uid=uid)
|
||||
key = admin.get("api_key")
|
||||
if not key:
|
||||
key = f"dbapikey_{uid}"
|
||||
users.update({"uid": uid, "api_key": key}, ["uid"])
|
||||
clear_user_cache(uid)
|
||||
refresh_snapshot()
|
||||
return {"X-API-KEY": key}
|
||||
|
||||
Reference in New Issue
Block a user