mirror of
https://github.com/django/django.git
synced 2024-11-29 06:03:25 +01:00
8662654d6d
Since the original ones in django.db.models.loading were kept only for backwards compatibility, there's no need to recreate them. However, many internals of Django still relied on them. They were also imported in django.db.models. They never appear in the documentation, except a quick mention of get_models and get_app in the 1.2 release notes to document an edge case in GIS. I don't think that makes them a public API. This commit doesn't change the overall amount of global state but clarifies that it's tied to the app_cache object instead of hiding it behind half a dozen functions.
126 lines
4.6 KiB
Python
126 lines
4.6 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import copy
|
|
import os
|
|
import sys
|
|
from unittest import TestCase
|
|
|
|
from django.apps import app_cache
|
|
from django.apps.cache import AppCache
|
|
from django.test.utils import override_settings
|
|
from django.utils._os import upath
|
|
|
|
|
|
class EggLoadingTest(TestCase):
|
|
|
|
def setUp(self):
|
|
self.old_path = sys.path[:]
|
|
self.egg_dir = '%s/eggs' % os.path.dirname(upath(__file__))
|
|
|
|
# This test adds dummy applications to the app cache. These
|
|
# need to be removed in order to prevent bad interactions
|
|
# with the flush operation in other tests.
|
|
self.old_app_models = copy.deepcopy(app_cache.app_models)
|
|
|
|
def tearDown(self):
|
|
sys.path = self.old_path
|
|
app_cache.app_models = self.old_app_models
|
|
|
|
def test_egg1(self):
|
|
"""Models module can be loaded from an app in an egg"""
|
|
egg_name = '%s/modelapp.egg' % self.egg_dir
|
|
sys.path.append(egg_name)
|
|
models = app_cache.load_app('app_with_models')
|
|
self.assertFalse(models is None)
|
|
|
|
def test_egg2(self):
|
|
"""Loading an app from an egg that has no models returns no models (and no error)"""
|
|
egg_name = '%s/nomodelapp.egg' % self.egg_dir
|
|
sys.path.append(egg_name)
|
|
models = app_cache.load_app('app_no_models')
|
|
self.assertTrue(models is None)
|
|
|
|
def test_egg3(self):
|
|
"""Models module can be loaded from an app located under an egg's top-level package"""
|
|
egg_name = '%s/omelet.egg' % self.egg_dir
|
|
sys.path.append(egg_name)
|
|
models = app_cache.load_app('omelet.app_with_models')
|
|
self.assertFalse(models is None)
|
|
|
|
def test_egg4(self):
|
|
"""Loading an app with no models from under the top-level egg package generates no error"""
|
|
egg_name = '%s/omelet.egg' % self.egg_dir
|
|
sys.path.append(egg_name)
|
|
models = app_cache.load_app('omelet.app_no_models')
|
|
self.assertTrue(models is None)
|
|
|
|
def test_egg5(self):
|
|
"""Loading an app from an egg that has an import error in its models module raises that error"""
|
|
egg_name = '%s/brokenapp.egg' % self.egg_dir
|
|
sys.path.append(egg_name)
|
|
self.assertRaises(ImportError, app_cache.load_app, 'broken_app')
|
|
raised = None
|
|
try:
|
|
app_cache.load_app('broken_app')
|
|
except ImportError as e:
|
|
raised = e
|
|
|
|
# Make sure the message is indicating the actual
|
|
# problem in the broken app.
|
|
self.assertTrue(raised is not None)
|
|
self.assertTrue("modelz" in raised.args[0])
|
|
|
|
def test_missing_app(self):
|
|
"""
|
|
Test that repeated app loading doesn't succeed in case there is an
|
|
error. Refs #17667.
|
|
"""
|
|
# AppCache is a Borg, so we can instantiate one and change its
|
|
# loaded to False to force the following code to actually try to
|
|
# populate the cache.
|
|
a = AppCache()
|
|
a.loaded = False
|
|
try:
|
|
with override_settings(INSTALLED_APPS=('notexists',)):
|
|
self.assertRaises(ImportError, app_cache.get_model, 'notexists', 'nomodel', seed_cache=True)
|
|
self.assertRaises(ImportError, app_cache.get_model, 'notexists', 'nomodel', seed_cache=True)
|
|
finally:
|
|
a.loaded = True
|
|
|
|
|
|
class GetModelsTest(TestCase):
|
|
def setUp(self):
|
|
from .not_installed import models
|
|
self.not_installed_module = models
|
|
|
|
def test_get_model_only_returns_installed_models(self):
|
|
self.assertEqual(
|
|
app_cache.get_model("not_installed", "NotInstalledModel"), None)
|
|
|
|
def test_get_model_with_not_installed(self):
|
|
self.assertEqual(
|
|
app_cache.get_model(
|
|
"not_installed", "NotInstalledModel", only_installed=False),
|
|
self.not_installed_module.NotInstalledModel)
|
|
|
|
def test_get_models_only_returns_installed_models(self):
|
|
self.assertFalse(
|
|
"NotInstalledModel" in
|
|
[m.__name__ for m in app_cache.get_models()])
|
|
|
|
def test_get_models_with_app_label_only_returns_installed_models(self):
|
|
self.assertEqual(app_cache.get_models(self.not_installed_module), [])
|
|
|
|
def test_get_models_with_not_installed(self):
|
|
self.assertTrue(
|
|
"NotInstalledModel" in [
|
|
m.__name__ for m in app_cache.get_models(only_installed=False)])
|
|
|
|
|
|
class NotInstalledModelsTest(TestCase):
|
|
def test_related_not_installed_model(self):
|
|
from .not_installed.models import NotInstalledModel
|
|
self.assertEqual(
|
|
set(NotInstalledModel._meta.get_all_field_names()),
|
|
set(["id", "relatedmodel", "m2mrelatedmodel"]))
|