← Back to branch summary

~mars/django-configglue/remove-root-logger

« back to all changes in this revision

Viewing changes to django_configglue/tests/helpers.py

  • Committer: Ricardo Kirkner
  • Date: 2011-08-05 16:12:53 UTC
  • Revision ID: ricardo.kirkner@canonical.com-20110805161253-rwletg8ge9zfqwf0
some fixes to the tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
django_configglue/tests/helpers.py
1
1
# Copyright 2010-2011 Canonical Ltd.  This software is licensed under the
2
2
# GNU Lesser General Public License version 3 (see the file LICENSE).
3
3
 
 
4
import logging
4
5
import os
5
6
import sys
6
7
import textwrap
22
23
    COMMAND = ''
23
24
 
24
25
    def setUp(self):
 
26
        # disable logging during tests
 
27
        self.level = logging.getLogger().level
 
28
        logging.disable(logging.ERROR)
 
29
 
25
30
        config = textwrap.dedent("""
26
31
            [django]
27
32
            database_engine = sqlite3
47
52
        self._DJANGO_SETTINGS_MODULE = self.load_settings()
48
53
 
49
54
    def tearDown(self):
 
55
        # re-enable logging
 
56
        logging.getLogger().setLevel(self.level)
 
57
 
50
58
        self.load_settings(self._DJANGO_SETTINGS_MODULE)
51
59
        self.assertEqual(os.environ['DJANGO_SETTINGS_MODULE'],
52
60
                         self._DJANGO_SETTINGS_MODULE)
126
134
 
127
135
 
128
136
class SchemaHelperTestCase(TestCase):
 
137
    def setUp(self):
 
138
        # disable logging during tests
 
139
        self.level = logging.getLogger().level
 
140
        logging.disable(logging.ERROR)
 
141
 
 
142
    def tearDown(self):
 
143
        # re-enable logging
 
144
        logging.getLogger().setLevel(self.level)
 
145
 
129
146
    def assert_schema_structure(self, schema_cls, version, options):
130
147
        self.assertTrue(issubclass(schema_cls, Schema))
131
148
        self.assertEqual(schema_cls.version, 'bogus')