import copy import os import sys import time from django.conf import Settings from django.db.models.loading import cache, load_app, get_model, get_models from django.utils.unittest import TestCase class InstalledAppsGlobbingTest(TestCase): def setUp(self): self.OLD_SYS_PATH = sys.path[:] sys.path.append(os.path.dirname(os.path.abspath(__file__))) self.OLD_TZ = os.environ.get("TZ") def test_globbing(self): settings = Settings('test_settings') self.assertEqual(settings.INSTALLED_APPS, ['parent.app', 'parent.app1', 'parent.app_2']) def tearDown(self): sys.path = self.OLD_SYS_PATH if hasattr(time, "tzset") and self.OLD_TZ: os.environ["TZ"] = self.OLD_TZ time.tzset() class EggLoadingTest(TestCase): def setUp(self): self.old_path = sys.path[:] self.egg_dir = '%s/eggs' % os.path.dirname(__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(cache.app_models) self.old_app_store = copy.deepcopy(cache.app_store) def tearDown(self): sys.path = self.old_path cache.app_models = self.old_app_models cache.app_store = self.old_app_store 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 = 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 = 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 = 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 = 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, load_app, 'broken_app') try: load_app('broken_app') except ImportError, e: # Make sure the message is indicating the actual # problem in the broken app. self.assertTrue("modelz" in e.args[0]) class GetModelsTest(TestCase): def setUp(self): import not_installed.models self.not_installed_module = not_installed.models def test_get_model_only_returns_installed_models(self): self.assertEqual( get_model("not_installed", "NotInstalledModel"), None) def test_get_models_only_returns_installed_models(self): self.assertFalse( "NotInstalledModel" in [m.__name__ for m in get_models()]) def test_get_models_with_app_label_only_returns_installed_models(self): self.assertEqual(get_models(self.not_installed_module), [])