Implementación de Pruebas Unitarias en Django Rest

Implementación de pruebas unitarias en Django REST Framework

Las pruebas unitarias son una parte fundamental en el desarrollo de aplicaciones con Django REST framework, ya que permiten asegurar la calidad del código y su correcto funcionamiento. A continuación te indicaré los pasos para implementar pruebas unitarias en Django REST framework:

1. Crear un archivo tests.py: Este archivo debe estar ubicado en el mismo directorio que el archivo que contiene la lógica principal de la API.
2. Importar las dependencias necesarias: Asegúrate de importar las siguientes dependencias al inicio del archivo tests.py:

from django.test import TestCase
from rest_framework.test import APIClient
from rest_framework import status
from django.urls import reverse

3. Crear la clase de pruebas: Esta será la clase principal donde se definirán las pruebas unitarias.

class TestAPI(TestCase):

4. Definir los métodos de prueba: Cada método de prueba debe empezar con la palabra clave test. En los métodos de prueba, debes definir los casos que quieres evaluar.

def test_get_all_users(self):
    client = APIClient()
    response = client.get(reverse('api-users'))
    self.assertEqual(response.status_code, status.HTTP_200_OK)

5. Ejecutar las pruebas: Una vez que has definido tus métodos de prueba, puedes ejecutarlas con el comando python manage.py test en la terminal. Este comando busca automáticamente todos los archivos de pruebas definidos en un proyecto Django y ejecuta todos los métodos de prueba de cada archivo. Es importante que al ejecutar las pruebas, se compruebe que todas las pruebas pasan satisfactoriamente para asegurar la correcta funcionalidad de la API.

Ejemplo de pruebas unitarias en Django Rest Framework

Claro, te puedo dar un ejemplo práctico de cómo implementar pruebas unitarias en Django Rest Framework. Imaginemos que tenemos un modelo de Django llamado "Producto" que tiene los campos "nombre", "precio" y "stock". Además, hemos implementado una API para agregar nuevos productos y listar todos los productos disponibles. A continuación, te mostraré cómo podemos escribir pruebas unitarias para los endpoints de la API:

# Importamos las herramientas necesarias para realizar las pruebas
from django.urls import reverse
from rest_framework import status
from rest_framework.test import APITestCase
from .models import Producto

class ProductoTests(APITestCase):
    def test_agregar_producto(self):
        """ Prueba que se puede agregar un nuevo producto a la base de datos """
        url = reverse('producto-list')
        data = {'nombre': 'Producto 1', 'precio': 9.99, 'stock': 20}
        response = self.client.post(url, data, format='json')
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(Producto.objects.count(), 1)
        self.assertEqual(Producto.objects.get().nombre, 'Producto 1')

    def test_listar_productos(self):
        """ Prueba que se pueden listar todos los productos disponibles """
        Producto.objects.create(nombre='Producto 1', precio=9.99, stock=20)
        Producto.objects.create(nombre='Producto 2', precio=19.99, stock=15)
        url = reverse('producto-list')
        response = self.client.get(url, format='json')
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(len(response.data), 2)
        self.assertEqual(response.data[0]['nombre'], 'Producto 1')
        self.assertEqual(response.data[0]['precio'], '9.99')
        self.assertEqual(response.data[0]['stock'], 20)

En el ejemplo anterior, hemos creado dos pruebas unitarias para nuestra API de productos: una para agregar un nuevo producto y otra para listar todos los productos disponibles. El método reverse se usa para obtener la URL de la API asociada a la vista, y el método client.post y client.get se utilizan para enviar solicitudes POST y GET a la API, respectivamente. En ambas pruebas, hemos verificado que la API responde correctamente y que las respuestas son válidas. Siguiendo de esta manera se pueden realizar pruebas unitarias para todas las acciones de nuestra API y tener una mejor confianza en la implementación de nuestra API.

Ejemplo de pruebas unitarias en Django Rest Framework

Claro, aquí te brindo un ejemplo práctico de cómo implementar pruebas unitarias en Django Rest Framework en Python. Supongamos que tenemos una API que gestiona un conjunto de tareas a realizar. Y queremos probar que la API responde correctamente a las solicitudes GET, POST, PUT y DELETE. Para ello, creamos un archivo de pruebas llamado tests.py en nuestro directorio de aplicaciones. Luego, importamos las bibliotecas necesarias y definimos una clase de prueba TaskApiTestCase que se encargará de enviar diferentes solicitudes HTTP y verificar las respuestas correspondientes. Aquí hay un ejemplo de cómo se vería:

from django.test import TestCase
from rest_framework.test import APIClient
from .models import Task

class TaskApiTestCase(TestCase):
    ''' Clase para probar la API de tareas '''
    def setUp(self):
        self.client = APIClient()
        # Creamos algunas tareas para las pruebas
        self.task1 = Task.objects.create(
            title='Tarea 1',
            description='Descripción de la tarea 1',
            completed=False
        )
        self.task2 = Task.objects.create(
            title='Tarea 2',
            description='Descripción de la tarea 2',
            completed=True
        )

    def test_get_tasks_list(self):
        # Verificamos que la API responda correctamente a la solicitud GET para obtener la lista de tareas
        response = self.client.get('/tasks/')
        self.assertEqual(response.status_code, 200)

    def test_get_task_detail(self):
        # Verificamos que la API responda correctamente a la solicitud GET para obtener el detalle de una tarea
        response = self.client.get('/tasks/{}/'.format(self.task1.id))
        self.assertEqual(response.status_code, 200)

    def test_create_task(self):
        # Verificamos que la API responda correctamente a la solicitud POST para crear una nueva tarea
        data = {'title': 'Tarea 3', 'description': 'Descripción de la tarea 3', 'completed': False}
        response = self.client.post('/tasks/', data)
        self.assertEqual(response.status_code, 201)

    def test_update_task(self):
        # Verificamos que la API responda correctamente a la solicitud PUT para actualizar una tarea existente
        data = {'completed': True}
        response = self.client.put('/tasks/{}/'.format(self.task1.id), data, format='json')
        self.assertEqual(response.status_code, 200)

    def test_delete_task(self):
        # Verificamos que la API responda correctamente a la solicitud DELETE para eliminar una tarea existente
        response = self.client.delete('/tasks/{}/'.format(self.task2.id))
        self.assertEqual(response.status_code, 204)

En este ejemplo, hemos creado una clase de prueba TaskApiTestCase que hereda de django.test.TestCase. En el método setUp() creamos un cliente de API, y algunas tareas de prueba para usar durante las pruebas. Luego, hemos definido cinco métodos de prueba que envían diferentes solicitudes HTTP a la API y verifican que las respuestas sean las esperadas. Por ejemplo, el método test_get_tasks_list() envía una solicitud GET a /tasks/ y verifica que la respuesta tenga un código de estado HTTP 200. Finalmente, para ejecutar las pruebas, solo necesitamos usar el comando python manage.py test en la terminal. Por ejemplo, python manage.py test tasks para probar la API de tareas.