def test_get_queryset_calls_helper(self, mock_helper): # Given a mock that returns a sentinel mock_helper.return_value = sentinel.activities # and a view user view = UserView() view.user = sentinel.user view.request = Mock(user=sentinel.other) # When getting the queryset activities = view.get_queryset() # Then the sentinel is returned, via the helper assert activities == sentinel.activities mock_helper.assert_called_once_with(sentinel.user, sentinel.other)
def test_get_context_data_populates_correctly(self, super_mock, mock_helper): # Given a mock that returns a sentinel summary mock_helper.return_value = sentinel.summary super_mock.return_value = dict(super=sentinel.super) # and a mock queryset that returns a sentinel view = UserView() view.user = sentinel.user view.get_queryset = Mock(return_value=sentinel.activity_query_set) # When getting the context data context = view.get_context_data() # Then the context contains the sentinel summary, via the mock assert context['summaries'] == sentinel.summary assert context['view_user'] == sentinel.user assert context['super'] == sentinel.super mock_helper.assert_called_once_with(sentinel.activity_query_set)
def test_get_adds_user_to_view(self, mock_get_user_model, get_mock): # Given a mock that returns a sentinel user users_mock = Mock() users_mock.objects.get.return_value = sentinel.user mock_get_user_model.return_value = users_mock get_mock.return_value = sentinel.response view = UserView() view.kwargs = dict(username=sentinel.username) # When getting the view response = view.get(sentinel.request) # Then the sentinel user is returned, via the mock assert view.user == sentinel.user users_mock.objects.get.assert_called_with(username=sentinel.username) assert response == sentinel.response get_mock.assert_called_once_with(sentinel.request)
def test_includes_upload_form_mixin(self): # Expect the view to have the upload form in the hierarchy view = UserView() assert isinstance(view, UploadFormMixin)