87 lines
3.1 KiB
Python
87 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
"""Seed the e2e test database with test fixtures.
|
|
|
|
Outputs JSON with the created test data (magic link tokens, usernames, etc.)
|
|
so the JS tests can use them.
|
|
|
|
Requires OIDC_OP_SQLITE_PATH env var pointing to the app's SQLite DB.
|
|
"""
|
|
|
|
import asyncio
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
import aiosqlite
|
|
|
|
from porchlight.authn.password import PasswordService
|
|
from porchlight.invite.service import MagicLinkService
|
|
from porchlight.models import PasswordCredential, User
|
|
from porchlight.store.sqlite.repositories import (
|
|
SQLiteCredentialRepository,
|
|
SQLiteMagicLinkRepository,
|
|
SQLiteUserRepository,
|
|
)
|
|
|
|
|
|
async def seed() -> None:
|
|
db_path = os.environ.get("OIDC_OP_SQLITE_PATH")
|
|
if not db_path:
|
|
print("OIDC_OP_SQLITE_PATH not set", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
db = await aiosqlite.connect(db_path)
|
|
db.row_factory = aiosqlite.Row
|
|
|
|
user_repo = SQLiteUserRepository(db)
|
|
cred_repo = SQLiteCredentialRepository(db)
|
|
magic_link_repo = SQLiteMagicLinkRepository(db)
|
|
password_service = PasswordService()
|
|
magic_link_service = MagicLinkService(repo=magic_link_repo)
|
|
|
|
result = {}
|
|
|
|
# 1. Create a magic link for registration test
|
|
link = await magic_link_service.create(username="newuser")
|
|
result["register_token"] = link.token
|
|
result["register_username"] = "newuser"
|
|
|
|
# 2. Create a user with a password for login test
|
|
user = User(userid="test-user-01", username="testuser", groups=["users"])
|
|
await user_repo.create(user)
|
|
password_hash = password_service.hash("testpassword123")
|
|
await cred_repo.create_password(PasswordCredential(user_id=user.userid, password_hash=password_hash))
|
|
result["login_username"] = "testuser"
|
|
result["login_password"] = "testpassword123"
|
|
|
|
# 3. Create a separate user for credentials management test
|
|
cred_user = User(userid="test-user-02", username="creduser", groups=["users"])
|
|
await user_repo.create(cred_user)
|
|
cred_password_hash = password_service.hash("credpassword123")
|
|
await cred_repo.create_password(PasswordCredential(user_id=cred_user.userid, password_hash=cred_password_hash))
|
|
result["cred_username"] = "creduser"
|
|
result["cred_password"] = "credpassword123"
|
|
|
|
# 5. Create a user with password for WebAuthn registration tests
|
|
# (login with password first, then register a passkey)
|
|
webauthn_user = User(userid="test-user-03", username="webauthnuser", groups=["users"])
|
|
await user_repo.create(webauthn_user)
|
|
webauthn_password_hash = password_service.hash("webauthnpass123")
|
|
await cred_repo.create_password(
|
|
PasswordCredential(user_id=webauthn_user.userid, password_hash=webauthn_password_hash)
|
|
)
|
|
result["webauthn_username"] = "webauthnuser"
|
|
result["webauthn_password"] = "webauthnpass123"
|
|
result["webauthn_userid"] = "test-user-03"
|
|
|
|
# 4. Create an expired/used magic link for negative test
|
|
expired_link = await magic_link_service.create(username="expired")
|
|
await magic_link_service.mark_used(expired_link.token)
|
|
result["used_token"] = expired_link.token
|
|
|
|
await db.commit()
|
|
await db.close()
|
|
print(json.dumps(result))
|
|
|
|
|
|
asyncio.run(seed())
|