mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-05-24 01:12:54 -04:00
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from tests.utils import api_routes
|
|
from tests.utils.factories import user_registration_factory
|
|
from tests.utils.fixture_schemas import TestUser
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def invite(api_client: TestClient, unique_user: TestUser) -> None:
|
|
# Test Creation
|
|
r = api_client.post(api_routes.groups_invitations, json={"uses": 2}, headers=unique_user.token)
|
|
assert r.status_code == 201
|
|
invitation = r.json()
|
|
return invitation["token"]
|
|
|
|
|
|
def test_get_all_invitation(api_client: TestClient, unique_user: TestUser, invite: str) -> None:
|
|
# Get All Invites
|
|
r = api_client.get(api_routes.groups_invitations, headers=unique_user.token)
|
|
|
|
assert r.status_code == 200
|
|
|
|
items = r.json()
|
|
|
|
assert len(items) == 1
|
|
|
|
for item in items:
|
|
assert item["groupId"] == unique_user.group_id
|
|
assert item["token"] == invite
|
|
|
|
|
|
def register_user(api_client, invite):
|
|
# Test User can Join Group
|
|
registration = user_registration_factory()
|
|
registration.group = ""
|
|
registration.group_token = invite
|
|
|
|
response = api_client.post(api_routes.users_register, json=registration.model_dump(by_alias=True))
|
|
return registration, response
|
|
|
|
|
|
def test_group_invitation_link(api_client: TestClient, unique_user: TestUser, invite: str):
|
|
registration, r = register_user(api_client, invite)
|
|
assert r.status_code == 201
|
|
|
|
# Login as new User
|
|
form_data = {"username": registration.email, "password": registration.password}
|
|
|
|
r = api_client.post(api_routes.auth_token, data=form_data)
|
|
assert r.status_code == 200
|
|
token = r.json().get("access_token")
|
|
assert token is not None
|
|
|
|
# Check user Group is Same
|
|
r = api_client.get(api_routes.users_self, headers={"Authorization": f"Bearer {token}"})
|
|
|
|
assert r.status_code == 200
|
|
assert r.json()["groupId"] == unique_user.group_id
|
|
|
|
|
|
def test_group_invitation_delete_after_uses(api_client: TestClient, invite: str) -> None:
|
|
# Register First User
|
|
_, r = register_user(api_client, invite)
|
|
assert r.status_code == 201
|
|
|
|
# Register Second User
|
|
_, r = register_user(api_client, invite)
|
|
assert r.status_code == 201
|
|
|
|
# Check Group Invitation is Deleted
|
|
_, r = register_user(api_client, invite)
|
|
assert r.status_code == 400
|