Steps and code to set up Django Rest Framework (DRF) test cases with database mocking.

Set up Django and DRF

Install Django and DRF:


pip install django djangorestframework

Create a Django project and app:


django-admin startproject projectname

cd projectname

python manage.py startapp appname

Define Models, Serializers, and Views

models.py (appname/models.py):


from django.db import models



class Item(models.Model):

    name = models.CharField(max_length=100)

    description = models.TextField()

serializers.py (appname/serializers.py):


from rest_framework import serializers

from .models import Item



class ItemSerializer(serializers.ModelSerializer):

    class Meta:

        model = Item

        fields = '__all__'

views.py (appname/views.py):


from rest_framework import viewsets

from .models import Item

from .serializers import ItemSerializer



class ItemViewSet(viewsets.ModelViewSet):

    queryset = Item.objects.all()

    serializer_class = ItemSerializer

urls.py (appname/urls.py):


from django.urls import path, include

from rest_framework.routers import DefaultRouter

from .views import ItemViewSet



router = DefaultRouter()

router.register(r'items', ItemViewSet)



urlpatterns = [

    path('', include(router.urls)),

]

projectname/urls.py:


from django.contrib import admin

from django.urls import path, include



urlpatterns = [

    path('admin/', admin.site.urls),

    path('api/', include('appname.urls')),

]

Migrate Database and Create Superuser


python manage.py makemigrations appname

python manage.py migrate

python manage.py createsuperuser

python manage.py runserver

Write Test Cases

tests.py (appname/tests.py):


from django.urls import reverse

from rest_framework import status

from rest_framework.test import APITestCase

from .models import Item

from .serializers import ItemSerializer



class ItemTests(APITestCase):



    def setUp(self):

        self.item1 = Item.objects.create(name='Item 1', description='Description 1')

        self.item2 = Item.objects.create(name='Item 2', description='Description 2')



    def test_get_items(self):

        url = reverse('item-list')

        response = self.client.get(url, format='json')

        items = Item.objects.all()

        serializer = ItemSerializer(items, many=True)

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

        self.assertEqual(response.data, serializer.data)



    def test_create_item(self):

        url = reverse('item-list')

        data = {'name': 'Item 3', 'description': 'Description 3'}

        response = self.client.post(url, data, format='json')

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

        self.assertEqual(Item.objects.count(), 3)

        self.assertEqual(Item.objects.get(id=3).name, 'Item 3')



    def test_update_item(self):

        url = reverse('item-detail', kwargs={'pk': self.item1.id})

        data = {'name': 'Updated Item 1', 'description': 'Updated Description 1'}

        response = self.client.put(url, data, format='json')

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

        self.item1.refresh_from_db()

        self.assertEqual(self.item1.name, 'Updated Item 1')



    def test_delete_item(self):

        url = reverse('item-detail', kwargs={'pk': self.item2.id})

        response = self.client.delete(url, format='json')

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

        self.assertEqual(Item.objects.count(), 1)

Run Tests


python manage.py test

This setup provides a basic Django project with DRF and test cases for CRUD operations using the database. The test cases mock the database operations, ensuring isolation and consistency during testing.

Now diving into some more feature tests with Mock, patch etc.

Here are steps and code to write Django Rest Framework (DRF) test cases using mocking and faking features for scenarios like credit card processing.

Set up Django and DRF