django-rest-framework-bulk/rest_framework_bulk/tests/test_generics.py

from __future__ import unicode_literals, print_function
import json
from django.core.urlresolvers import reverse
from django.test import TestCase
from django.test.client import RequestFactory
from rest_framework import status

from .simple_app.models import SimpleModel, UniqueTogetherModel
from .simple_app.views import FilteredBulkAPIView, SimpleBulkAPIView


class TestBulkAPIView(TestCase):
    def setUp(self):
        super(TestBulkAPIView, self).setUp()
        self.view = SimpleBulkAPIView.as_view()
        self.request = RequestFactory()

    def test_get(self):
        """
        Test that GET request is successful on bulk view.
        """
        response = self.view(self.request.get(''))

        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_post_single(self):
        """
        Test that POST request with single resource only creates a single resource.
        """
        response = self.view(self.request.post(
            '',
            json.dumps({'contents': 'hello world', 'number': 1}),
            content_type='application/json',
        ))

        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(SimpleModel.objects.count(), 1)
        self.assertEqual(SimpleModel.objects.get().contents, 'hello world')

    def test_post_bulk(self):
        """
        Test that POST request with multiple resources creates all posted resources.
        """
        response = self.view(self.request.post(
            '',
            json.dumps([
                {'contents': 'hello world', 'number': 1},
                {'contents': 'hello mars', 'number': 2},
            ]),
            content_type='application/json',
        ))

        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(SimpleModel.objects.count(), 2)
        self.assertEqual(list(SimpleModel.objects.all().values_list('contents', flat=True)), [
            'hello world',
            'hello mars',
        ])

    def test_put(self):
        """
        Test that PUT request updates all submitted resources.
        """
        obj1 = SimpleModel.objects.create(contents='hello world', number=1)
        obj2 = SimpleModel.objects.create(contents='hello mars', number=2)

        response = self.view(self.request.put(
            '',
            json.dumps([
                {'contents': 'foo', 'number': 3, 'id': obj1.pk},
                {'contents': 'bar', 'number': 4, 'id': obj2.pk},
            ]),
            content_type='application/json',
        ))

        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(SimpleModel.objects.count(), 2)
        self.assertEqual(
            list(SimpleModel.objects.all().values_list('id', 'contents', 'number')),
            [
                (obj1.pk, 'foo', 3),
                (obj2.pk, 'bar', 4),
            ]
        )

    def test_patch(self):
        """
        Test that PATCH request partially updates all submitted resources.
        """
        obj1 = SimpleModel.objects.create(contents='hello world', number=1)
        obj2 = SimpleModel.objects.create(contents='hello mars', number=2)

        response = self.view(self.request.patch(
            '',
            json.dumps([
                {'contents': 'foo', 'id': obj1.pk},
                {'contents': 'bar', 'id': obj2.pk},
            ]),
            content_type='application/json',
        ))

        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(SimpleModel.objects.count(), 2)
        self.assertEqual(
            list(SimpleModel.objects.all().values_list('id', 'contents', 'number')),
            [
                (obj1.pk, 'foo', 1),
                (obj2.pk, 'bar', 2),
            ]
        )

    def test_delete_not_filtered(self):
        """
        Test that DELETE is not allowed when results are not filtered.
        """
        SimpleModel.objects.create(contents='hello world', number=1)
        SimpleModel.objects.create(contents='hello mars', number=10)

        response = self.view(self.request.delete(''))

        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)

    def test_delete_filtered(self):
        """
        Test that DELETE removes all filtered resources.
        """
        SimpleModel.objects.create(contents='hello world', number=1)
        SimpleModel.objects.create(contents='hello mars', number=10)

        response = FilteredBulkAPIView.as_view()(self.request.delete(''))

        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
        self.assertEqual(SimpleModel.objects.count(), 1)
        self.assertEqual(SimpleModel.objects.get().contents, 'hello world')

    def test_options(self):
        """
        Test that OPTIONS request is successful on bulk view.
        """
        response = self.view(self.request.options(''))

        self.assertEqual(response.status_code, status.HTTP_200_OK)


class TestBulkAPIViewSet(TestCase):
    """
    Integration class testing that viewset requests are correctly
    routed via bulk router and that expected status code is returned.
    """

    def setUp(self):
        super(TestBulkAPIViewSet, self).setUp()
        self.url = reverse('api:simple-list')

    def test_get_single(self):
        """
        Test that we are still able to query single resource
        """
        response = self.client.get(self.url)

        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_get(self):
        """
        Test that GET returns 200
        """
        obj = SimpleModel.objects.create(contents='hello world', number=7)

        response = self.client.get(reverse('api:simple-detail', args=[obj.pk]))

        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_post_single(self):
        """
        Test that POST with single resource returns 201
        """
        response = self.client.post(
            self.url,
            data=json.dumps({
                'contents': 'hello world',
                'number': 1,
            }),
            content_type='application/json',
        )

        self.assertEqual(response.status_code, status.HTTP_201_CREATED)

    def test_post_bulk(self):
        """
        Test that POST with multiple resources returns 201
        """
        response = self.client.post(
            self.url,
            data=json.dumps([
                {
                    'contents': 'hello world',
                    'number': 1,
                },
                {
                    'contents': 'hello mars',
                    'number': 2,
                },
            ]),
            content_type='application/json',
        )

        self.assertEqual(response.status_code, status.HTTP_201_CREATED)

    def test_put(self):
        """
        Test that PUT with multiple resources returns 200
        """
        obj1 = SimpleModel.objects.create(contents='hello world', number=7)
        obj2 = SimpleModel.objects.create(contents='hello mars', number=10)

        response = self.client.put(
            self.url,
            data=json.dumps([
                {
                    'contents': 'foo',
                    'number': 1,
                    'id': obj1.pk,
                },
                {
                    'contents': 'bar',
                    'number': 2,
                    'id': obj2.pk,
                },
            ]),
            content_type='application/json',
        )

        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_patch(self):
        """
        Test that PATCH with multiple partial resources returns 200
        """
        obj1 = SimpleModel.objects.create(contents='hello world', number=7)
        obj2 = SimpleModel.objects.create(contents='hello mars', number=10)

        response = self.client.patch(
            self.url,
            data=json.dumps([
                {
                    'contents': 'foo',
                    'id': obj1.pk,
                },
                {
                    'contents': 'bar',
                    'id': obj2.pk,
                },
            ]),
            content_type='application/json',
        )

        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_patch_unique_together(self):
        """
        Test that PATCH with multiple partial resources returns 200
        even on model with unique together columns
        """
        obj1 = UniqueTogetherModel.objects.create(foo=1, bar=2)
        obj2 = UniqueTogetherModel.objects.create(foo=3, bar=4)

        response = self.client.patch(
            reverse('api:unique-together-list'),
            data=json.dumps([
                {'foo': 5, 'id': obj1.pk},
                {'foo': 6, 'id': obj2.pk},
            ]),
            content_type='application/json',
        )

        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_delete(self):
        """
        Test that PATCH with multiple partial resources returns 200
        """
        SimpleModel.objects.create(contents='hello world', number=7)
        SimpleModel.objects.create(contents='hello mars', number=10)

        response = self.client.delete(self.url)

        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)