import copy

import os

import sys

import time

from unittest import TestCase

from django.conf import Settings

from django.db.models.loading import cache, load_app

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.assertEquals(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])