17
17
"""Tests for bzrlib.debug"""
27
class TestDebugFlags(tests.TestCaseInTempDir):
23
from bzrlib import debug
24
from bzrlib.config import config_filename, ensure_config_dir_exists
25
from bzrlib.transport import get_transport
26
from bzrlib.tests import TestCaseInTempDir
29
class TestDebugFlags(TestCaseInTempDir):
29
31
def test_set_debug_flags_from_config(self):
30
32
# test both combinations because configobject automatically splits up
31
33
# comma-separated lists
32
self.assertDebugFlags(['hpss', 'error'], 'debug_flags = hpss, error\n')
33
self.assertDebugFlags(['hpss'], 'debug_flags = hpss\n')
35
def assertDebugFlags(self, expected_flags, conf_bytes):
36
conf = config.GlobalStack()
37
conf.store._load_from_string(conf_bytes)
39
self.overrideAttr(debug, 'debug_flags', set())
40
debug.set_debug_flags_from_config()
41
self.assertEqual(set(expected_flags), debug.debug_flags)
34
if os.path.isfile(config_filename()):
35
# Something is wrong in environment,
36
# we risk overwriting users config
37
self.assert_(config_filename() + "exists, abort")
40
"""debug_flags = hpss, error\n""",
41
set(['hpss', 'error']))
44
"""debug_flags = hpss\n""",
47
def try_debug_flags(self, conf_bytes, expected_flags):
48
ensure_config_dir_exists()
49
f = open(config_filename(), 'wb')
54
saved_debug = set(debug.debug_flags)
55
debug.debug_flags.clear()
57
debug.set_debug_flags_from_config()
58
self.assertEqual(expected_flags,
61
# restore without rebinding the variable
62
debug.debug_flags.clear()
63
debug.debug_flags.update(saved_debug)