summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorClaude Paroz <claude@2xlibre.net>2014-12-17 17:25:28 +0100
committerClaude Paroz <claude@2xlibre.net>2014-12-17 18:33:32 +0100
commit108b8bf852c76855ed98f5abe55db1da845598e7 (patch)
tree5d2edc8ea3e1daf32eb068e8647c23ae76973952 /tests
parentf3eed95175f8a8a365218a09fdc2dc864ff75d0c (diff)
Fixed #24007 -- Ensure apps registry's ready before unpickling models
This prevents AppRegistryNotReady errors when unpickling Django models from an external script.
Diffstat (limited to 'tests')
-rw-r--r--tests/model_regress/test_pickle.py35
1 files changed, 35 insertions, 0 deletions
diff --git a/tests/model_regress/test_pickle.py b/tests/model_regress/test_pickle.py
index 8dc90893bf..ab70b4ec91 100644
--- a/tests/model_regress/test_pickle.py
+++ b/tests/model_regress/test_pickle.py
@@ -1,4 +1,7 @@
+import datetime
import pickle
+import subprocess
+import tempfile
import warnings
from django.db import models, DJANGO_VERSION_PICKLE_KEY
@@ -6,6 +9,8 @@ from django.test import TestCase
from django.utils.encoding import force_text
from django.utils.version import get_major_version, get_version
+from .models import Article
+
class ModelPickleTestCase(TestCase):
def test_missing_django_version_unpickling(self):
@@ -51,3 +56,33 @@ class ModelPickleTestCase(TestCase):
"Pickled model instance's Django version %s does not "
"match the current version %s."
% (str(float(get_major_version()) - 0.1), get_version()))
+
+ def test_unpickling_when_appregistrynotready(self):
+ """
+ #24007 -- Verifies that a pickled model can be unpickled without having
+ to manually setup the apps registry beforehand.
+ """
+ script_template = """#!/usr/bin/env python
+import pickle
+
+from django.conf import settings
+
+data = %r
+
+settings.configure(DEBUG=False, INSTALLED_APPS=('model_regress',), SECRET_KEY = "blah")
+article = pickle.loads(data)
+print(article.headline)"""
+ a = Article.objects.create(
+ headline="Some object",
+ pub_date=datetime.datetime.now(),
+ article_text="This is an article",
+ )
+
+ with tempfile.NamedTemporaryFile(mode='w+', suffix=".py", dir='.', delete=True) as script:
+ script.write(script_template % pickle.dumps(a))
+ script.flush()
+ try:
+ result = subprocess.check_output(['python', script.name])
+ except subprocess.CalledProcessError:
+ self.fail("Unable to reload model pickled data")
+ self.assertEqual(result.strip().decode(), "Some object")