def test_get_subscription_formset(self):
"""should return a formset with forms for a user."""
user = mommy.make('accounts.User')
group1 = mommy.make('groups.Group')
group2 = mommy.make('groups.Group')
group3 = mommy.make('groups.Group')
user.add_to_group(group1.pk)
user.add_to_group(group2.pk)
formset = get_subscription_formset(user)
groups_with_forms = []
for form in formset:
groups_with_forms.append(form.instance.group)
self.assertEqual(form.group_name, form.instance.group.group.name)
self.assertIn(group1, groups_with_forms)
self.assertIn(group2, groups_with_forms)
self.assertNotIn(group3, groups_with_forms)
def get_context_data(self, **kwargs):
"""Set the active nav item to the current object."""
context = super(UserDetailView, self).get_context_data(**kwargs)
context['nav_active_item'] = self.object
context['dd_active_item'] = 'Profile'
context['show_banned_warning'] = self.show_banned_warning()
context['subscription_formset'] = get_subscription_formset(
user=self.request.user)
context['groups_joined'] = Group.objects.filter(
group__user__id=self.object.pk).select_related('image', 'group')
context['subscribed_ids'] = self.request.user.groups_joined.values_list(
'pk', flat=True)
# pylint: disable=line-too-long
context['show_message_button'] = self.request.user.can_direct_message_user(self.object)
context['profile_is_self'] = self.request.user.pk == self.object.pk
context['mod_update_form'] = ModNotificationUpdateForm(
instance=self.request.user)
return context
def get_context_data(self, **kwargs):
"""Set the active nav item to the current object."""
context = super(UserDetailView, self).get_context_data(**kwargs)
context['nav_active_item'] = self.object
context['dd_active_item'] = 'Profile'
context['show_banned_warning'] = self.show_banned_warning()
context['subscription_formset'] = get_subscription_formset(
user=self.request.user)
context['groups_joined'] = Group.objects.filter(
group__user__id=self.object.pk).select_related('image', 'group')
context[
'subscribed_ids'] = self.request.user.groups_joined.values_list(
'pk', flat=True)
# pylint: disable=line-too-long
context[
'show_message_button'] = self.request.user.can_direct_message_user(
self.object)
context['profile_is_self'] = self.request.user.pk == self.object.pk
context['mod_update_form'] = ModNotificationUpdateForm(
instance=self.request.user)
return context
