Refs #33012 -- Moved PickleSerializer to django.core.serializers.base and added tests.

1 files changed, 17 insertions, 1 deletions

diff --git a/tests/serializers/tests.py b/tests/serializers/tests.py
index 439464980d..4bd6acd982 100644
--- a/tests/serializers/tests.py
+++ b/tests/serializers/tests.py
@@ -1,3 +1,4 @@
+import pickle
 from datetime import datetime
 from functools import partialmethod
 from io import StringIO
@@ -5,7 +6,7 @@ from unittest import mock, skipIf
 
 from django.core import serializers
 from django.core.serializers import SerializerDoesNotExist
-from django.core.serializers.base import ProgressBar
+from django.core.serializers.base import PickleSerializer, ProgressBar
 from django.db import connection, transaction
 from django.http import HttpResponse
 from django.test import SimpleTestCase, override_settings, skipUnlessDBFeature
@@ -418,6 +419,21 @@ class SerializersTransactionTestBase:
         self.assertEqual(art_obj.author.name, "Agnes")
 
 
+class PickleSerializerTests(SimpleTestCase):
+    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)
+
+    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 register_tests(test_class, method_name, test_func, exclude=()):
     """
     Dynamically create serializer tests to ensure that all registered