34 lines
927 B
Python
34 lines
927 B
Python
import pytest
|
|
from rest_framework.test import APIRequestFactory
|
|
|
|
from lms.users.api.views import UserViewSet
|
|
from lms.users.models import User
|
|
|
|
|
|
class TestUserViewSet:
|
|
@pytest.fixture
|
|
def api_rf(self) -> APIRequestFactory:
|
|
return APIRequestFactory()
|
|
|
|
def test_get_queryset(self, user: User, api_rf: APIRequestFactory):
|
|
view = UserViewSet()
|
|
request = api_rf.get("/fake-url/")
|
|
request.user = user
|
|
|
|
view.request = request
|
|
|
|
assert user in view.get_queryset()
|
|
|
|
def test_me(self, user: User, api_rf: APIRequestFactory):
|
|
view = UserViewSet()
|
|
request = api_rf.get("/fake-url/")
|
|
request.user = user
|
|
|
|
view.request = request
|
|
|
|
response = view.me(request) # type: ignore[call-arg, arg-type, misc]
|
|
|
|
assert response.data == {
|
|
"url": f"http://testserver/api/users/{user.pk}/",
|
|
"name": user.name,
|
|
}
|