first implementation

backend/tests/helpers.py

@@ -1,7 +1,51 @@
 import uuid
 
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from models import User, Invite, InviteStatus, ROLE_ADMIN
+from invite_utils import generate_invite_identifier
+
 
 def unique_email(prefix: str = "test") -> str:
     """Generate a unique email for tests sharing the same database."""
     return f"{prefix}-{uuid.uuid4().hex[:8]}@example.com"
 
+
+async def create_invite_for_registration(db: AsyncSession, godfather_email: str) -> str:
+    """
+    Create an invite that can be used for registration.
+    Returns the invite identifier.
+    """
+    # Find godfather
+    result = await db.execute(select(User).where(User.email == godfather_email))
+    godfather = result.scalar_one_or_none()
+    
+    if not godfather:
+        # Create a godfather user (admin can create invites)
+        from auth import get_password_hash
+        from models import Role
+        
+        result = await db.execute(select(Role).where(Role.name == ROLE_ADMIN))
+        admin_role = result.scalar_one_or_none()
+        
+        godfather = User(
+            email=godfather_email,
+            hashed_password=get_password_hash("password123"),
+            roles=[admin_role] if admin_role else [],
+        )
+        db.add(godfather)
+        await db.flush()
+    
+    # Create invite
+    identifier = generate_invite_identifier()
+    invite = Invite(
+        identifier=identifier,
+        godfather_id=godfather.id,
+        status=InviteStatus.READY,
+    )
+    db.add(invite)
+    await db.commit()
+    
+    return identifier
+