edan_f0/tests/unit/core/customisation_tests.py

import os
import tempfile

from django.test import TestCase
from django.conf import settings

from oscar.core import customisation

VALID_FOLDER_PATH = 'tests/_site/apps'


class TestUtilities(TestCase):

    def test_subfolder_extraction(self):
        folders = list(customisation.subfolders('/var/www/eggs'))
        self.assertEqual(folders, ['/var', '/var/www', '/var/www/eggs'])


class TestForkAppFunction(TestCase):

    def setUp(self):
        self.tmp_folder = tempfile.mkdtemp()

    def test_raises_exception_for_nonexistant_app_label(self):
        with self.assertRaises(ValueError):
            customisation.fork_app('sillytown', 'somefolder')

    def test_raises_exception_if_app_has_already_been_forked(self):
        # We piggyback on another test which means a custom app is already in
        # the settings we use for the test suite. We just check that's still
        # the case here.
        self.assertIn('tests._site.apps.partner', settings.INSTALLED_APPS)
        with self.assertRaises(ValueError):
            customisation.fork_app('partner', VALID_FOLDER_PATH)

    def test_creates_new_folder(self):
        customisation.fork_app('order', self.tmp_folder)
        new_folder_path = os.path.join(self.tmp_folder, 'order')
        self.assertTrue(os.path.exists(new_folder_path))

    def test_creates_init_file(self):
        customisation.fork_app('order', self.tmp_folder)
        filepath = os.path.join(self.tmp_folder, 'order', '__init__.py')
        self.assertTrue(os.path.exists(filepath))

    def test_handles_dashboard_app(self):
        # Dashboard apps are fiddly as they aren't identified by a single app
        # label.
        customisation.fork_app('dashboard.catalogue', self.tmp_folder)
        # Check __init__.py created (and supporting folders)
        init_path = os.path.join(self.tmp_folder,
                                 'dashboard/catalogue/__init__.py')
        self.assertTrue(os.path.exists(init_path))

    def test_creates_models_and_admin_file(self):
        customisation.fork_app('order', self.tmp_folder)
        for module, expected_string in [
            ('models', 'from oscar.apps.order.models import *'),
            ('admin', 'from oscar.apps.order.admin import *'),
            ('config', 'OrderConfig')
        ]:
            filepath = os.path.join(self.tmp_folder, 'order', '%s.py' % module)
            self.assertTrue(os.path.exists(filepath))

            contents = open(filepath).read()
            self.assertTrue(expected_string in contents)

    def test_copies_in_migrations_when_needed(self):
        for app, has_models in [('order', True), ('search', False)]:
            customisation.fork_app(app, self.tmp_folder)
            native_migration_path = os.path.join(
                self.tmp_folder, app, 'migrations')
            self.assertEqual(has_models, os.path.exists(native_migration_path))