PageRenderTime 15ms CodeModel.GetById 8ms app.highlight 4ms RepoModel.GetById 2ms app.codeStats 0ms

/tests/regressiontests/app_loading/tests.py

https://code.google.com/p/mango-py/
Python | 83 lines | 65 code | 13 blank | 5 comment | 2 complexity | 5efec2ac19396c7959e1f1ce4d367dd9 MD5 | raw file
 1import copy
 2import os
 3import sys
 4import time
 5
 6from django.conf import Settings
 7from django.db.models.loading import cache, load_app
 8from django.utils.unittest import TestCase
 9
10
11class InstalledAppsGlobbingTest(TestCase):
12    def setUp(self):
13        self.OLD_SYS_PATH = sys.path[:]
14        sys.path.append(os.path.dirname(os.path.abspath(__file__)))
15        self.OLD_TZ = os.environ.get("TZ")
16
17    def test_globbing(self):
18        settings = Settings('test_settings')
19        self.assertEqual(settings.INSTALLED_APPS, ['parent.app', 'parent.app1', 'parent.app_2'])
20
21    def tearDown(self):
22        sys.path = self.OLD_SYS_PATH
23        if hasattr(time, "tzset") and self.OLD_TZ:
24            os.environ["TZ"] = self.OLD_TZ
25            time.tzset()
26
27
28class EggLoadingTest(TestCase):
29
30    def setUp(self):
31        self.old_path = sys.path[:]
32        self.egg_dir = '%s/eggs' % os.path.dirname(__file__)
33
34        # This test adds dummy applications to the app cache. These
35        # need to be removed in order to prevent bad interactions
36        # with the flush operation in other tests.
37        self.old_app_models = copy.deepcopy(cache.app_models)
38        self.old_app_store = copy.deepcopy(cache.app_store)
39
40    def tearDown(self):
41        sys.path = self.old_path
42        cache.app_models = self.old_app_models
43        cache.app_store = self.old_app_store
44
45    def test_egg1(self):
46        """Models module can be loaded from an app in an egg"""
47        egg_name = '%s/modelapp.egg' % self.egg_dir
48        sys.path.append(egg_name)
49        models = load_app('app_with_models')
50        self.assertFalse(models is None)
51
52    def test_egg2(self):
53        """Loading an app from an egg that has no models returns no models (and no error)"""
54        egg_name = '%s/nomodelapp.egg' % self.egg_dir
55        sys.path.append(egg_name)
56        models = load_app('app_no_models')
57        self.assertTrue(models is None)
58
59    def test_egg3(self):
60        """Models module can be loaded from an app located under an egg's top-level package"""
61        egg_name = '%s/omelet.egg' % self.egg_dir
62        sys.path.append(egg_name)
63        models = load_app('omelet.app_with_models')
64        self.assertFalse(models is None)
65
66    def test_egg4(self):
67        """Loading an app with no models from under the top-level egg package generates no error"""
68        egg_name = '%s/omelet.egg' % self.egg_dir
69        sys.path.append(egg_name)
70        models = load_app('omelet.app_no_models')
71        self.assertTrue(models is None)
72
73    def test_egg5(self):
74        """Loading an app from an egg that has an import error in its models module raises that error"""
75        egg_name = '%s/brokenapp.egg' % self.egg_dir
76        sys.path.append(egg_name)
77        self.assertRaises(ImportError, load_app, 'broken_app')
78        try:
79            load_app('broken_app')
80        except ImportError, e:
81            # Make sure the message is indicating the actual
82            # problem in the broken app.
83            self.assertTrue("modelz" in e.args[0])