mirror of
				https://github.com/encode/django-rest-framework.git
				synced 2025-11-04 01:47:59 +03:00 
			
		
		
		
	
		
			
				
	
	
		
			124 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			124 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
"""
 | 
						|
Tests to cover bulk create and update using serializers.
 | 
						|
"""
 | 
						|
from __future__ import unicode_literals
 | 
						|
from django.test import TestCase
 | 
						|
from django.utils import six
 | 
						|
from rest_framework import serializers
 | 
						|
 | 
						|
 | 
						|
class BulkCreateSerializerTests(TestCase):
 | 
						|
    """
 | 
						|
    Creating multiple instances using serializers.
 | 
						|
    """
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        class BookSerializer(serializers.Serializer):
 | 
						|
            id = serializers.IntegerField()
 | 
						|
            title = serializers.CharField(max_length=100)
 | 
						|
            author = serializers.CharField(max_length=100)
 | 
						|
 | 
						|
        self.BookSerializer = BookSerializer
 | 
						|
 | 
						|
    def test_bulk_create_success(self):
 | 
						|
        """
 | 
						|
        Correct bulk update serialization should return the input data.
 | 
						|
        """
 | 
						|
 | 
						|
        data = [
 | 
						|
            {
 | 
						|
                'id': 0,
 | 
						|
                'title': 'The electric kool-aid acid test',
 | 
						|
                'author': 'Tom Wolfe'
 | 
						|
            }, {
 | 
						|
                'id': 1,
 | 
						|
                'title': 'If this is a man',
 | 
						|
                'author': 'Primo Levi'
 | 
						|
            }, {
 | 
						|
                'id': 2,
 | 
						|
                'title': 'The wind-up bird chronicle',
 | 
						|
                'author': 'Haruki Murakami'
 | 
						|
            }
 | 
						|
        ]
 | 
						|
 | 
						|
        serializer = self.BookSerializer(data=data, many=True)
 | 
						|
        self.assertEqual(serializer.is_valid(), True)
 | 
						|
        self.assertEqual(serializer.validated_data, data)
 | 
						|
 | 
						|
    def test_bulk_create_errors(self):
 | 
						|
        """
 | 
						|
        Incorrect bulk create serialization should return errors.
 | 
						|
        """
 | 
						|
 | 
						|
        data = [
 | 
						|
            {
 | 
						|
                'id': 0,
 | 
						|
                'title': 'The electric kool-aid acid test',
 | 
						|
                'author': 'Tom Wolfe'
 | 
						|
            }, {
 | 
						|
                'id': 1,
 | 
						|
                'title': 'If this is a man',
 | 
						|
                'author': 'Primo Levi'
 | 
						|
            }, {
 | 
						|
                'id': 'foo',
 | 
						|
                'title': 'The wind-up bird chronicle',
 | 
						|
                'author': 'Haruki Murakami'
 | 
						|
            }
 | 
						|
        ]
 | 
						|
        expected_errors = [
 | 
						|
            {},
 | 
						|
            {},
 | 
						|
            {'id': ['A valid integer is required.']}
 | 
						|
        ]
 | 
						|
 | 
						|
        serializer = self.BookSerializer(data=data, many=True)
 | 
						|
        self.assertEqual(serializer.is_valid(), False)
 | 
						|
        self.assertEqual(serializer.errors, expected_errors)
 | 
						|
 | 
						|
    def test_invalid_list_datatype(self):
 | 
						|
        """
 | 
						|
        Data containing list of incorrect data type should return errors.
 | 
						|
        """
 | 
						|
        data = ['foo', 'bar', 'baz']
 | 
						|
        serializer = self.BookSerializer(data=data, many=True)
 | 
						|
        self.assertEqual(serializer.is_valid(), False)
 | 
						|
 | 
						|
        text_type_string = six.text_type.__name__
 | 
						|
        message = 'Invalid data. Expected a dictionary, but got %s.' % text_type_string
 | 
						|
        expected_errors = [
 | 
						|
            {'non_field_errors': [message]},
 | 
						|
            {'non_field_errors': [message]},
 | 
						|
            {'non_field_errors': [message]}
 | 
						|
        ]
 | 
						|
 | 
						|
        self.assertEqual(serializer.errors, expected_errors)
 | 
						|
 | 
						|
    def test_invalid_single_datatype(self):
 | 
						|
        """
 | 
						|
        Data containing a single incorrect data type should return errors.
 | 
						|
        """
 | 
						|
        data = 123
 | 
						|
        serializer = self.BookSerializer(data=data, many=True)
 | 
						|
        self.assertEqual(serializer.is_valid(), False)
 | 
						|
 | 
						|
        expected_errors = {'non_field_errors': ['Expected a list of items but got type "int".']}
 | 
						|
 | 
						|
        self.assertEqual(serializer.errors, expected_errors)
 | 
						|
 | 
						|
    def test_invalid_single_object(self):
 | 
						|
        """
 | 
						|
        Data containing only a single object, instead of a list of objects
 | 
						|
        should return errors.
 | 
						|
        """
 | 
						|
        data = {
 | 
						|
            'id': 0,
 | 
						|
            'title': 'The electric kool-aid acid test',
 | 
						|
            'author': 'Tom Wolfe'
 | 
						|
        }
 | 
						|
        serializer = self.BookSerializer(data=data, many=True)
 | 
						|
        self.assertEqual(serializer.is_valid(), False)
 | 
						|
 | 
						|
        expected_errors = {'non_field_errors': ['Expected a list of items but got type "dict".']}
 | 
						|
 | 
						|
        self.assertEqual(serializer.errors, expected_errors)
 |