Misago/misago/conf/tests/test_migrationutils.py

from django.test import TestCase
from misago.core import threadstore
from misago.conf import migrationutils
from misago.conf.models import SettingsGroup, Setting


class MigrationUtilsTests(TestCase):
    def test_with_conf_models(self):
        """with_conf_models builds correct dict of models"""
        models = {
            u'core.cacheversion': {
                'Meta': {'object_name': 'CacheVersion'},
                'cache': ('django.db.models.fields.CharField', [], {'max_length': '128'}),
                u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
                'version': ('django.db.models.fields.PositiveIntegerField', [], {'default': '0'})
            }
        }

        final_models = migrationutils.with_conf_models('0001_initial')
        self.assertTrue('conf.settingsgroup' in final_models),
        self.assertTrue('conf.setting' in final_models),

        final_models = migrationutils.with_conf_models('0001_initial', models)
        self.assertTrue('conf.settingsgroup' in final_models),
        self.assertTrue('conf.setting' in final_models),
        self.assertTrue('core.cacheversion' in final_models),


class DBConfMigrationUtilsTests(TestCase):
    def setUp(self):
        self.orm = {
            'conf.SettingsGroup': SettingsGroup,
            'conf.Setting': Setting,
        }

        self.test_group = {
            'key': 'test_group',
            'name': "Test settings",
            'description': "Those are test settings.",
            'settings': (
                {
                    'setting': 'fish_name',
                    'name': "Fish's name",
                    'value': "Eric",
                    'field_extra': {
                           'min_length': 2,
                           'max_length': 255
                        },
                },
                {
                    'setting': 'fish_license_no',
                    'name': "Fish's license number",
                    'default_value': '123-456',
                    'field_extra': {
                            'max_length': 255
                        },
                },
            )
        }

        migrationutils.migrate_settings_group(self.orm, self.test_group)
        self.groups_count = SettingsGroup.objects.count()

    def tearDown(self):
        threadstore.clear()

    def test_get_custom_group_and_settings(self):
        """tests setup created settings group"""
        custom_group = migrationutils.get_group(self.orm,
                                                self.test_group['key'])

        self.assertEqual(custom_group.key, self.test_group['key'])
        self.assertEqual(custom_group.name, self.test_group['name'])
        self.assertEqual(custom_group.description,
                         self.test_group['description'])

        custom_settings = migrationutils.get_custom_settings_values(
            self.orm, custom_group)

        self.assertEqual(custom_settings['fish_name'], 'Eric')
        self.assertTrue('fish_license_no' not in custom_settings)

    def test_change_group_key(self):
        """migrate_settings_group changed group key"""

        new_group = {
            'key': 'new_test_group',
            'name': "New test settings",
            'description': "Those are updated test settings.",
            'settings': (
                {
                    'setting': 'fish_new_name',
                    'name': "Fish's new name",
                    'value': "Eric",
                    'field_extra': {
                            'min_length': 2,
                            'max_length': 255
                        },
                },
                {
                    'setting': 'fish_new_license_no',
                    'name': "Fish's changed license number",
                    'default_value': '123-456',
                    'field_extra': {
                            'max_length': 255
                        },
                },
            )
        }

        migrationutils.migrate_settings_group(
            self.orm, new_group, old_group_key=self.test_group['key'])
        db_group = migrationutils.get_group(self.orm, new_group['key'])

        self.assertEqual(SettingsGroup.objects.count(), self.groups_count)
        self.assertEqual(db_group.key, new_group['key'])
        self.assertEqual(db_group.name, new_group['name'])
        self.assertEqual(db_group.description,
                         new_group['description'])

        for setting in new_group['settings']:
            db_setting = db_group.setting_set.get(setting=setting['setting'])
            self.assertEqual(db_setting.name, setting['name'])