|
@@ -1,4 +1,3 @@
|
|
|
-import pickle
|
|
|
from datetime import datetime
|
|
|
from functools import partialmethod
|
|
|
from io import StringIO
|
|
@@ -6,12 +5,11 @@ from unittest import mock, skipIf
|
|
|
|
|
|
from django.core import serializers
|
|
|
from django.core.serializers import SerializerDoesNotExist
|
|
|
-from django.core.serializers.base import PickleSerializer, ProgressBar
|
|
|
+from django.core.serializers.base import ProgressBar
|
|
|
from django.db import connection, transaction
|
|
|
from django.http import HttpResponse
|
|
|
from django.test import SimpleTestCase, override_settings, skipUnlessDBFeature
|
|
|
-from django.test.utils import Approximate, ignore_warnings
|
|
|
-from django.utils.deprecation import RemovedInDjango50Warning
|
|
|
+from django.test.utils import Approximate
|
|
|
|
|
|
from .models import (
|
|
|
Actor,
|
|
@@ -487,31 +485,6 @@ class SerializersTransactionTestBase:
|
|
|
self.assertEqual(art_obj.author.name, "Agnes")
|
|
|
|
|
|
|
|
|
-class PickleSerializerTests(SimpleTestCase):
|
|
|
- @ignore_warnings(category=RemovedInDjango50Warning)
|
|
|
- def test_serializer_protocol(self):
|
|
|
- serializer = PickleSerializer(protocol=3)
|
|
|
- self.assertEqual(serializer.protocol, 3)
|
|
|
- # If protocol is not provided, it defaults to pickle.HIGHEST_PROTOCOL
|
|
|
- serializer = PickleSerializer()
|
|
|
- self.assertEqual(serializer.protocol, pickle.HIGHEST_PROTOCOL)
|
|
|
-
|
|
|
- @ignore_warnings(category=RemovedInDjango50Warning)
|
|
|
- def test_serializer_loads_dumps(self):
|
|
|
- serializer = PickleSerializer()
|
|
|
- test_data = "test data"
|
|
|
- dump = serializer.dumps(test_data)
|
|
|
- self.assertEqual(serializer.loads(dump), test_data)
|
|
|
-
|
|
|
- def test_serializer_warning(self):
|
|
|
- msg = (
|
|
|
- "PickleSerializer is deprecated due to its security risk. Use "
|
|
|
- "JSONSerializer instead."
|
|
|
- )
|
|
|
- with self.assertRaisesMessage(RemovedInDjango50Warning, msg):
|
|
|
- PickleSerializer()
|
|
|
-
|
|
|
-
|
|
|
def register_tests(test_class, method_name, test_func, exclude=()):
|
|
|
"""
|
|
|
Dynamically create serializer tests to ensure that all registered
|