mirror of
https://github.com/ae-utbm/sith.git
synced 2026-03-13 15:15:03 +00:00
feat: API route to get user memberships
This commit is contained in:
28
club/api.py
28
club/api.py
@@ -6,9 +6,15 @@ from ninja_extra.pagination import PageNumberPaginationExtra
|
||||
from ninja_extra.schemas import PaginatedResponseSchema
|
||||
|
||||
from api.auth import ApiKeyAuth
|
||||
from api.permissions import CanAccessLookup, HasPerm
|
||||
from api.permissions import CanAccessLookup, CanView, HasPerm
|
||||
from club.models import Club, Membership
|
||||
from club.schemas import ClubSchema, ClubSearchFilterSchema, SimpleClubSchema
|
||||
from club.schemas import (
|
||||
ClubSchema,
|
||||
ClubSearchFilterSchema,
|
||||
SimpleClubSchema,
|
||||
UserMembershipSchema,
|
||||
)
|
||||
from core.models import User
|
||||
|
||||
|
||||
@api_controller("/club")
|
||||
@@ -38,3 +44,21 @@ class ClubController(ControllerBase):
|
||||
return self.get_object_or_exception(
|
||||
Club.objects.prefetch_related(prefetch), id=club_id
|
||||
)
|
||||
|
||||
|
||||
@api_controller("/user/{int:user_id}/club")
|
||||
class UserClubController(ControllerBase):
|
||||
@route.get(
|
||||
"",
|
||||
response=list[UserMembershipSchema],
|
||||
auth=[ApiKeyAuth(), SessionAuth()],
|
||||
permissions=[CanView],
|
||||
url_name="fetch_user_clubs",
|
||||
)
|
||||
def search_club(self, user_id: int):
|
||||
user = self.get_object_or_exception(User, id=user_id)
|
||||
return (
|
||||
Membership.objects.ongoing()
|
||||
.filter(user=user)
|
||||
.select_related("club", "user")
|
||||
)
|
||||
|
||||
@@ -40,6 +40,8 @@ class ClubProfileSchema(ModelSchema):
|
||||
|
||||
|
||||
class ClubMemberSchema(ModelSchema):
|
||||
"""A schema to represent all memberships in a club."""
|
||||
|
||||
class Meta:
|
||||
model = Membership
|
||||
fields = ["start_date", "end_date", "role", "description"]
|
||||
@@ -53,3 +55,13 @@ class ClubSchema(ModelSchema):
|
||||
fields = ["id", "name", "logo", "is_active", "short_description", "address"]
|
||||
|
||||
members: list[ClubMemberSchema]
|
||||
|
||||
|
||||
class UserMembershipSchema(ModelSchema):
|
||||
"""A schema to represent the active club memberships of a user."""
|
||||
|
||||
class Meta:
|
||||
model = Membership
|
||||
fields = ["id", "start_date", "role", "description"]
|
||||
|
||||
club: SimpleClubSchema
|
||||
|
||||
50
club/tests/test_user_club_controller.py
Normal file
50
club/tests/test_user_club_controller.py
Normal file
@@ -0,0 +1,50 @@
|
||||
from datetime import timedelta
|
||||
|
||||
from django.test import TestCase
|
||||
from django.urls import reverse
|
||||
from django.utils.timezone import localdate
|
||||
from model_bakery import baker
|
||||
from model_bakery.recipe import Recipe
|
||||
|
||||
from club.models import Club, Membership
|
||||
from club.schemas import UserMembershipSchema
|
||||
from core.baker_recipes import subscriber_user
|
||||
from core.models import Page
|
||||
|
||||
|
||||
class TestFetchClub(TestCase):
|
||||
@classmethod
|
||||
def setUpTestData(cls):
|
||||
cls.user = subscriber_user.make()
|
||||
pages = baker.make(Page, _quantity=3, _bulk_create=True)
|
||||
clubs = baker.make(Club, page=iter(pages), _quantity=3, _bulk_create=True)
|
||||
recipe = Recipe(
|
||||
Membership, user=cls.user, start_date=localdate() - timedelta(days=2)
|
||||
)
|
||||
cls.members = Membership.objects.bulk_create(
|
||||
[
|
||||
recipe.prepare(club=clubs[0]),
|
||||
recipe.prepare(club=clubs[1], end_date=localdate() - timedelta(days=1)),
|
||||
recipe.prepare(club=clubs[1]),
|
||||
]
|
||||
)
|
||||
|
||||
def test_fetch_memberships(self):
|
||||
self.client.force_login(subscriber_user.make())
|
||||
res = self.client.get(
|
||||
reverse("api:fetch_user_clubs", kwargs={"user_id": self.user.id})
|
||||
)
|
||||
assert res.status_code == 200
|
||||
assert [UserMembershipSchema.model_validate(m) for m in res.json()] == [
|
||||
UserMembershipSchema.from_orm(m) for m in (self.members[0], self.members[2])
|
||||
]
|
||||
|
||||
def test_fetch_club_nb_queries(self):
|
||||
self.client.force_login(subscriber_user.make())
|
||||
with self.assertNumQueries(6):
|
||||
# - 5 queries for authentication
|
||||
# - 1 query for the actual data
|
||||
res = self.client.get(
|
||||
reverse("api:fetch_user_clubs", kwargs={"user_id": self.user.id})
|
||||
)
|
||||
assert res.status_code == 200
|
||||
Reference in New Issue
Block a user