diff options
| author | David Wobrock <david.wobrock@gmail.com> | 2024-02-26 17:18:48 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-26 17:18:48 +0100 |
| commit | ef2434f8508551fee183079ab471b1dc325c7acb (patch) | |
| tree | 503a6bd98f9c547b363b8149a6cce7e910ebce73 /tests/test_utils | |
| parent | 18d79033b90902a6d6b615b42051191fd1b37892 (diff) | |
Refs #32114 -- Fixed test crash on non-picklable objects in subtests when PickleError is raised.
Related to the https://github.com/python/cpython/issues/73373.
Follow up to c09e8f5fd8f977bf16e9ec5d11b370151fc81ea8.
Diffstat (limited to 'tests/test_utils')
| -rw-r--r-- | tests/test_utils/test_testcase.py | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/tests/test_utils/test_testcase.py b/tests/test_utils/test_testcase.py index 0f41f29a23..efca01e29e 100644 --- a/tests/test_utils/test_testcase.py +++ b/tests/test_utils/test_testcase.py @@ -3,17 +3,31 @@ from functools import wraps from django.db import IntegrityError, connections, transaction from django.test import TestCase, skipUnlessDBFeature -from django.test.testcases import DatabaseOperationForbidden, SimpleTestCase, TestData +from django.test.testcases import ( + DatabaseOperationForbidden, + SimpleTestCase, + TestData, + is_pickable, +) from .models import Car, Person, PossessedCar +class UnpicklableObject: + def __getstate__(self): + raise pickle.PickleError("cannot be pickled for testing reasons") + + class TestSimpleTestCase(SimpleTestCase): def test_is_picklable_with_non_picklable_properties(self): """ParallelTestSuite requires that all TestCases are picklable.""" self.non_picklable = lambda: 0 self.assertEqual(self, pickle.loads(pickle.dumps(self))) + def test_is_picklable_with_non_picklable_object(self): + unpicklable_obj = UnpicklableObject() + self.assertEqual(is_pickable(unpicklable_obj), False) + class TestTestCase(TestCase): @skipUnlessDBFeature("can_defer_constraint_checks") |
