def test_get_queryset_non_public(self):
viewset = UserProfileViewSet()
viewset.request = Mock()
viewset.request.privacy_level = MOZILLIANS
with patch('mozillians.users.api.v2.UserProfile') as userprofile_mock:
viewset.get_queryset()
ok_(userprofile_mock.objects.complete.called)
userprofile_mock.objects.complete().privacy_level.assert_called_with(MOZILLIANS)
def test_retrieve_base(self):
viewset = UserProfileViewSet()
viewset.request = Mock()
viewset.request.privacy_level = MOZILLIANS
user = UserFactory.create()
with patch('mozillians.users.api.v2.UserProfileDetailedSerializer') as serializer_mock:
viewset.retrieve(None, user.userprofile.id)
serializer_mock.assert_called_with(user.userprofile, context=ANY)
def test_get_queryset_public(self):
viewset = UserProfileViewSet()
viewset.request = Mock()
viewset.request.privacy_level = PUBLIC
with patch("mozillians.users.api.v2.UserProfile") as userprofile_mock:
viewset.get_queryset()
ok_(userprofile_mock.objects.complete.called)
ok_(userprofile_mock.objects.complete().public.called)
userprofile_mock.objects.complete().public(
).privacy_level.assert_called_with(PUBLIC)
def test_retrieve_non_existent(self):
viewset = UserProfileViewSet()
viewset.request = Mock()
viewset.request.privacy_level = MOZILLIANS
self.assertRaises(Http404, viewset.retrieve, viewset.request, -1)
