53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
import json
|
|
|
|
import pytest
|
|
|
|
from src.huesoporro.infra.repos import UserRepo
|
|
from src.huesoporro.models import User
|
|
|
|
|
|
@pytest.fixture
|
|
async def user_repo(s, db, user: User):
|
|
async with db.get_client() as client:
|
|
await client.execute(
|
|
"INSERT INTO users (user, external_auth) VALUES (?, ?)",
|
|
(user.user, json.dumps(user.external_auth)),
|
|
)
|
|
|
|
return UserRepo(s=s)
|
|
|
|
|
|
async def test_get_user(user_repo: UserRepo, user: User):
|
|
db_user = await user_repo.get_by_user(user.user)
|
|
assert db_user == user
|
|
|
|
|
|
async def test_get_user_returns_none(user_repo: UserRepo):
|
|
assert await user_repo.get_by_user("unknown_user") is None
|
|
|
|
|
|
async def test_create_user(user_repo: UserRepo):
|
|
new_user = User(
|
|
user="new_user", external_auth={"twitch": {"token": "twitch_token"}}
|
|
)
|
|
assert await user_repo.create(new_user) == new_user
|
|
|
|
|
|
async def test_update_users_tokens(user_repo: UserRepo, user: User):
|
|
new_tokens = {"twitch": {"token": "new_tokens"}}
|
|
user.external_auth = new_tokens # type: ignore[assignment]
|
|
assert await user_repo.update(user) == user
|
|
|
|
|
|
async def test_update_non_existing_user_raises_value_error(user_repo: UserRepo):
|
|
with pytest.raises(ValueError, match="User unknown_user does not exist"):
|
|
await user_repo.update(
|
|
User(
|
|
user="unknown_user", external_auth={"twitch": {"token": "twitch_token"}}
|
|
)
|
|
)
|
|
|
|
|
|
async def test_delete_user(user_repo: UserRepo, user: User):
|
|
assert await user_repo.delete(user) is None
|
|
assert await user_repo.get_by_user(user.user) is None
|