How to Test Django, Django Rest-Framework, and Djoser

settings/      __init__.py      base.py      production.py      development.py      testing.py

__init__.pybase.pyproduction.pydevelopment.pytesting.py

BASE_DIRSECRET_KEY # read from environment variableINSTALLED_APPSMIDDLEWAREROOT_URLCONFTEMPLATESWSGI_APPLICATIONAUTH_PASSWORD_VALIDATORSLANGUAGE_CODETIME_ZONEUSE_I18NUSE_TZSTATIC_URLDEFAULT_AUTO_FIELDAUTH_USER_MODEL# emial related parameter are read from env in my caseEMAIL_BACKENDEMAIL_HOSTEMAIL_PORTEMAIL_HOST_USEREMAIL_HOST_PASSWORDEMAIL_USE_TLSREST_FRAMEWORK # depends on your choice of drf settingsDJOSER # common setting only we will extend it in other files

from .base import *DEBUG = FalseALLOWED_HOSTS = ['yoursite.com'] # up to you choice what to allowDATABASES # i use different db for prodDJOSER.update({....}) # i use different conf for prod 

from .base import *DEBUG = TrueALLOWED_HOSTS = ['*'] # up to you choice what to allowDATABASES # i use different db for devDJOSER.update({....}) # i use different conf for dev

from .base import *DEBUG = TrueALLOWED_HOSTS = ['*'] # up to you choice what to allowEMAIL_BACKEND = 'django.core.mail.backends.console.EmailBackend'DATABASES # i use different db for testDJOSER.update({....}) # i use different conf for test

import osfrom . import baseenviroment = os.environ.get('ENVIROMENT', 'development')if enviroment == 'production':    from .production import *elif enviroment == 'testing':    from .testing import *else:    from .development import *

from .base import *DEBUG = TrueALLOWED_HOSTS = ['*']EMAIL_BACKEND = 'django.core.mail.backends.console.EmailBackend'DATABASES = {    # 'default': {    #     'ENGINE': 'django.db.backends.sqlite3',    #     'NAME': os.path.join(BASE_DIR, 'db.sqlite3'),    # }    # for postgresql: i use postgres     'default': {        'ENGINE': 'django.db.backends.postgresql',        'NAME': 'djoser',        'USER': 'djoser',        'PASSWORD': 'djoser',        'HOST': 'localhost',        'PORT': '5432',    }    # for mysql:    # 'default': {    #     'ENGINE': 'django.db.backends.mysql',    #     'NAME': 'djoser',    #     'USER': 'djoser',    #     'PASSWORD': 'djoser',    #     'HOST': 'localhost',    #     'PORT': '3306',    # }}DJOSER_EMAIL = {    'activation': 'your-authentication-app.email.ActivationEmail'}if 'EMAIL' in DJOSER:    DJOSER['EMAIL'].update(DJOSER_EMAIL)else:    DJOSER['EMAIL'] = DJOSER_EMAIL

DJOSER_EMAIL = {    'activation': 'your-authentication-app.email.ActivationEmail'}

from django.contrib.auth.tokens import default_token_generator# djoser importsfrom templated_mail.mail import BaseEmailMessagefrom djoser import utilsfrom djoser.conf import settingsEMAILS = {}class ActivationEmail(BaseEmailMessage):    """Email Activation Token Generator    """    template_name = "email/activation.html"    def get_context_data(self):        # ActivationEmail can be deleted        context = super().get_context_data()        user = context.get("user")        context["uid"] = utils.encode_uid(user.pk)        context["token"] = default_token_generator.make_token(user)        context["url"] = settings.ACTIVATION_URL.format(**context)        uid, token = context['uid'], context['token']        # here we store all the requested tokens in a dictionary for later use        EMAILS[user.email] = {'uid': uid, 'token': token}        return context

EMAIL = {}...EMAILS[user.email] = {'uid': uid, 'token': token}

from django.urls import reversefrom django.contrib.auth import get_user_modelfrom rest_framework import statusfrom rest_framework.test import APITestCaseUser = get_user_model()class UserViewSetTest(APITestCase):    def setUp(self):        """        Set up method which is used to initialize before any test run.        """        self.user_info = self.generate_user_info()    def generate_user_info(self):        """Generate user data for new user.        Returns:            Dict: dictionary of the test user data.        """        return {            "first_name": "fake.first_name()",            "last_name": "fake.last_name()",            "username": "fake.user_name()",            "password": "fake.password()",        }    def test_create_user(self):        """        Test for creating users using API.        """        url = reverse("user-list")        response = self.client.post(            url,            self.user_info,        )        self.assertEqual(response.status_code, status.HTTP_201_CREATED)        user = User.objects.get(id=response.data['id'])        self.assertEqual(user.email, self.user_info["email"])        self.assertEqual(user.username, self.user_info["username"])        # self.assertEqual(user.ssn, self.user_info["ssn"])        self.assertTrue(user.password is not self.user_info["password"])        self.assertTrue(user.is_deleted is not True)        self.assertTrue(user.father_first_name is None)        self.assertTrue(user.mother_first_name is None)        self.assertTrue(user.password is not None)        self.assertTrue(user.birth_date is not None)    def test_get_token(self):        """        This test is used to test the login API. getting token and testing the token.        """        # Create a new user to login        user_info = self.generate_user_info()        new_user = self.client.post(            reverse("user-list"),            user_info,        )        self.assertEqual(new_user.status_code, status.HTTP_201_CREATED)        # Activation of User        from authentications.email import EMAILS        activation_url = reverse('user-activation')        activation_data = EMAILS[user_info["email"]]        self.client.post(activation_url, activation_data)        url = reverse("jwt-create")        data = {            "username": user_info["username"],            "password": user_info["password"],        }        response = self.client.post(url, data)        self.assertTrue(response.status_code, status.HTTP_200_OK)        self.assertTrue(response.data["access"] is not None)    def test_get_user(self):        """        This test for retrieving single user using API.        """        # Create a new user to login        new_user = self.client.post(            reverse("user-list"),            self.user_info,        )        self.assertEqual(new_user.status_code, status.HTTP_201_CREATED)        # Activate User        from authentications.email import EMAILS        activation_url = "http://127.0.0.1:8000/auth/users/activation/"        activation_data = EMAILS[self.user_info["email"]]        self.client.post(activation_url, activation_data)        # Get token        url = reverse("jwt-create")        data = {            "username": self.user_info["username"],            "password": self.user_info["password"],        }        response = self.client.post(url, data)        self.assertTrue(response.status_code, status.HTTP_200_OK)        # Get user        token = response.data["access"]        self.client.credentials(HTTP_AUTHORIZATION=f"JWT {token}")        url = reverse('user-list', kwargs={'id':new_user.data["id"]})        get_user = self.client.get(url)        self.assertEqual(get_user.status_code, status.HTTP_200_OK)        self.assertEqual(get_user.data["id"], new_user.data["id"])        self.assertEqual(get_user.data["email"], new_user.data["email"])        test_user = self.client.post(            reverse("user-list"),            self.generate_user_info(),        )        url = url = reverse('user-list', kwargs={'id': test_user.data['id'] })        get_user = self.client.get(url)        self.assertEqual(get_user.status_code, status.HTTP_404_NOT_FOUND)