← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_debug.py

  • Committer: Patch Queue Manager
  • Date: 2015-09-30 16:43:21 UTC
  • mfrom: (6603.2.2 fix-keep-dirty)
  • Revision ID: pqm@pqm.ubuntu.com-20150930164321-ct2v2qnmvimqt8qf
(vila) Avoid associating dirty patch headers with the previous file in the
 patch. (Colin Watson)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_debug.py
1
 
# Copyright (C) 2009 Canonical Ltd
 
1
# Copyright (C) 2009, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
17
17
"""Tests for bzrlib.debug"""
18
18
 
19
19
 
20
 
import os
21
 
 
22
 
 
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
27
 
 
28
 
 
29
 
class TestDebugFlags(TestCaseInTempDir):
30
 
 
31
 
    def test_set_debug_flags_from_config(self):
32
 
        # test both combinations because configobject automatically splits up
33
 
        # comma-separated lists
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")
38
 
 
39
 
        self.try_debug_flags(
40
 
            """debug_flags = hpss, error\n""",
41
 
            set(['hpss', 'error']))
42
 
 
43
 
        self.try_debug_flags(
44
 
            """debug_flags = hpss\n""",
45
 
            set(['hpss']))
46
 
 
47
 
    def try_debug_flags(self, conf_bytes, expected_flags):
48
 
        ensure_config_dir_exists()
49
 
        f = open(config_filename(), 'wb')
50
 
        try:
51
 
            f.write(conf_bytes)
52
 
        finally:
53
 
            f.close()
54
 
        saved_debug = set(debug.debug_flags)
55
 
        debug.debug_flags.clear()
56
 
        try:
57
 
            debug.set_debug_flags_from_config()
58
 
            self.assertEqual(expected_flags,
59
 
                debug.debug_flags)
60
 
        finally:
61
 
            # restore without rebinding the variable
62
 
            debug.debug_flags.clear()
63
 
            debug.debug_flags.update(saved_debug)
 
20
from bzrlib import (
 
21
    config,
 
22
    debug,
 
23
    tests,
 
24
    )
 
25
 
 
26
 
 
27
class TestDebugFlags(tests.TestCaseInTempDir):
 
28
 
 
29
    def test_set_no_debug_flags_from_config(self):
 
30
        self.assertDebugFlags([], '')
 
31
 
 
32
    def test_set_single_debug_flags_from_config(self):
 
33
        self.assertDebugFlags(['hpss'], 'debug_flags = hpss\n')
 
34
 
 
35
    def test_set_multiple_debug_flags_from_config(self):
 
36
        self.assertDebugFlags(['hpss', 'error'], 'debug_flags = hpss, error\n')
 
37
 
 
38
    def assertDebugFlags(self, expected_flags, conf_bytes):
 
39
        conf = config.GlobalStack()
 
40
        conf.store._load_from_string('[DEFAULT]\n' + conf_bytes)
 
41
        conf.store.save()
 
42
        self.overrideAttr(debug, 'debug_flags', set())
 
43
        debug.set_debug_flags_from_config()
 
44
        self.assertEqual(set(expected_flags), debug.debug_flags)