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