Django CBV - Как протестировать get_context_data с uuid в url?
Я использую UUID в url вместо первичного ключа. Я предполагаю, но не уверен, что это является причиной моей проблемы при тестировании CBVs.
Мой взгляд на профиль пользователя :
class ProfileView(DetailView):
slug_url_kwarg = 'uuid'
slug_field = 'uuid'
model = User
template_name = 'users/profile.html'
context_object_name = 'user_profile'
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context['uuid'] = self.kwargs.get("uuid")
return context
Мой url :
path(
route='profile/<uuid:uuid>',
view=views.ProfileView.as_view(),
name='profile',
),
Я не могу проверить get_context_data, Django говорит мне, что у моего представления нет атрибута "object". Возможно, мне нужно переопределить get_object, но мой поиск ничего не нашел.
Мой тест :
class BaseTest(TestCase):
def setUp(self):
# Set up non-modified objects used by all test methods
self.factory = RequestFactory()
self.user2 = User.objects.create_user(
email='caroline.dupont@free.fr',
password='fhh456GG455t',
status='VALIDATED',
)
return super().setUp()
def profile_view_instance(self, test_user):
request = self.factory.get(reverse('profile', args=(test_user.uuid,)))
request.user = test_user
view = ProfileView()
view.setup(request)
return view
class ProfileViewTestCase(BaseTest):
def test_get_context_data(self):
self.client.force_login(self.user2)
context = self.profile_view_instance(self.user2).get_context_data()
self.assertIn('uuid', context)
Ошибка :
ERROR: test_get_context_data (tests.appusers.test_views.ProfileViewTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
File "D:\Developpement\projet13\tests\appusers\test_views.py", line 75, in test_get_context_data
context = self.profile_view_instance(self.user2).get_context_data()
File "D:\Developpement\projet13\users\views.py", line 66, in get_context_data
context = super().get_context_data(**kwargs)
File "D:\Developpement\projet13\venvp13\lib\site-packages\django\views\generic\detail.py", line 94, in get_context_data
if self.object:
AttributeError: 'ProfileView' object has no attribute 'object'
Недостаточно profile_view_instance
: DetailView
имеет некоторую шаблонную логику в методе .get(…)
, которая должна быть выполнена до того, как сработает .get_context_data(…)
. Также не очень хорошая идея реализовывать эту логику в тестовой функции, поскольку тогда вы дублируете логику, и будет мучительно обновлять ее каждый раз, когда вы обновляете версию Django.
Django создал клиент, который можно использовать для запуска представления. Вы можете определить тест с помощью:
class ProfileViewTestCase(BaseTest):
def test_get_context_data(self):
self.client.force_login(self.user2)
response = self.client.get(f'/profile/{self.user2.uuid}')
context = response.context
self.assertIn('uuid', context)