mirror of
https://github.com/django/django.git
synced 2024-11-29 06:03:25 +01:00
ba7206cd81
Returning None on errors required unpythonic error checking and was inconsistent with get_app_config. get_model was a private API until the previous commit, but given that it was certainly used in third party software, the change is explained in the release notes. Applied the same change to get_registered_model, which is a new private API introduced during the recent refactoring.
98 lines
3.7 KiB
Python
98 lines
3.7 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import sys
|
|
|
|
from django.apps import apps
|
|
from django.test import TestCase
|
|
from django.utils._os import upath
|
|
from django.utils import six
|
|
|
|
|
|
class EggLoadingTest(TestCase):
|
|
|
|
def setUp(self):
|
|
self.old_path = sys.path[:]
|
|
self.egg_dir = '%s/eggs' % os.path.dirname(upath(__file__))
|
|
|
|
# The models need to be removed after the test in order to prevent bad
|
|
# interactions with the flush operation in other tests.
|
|
self._old_models = apps.all_models['app_loading'].copy()
|
|
|
|
def tearDown(self):
|
|
apps.all_models['app_loading'] = self._old_models
|
|
apps.get_models.cache_clear()
|
|
|
|
sys.path = self.old_path
|
|
|
|
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)
|
|
with self.settings(INSTALLED_APPS=['app_with_models']):
|
|
models_module = apps.get_app_config('app_with_models').models_module
|
|
self.assertIsNotNone(models_module)
|
|
|
|
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)
|
|
with self.settings(INSTALLED_APPS=['app_no_models']):
|
|
models_module = apps.get_app_config('app_no_models').models_module
|
|
self.assertIsNone(models_module)
|
|
|
|
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)
|
|
with self.settings(INSTALLED_APPS=['omelet.app_with_models']):
|
|
models_module = apps.get_app_config('app_with_models').models_module
|
|
self.assertIsNotNone(models_module)
|
|
|
|
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)
|
|
with self.settings(INSTALLED_APPS=['omelet.app_no_models']):
|
|
models_module = apps.get_app_config('app_no_models').models_module
|
|
self.assertIsNone(models_module)
|
|
|
|
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)
|
|
with six.assertRaisesRegex(self, ImportError, 'modelz'):
|
|
with self.settings(INSTALLED_APPS=['broken_app']):
|
|
pass
|
|
|
|
|
|
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):
|
|
with self.assertRaises(LookupError):
|
|
apps.get_model("not_installed", "NotInstalledModel")
|
|
|
|
def test_get_models_only_returns_installed_models(self):
|
|
self.assertNotIn(
|
|
"NotInstalledModel",
|
|
[m.__name__ for m in apps.get_models()])
|
|
|
|
def test_get_models_with_app_label_only_returns_installed_models(self):
|
|
self.assertEqual(apps.get_models(self.not_installed_module), [])
|
|
|
|
def test_get_models_with_not_installed(self):
|
|
self.assertIn(
|
|
"NotInstalledModel",
|
|
[m.__name__ for m in apps.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"]))
|