31 lines
709 B
Python
31 lines
709 B
Python
import uuid
|
|
|
|
from app.models.user import User
|
|
from app.services.auth import hash_password
|
|
|
|
|
|
def _make_user(
|
|
user_id: str | uuid.UUID | None = None,
|
|
email: str = "test@example.com",
|
|
plan: str = "free",
|
|
) -> User:
|
|
uid = user_id or str(uuid.uuid4())
|
|
user = User(
|
|
id=str(uid),
|
|
email=email,
|
|
password=hash_password("Test@123456"),
|
|
firstName="Test",
|
|
lastName="User",
|
|
isActive=True,
|
|
emailVerified=True,
|
|
)
|
|
user.plan = plan
|
|
user.max_queries = 50 if plan != "free" else 5
|
|
return user
|
|
|
|
|
|
def _to_uuid(value: str | uuid.UUID) -> uuid.UUID:
|
|
if isinstance(value, uuid.UUID):
|
|
return value
|
|
return uuid.UUID(str(value))
|