/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_config.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2011-09-01 10:55:26 UTC
  • mfrom: (6110.6.3 haslayout-noversioneddirs)
  • Revision ID: pqm@pqm.ubuntu.com-20110901105526-jnqnr84fhdriyjee
(jelmer) Support repository formats without explicitly versioned directories
 in HasLayout. (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005-2011 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
19
19
from cStringIO import StringIO
20
20
import os
21
21
import sys
 
22
import threading
 
23
 
 
24
 
 
25
from testtools import matchers
22
26
 
23
27
#import bzrlib specific imports here
24
28
from bzrlib import (
31
35
    mail_client,
32
36
    ui,
33
37
    urlutils,
 
38
    remote,
34
39
    tests,
35
40
    trace,
36
 
    transport,
 
41
    )
 
42
from bzrlib.symbol_versioning import (
 
43
    deprecated_in,
 
44
    )
 
45
from bzrlib.transport import remote as transport_remote
 
46
from bzrlib.tests import (
 
47
    features,
 
48
    scenarios,
 
49
    test_server,
37
50
    )
38
51
from bzrlib.util.configobj import configobj
39
52
 
40
53
 
 
54
def lockable_config_scenarios():
 
55
    return [
 
56
        ('global',
 
57
         {'config_class': config.GlobalConfig,
 
58
          'config_args': [],
 
59
          'config_section': 'DEFAULT'}),
 
60
        ('locations',
 
61
         {'config_class': config.LocationConfig,
 
62
          'config_args': ['.'],
 
63
          'config_section': '.'}),]
 
64
 
 
65
 
 
66
load_tests = scenarios.load_tests_apply_scenarios
 
67
 
 
68
# Register helpers to build stores
 
69
config.test_store_builder_registry.register(
 
70
    'configobj', lambda test: config.IniFileStore(test.get_transport(),
 
71
                                                  'configobj.conf'))
 
72
config.test_store_builder_registry.register(
 
73
    'bazaar', lambda test: config.GlobalStore())
 
74
config.test_store_builder_registry.register(
 
75
    'location', lambda test: config.LocationStore())
 
76
 
 
77
 
 
78
def build_backing_branch(test, relpath,
 
79
                         transport_class=None, server_class=None):
 
80
    """Test helper to create a backing branch only once.
 
81
 
 
82
    Some tests needs multiple stores/stacks to check concurrent update
 
83
    behaviours. As such, they need to build different branch *objects* even if
 
84
    they share the branch on disk.
 
85
 
 
86
    :param relpath: The relative path to the branch. (Note that the helper
 
87
        should always specify the same relpath).
 
88
 
 
89
    :param transport_class: The Transport class the test needs to use.
 
90
 
 
91
    :param server_class: The server associated with the ``transport_class``
 
92
        above.
 
93
 
 
94
    Either both or neither of ``transport_class`` and ``server_class`` should
 
95
    be specified.
 
96
    """
 
97
    if transport_class is not None and server_class is not None:
 
98
        test.transport_class = transport_class
 
99
        test.transport_server = server_class
 
100
    elif not (transport_class is None and server_class is None):
 
101
        raise AssertionError('Specify both ``transport_class`` and '
 
102
                             '``server_class`` or neither of them')
 
103
    if getattr(test, 'backing_branch', None) is None:
 
104
        # First call, let's build the branch on disk
 
105
        test.backing_branch = test.make_branch(relpath)
 
106
 
 
107
 
 
108
def build_branch_store(test):
 
109
    build_backing_branch(test, 'branch')
 
110
    b = branch.Branch.open('branch')
 
111
    return config.BranchStore(b)
 
112
config.test_store_builder_registry.register('branch', build_branch_store)
 
113
 
 
114
 
 
115
def build_control_store(test):
 
116
    build_backing_branch(test, 'branch')
 
117
    b = bzrdir.BzrDir.open('branch')
 
118
    return config.ControlStore(b)
 
119
config.test_store_builder_registry.register('control', build_control_store)
 
120
 
 
121
 
 
122
def build_remote_branch_store(test):
 
123
    # There is only one permutation (but we won't be able to handle more with
 
124
    # this design anyway)
 
125
    (transport_class,
 
126
     server_class) = transport_remote.get_test_permutations()[0]
 
127
    build_backing_branch(test, 'branch', transport_class, server_class)
 
128
    b = branch.Branch.open(test.get_url('branch'))
 
129
    return config.BranchStore(b)
 
130
config.test_store_builder_registry.register('remote_branch',
 
131
                                            build_remote_branch_store)
 
132
 
 
133
 
 
134
config.test_stack_builder_registry.register(
 
135
    'bazaar', lambda test: config.GlobalStack())
 
136
config.test_stack_builder_registry.register(
 
137
    'location', lambda test: config.LocationStack('.'))
 
138
 
 
139
 
 
140
def build_branch_stack(test):
 
141
    build_backing_branch(test, 'branch')
 
142
    b = branch.Branch.open('branch')
 
143
    return config.BranchStack(b)
 
144
config.test_stack_builder_registry.register('branch', build_branch_stack)
 
145
 
 
146
 
 
147
def build_remote_branch_stack(test):
 
148
    # There is only one permutation (but we won't be able to handle more with
 
149
    # this design anyway)
 
150
    (transport_class,
 
151
     server_class) = transport_remote.get_test_permutations()[0]
 
152
    build_backing_branch(test, 'branch', transport_class, server_class)
 
153
    b = branch.Branch.open(test.get_url('branch'))
 
154
    return config.RemoteBranchStack(b)
 
155
config.test_stack_builder_registry.register('remote_branch',
 
156
                                            build_remote_branch_stack)
 
157
 
 
158
def build_remote_control_stack(test):
 
159
    # There is only one permutation (but we won't be able to handle more with
 
160
    # this design anyway)
 
161
    (transport_class,
 
162
     server_class) = transport_remote.get_test_permutations()[0]
 
163
    # We need only a bzrdir for this, not a full branch, but it's not worth
 
164
    # creating a dedicated helper to create only the bzrdir
 
165
    build_backing_branch(test, 'branch', transport_class, server_class)
 
166
    b = branch.Branch.open(test.get_url('branch'))
 
167
    return config.RemoteControlStack(b.bzrdir)
 
168
config.test_stack_builder_registry.register('remote_control',
 
169
                                            build_remote_control_stack)
 
170
 
 
171
 
41
172
sample_long_alias="log -r-15..-1 --line"
42
173
sample_config_text = u"""
43
174
[DEFAULT]
45
176
editor=vim
46
177
change_editor=vimdiff -of @new_path @old_path
47
178
gpg_signing_command=gnome-gpg
 
179
gpg_signing_key=DD4D5088
48
180
log_format=short
 
181
validate_signatures_in_log=true
 
182
acceptable_keys=amy
49
183
user_global_option=something
 
184
bzr.mergetool.sometool=sometool {base} {this} {other} -o {result}
 
185
bzr.mergetool.funkytool=funkytool "arg with spaces" {this_temp}
 
186
bzr.default_mergetool=sometool
50
187
[ALIASES]
51
188
h=help
52
189
ll=""" + sample_long_alias + "\n"
94
231
[/a/]
95
232
check_signatures=check-available
96
233
gpg_signing_command=false
 
234
gpg_signing_key=default
97
235
user_local_option=local
98
236
# test trailing / matching
99
237
[/a/*]
105
243
"""
106
244
 
107
245
 
 
246
def create_configs(test):
 
247
    """Create configuration files for a given test.
 
248
 
 
249
    This requires creating a tree (and populate the ``test.tree`` attribute)
 
250
    and its associated branch and will populate the following attributes:
 
251
 
 
252
    - branch_config: A BranchConfig for the associated branch.
 
253
 
 
254
    - locations_config : A LocationConfig for the associated branch
 
255
 
 
256
    - bazaar_config: A GlobalConfig.
 
257
 
 
258
    The tree and branch are created in a 'tree' subdirectory so the tests can
 
259
    still use the test directory to stay outside of the branch.
 
260
    """
 
261
    tree = test.make_branch_and_tree('tree')
 
262
    test.tree = tree
 
263
    test.branch_config = config.BranchConfig(tree.branch)
 
264
    test.locations_config = config.LocationConfig(tree.basedir)
 
265
    test.bazaar_config = config.GlobalConfig()
 
266
 
 
267
 
 
268
def create_configs_with_file_option(test):
 
269
    """Create configuration files with a ``file`` option set in each.
 
270
 
 
271
    This builds on ``create_configs`` and add one ``file`` option in each
 
272
    configuration with a value which allows identifying the configuration file.
 
273
    """
 
274
    create_configs(test)
 
275
    test.bazaar_config.set_user_option('file', 'bazaar')
 
276
    test.locations_config.set_user_option('file', 'locations')
 
277
    test.branch_config.set_user_option('file', 'branch')
 
278
 
 
279
 
 
280
class TestOptionsMixin:
 
281
 
 
282
    def assertOptions(self, expected, conf):
 
283
        # We don't care about the parser (as it will make tests hard to write
 
284
        # and error-prone anyway)
 
285
        self.assertThat([opt[:4] for opt in conf._get_options()],
 
286
                        matchers.Equals(expected))
 
287
 
 
288
 
108
289
class InstrumentedConfigObj(object):
109
290
    """A config obj look-enough-alike to record calls made to it."""
110
291
 
129
310
        self._calls.append(('keys',))
130
311
        return []
131
312
 
 
313
    def reload(self):
 
314
        self._calls.append(('reload',))
 
315
 
132
316
    def write(self, arg):
133
317
        self._calls.append(('write',))
134
318
 
240
424
        """
241
425
        co = config.ConfigObj()
242
426
        co['test'] = 'foo#bar'
243
 
        lines = co.write()
 
427
        outfile = StringIO()
 
428
        co.write(outfile=outfile)
 
429
        lines = outfile.getvalue().splitlines()
244
430
        self.assertEqual(lines, ['test = "foo#bar"'])
245
431
        co2 = config.ConfigObj(lines)
246
432
        self.assertEqual(co2['test'], 'foo#bar')
247
433
 
 
434
    def test_triple_quotes(self):
 
435
        # Bug #710410: if the value string has triple quotes
 
436
        # then ConfigObj versions up to 4.7.2 will quote them wrong
 
437
        # and won't able to read them back
 
438
        triple_quotes_value = '''spam
 
439
""" that's my spam """
 
440
eggs'''
 
441
        co = config.ConfigObj()
 
442
        co['test'] = triple_quotes_value
 
443
        # While writing this test another bug in ConfigObj has been found:
 
444
        # method co.write() without arguments produces list of lines
 
445
        # one option per line, and multiline values are not split
 
446
        # across multiple lines,
 
447
        # and that breaks the parsing these lines back by ConfigObj.
 
448
        # This issue only affects test, but it's better to avoid
 
449
        # `co.write()` construct at all.
 
450
        # [bialix 20110222] bug report sent to ConfigObj's author
 
451
        outfile = StringIO()
 
452
        co.write(outfile=outfile)
 
453
        output = outfile.getvalue()
 
454
        # now we're trying to read it back
 
455
        co2 = config.ConfigObj(StringIO(output))
 
456
        self.assertEquals(triple_quotes_value, co2['test'])
 
457
 
248
458
 
249
459
erroneous_config = """[section] # line 1
250
460
good=good # line 2
271
481
        config.Config()
272
482
 
273
483
    def test_no_default_editor(self):
274
 
        self.assertRaises(NotImplementedError, config.Config().get_editor)
 
484
        self.assertRaises(
 
485
            NotImplementedError,
 
486
            self.applyDeprecated, deprecated_in((2, 4, 0)),
 
487
            config.Config().get_editor)
275
488
 
276
489
    def test_user_email(self):
277
490
        my_config = InstrumentedConfig()
320
533
        my_config = config.Config()
321
534
        self.assertEqual('long', my_config.log_format())
322
535
 
 
536
    def test_acceptable_keys_default(self):
 
537
        my_config = config.Config()
 
538
        self.assertEqual(None, my_config.acceptable_keys())
 
539
 
 
540
    def test_validate_signatures_in_log_default(self):
 
541
        my_config = config.Config()
 
542
        self.assertEqual(False, my_config.validate_signatures_in_log())
 
543
 
323
544
    def test_get_change_editor(self):
324
545
        my_config = InstrumentedConfig()
325
546
        change_editor = my_config.get_change_editor('old_tree', 'new_tree')
333
554
 
334
555
    def setUp(self):
335
556
        super(TestConfigPath, self).setUp()
336
 
        os.environ['HOME'] = '/home/bogus'
337
 
        os.environ['XDG_CACHE_DIR'] = ''
 
557
        self.overrideEnv('HOME', '/home/bogus')
 
558
        self.overrideEnv('XDG_CACHE_DIR', '')
338
559
        if sys.platform == 'win32':
339
 
            os.environ['BZR_HOME'] = \
340
 
                r'C:\Documents and Settings\bogus\Application Data'
 
560
            self.overrideEnv(
 
561
                'BZR_HOME', r'C:\Documents and Settings\bogus\Application Data')
341
562
            self.bzr_home = \
342
563
                'C:/Documents and Settings/bogus/Application Data/bazaar/2.0'
343
564
        else:
350
571
        self.assertEqual(config.config_filename(),
351
572
                         self.bzr_home + '/bazaar.conf')
352
573
 
353
 
    def test_branches_config_filename(self):
354
 
        self.assertEqual(config.branches_config_filename(),
355
 
                         self.bzr_home + '/branches.conf')
356
 
 
357
574
    def test_locations_config_filename(self):
358
575
        self.assertEqual(config.locations_config_filename(),
359
576
                         self.bzr_home + '/locations.conf')
367
584
            '/home/bogus/.cache')
368
585
 
369
586
 
370
 
class TestIniConfig(tests.TestCase):
 
587
class TestXDGConfigDir(tests.TestCaseInTempDir):
 
588
    # must be in temp dir because config tests for the existence of the bazaar
 
589
    # subdirectory of $XDG_CONFIG_HOME
 
590
 
 
591
    def setUp(self):
 
592
        if sys.platform in ('darwin', 'win32'):
 
593
            raise tests.TestNotApplicable(
 
594
                'XDG config dir not used on this platform')
 
595
        super(TestXDGConfigDir, self).setUp()
 
596
        self.overrideEnv('HOME', self.test_home_dir)
 
597
        # BZR_HOME overrides everything we want to test so unset it.
 
598
        self.overrideEnv('BZR_HOME', None)
 
599
 
 
600
    def test_xdg_config_dir_exists(self):
 
601
        """When ~/.config/bazaar exists, use it as the config dir."""
 
602
        newdir = osutils.pathjoin(self.test_home_dir, '.config', 'bazaar')
 
603
        os.makedirs(newdir)
 
604
        self.assertEqual(config.config_dir(), newdir)
 
605
 
 
606
    def test_xdg_config_home(self):
 
607
        """When XDG_CONFIG_HOME is set, use it."""
 
608
        xdgconfigdir = osutils.pathjoin(self.test_home_dir, 'xdgconfig')
 
609
        self.overrideEnv('XDG_CONFIG_HOME', xdgconfigdir)
 
610
        newdir = osutils.pathjoin(xdgconfigdir, 'bazaar')
 
611
        os.makedirs(newdir)
 
612
        self.assertEqual(config.config_dir(), newdir)
 
613
 
 
614
 
 
615
class TestIniConfig(tests.TestCaseInTempDir):
371
616
 
372
617
    def make_config_parser(self, s):
373
 
        conf = config.IniBasedConfig(None)
374
 
        parser = conf._get_parser(file=StringIO(s.encode('utf-8')))
375
 
        return conf, parser
 
618
        conf = config.IniBasedConfig.from_string(s)
 
619
        return conf, conf._get_parser()
376
620
 
377
621
 
378
622
class TestIniConfigBuilding(TestIniConfig):
379
623
 
380
624
    def test_contructs(self):
381
 
        my_config = config.IniBasedConfig("nothing")
 
625
        my_config = config.IniBasedConfig()
382
626
 
383
627
    def test_from_fp(self):
384
 
        config_file = StringIO(sample_config_text.encode('utf-8'))
385
 
        my_config = config.IniBasedConfig(None)
386
 
        self.failUnless(
387
 
            isinstance(my_config._get_parser(file=config_file),
388
 
                        configobj.ConfigObj))
 
628
        my_config = config.IniBasedConfig.from_string(sample_config_text)
 
629
        self.assertIsInstance(my_config._get_parser(), configobj.ConfigObj)
389
630
 
390
631
    def test_cached(self):
 
632
        my_config = config.IniBasedConfig.from_string(sample_config_text)
 
633
        parser = my_config._get_parser()
 
634
        self.assertTrue(my_config._get_parser() is parser)
 
635
 
 
636
    def _dummy_chown(self, path, uid, gid):
 
637
        self.path, self.uid, self.gid = path, uid, gid
 
638
 
 
639
    def test_ini_config_ownership(self):
 
640
        """Ensure that chown is happening during _write_config_file"""
 
641
        self.requireFeature(features.chown_feature)
 
642
        self.overrideAttr(os, 'chown', self._dummy_chown)
 
643
        self.path = self.uid = self.gid = None
 
644
        conf = config.IniBasedConfig(file_name='./foo.conf')
 
645
        conf._write_config_file()
 
646
        self.assertEquals(self.path, './foo.conf')
 
647
        self.assertTrue(isinstance(self.uid, int))
 
648
        self.assertTrue(isinstance(self.gid, int))
 
649
 
 
650
    def test_get_filename_parameter_is_deprecated_(self):
 
651
        conf = self.callDeprecated([
 
652
            'IniBasedConfig.__init__(get_filename) was deprecated in 2.3.'
 
653
            ' Use file_name instead.'],
 
654
            config.IniBasedConfig, lambda: 'ini.conf')
 
655
        self.assertEqual('ini.conf', conf.file_name)
 
656
 
 
657
    def test_get_parser_file_parameter_is_deprecated_(self):
391
658
        config_file = StringIO(sample_config_text.encode('utf-8'))
392
 
        my_config = config.IniBasedConfig(None)
393
 
        parser = my_config._get_parser(file=config_file)
394
 
        self.failUnless(my_config._get_parser() is parser)
 
659
        conf = config.IniBasedConfig.from_string(sample_config_text)
 
660
        conf = self.callDeprecated([
 
661
            'IniBasedConfig._get_parser(file=xxx) was deprecated in 2.3.'
 
662
            ' Use IniBasedConfig(_content=xxx) instead.'],
 
663
            conf._get_parser, file=config_file)
 
664
 
 
665
 
 
666
class TestIniConfigSaving(tests.TestCaseInTempDir):
 
667
 
 
668
    def test_cant_save_without_a_file_name(self):
 
669
        conf = config.IniBasedConfig()
 
670
        self.assertRaises(AssertionError, conf._write_config_file)
 
671
 
 
672
    def test_saved_with_content(self):
 
673
        content = 'foo = bar\n'
 
674
        conf = config.IniBasedConfig.from_string(
 
675
            content, file_name='./test.conf', save=True)
 
676
        self.assertFileEqual(content, 'test.conf')
 
677
 
 
678
 
 
679
class TestIniConfigOptionExpansionDefaultValue(tests.TestCaseInTempDir):
 
680
    """What is the default value of expand for config options.
 
681
 
 
682
    This is an opt-in beta feature used to evaluate whether or not option
 
683
    references can appear in dangerous place raising exceptions, disapearing
 
684
    (and as such corrupting data) or if it's safe to activate the option by
 
685
    default.
 
686
 
 
687
    Note that these tests relies on config._expand_default_value being already
 
688
    overwritten in the parent class setUp.
 
689
    """
 
690
 
 
691
    def setUp(self):
 
692
        super(TestIniConfigOptionExpansionDefaultValue, self).setUp()
 
693
        self.config = None
 
694
        self.warnings = []
 
695
        def warning(*args):
 
696
            self.warnings.append(args[0] % args[1:])
 
697
        self.overrideAttr(trace, 'warning', warning)
 
698
 
 
699
    def get_config(self, expand):
 
700
        c = config.GlobalConfig.from_string('bzr.config.expand=%s' % (expand,),
 
701
                                            save=True)
 
702
        return c
 
703
 
 
704
    def assertExpandIs(self, expected):
 
705
        actual = config._get_expand_default_value()
 
706
        #self.config.get_user_option_as_bool('bzr.config.expand')
 
707
        self.assertEquals(expected, actual)
 
708
 
 
709
    def test_default_is_None(self):
 
710
        self.assertEquals(None, config._expand_default_value)
 
711
 
 
712
    def test_default_is_False_even_if_None(self):
 
713
        self.config = self.get_config(None)
 
714
        self.assertExpandIs(False)
 
715
 
 
716
    def test_default_is_False_even_if_invalid(self):
 
717
        self.config = self.get_config('<your choice>')
 
718
        self.assertExpandIs(False)
 
719
        # ...
 
720
        # Huh ? My choice is False ? Thanks, always happy to hear that :D
 
721
        # Wait, you've been warned !
 
722
        self.assertLength(1, self.warnings)
 
723
        self.assertEquals(
 
724
            'Value "<your choice>" is not a boolean for "bzr.config.expand"',
 
725
            self.warnings[0])
 
726
 
 
727
    def test_default_is_True(self):
 
728
        self.config = self.get_config(True)
 
729
        self.assertExpandIs(True)
 
730
 
 
731
    def test_default_is_False(self):
 
732
        self.config = self.get_config(False)
 
733
        self.assertExpandIs(False)
 
734
 
 
735
 
 
736
class TestIniConfigOptionExpansion(tests.TestCase):
 
737
    """Test option expansion from the IniConfig level.
 
738
 
 
739
    What we really want here is to test the Config level, but the class being
 
740
    abstract as far as storing values is concerned, this can't be done
 
741
    properly (yet).
 
742
    """
 
743
    # FIXME: This should be rewritten when all configs share a storage
 
744
    # implementation -- vila 2011-02-18
 
745
 
 
746
    def get_config(self, string=None):
 
747
        if string is None:
 
748
            string = ''
 
749
        c = config.IniBasedConfig.from_string(string)
 
750
        return c
 
751
 
 
752
    def assertExpansion(self, expected, conf, string, env=None):
 
753
        self.assertEquals(expected, conf.expand_options(string, env))
 
754
 
 
755
    def test_no_expansion(self):
 
756
        c = self.get_config('')
 
757
        self.assertExpansion('foo', c, 'foo')
 
758
 
 
759
    def test_env_adding_options(self):
 
760
        c = self.get_config('')
 
761
        self.assertExpansion('bar', c, '{foo}', {'foo': 'bar'})
 
762
 
 
763
    def test_env_overriding_options(self):
 
764
        c = self.get_config('foo=baz')
 
765
        self.assertExpansion('bar', c, '{foo}', {'foo': 'bar'})
 
766
 
 
767
    def test_simple_ref(self):
 
768
        c = self.get_config('foo=xxx')
 
769
        self.assertExpansion('xxx', c, '{foo}')
 
770
 
 
771
    def test_unknown_ref(self):
 
772
        c = self.get_config('')
 
773
        self.assertRaises(errors.ExpandingUnknownOption,
 
774
                          c.expand_options, '{foo}')
 
775
 
 
776
    def test_indirect_ref(self):
 
777
        c = self.get_config('''
 
778
foo=xxx
 
779
bar={foo}
 
780
''')
 
781
        self.assertExpansion('xxx', c, '{bar}')
 
782
 
 
783
    def test_embedded_ref(self):
 
784
        c = self.get_config('''
 
785
foo=xxx
 
786
bar=foo
 
787
''')
 
788
        self.assertExpansion('xxx', c, '{{bar}}')
 
789
 
 
790
    def test_simple_loop(self):
 
791
        c = self.get_config('foo={foo}')
 
792
        self.assertRaises(errors.OptionExpansionLoop, c.expand_options, '{foo}')
 
793
 
 
794
    def test_indirect_loop(self):
 
795
        c = self.get_config('''
 
796
foo={bar}
 
797
bar={baz}
 
798
baz={foo}''')
 
799
        e = self.assertRaises(errors.OptionExpansionLoop,
 
800
                              c.expand_options, '{foo}')
 
801
        self.assertEquals('foo->bar->baz', e.refs)
 
802
        self.assertEquals('{foo}', e.string)
 
803
 
 
804
    def test_list(self):
 
805
        conf = self.get_config('''
 
806
foo=start
 
807
bar=middle
 
808
baz=end
 
809
list={foo},{bar},{baz}
 
810
''')
 
811
        self.assertEquals(['start', 'middle', 'end'],
 
812
                           conf.get_user_option('list', expand=True))
 
813
 
 
814
    def test_cascading_list(self):
 
815
        conf = self.get_config('''
 
816
foo=start,{bar}
 
817
bar=middle,{baz}
 
818
baz=end
 
819
list={foo}
 
820
''')
 
821
        self.assertEquals(['start', 'middle', 'end'],
 
822
                           conf.get_user_option('list', expand=True))
 
823
 
 
824
    def test_pathological_hidden_list(self):
 
825
        conf = self.get_config('''
 
826
foo=bin
 
827
bar=go
 
828
start={foo
 
829
middle=},{
 
830
end=bar}
 
831
hidden={start}{middle}{end}
 
832
''')
 
833
        # Nope, it's either a string or a list, and the list wins as soon as a
 
834
        # ',' appears, so the string concatenation never occur.
 
835
        self.assertEquals(['{foo', '}', '{', 'bar}'],
 
836
                          conf.get_user_option('hidden', expand=True))
 
837
 
 
838
class TestLocationConfigOptionExpansion(tests.TestCaseInTempDir):
 
839
 
 
840
    def get_config(self, location, string=None):
 
841
        if string is None:
 
842
            string = ''
 
843
        # Since we don't save the config we won't strictly require to inherit
 
844
        # from TestCaseInTempDir, but an error occurs so quickly...
 
845
        c = config.LocationConfig.from_string(string, location)
 
846
        return c
 
847
 
 
848
    def test_dont_cross_unrelated_section(self):
 
849
        c = self.get_config('/another/branch/path','''
 
850
[/one/branch/path]
 
851
foo = hello
 
852
bar = {foo}/2
 
853
 
 
854
[/another/branch/path]
 
855
bar = {foo}/2
 
856
''')
 
857
        self.assertRaises(errors.ExpandingUnknownOption,
 
858
                          c.get_user_option, 'bar', expand=True)
 
859
 
 
860
    def test_cross_related_sections(self):
 
861
        c = self.get_config('/project/branch/path','''
 
862
[/project]
 
863
foo = qu
 
864
 
 
865
[/project/branch/path]
 
866
bar = {foo}ux
 
867
''')
 
868
        self.assertEquals('quux', c.get_user_option('bar', expand=True))
 
869
 
 
870
 
 
871
class TestIniBaseConfigOnDisk(tests.TestCaseInTempDir):
 
872
 
 
873
    def test_cannot_reload_without_name(self):
 
874
        conf = config.IniBasedConfig.from_string(sample_config_text)
 
875
        self.assertRaises(AssertionError, conf.reload)
 
876
 
 
877
    def test_reload_see_new_value(self):
 
878
        c1 = config.IniBasedConfig.from_string('editor=vim\n',
 
879
                                               file_name='./test/conf')
 
880
        c1._write_config_file()
 
881
        c2 = config.IniBasedConfig.from_string('editor=emacs\n',
 
882
                                               file_name='./test/conf')
 
883
        c2._write_config_file()
 
884
        self.assertEqual('vim', c1.get_user_option('editor'))
 
885
        self.assertEqual('emacs', c2.get_user_option('editor'))
 
886
        # Make sure we get the Right value
 
887
        c1.reload()
 
888
        self.assertEqual('emacs', c1.get_user_option('editor'))
 
889
 
 
890
 
 
891
class TestLockableConfig(tests.TestCaseInTempDir):
 
892
 
 
893
    scenarios = lockable_config_scenarios()
 
894
 
 
895
    # Set by load_tests
 
896
    config_class = None
 
897
    config_args = None
 
898
    config_section = None
 
899
 
 
900
    def setUp(self):
 
901
        super(TestLockableConfig, self).setUp()
 
902
        self._content = '[%s]\none=1\ntwo=2\n' % (self.config_section,)
 
903
        self.config = self.create_config(self._content)
 
904
 
 
905
    def get_existing_config(self):
 
906
        return self.config_class(*self.config_args)
 
907
 
 
908
    def create_config(self, content):
 
909
        kwargs = dict(save=True)
 
910
        c = self.config_class.from_string(content, *self.config_args, **kwargs)
 
911
        return c
 
912
 
 
913
    def test_simple_read_access(self):
 
914
        self.assertEquals('1', self.config.get_user_option('one'))
 
915
 
 
916
    def test_simple_write_access(self):
 
917
        self.config.set_user_option('one', 'one')
 
918
        self.assertEquals('one', self.config.get_user_option('one'))
 
919
 
 
920
    def test_listen_to_the_last_speaker(self):
 
921
        c1 = self.config
 
922
        c2 = self.get_existing_config()
 
923
        c1.set_user_option('one', 'ONE')
 
924
        c2.set_user_option('two', 'TWO')
 
925
        self.assertEquals('ONE', c1.get_user_option('one'))
 
926
        self.assertEquals('TWO', c2.get_user_option('two'))
 
927
        # The second update respect the first one
 
928
        self.assertEquals('ONE', c2.get_user_option('one'))
 
929
 
 
930
    def test_last_speaker_wins(self):
 
931
        # If the same config is not shared, the same variable modified twice
 
932
        # can only see a single result.
 
933
        c1 = self.config
 
934
        c2 = self.get_existing_config()
 
935
        c1.set_user_option('one', 'c1')
 
936
        c2.set_user_option('one', 'c2')
 
937
        self.assertEquals('c2', c2._get_user_option('one'))
 
938
        # The first modification is still available until another refresh
 
939
        # occur
 
940
        self.assertEquals('c1', c1._get_user_option('one'))
 
941
        c1.set_user_option('two', 'done')
 
942
        self.assertEquals('c2', c1._get_user_option('one'))
 
943
 
 
944
    def test_writes_are_serialized(self):
 
945
        c1 = self.config
 
946
        c2 = self.get_existing_config()
 
947
 
 
948
        # We spawn a thread that will pause *during* the write
 
949
        before_writing = threading.Event()
 
950
        after_writing = threading.Event()
 
951
        writing_done = threading.Event()
 
952
        c1_orig = c1._write_config_file
 
953
        def c1_write_config_file():
 
954
            before_writing.set()
 
955
            c1_orig()
 
956
            # The lock is held. We wait for the main thread to decide when to
 
957
            # continue
 
958
            after_writing.wait()
 
959
        c1._write_config_file = c1_write_config_file
 
960
        def c1_set_option():
 
961
            c1.set_user_option('one', 'c1')
 
962
            writing_done.set()
 
963
        t1 = threading.Thread(target=c1_set_option)
 
964
        # Collect the thread after the test
 
965
        self.addCleanup(t1.join)
 
966
        # Be ready to unblock the thread if the test goes wrong
 
967
        self.addCleanup(after_writing.set)
 
968
        t1.start()
 
969
        before_writing.wait()
 
970
        self.assertTrue(c1._lock.is_held)
 
971
        self.assertRaises(errors.LockContention,
 
972
                          c2.set_user_option, 'one', 'c2')
 
973
        self.assertEquals('c1', c1.get_user_option('one'))
 
974
        # Let the lock be released
 
975
        after_writing.set()
 
976
        writing_done.wait()
 
977
        c2.set_user_option('one', 'c2')
 
978
        self.assertEquals('c2', c2.get_user_option('one'))
 
979
 
 
980
    def test_read_while_writing(self):
 
981
       c1 = self.config
 
982
       # We spawn a thread that will pause *during* the write
 
983
       ready_to_write = threading.Event()
 
984
       do_writing = threading.Event()
 
985
       writing_done = threading.Event()
 
986
       c1_orig = c1._write_config_file
 
987
       def c1_write_config_file():
 
988
           ready_to_write.set()
 
989
           # The lock is held. We wait for the main thread to decide when to
 
990
           # continue
 
991
           do_writing.wait()
 
992
           c1_orig()
 
993
           writing_done.set()
 
994
       c1._write_config_file = c1_write_config_file
 
995
       def c1_set_option():
 
996
           c1.set_user_option('one', 'c1')
 
997
       t1 = threading.Thread(target=c1_set_option)
 
998
       # Collect the thread after the test
 
999
       self.addCleanup(t1.join)
 
1000
       # Be ready to unblock the thread if the test goes wrong
 
1001
       self.addCleanup(do_writing.set)
 
1002
       t1.start()
 
1003
       # Ensure the thread is ready to write
 
1004
       ready_to_write.wait()
 
1005
       self.assertTrue(c1._lock.is_held)
 
1006
       self.assertEquals('c1', c1.get_user_option('one'))
 
1007
       # If we read during the write, we get the old value
 
1008
       c2 = self.get_existing_config()
 
1009
       self.assertEquals('1', c2.get_user_option('one'))
 
1010
       # Let the writing occur and ensure it occurred
 
1011
       do_writing.set()
 
1012
       writing_done.wait()
 
1013
       # Now we get the updated value
 
1014
       c3 = self.get_existing_config()
 
1015
       self.assertEquals('c1', c3.get_user_option('one'))
395
1016
 
396
1017
 
397
1018
class TestGetUserOptionAs(TestIniConfig):
430
1051
        # automatically cast to list
431
1052
        self.assertEqual(['x'], get_list('one_item'))
432
1053
 
 
1054
    def test_get_user_option_as_int_from_SI(self):
 
1055
        conf, parser = self.make_config_parser("""
 
1056
plain = 100
 
1057
si_k = 5k,
 
1058
si_kb = 5kb,
 
1059
si_m = 5M,
 
1060
si_mb = 5MB,
 
1061
si_g = 5g,
 
1062
si_gb = 5gB,
 
1063
""")
 
1064
        get_si = conf.get_user_option_as_int_from_SI
 
1065
        self.assertEqual(100, get_si('plain'))
 
1066
        self.assertEqual(5000, get_si('si_k'))
 
1067
        self.assertEqual(5000, get_si('si_kb'))
 
1068
        self.assertEqual(5000000, get_si('si_m'))
 
1069
        self.assertEqual(5000000, get_si('si_mb'))
 
1070
        self.assertEqual(5000000000, get_si('si_g'))
 
1071
        self.assertEqual(5000000000, get_si('si_gb'))
 
1072
        self.assertEqual(None, get_si('non-exist'))
 
1073
        self.assertEqual(42, get_si('non-exist-with-default',  42))
433
1074
 
434
1075
class TestSupressWarning(TestIniConfig):
435
1076
 
462
1103
            parser = my_config._get_parser()
463
1104
        finally:
464
1105
            config.ConfigObj = oldparserclass
465
 
        self.failUnless(isinstance(parser, InstrumentedConfigObj))
 
1106
        self.assertIsInstance(parser, InstrumentedConfigObj)
466
1107
        self.assertEqual(parser._calls, [('__init__', config.config_filename(),
467
1108
                                          'utf-8')])
468
1109
 
479
1120
        my_config = config.BranchConfig(branch)
480
1121
        location_config = my_config._get_location_config()
481
1122
        self.assertEqual(branch.base, location_config.location)
482
 
        self.failUnless(location_config is my_config._get_location_config())
 
1123
        self.assertIs(location_config, my_config._get_location_config())
483
1124
 
484
1125
    def test_get_config(self):
485
1126
        """The Branch.get_config method works properly"""
505
1146
        branch = self.make_branch('branch')
506
1147
        self.assertEqual('branch', branch.nick)
507
1148
 
508
 
        locations = config.locations_config_filename()
509
 
        config.ensure_config_dir_exists()
510
1149
        local_url = urlutils.local_path_to_url('branch')
511
 
        open(locations, 'wb').write('[%s]\nnickname = foobar'
512
 
                                    % (local_url,))
 
1150
        conf = config.LocationConfig.from_string(
 
1151
            '[%s]\nnickname = foobar' % (local_url,),
 
1152
            local_url, save=True)
513
1153
        self.assertEqual('foobar', branch.nick)
514
1154
 
515
1155
    def test_config_local_path(self):
517
1157
        branch = self.make_branch('branch')
518
1158
        self.assertEqual('branch', branch.nick)
519
1159
 
520
 
        locations = config.locations_config_filename()
521
 
        config.ensure_config_dir_exists()
522
 
        open(locations, 'wb').write('[%s/branch]\nnickname = barry'
523
 
                                    % (osutils.getcwd().encode('utf8'),))
 
1160
        local_path = osutils.getcwd().encode('utf8')
 
1161
        conf = config.LocationConfig.from_string(
 
1162
            '[%s/branch]\nnickname = barry' % (local_path,),
 
1163
            'branch',  save=True)
524
1164
        self.assertEqual('barry', branch.nick)
525
1165
 
526
1166
    def test_config_creates_local(self):
527
1167
        """Creating a new entry in config uses a local path."""
528
1168
        branch = self.make_branch('branch', format='knit')
529
1169
        branch.set_push_location('http://foobar')
530
 
        locations = config.locations_config_filename()
531
1170
        local_path = osutils.getcwd().encode('utf8')
532
1171
        # Surprisingly ConfigObj doesn't create a trailing newline
533
 
        self.check_file_contents(locations,
 
1172
        self.check_file_contents(config.locations_config_filename(),
534
1173
                                 '[%s/branch]\n'
535
1174
                                 'push_location = http://foobar\n'
536
1175
                                 'push_location:policy = norecurse\n'
541
1180
        self.assertEqual('!repo', b.get_config().get_nickname())
542
1181
 
543
1182
    def test_warn_if_masked(self):
544
 
        _warning = trace.warning
545
1183
        warnings = []
546
1184
        def warning(*args):
547
1185
            warnings.append(args[0] % args[1:])
 
1186
        self.overrideAttr(trace, 'warning', warning)
548
1187
 
549
1188
        def set_option(store, warn_masked=True):
550
1189
            warnings[:] = []
556
1195
            else:
557
1196
                self.assertEqual(1, len(warnings))
558
1197
                self.assertEqual(warning, warnings[0])
559
 
        trace.warning = warning
560
 
        try:
561
 
            branch = self.make_branch('.')
562
 
            conf = branch.get_config()
563
 
            set_option(config.STORE_GLOBAL)
564
 
            assertWarning(None)
565
 
            set_option(config.STORE_BRANCH)
566
 
            assertWarning(None)
567
 
            set_option(config.STORE_GLOBAL)
568
 
            assertWarning('Value "4" is masked by "3" from branch.conf')
569
 
            set_option(config.STORE_GLOBAL, warn_masked=False)
570
 
            assertWarning(None)
571
 
            set_option(config.STORE_LOCATION)
572
 
            assertWarning(None)
573
 
            set_option(config.STORE_BRANCH)
574
 
            assertWarning('Value "3" is masked by "0" from locations.conf')
575
 
            set_option(config.STORE_BRANCH, warn_masked=False)
576
 
            assertWarning(None)
577
 
        finally:
578
 
            trace.warning = _warning
579
 
 
580
 
 
581
 
class TestGlobalConfigItems(tests.TestCase):
 
1198
        branch = self.make_branch('.')
 
1199
        conf = branch.get_config()
 
1200
        set_option(config.STORE_GLOBAL)
 
1201
        assertWarning(None)
 
1202
        set_option(config.STORE_BRANCH)
 
1203
        assertWarning(None)
 
1204
        set_option(config.STORE_GLOBAL)
 
1205
        assertWarning('Value "4" is masked by "3" from branch.conf')
 
1206
        set_option(config.STORE_GLOBAL, warn_masked=False)
 
1207
        assertWarning(None)
 
1208
        set_option(config.STORE_LOCATION)
 
1209
        assertWarning(None)
 
1210
        set_option(config.STORE_BRANCH)
 
1211
        assertWarning('Value "3" is masked by "0" from locations.conf')
 
1212
        set_option(config.STORE_BRANCH, warn_masked=False)
 
1213
        assertWarning(None)
 
1214
 
 
1215
 
 
1216
class TestGlobalConfigItems(tests.TestCaseInTempDir):
582
1217
 
583
1218
    def test_user_id(self):
584
 
        config_file = StringIO(sample_config_text.encode('utf-8'))
585
 
        my_config = config.GlobalConfig()
586
 
        my_config._parser = my_config._get_parser(file=config_file)
 
1219
        my_config = config.GlobalConfig.from_string(sample_config_text)
587
1220
        self.assertEqual(u"Erik B\u00e5gfors <erik@bagfors.nu>",
588
1221
                         my_config._get_user_id())
589
1222
 
590
1223
    def test_absent_user_id(self):
591
 
        config_file = StringIO("")
592
1224
        my_config = config.GlobalConfig()
593
 
        my_config._parser = my_config._get_parser(file=config_file)
594
1225
        self.assertEqual(None, my_config._get_user_id())
595
1226
 
596
1227
    def test_configured_editor(self):
597
 
        config_file = StringIO(sample_config_text.encode('utf-8'))
598
 
        my_config = config.GlobalConfig()
599
 
        my_config._parser = my_config._get_parser(file=config_file)
600
 
        self.assertEqual("vim", my_config.get_editor())
 
1228
        my_config = config.GlobalConfig.from_string(sample_config_text)
 
1229
        editor = self.applyDeprecated(
 
1230
            deprecated_in((2, 4, 0)), my_config.get_editor)
 
1231
        self.assertEqual('vim', editor)
601
1232
 
602
1233
    def test_signatures_always(self):
603
 
        config_file = StringIO(sample_always_signatures)
604
 
        my_config = config.GlobalConfig()
605
 
        my_config._parser = my_config._get_parser(file=config_file)
 
1234
        my_config = config.GlobalConfig.from_string(sample_always_signatures)
606
1235
        self.assertEqual(config.CHECK_NEVER,
607
1236
                         my_config.signature_checking())
608
1237
        self.assertEqual(config.SIGN_ALWAYS,
610
1239
        self.assertEqual(True, my_config.signature_needed())
611
1240
 
612
1241
    def test_signatures_if_possible(self):
613
 
        config_file = StringIO(sample_maybe_signatures)
614
 
        my_config = config.GlobalConfig()
615
 
        my_config._parser = my_config._get_parser(file=config_file)
 
1242
        my_config = config.GlobalConfig.from_string(sample_maybe_signatures)
616
1243
        self.assertEqual(config.CHECK_NEVER,
617
1244
                         my_config.signature_checking())
618
1245
        self.assertEqual(config.SIGN_WHEN_REQUIRED,
620
1247
        self.assertEqual(False, my_config.signature_needed())
621
1248
 
622
1249
    def test_signatures_ignore(self):
623
 
        config_file = StringIO(sample_ignore_signatures)
624
 
        my_config = config.GlobalConfig()
625
 
        my_config._parser = my_config._get_parser(file=config_file)
 
1250
        my_config = config.GlobalConfig.from_string(sample_ignore_signatures)
626
1251
        self.assertEqual(config.CHECK_ALWAYS,
627
1252
                         my_config.signature_checking())
628
1253
        self.assertEqual(config.SIGN_NEVER,
630
1255
        self.assertEqual(False, my_config.signature_needed())
631
1256
 
632
1257
    def _get_sample_config(self):
633
 
        config_file = StringIO(sample_config_text.encode('utf-8'))
634
 
        my_config = config.GlobalConfig()
635
 
        my_config._parser = my_config._get_parser(file=config_file)
 
1258
        my_config = config.GlobalConfig.from_string(sample_config_text)
636
1259
        return my_config
637
1260
 
638
1261
    def test_gpg_signing_command(self):
640
1263
        self.assertEqual("gnome-gpg", my_config.gpg_signing_command())
641
1264
        self.assertEqual(False, my_config.signature_needed())
642
1265
 
 
1266
    def test_gpg_signing_key(self):
 
1267
        my_config = self._get_sample_config()
 
1268
        self.assertEqual("DD4D5088", my_config.gpg_signing_key())
 
1269
 
643
1270
    def _get_empty_config(self):
644
 
        config_file = StringIO("")
645
1271
        my_config = config.GlobalConfig()
646
 
        my_config._parser = my_config._get_parser(file=config_file)
647
1272
        return my_config
648
1273
 
649
1274
    def test_gpg_signing_command_unset(self):
667
1292
        my_config = self._get_sample_config()
668
1293
        self.assertEqual("short", my_config.log_format())
669
1294
 
 
1295
    def test_configured_acceptable_keys(self):
 
1296
        my_config = self._get_sample_config()
 
1297
        self.assertEqual("amy", my_config.acceptable_keys())
 
1298
 
 
1299
    def test_configured_validate_signatures_in_log(self):
 
1300
        my_config = self._get_sample_config()
 
1301
        self.assertEqual(True, my_config.validate_signatures_in_log())
 
1302
 
670
1303
    def test_get_alias(self):
671
1304
        my_config = self._get_sample_config()
672
1305
        self.assertEqual('help', my_config.get_alias('h'))
699
1332
        change_editor = my_config.get_change_editor('old', 'new')
700
1333
        self.assertIs(None, change_editor)
701
1334
 
 
1335
    def test_get_merge_tools(self):
 
1336
        conf = self._get_sample_config()
 
1337
        tools = conf.get_merge_tools()
 
1338
        self.log(repr(tools))
 
1339
        self.assertEqual(
 
1340
            {u'funkytool' : u'funkytool "arg with spaces" {this_temp}',
 
1341
            u'sometool' : u'sometool {base} {this} {other} -o {result}'},
 
1342
            tools)
 
1343
 
 
1344
    def test_get_merge_tools_empty(self):
 
1345
        conf = self._get_empty_config()
 
1346
        tools = conf.get_merge_tools()
 
1347
        self.assertEqual({}, tools)
 
1348
 
 
1349
    def test_find_merge_tool(self):
 
1350
        conf = self._get_sample_config()
 
1351
        cmdline = conf.find_merge_tool('sometool')
 
1352
        self.assertEqual('sometool {base} {this} {other} -o {result}', cmdline)
 
1353
 
 
1354
    def test_find_merge_tool_not_found(self):
 
1355
        conf = self._get_sample_config()
 
1356
        cmdline = conf.find_merge_tool('DOES NOT EXIST')
 
1357
        self.assertIs(cmdline, None)
 
1358
 
 
1359
    def test_find_merge_tool_known(self):
 
1360
        conf = self._get_empty_config()
 
1361
        cmdline = conf.find_merge_tool('kdiff3')
 
1362
        self.assertEquals('kdiff3 {base} {this} {other} -o {result}', cmdline)
 
1363
 
 
1364
    def test_find_merge_tool_override_known(self):
 
1365
        conf = self._get_empty_config()
 
1366
        conf.set_user_option('bzr.mergetool.kdiff3', 'kdiff3 blah')
 
1367
        cmdline = conf.find_merge_tool('kdiff3')
 
1368
        self.assertEqual('kdiff3 blah', cmdline)
 
1369
 
702
1370
 
703
1371
class TestGlobalConfigSavingOptions(tests.TestCaseInTempDir):
704
1372
 
722
1390
        self.assertIs(None, new_config.get_alias('commit'))
723
1391
 
724
1392
 
725
 
class TestLocationConfig(tests.TestCaseInTempDir):
 
1393
class TestLocationConfig(tests.TestCaseInTempDir, TestOptionsMixin):
726
1394
 
727
1395
    def test_constructs(self):
728
1396
        my_config = config.LocationConfig('http://example.com')
740
1408
            parser = my_config._get_parser()
741
1409
        finally:
742
1410
            config.ConfigObj = oldparserclass
743
 
        self.failUnless(isinstance(parser, InstrumentedConfigObj))
 
1411
        self.assertIsInstance(parser, InstrumentedConfigObj)
744
1412
        self.assertEqual(parser._calls,
745
1413
                         [('__init__', config.locations_config_filename(),
746
1414
                           'utf-8')])
747
 
        config.ensure_config_dir_exists()
748
 
        #os.mkdir(config.config_dir())
749
 
        f = file(config.branches_config_filename(), 'wb')
750
 
        f.write('')
751
 
        f.close()
752
 
        oldparserclass = config.ConfigObj
753
 
        config.ConfigObj = InstrumentedConfigObj
754
 
        try:
755
 
            my_config = config.LocationConfig('http://www.example.com')
756
 
            parser = my_config._get_parser()
757
 
        finally:
758
 
            config.ConfigObj = oldparserclass
759
1415
 
760
1416
    def test_get_global_config(self):
761
1417
        my_config = config.BranchConfig(FakeBranch('http://example.com'))
762
1418
        global_config = my_config._get_global_config()
763
 
        self.failUnless(isinstance(global_config, config.GlobalConfig))
764
 
        self.failUnless(global_config is my_config._get_global_config())
 
1419
        self.assertIsInstance(global_config, config.GlobalConfig)
 
1420
        self.assertIs(global_config, my_config._get_global_config())
 
1421
 
 
1422
    def assertLocationMatching(self, expected):
 
1423
        self.assertEqual(expected,
 
1424
                         list(self.my_location_config._get_matching_sections()))
765
1425
 
766
1426
    def test__get_matching_sections_no_match(self):
767
1427
        self.get_branch_config('/')
768
 
        self.assertEqual([], self.my_location_config._get_matching_sections())
 
1428
        self.assertLocationMatching([])
769
1429
 
770
1430
    def test__get_matching_sections_exact(self):
771
1431
        self.get_branch_config('http://www.example.com')
772
 
        self.assertEqual([('http://www.example.com', '')],
773
 
                         self.my_location_config._get_matching_sections())
 
1432
        self.assertLocationMatching([('http://www.example.com', '')])
774
1433
 
775
1434
    def test__get_matching_sections_suffix_does_not(self):
776
1435
        self.get_branch_config('http://www.example.com-com')
777
 
        self.assertEqual([], self.my_location_config._get_matching_sections())
 
1436
        self.assertLocationMatching([])
778
1437
 
779
1438
    def test__get_matching_sections_subdir_recursive(self):
780
1439
        self.get_branch_config('http://www.example.com/com')
781
 
        self.assertEqual([('http://www.example.com', 'com')],
782
 
                         self.my_location_config._get_matching_sections())
 
1440
        self.assertLocationMatching([('http://www.example.com', 'com')])
783
1441
 
784
1442
    def test__get_matching_sections_ignoreparent(self):
785
1443
        self.get_branch_config('http://www.example.com/ignoreparent')
786
 
        self.assertEqual([('http://www.example.com/ignoreparent', '')],
787
 
                         self.my_location_config._get_matching_sections())
 
1444
        self.assertLocationMatching([('http://www.example.com/ignoreparent',
 
1445
                                      '')])
788
1446
 
789
1447
    def test__get_matching_sections_ignoreparent_subdir(self):
790
1448
        self.get_branch_config(
791
1449
            'http://www.example.com/ignoreparent/childbranch')
792
 
        self.assertEqual([('http://www.example.com/ignoreparent',
793
 
                           'childbranch')],
794
 
                         self.my_location_config._get_matching_sections())
 
1450
        self.assertLocationMatching([('http://www.example.com/ignoreparent',
 
1451
                                      'childbranch')])
795
1452
 
796
1453
    def test__get_matching_sections_subdir_trailing_slash(self):
797
1454
        self.get_branch_config('/b')
798
 
        self.assertEqual([('/b/', '')],
799
 
                         self.my_location_config._get_matching_sections())
 
1455
        self.assertLocationMatching([('/b/', '')])
800
1456
 
801
1457
    def test__get_matching_sections_subdir_child(self):
802
1458
        self.get_branch_config('/a/foo')
803
 
        self.assertEqual([('/a/*', ''), ('/a/', 'foo')],
804
 
                         self.my_location_config._get_matching_sections())
 
1459
        self.assertLocationMatching([('/a/*', ''), ('/a/', 'foo')])
805
1460
 
806
1461
    def test__get_matching_sections_subdir_child_child(self):
807
1462
        self.get_branch_config('/a/foo/bar')
808
 
        self.assertEqual([('/a/*', 'bar'), ('/a/', 'foo/bar')],
809
 
                         self.my_location_config._get_matching_sections())
 
1463
        self.assertLocationMatching([('/a/*', 'bar'), ('/a/', 'foo/bar')])
810
1464
 
811
1465
    def test__get_matching_sections_trailing_slash_with_children(self):
812
1466
        self.get_branch_config('/a/')
813
 
        self.assertEqual([('/a/', '')],
814
 
                         self.my_location_config._get_matching_sections())
 
1467
        self.assertLocationMatching([('/a/', '')])
815
1468
 
816
1469
    def test__get_matching_sections_explicit_over_glob(self):
817
1470
        # XXX: 2006-09-08 jamesh
819
1472
        # was a config section for '/a/?', it would get precedence
820
1473
        # over '/a/c'.
821
1474
        self.get_branch_config('/a/c')
822
 
        self.assertEqual([('/a/c', ''), ('/a/*', ''), ('/a/', 'c')],
823
 
                         self.my_location_config._get_matching_sections())
 
1475
        self.assertLocationMatching([('/a/c', ''), ('/a/*', ''), ('/a/', 'c')])
824
1476
 
825
1477
    def test__get_option_policy_normal(self):
826
1478
        self.get_branch_config('http://www.example.com')
848
1500
            'http://www.example.com', 'appendpath_option'),
849
1501
            config.POLICY_APPENDPATH)
850
1502
 
 
1503
    def test__get_options_with_policy(self):
 
1504
        self.get_branch_config('/dir/subdir',
 
1505
                               location_config="""\
 
1506
[/dir]
 
1507
other_url = /other-dir
 
1508
other_url:policy = appendpath
 
1509
[/dir/subdir]
 
1510
other_url = /other-subdir
 
1511
""")
 
1512
        self.assertOptions(
 
1513
            [(u'other_url', u'/other-subdir', u'/dir/subdir', 'locations'),
 
1514
             (u'other_url', u'/other-dir', u'/dir', 'locations'),
 
1515
             (u'other_url:policy', u'appendpath', u'/dir', 'locations')],
 
1516
            self.my_location_config)
 
1517
 
851
1518
    def test_location_without_username(self):
852
1519
        self.get_branch_config('http://www.example.com/ignoreparent')
853
1520
        self.assertEqual(u'Erik B\u00e5gfors <erik@bagfors.nu>',
895
1562
        self.get_branch_config('/a')
896
1563
        self.assertEqual("false", self.my_config.gpg_signing_command())
897
1564
 
 
1565
    def test_gpg_signing_key(self):
 
1566
        self.get_branch_config('/b')
 
1567
        self.assertEqual("DD4D5088", self.my_config.gpg_signing_key())
 
1568
 
 
1569
    def test_gpg_signing_key_default(self):
 
1570
        self.get_branch_config('/a')
 
1571
        self.assertEqual("erik@bagfors.nu", self.my_config.gpg_signing_key())
 
1572
 
898
1573
    def test_get_user_option_global(self):
899
1574
        self.get_branch_config('/a')
900
1575
        self.assertEqual('something',
989
1664
        self.assertEqual('bzrlib.tests.test_config.post_commit',
990
1665
                         self.my_config.post_commit())
991
1666
 
992
 
    def get_branch_config(self, location, global_config=None):
 
1667
    def get_branch_config(self, location, global_config=None,
 
1668
                          location_config=None):
 
1669
        my_branch = FakeBranch(location)
993
1670
        if global_config is None:
994
 
            global_file = StringIO(sample_config_text.encode('utf-8'))
995
 
        else:
996
 
            global_file = StringIO(global_config.encode('utf-8'))
997
 
        branches_file = StringIO(sample_branches_text.encode('utf-8'))
998
 
        self.my_config = config.BranchConfig(FakeBranch(location))
999
 
        # Force location config to use specified file
1000
 
        self.my_location_config = self.my_config._get_location_config()
1001
 
        self.my_location_config._get_parser(branches_file)
1002
 
        # Force global config to use specified file
1003
 
        self.my_config._get_global_config()._get_parser(global_file)
 
1671
            global_config = sample_config_text
 
1672
        if location_config is None:
 
1673
            location_config = sample_branches_text
 
1674
 
 
1675
        my_global_config = config.GlobalConfig.from_string(global_config,
 
1676
                                                           save=True)
 
1677
        my_location_config = config.LocationConfig.from_string(
 
1678
            location_config, my_branch.base, save=True)
 
1679
        my_config = config.BranchConfig(my_branch)
 
1680
        self.my_config = my_config
 
1681
        self.my_location_config = my_config._get_location_config()
1004
1682
 
1005
1683
    def test_set_user_setting_sets_and_saves(self):
1006
1684
        self.get_branch_config('/a/c')
1007
1685
        record = InstrumentedConfigObj("foo")
1008
1686
        self.my_location_config._parser = record
1009
1687
 
1010
 
        real_mkdir = os.mkdir
1011
 
        self.created = False
1012
 
        def checked_mkdir(path, mode=0777):
1013
 
            self.log('making directory: %s', path)
1014
 
            real_mkdir(path, mode)
1015
 
            self.created = True
1016
 
 
1017
 
        os.mkdir = checked_mkdir
1018
 
        try:
1019
 
            self.callDeprecated(['The recurse option is deprecated as of '
1020
 
                                 '0.14.  The section "/a/c" has been '
1021
 
                                 'converted to use policies.'],
1022
 
                                self.my_config.set_user_option,
1023
 
                                'foo', 'bar', store=config.STORE_LOCATION)
1024
 
        finally:
1025
 
            os.mkdir = real_mkdir
1026
 
 
1027
 
        self.failUnless(self.created, 'Failed to create ~/.bazaar')
1028
 
        self.assertEqual([('__contains__', '/a/c'),
 
1688
        self.callDeprecated(['The recurse option is deprecated as of '
 
1689
                             '0.14.  The section "/a/c" has been '
 
1690
                             'converted to use policies.'],
 
1691
                            self.my_config.set_user_option,
 
1692
                            'foo', 'bar', store=config.STORE_LOCATION)
 
1693
        self.assertEqual([('reload',),
 
1694
                          ('__contains__', '/a/c'),
1029
1695
                          ('__contains__', '/a/c/'),
1030
1696
                          ('__setitem__', '/a/c', {}),
1031
1697
                          ('__getitem__', '/a/c'),
1060
1726
        self.assertEqual('bzr', my_config.get_bzr_remote_path())
1061
1727
        my_config.set_user_option('bzr_remote_path', '/path-bzr')
1062
1728
        self.assertEqual('/path-bzr', my_config.get_bzr_remote_path())
1063
 
        os.environ['BZR_REMOTE_PATH'] = '/environ-bzr'
 
1729
        self.overrideEnv('BZR_REMOTE_PATH', '/environ-bzr')
1064
1730
        self.assertEqual('/environ-bzr', my_config.get_bzr_remote_path())
1065
1731
 
1066
1732
 
1074
1740
option = exact
1075
1741
"""
1076
1742
 
1077
 
 
1078
1743
class TestBranchConfigItems(tests.TestCaseInTempDir):
1079
1744
 
1080
1745
    def get_branch_config(self, global_config=None, location=None,
1081
1746
                          location_config=None, branch_data_config=None):
1082
 
        my_config = config.BranchConfig(FakeBranch(location))
 
1747
        my_branch = FakeBranch(location)
1083
1748
        if global_config is not None:
1084
 
            global_file = StringIO(global_config.encode('utf-8'))
1085
 
            my_config._get_global_config()._get_parser(global_file)
1086
 
        self.my_location_config = my_config._get_location_config()
 
1749
            my_global_config = config.GlobalConfig.from_string(global_config,
 
1750
                                                               save=True)
1087
1751
        if location_config is not None:
1088
 
            location_file = StringIO(location_config.encode('utf-8'))
1089
 
            self.my_location_config._get_parser(location_file)
 
1752
            my_location_config = config.LocationConfig.from_string(
 
1753
                location_config, my_branch.base, save=True)
 
1754
        my_config = config.BranchConfig(my_branch)
1090
1755
        if branch_data_config is not None:
1091
1756
            my_config.branch.control_files.files['branch.conf'] = \
1092
1757
                branch_data_config
1106
1771
                         my_config.username())
1107
1772
 
1108
1773
    def test_not_set_in_branch(self):
1109
 
        my_config = self.get_branch_config(sample_config_text)
 
1774
        my_config = self.get_branch_config(global_config=sample_config_text)
1110
1775
        self.assertEqual(u"Erik B\u00e5gfors <erik@bagfors.nu>",
1111
1776
                         my_config._get_user_id())
1112
1777
        my_config.branch.control_files.files['email'] = "John"
1113
1778
        self.assertEqual("John", my_config._get_user_id())
1114
1779
 
1115
1780
    def test_BZR_EMAIL_OVERRIDES(self):
1116
 
        os.environ['BZR_EMAIL'] = "Robert Collins <robertc@example.org>"
 
1781
        self.overrideEnv('BZR_EMAIL', "Robert Collins <robertc@example.org>")
1117
1782
        branch = FakeBranch()
1118
1783
        my_config = config.BranchConfig(branch)
1119
1784
        self.assertEqual("Robert Collins <robertc@example.org>",
1136
1801
 
1137
1802
    def test_gpg_signing_command(self):
1138
1803
        my_config = self.get_branch_config(
 
1804
            global_config=sample_config_text,
1139
1805
            # branch data cannot set gpg_signing_command
1140
1806
            branch_data_config="gpg_signing_command=pgp")
1141
 
        config_file = StringIO(sample_config_text.encode('utf-8'))
1142
 
        my_config._get_global_config()._get_parser(config_file)
1143
1807
        self.assertEqual('gnome-gpg', my_config.gpg_signing_command())
1144
1808
 
1145
1809
    def test_get_user_option_global(self):
1146
 
        branch = FakeBranch()
1147
 
        my_config = config.BranchConfig(branch)
1148
 
        config_file = StringIO(sample_config_text.encode('utf-8'))
1149
 
        (my_config._get_global_config()._get_parser(config_file))
 
1810
        my_config = self.get_branch_config(global_config=sample_config_text)
1150
1811
        self.assertEqual('something',
1151
1812
                         my_config.get_user_option('user_global_option'))
1152
1813
 
1153
1814
    def test_post_commit_default(self):
1154
 
        branch = FakeBranch()
1155
 
        my_config = self.get_branch_config(sample_config_text, '/a/c',
1156
 
                                           sample_branches_text)
 
1815
        my_config = self.get_branch_config(global_config=sample_config_text,
 
1816
                                      location='/a/c',
 
1817
                                      location_config=sample_branches_text)
1157
1818
        self.assertEqual(my_config.branch.base, '/a/c')
1158
1819
        self.assertEqual('bzrlib.tests.test_config.post_commit',
1159
1820
                         my_config.post_commit())
1160
1821
        my_config.set_user_option('post_commit', 'rmtree_root')
1161
 
        # post-commit is ignored when bresent in branch data
 
1822
        # post-commit is ignored when present in branch data
1162
1823
        self.assertEqual('bzrlib.tests.test_config.post_commit',
1163
1824
                         my_config.post_commit())
1164
1825
        my_config.set_user_option('post_commit', 'rmtree_root',
1166
1827
        self.assertEqual('rmtree_root', my_config.post_commit())
1167
1828
 
1168
1829
    def test_config_precedence(self):
 
1830
        # FIXME: eager test, luckily no persitent config file makes it fail
 
1831
        # -- vila 20100716
1169
1832
        my_config = self.get_branch_config(global_config=precedence_global)
1170
1833
        self.assertEqual(my_config.get_user_option('option'), 'global')
1171
1834
        my_config = self.get_branch_config(global_config=precedence_global,
1172
 
                                      branch_data_config=precedence_branch)
 
1835
                                           branch_data_config=precedence_branch)
1173
1836
        self.assertEqual(my_config.get_user_option('option'), 'branch')
1174
 
        my_config = self.get_branch_config(global_config=precedence_global,
1175
 
                                      branch_data_config=precedence_branch,
1176
 
                                      location_config=precedence_location)
 
1837
        my_config = self.get_branch_config(
 
1838
            global_config=precedence_global,
 
1839
            branch_data_config=precedence_branch,
 
1840
            location_config=precedence_location)
1177
1841
        self.assertEqual(my_config.get_user_option('option'), 'recurse')
1178
 
        my_config = self.get_branch_config(global_config=precedence_global,
1179
 
                                      branch_data_config=precedence_branch,
1180
 
                                      location_config=precedence_location,
1181
 
                                      location='http://example.com/specific')
 
1842
        my_config = self.get_branch_config(
 
1843
            global_config=precedence_global,
 
1844
            branch_data_config=precedence_branch,
 
1845
            location_config=precedence_location,
 
1846
            location='http://example.com/specific')
1182
1847
        self.assertEqual(my_config.get_user_option('option'), 'exact')
1183
1848
 
1184
1849
    def test_get_mail_client(self):
1274
1939
 
1275
1940
class TestTransportConfig(tests.TestCaseWithTransport):
1276
1941
 
 
1942
    def test_load_utf8(self):
 
1943
        """Ensure we can load an utf8-encoded file."""
 
1944
        t = self.get_transport()
 
1945
        unicode_user = u'b\N{Euro Sign}ar'
 
1946
        unicode_content = u'user=%s' % (unicode_user,)
 
1947
        utf8_content = unicode_content.encode('utf8')
 
1948
        # Store the raw content in the config file
 
1949
        t.put_bytes('foo.conf', utf8_content)
 
1950
        conf = config.TransportConfig(t, 'foo.conf')
 
1951
        self.assertEquals(unicode_user, conf.get_option('user'))
 
1952
 
 
1953
    def test_load_non_ascii(self):
 
1954
        """Ensure we display a proper error on non-ascii, non utf-8 content."""
 
1955
        t = self.get_transport()
 
1956
        t.put_bytes('foo.conf', 'user=foo\n#\xff\n')
 
1957
        conf = config.TransportConfig(t, 'foo.conf')
 
1958
        self.assertRaises(errors.ConfigContentError, conf._get_configobj)
 
1959
 
 
1960
    def test_load_erroneous_content(self):
 
1961
        """Ensure we display a proper error on content that can't be parsed."""
 
1962
        t = self.get_transport()
 
1963
        t.put_bytes('foo.conf', '[open_section\n')
 
1964
        conf = config.TransportConfig(t, 'foo.conf')
 
1965
        self.assertRaises(errors.ParseConfigError, conf._get_configobj)
 
1966
 
 
1967
    def test_load_permission_denied(self):
 
1968
        """Ensure we get an empty config file if the file is inaccessible."""
 
1969
        warnings = []
 
1970
        def warning(*args):
 
1971
            warnings.append(args[0] % args[1:])
 
1972
        self.overrideAttr(trace, 'warning', warning)
 
1973
 
 
1974
        class DenyingTransport(object):
 
1975
 
 
1976
            def __init__(self, base):
 
1977
                self.base = base
 
1978
 
 
1979
            def get_bytes(self, relpath):
 
1980
                raise errors.PermissionDenied(relpath, "")
 
1981
 
 
1982
        cfg = config.TransportConfig(
 
1983
            DenyingTransport("nonexisting://"), 'control.conf')
 
1984
        self.assertIs(None, cfg.get_option('non-existant', 'SECTION'))
 
1985
        self.assertEquals(
 
1986
            warnings,
 
1987
            [u'Permission denied while trying to open configuration file '
 
1988
             u'nonexisting:///control.conf.'])
 
1989
 
1277
1990
    def test_get_value(self):
1278
1991
        """Test that retreiving a value from a section is possible"""
1279
 
        bzrdir_config = config.TransportConfig(transport.get_transport('.'),
 
1992
        bzrdir_config = config.TransportConfig(self.get_transport('.'),
1280
1993
                                               'control.conf')
1281
1994
        bzrdir_config.set_option('value', 'key', 'SECTION')
1282
1995
        bzrdir_config.set_option('value2', 'key2')
1312
2025
        self.assertIs(None, bzrdir_config.get_default_stack_on())
1313
2026
 
1314
2027
 
 
2028
class TestOldConfigHooks(tests.TestCaseWithTransport):
 
2029
 
 
2030
    def setUp(self):
 
2031
        super(TestOldConfigHooks, self).setUp()
 
2032
        create_configs_with_file_option(self)
 
2033
 
 
2034
    def assertGetHook(self, conf, name, value):
 
2035
        calls = []
 
2036
        def hook(*args):
 
2037
            calls.append(args)
 
2038
        config.OldConfigHooks.install_named_hook('get', hook, None)
 
2039
        self.addCleanup(
 
2040
            config.OldConfigHooks.uninstall_named_hook, 'get', None)
 
2041
        self.assertLength(0, calls)
 
2042
        actual_value = conf.get_user_option(name)
 
2043
        self.assertEquals(value, actual_value)
 
2044
        self.assertLength(1, calls)
 
2045
        self.assertEquals((conf, name, value), calls[0])
 
2046
 
 
2047
    def test_get_hook_bazaar(self):
 
2048
        self.assertGetHook(self.bazaar_config, 'file', 'bazaar')
 
2049
 
 
2050
    def test_get_hook_locations(self):
 
2051
        self.assertGetHook(self.locations_config, 'file', 'locations')
 
2052
 
 
2053
    def test_get_hook_branch(self):
 
2054
        # Since locations masks branch, we define a different option
 
2055
        self.branch_config.set_user_option('file2', 'branch')
 
2056
        self.assertGetHook(self.branch_config, 'file2', 'branch')
 
2057
 
 
2058
    def assertSetHook(self, conf, name, value):
 
2059
        calls = []
 
2060
        def hook(*args):
 
2061
            calls.append(args)
 
2062
        config.OldConfigHooks.install_named_hook('set', hook, None)
 
2063
        self.addCleanup(
 
2064
            config.OldConfigHooks.uninstall_named_hook, 'set', None)
 
2065
        self.assertLength(0, calls)
 
2066
        conf.set_user_option(name, value)
 
2067
        self.assertLength(1, calls)
 
2068
        # We can't assert the conf object below as different configs use
 
2069
        # different means to implement set_user_option and we care only about
 
2070
        # coverage here.
 
2071
        self.assertEquals((name, value), calls[0][1:])
 
2072
 
 
2073
    def test_set_hook_bazaar(self):
 
2074
        self.assertSetHook(self.bazaar_config, 'foo', 'bazaar')
 
2075
 
 
2076
    def test_set_hook_locations(self):
 
2077
        self.assertSetHook(self.locations_config, 'foo', 'locations')
 
2078
 
 
2079
    def test_set_hook_branch(self):
 
2080
        self.assertSetHook(self.branch_config, 'foo', 'branch')
 
2081
 
 
2082
    def assertRemoveHook(self, conf, name, section_name=None):
 
2083
        calls = []
 
2084
        def hook(*args):
 
2085
            calls.append(args)
 
2086
        config.OldConfigHooks.install_named_hook('remove', hook, None)
 
2087
        self.addCleanup(
 
2088
            config.OldConfigHooks.uninstall_named_hook, 'remove', None)
 
2089
        self.assertLength(0, calls)
 
2090
        conf.remove_user_option(name, section_name)
 
2091
        self.assertLength(1, calls)
 
2092
        # We can't assert the conf object below as different configs use
 
2093
        # different means to implement remove_user_option and we care only about
 
2094
        # coverage here.
 
2095
        self.assertEquals((name,), calls[0][1:])
 
2096
 
 
2097
    def test_remove_hook_bazaar(self):
 
2098
        self.assertRemoveHook(self.bazaar_config, 'file')
 
2099
 
 
2100
    def test_remove_hook_locations(self):
 
2101
        self.assertRemoveHook(self.locations_config, 'file',
 
2102
                              self.locations_config.location)
 
2103
 
 
2104
    def test_remove_hook_branch(self):
 
2105
        self.assertRemoveHook(self.branch_config, 'file')
 
2106
 
 
2107
    def assertLoadHook(self, name, conf_class, *conf_args):
 
2108
        calls = []
 
2109
        def hook(*args):
 
2110
            calls.append(args)
 
2111
        config.OldConfigHooks.install_named_hook('load', hook, None)
 
2112
        self.addCleanup(
 
2113
            config.OldConfigHooks.uninstall_named_hook, 'load', None)
 
2114
        self.assertLength(0, calls)
 
2115
        # Build a config
 
2116
        conf = conf_class(*conf_args)
 
2117
        # Access an option to trigger a load
 
2118
        conf.get_user_option(name)
 
2119
        self.assertLength(1, calls)
 
2120
        # Since we can't assert about conf, we just use the number of calls ;-/
 
2121
 
 
2122
    def test_load_hook_bazaar(self):
 
2123
        self.assertLoadHook('file', config.GlobalConfig)
 
2124
 
 
2125
    def test_load_hook_locations(self):
 
2126
        self.assertLoadHook('file', config.LocationConfig, self.tree.basedir)
 
2127
 
 
2128
    def test_load_hook_branch(self):
 
2129
        self.assertLoadHook('file', config.BranchConfig, self.tree.branch)
 
2130
 
 
2131
    def assertSaveHook(self, conf):
 
2132
        calls = []
 
2133
        def hook(*args):
 
2134
            calls.append(args)
 
2135
        config.OldConfigHooks.install_named_hook('save', hook, None)
 
2136
        self.addCleanup(
 
2137
            config.OldConfigHooks.uninstall_named_hook, 'save', None)
 
2138
        self.assertLength(0, calls)
 
2139
        # Setting an option triggers a save
 
2140
        conf.set_user_option('foo', 'bar')
 
2141
        self.assertLength(1, calls)
 
2142
        # Since we can't assert about conf, we just use the number of calls ;-/
 
2143
 
 
2144
    def test_save_hook_bazaar(self):
 
2145
        self.assertSaveHook(self.bazaar_config)
 
2146
 
 
2147
    def test_save_hook_locations(self):
 
2148
        self.assertSaveHook(self.locations_config)
 
2149
 
 
2150
    def test_save_hook_branch(self):
 
2151
        self.assertSaveHook(self.branch_config)
 
2152
 
 
2153
 
 
2154
class TestOldConfigHooksForRemote(tests.TestCaseWithTransport):
 
2155
    """Tests config hooks for remote configs.
 
2156
 
 
2157
    No tests for the remove hook as this is not implemented there.
 
2158
    """
 
2159
 
 
2160
    def setUp(self):
 
2161
        super(TestOldConfigHooksForRemote, self).setUp()
 
2162
        self.transport_server = test_server.SmartTCPServer_for_testing
 
2163
        create_configs_with_file_option(self)
 
2164
 
 
2165
    def assertGetHook(self, conf, name, value):
 
2166
        calls = []
 
2167
        def hook(*args):
 
2168
            calls.append(args)
 
2169
        config.OldConfigHooks.install_named_hook('get', hook, None)
 
2170
        self.addCleanup(
 
2171
            config.OldConfigHooks.uninstall_named_hook, 'get', None)
 
2172
        self.assertLength(0, calls)
 
2173
        actual_value = conf.get_option(name)
 
2174
        self.assertEquals(value, actual_value)
 
2175
        self.assertLength(1, calls)
 
2176
        self.assertEquals((conf, name, value), calls[0])
 
2177
 
 
2178
    def test_get_hook_remote_branch(self):
 
2179
        remote_branch = branch.Branch.open(self.get_url('tree'))
 
2180
        self.assertGetHook(remote_branch._get_config(), 'file', 'branch')
 
2181
 
 
2182
    def test_get_hook_remote_bzrdir(self):
 
2183
        remote_bzrdir = bzrdir.BzrDir.open(self.get_url('tree'))
 
2184
        conf = remote_bzrdir._get_config()
 
2185
        conf.set_option('remotedir', 'file')
 
2186
        self.assertGetHook(conf, 'file', 'remotedir')
 
2187
 
 
2188
    def assertSetHook(self, conf, name, value):
 
2189
        calls = []
 
2190
        def hook(*args):
 
2191
            calls.append(args)
 
2192
        config.OldConfigHooks.install_named_hook('set', hook, None)
 
2193
        self.addCleanup(
 
2194
            config.OldConfigHooks.uninstall_named_hook, 'set', None)
 
2195
        self.assertLength(0, calls)
 
2196
        conf.set_option(value, name)
 
2197
        self.assertLength(1, calls)
 
2198
        # We can't assert the conf object below as different configs use
 
2199
        # different means to implement set_user_option and we care only about
 
2200
        # coverage here.
 
2201
        self.assertEquals((name, value), calls[0][1:])
 
2202
 
 
2203
    def test_set_hook_remote_branch(self):
 
2204
        remote_branch = branch.Branch.open(self.get_url('tree'))
 
2205
        self.addCleanup(remote_branch.lock_write().unlock)
 
2206
        self.assertSetHook(remote_branch._get_config(), 'file', 'remote')
 
2207
 
 
2208
    def test_set_hook_remote_bzrdir(self):
 
2209
        remote_branch = branch.Branch.open(self.get_url('tree'))
 
2210
        self.addCleanup(remote_branch.lock_write().unlock)
 
2211
        remote_bzrdir = bzrdir.BzrDir.open(self.get_url('tree'))
 
2212
        self.assertSetHook(remote_bzrdir._get_config(), 'file', 'remotedir')
 
2213
 
 
2214
    def assertLoadHook(self, expected_nb_calls, name, conf_class, *conf_args):
 
2215
        calls = []
 
2216
        def hook(*args):
 
2217
            calls.append(args)
 
2218
        config.OldConfigHooks.install_named_hook('load', hook, None)
 
2219
        self.addCleanup(
 
2220
            config.OldConfigHooks.uninstall_named_hook, 'load', None)
 
2221
        self.assertLength(0, calls)
 
2222
        # Build a config
 
2223
        conf = conf_class(*conf_args)
 
2224
        # Access an option to trigger a load
 
2225
        conf.get_option(name)
 
2226
        self.assertLength(expected_nb_calls, calls)
 
2227
        # Since we can't assert about conf, we just use the number of calls ;-/
 
2228
 
 
2229
    def test_load_hook_remote_branch(self):
 
2230
        remote_branch = branch.Branch.open(self.get_url('tree'))
 
2231
        self.assertLoadHook(1, 'file', remote.RemoteBranchConfig, remote_branch)
 
2232
 
 
2233
    def test_load_hook_remote_bzrdir(self):
 
2234
        remote_bzrdir = bzrdir.BzrDir.open(self.get_url('tree'))
 
2235
        # The config file doesn't exist, set an option to force its creation
 
2236
        conf = remote_bzrdir._get_config()
 
2237
        conf.set_option('remotedir', 'file')
 
2238
        # We get one call for the server and one call for the client, this is
 
2239
        # caused by the differences in implementations betwen
 
2240
        # SmartServerBzrDirRequestConfigFile (in smart/bzrdir.py) and
 
2241
        # SmartServerBranchGetConfigFile (in smart/branch.py)
 
2242
        self.assertLoadHook(2 ,'file', remote.RemoteBzrDirConfig, remote_bzrdir)
 
2243
 
 
2244
    def assertSaveHook(self, conf):
 
2245
        calls = []
 
2246
        def hook(*args):
 
2247
            calls.append(args)
 
2248
        config.OldConfigHooks.install_named_hook('save', hook, None)
 
2249
        self.addCleanup(
 
2250
            config.OldConfigHooks.uninstall_named_hook, 'save', None)
 
2251
        self.assertLength(0, calls)
 
2252
        # Setting an option triggers a save
 
2253
        conf.set_option('foo', 'bar')
 
2254
        self.assertLength(1, calls)
 
2255
        # Since we can't assert about conf, we just use the number of calls ;-/
 
2256
 
 
2257
    def test_save_hook_remote_branch(self):
 
2258
        remote_branch = branch.Branch.open(self.get_url('tree'))
 
2259
        self.addCleanup(remote_branch.lock_write().unlock)
 
2260
        self.assertSaveHook(remote_branch._get_config())
 
2261
 
 
2262
    def test_save_hook_remote_bzrdir(self):
 
2263
        remote_branch = branch.Branch.open(self.get_url('tree'))
 
2264
        self.addCleanup(remote_branch.lock_write().unlock)
 
2265
        remote_bzrdir = bzrdir.BzrDir.open(self.get_url('tree'))
 
2266
        self.assertSaveHook(remote_bzrdir._get_config())
 
2267
 
 
2268
 
 
2269
class TestOption(tests.TestCase):
 
2270
 
 
2271
    def test_default_value(self):
 
2272
        opt = config.Option('foo', default='bar')
 
2273
        self.assertEquals('bar', opt.get_default())
 
2274
 
 
2275
    def test_default_value_from_env(self):
 
2276
        opt = config.Option('foo', default='bar', default_from_env=['FOO'])
 
2277
        self.overrideEnv('FOO', 'quux')
 
2278
        # Env variable provides a default taking over the option one
 
2279
        self.assertEquals('quux', opt.get_default())
 
2280
 
 
2281
    def test_first_default_value_from_env_wins(self):
 
2282
        opt = config.Option('foo', default='bar',
 
2283
                            default_from_env=['NO_VALUE', 'FOO', 'BAZ'])
 
2284
        self.overrideEnv('FOO', 'foo')
 
2285
        self.overrideEnv('BAZ', 'baz')
 
2286
        # The first env var set wins
 
2287
        self.assertEquals('foo', opt.get_default())
 
2288
 
 
2289
    def test_not_supported_list_default_value(self):
 
2290
        self.assertRaises(AssertionError, config.Option, 'foo', default=[1])
 
2291
 
 
2292
    def test_not_supported_object_default_value(self):
 
2293
        self.assertRaises(AssertionError, config.Option, 'foo',
 
2294
                          default=object())
 
2295
 
 
2296
 
 
2297
class TestOptionConverterMixin(object):
 
2298
 
 
2299
    def assertConverted(self, expected, opt, value):
 
2300
        self.assertEquals(expected, opt.convert_from_unicode(value))
 
2301
 
 
2302
    def assertWarns(self, opt, value):
 
2303
        warnings = []
 
2304
        def warning(*args):
 
2305
            warnings.append(args[0] % args[1:])
 
2306
        self.overrideAttr(trace, 'warning', warning)
 
2307
        self.assertEquals(None, opt.convert_from_unicode(value))
 
2308
        self.assertLength(1, warnings)
 
2309
        self.assertEquals(
 
2310
            'Value "%s" is not valid for "%s"' % (value, opt.name),
 
2311
            warnings[0])
 
2312
 
 
2313
    def assertErrors(self, opt, value):
 
2314
        self.assertRaises(errors.ConfigOptionValueError,
 
2315
                          opt.convert_from_unicode, value)
 
2316
 
 
2317
    def assertConvertInvalid(self, opt, invalid_value):
 
2318
        opt.invalid = None
 
2319
        self.assertEquals(None, opt.convert_from_unicode(invalid_value))
 
2320
        opt.invalid = 'warning'
 
2321
        self.assertWarns(opt, invalid_value)
 
2322
        opt.invalid = 'error'
 
2323
        self.assertErrors(opt, invalid_value)
 
2324
 
 
2325
 
 
2326
class TestOptionWithBooleanConverter(tests.TestCase, TestOptionConverterMixin):
 
2327
 
 
2328
    def get_option(self):
 
2329
        return config.Option('foo', help='A boolean.',
 
2330
                             from_unicode=config.bool_from_store)
 
2331
 
 
2332
    def test_convert_invalid(self):
 
2333
        opt = self.get_option()
 
2334
        # A string that is not recognized as a boolean
 
2335
        self.assertConvertInvalid(opt, u'invalid-boolean')
 
2336
        # A list of strings is never recognized as a boolean
 
2337
        self.assertConvertInvalid(opt, [u'not', u'a', u'boolean'])
 
2338
 
 
2339
    def test_convert_valid(self):
 
2340
        opt = self.get_option()
 
2341
        self.assertConverted(True, opt, u'True')
 
2342
        self.assertConverted(True, opt, u'1')
 
2343
        self.assertConverted(False, opt, u'False')
 
2344
 
 
2345
 
 
2346
class TestOptionWithIntegerConverter(tests.TestCase, TestOptionConverterMixin):
 
2347
 
 
2348
    def get_option(self):
 
2349
        return config.Option('foo', help='An integer.',
 
2350
                             from_unicode=config.int_from_store)
 
2351
 
 
2352
    def test_convert_invalid(self):
 
2353
        opt = self.get_option()
 
2354
        # A string that is not recognized as an integer
 
2355
        self.assertConvertInvalid(opt, u'forty-two')
 
2356
        # A list of strings is never recognized as an integer
 
2357
        self.assertConvertInvalid(opt, [u'a', u'list'])
 
2358
 
 
2359
    def test_convert_valid(self):
 
2360
        opt = self.get_option()
 
2361
        self.assertConverted(16, opt, u'16')
 
2362
 
 
2363
class TestOptionWithListConverter(tests.TestCase, TestOptionConverterMixin):
 
2364
 
 
2365
    def get_option(self):
 
2366
        return config.Option('foo', help='A list.',
 
2367
                             from_unicode=config.list_from_store)
 
2368
 
 
2369
    def test_convert_invalid(self):
 
2370
        # No string is invalid as all forms can be converted to a list
 
2371
        pass
 
2372
 
 
2373
    def test_convert_valid(self):
 
2374
        opt = self.get_option()
 
2375
        # An empty string is an empty list
 
2376
        self.assertConverted([], opt, '') # Using a bare str() just in case
 
2377
        self.assertConverted([], opt, u'')
 
2378
        # A boolean
 
2379
        self.assertConverted([u'True'], opt, u'True')
 
2380
        # An integer
 
2381
        self.assertConverted([u'42'], opt, u'42')
 
2382
        # A single string
 
2383
        self.assertConverted([u'bar'], opt, u'bar')
 
2384
        # A list remains a list (configObj will turn a string containing commas
 
2385
        # into a list, but that's not what we're testing here)
 
2386
        self.assertConverted([u'foo', u'1', u'True'],
 
2387
                             opt, [u'foo', u'1', u'True'])
 
2388
 
 
2389
 
 
2390
class TestOptionRegistry(tests.TestCase):
 
2391
 
 
2392
    def setUp(self):
 
2393
        super(TestOptionRegistry, self).setUp()
 
2394
        # Always start with an empty registry
 
2395
        self.overrideAttr(config, 'option_registry', config.OptionRegistry())
 
2396
        self.registry = config.option_registry
 
2397
 
 
2398
    def test_register(self):
 
2399
        opt = config.Option('foo')
 
2400
        self.registry.register(opt)
 
2401
        self.assertIs(opt, self.registry.get('foo'))
 
2402
 
 
2403
    def test_registered_help(self):
 
2404
        opt = config.Option('foo', help='A simple option')
 
2405
        self.registry.register(opt)
 
2406
        self.assertEquals('A simple option', self.registry.get_help('foo'))
 
2407
 
 
2408
    lazy_option = config.Option('lazy_foo', help='Lazy help')
 
2409
 
 
2410
    def test_register_lazy(self):
 
2411
        self.registry.register_lazy('lazy_foo', self.__module__,
 
2412
                                    'TestOptionRegistry.lazy_option')
 
2413
        self.assertIs(self.lazy_option, self.registry.get('lazy_foo'))
 
2414
 
 
2415
    def test_registered_lazy_help(self):
 
2416
        self.registry.register_lazy('lazy_foo', self.__module__,
 
2417
                                    'TestOptionRegistry.lazy_option')
 
2418
        self.assertEquals('Lazy help', self.registry.get_help('lazy_foo'))
 
2419
 
 
2420
 
 
2421
class TestRegisteredOptions(tests.TestCase):
 
2422
    """All registered options should verify some constraints."""
 
2423
 
 
2424
    scenarios = [(key, {'option_name': key, 'option': option}) for key, option
 
2425
                 in config.option_registry.iteritems()]
 
2426
 
 
2427
    def setUp(self):
 
2428
        super(TestRegisteredOptions, self).setUp()
 
2429
        self.registry = config.option_registry
 
2430
 
 
2431
    def test_proper_name(self):
 
2432
        # An option should be registered under its own name, this can't be
 
2433
        # checked at registration time for the lazy ones.
 
2434
        self.assertEquals(self.option_name, self.option.name)
 
2435
 
 
2436
    def test_help_is_set(self):
 
2437
        option_help = self.registry.get_help(self.option_name)
 
2438
        self.assertNotEquals(None, option_help)
 
2439
        # Come on, think about the user, he really wants to know what the
 
2440
        # option is about
 
2441
        self.assertIsNot(None, option_help)
 
2442
        self.assertNotEquals('', option_help)
 
2443
 
 
2444
 
 
2445
class TestSection(tests.TestCase):
 
2446
 
 
2447
    # FIXME: Parametrize so that all sections produced by Stores run these
 
2448
    # tests -- vila 2011-04-01
 
2449
 
 
2450
    def test_get_a_value(self):
 
2451
        a_dict = dict(foo='bar')
 
2452
        section = config.Section('myID', a_dict)
 
2453
        self.assertEquals('bar', section.get('foo'))
 
2454
 
 
2455
    def test_get_unknown_option(self):
 
2456
        a_dict = dict()
 
2457
        section = config.Section(None, a_dict)
 
2458
        self.assertEquals('out of thin air',
 
2459
                          section.get('foo', 'out of thin air'))
 
2460
 
 
2461
    def test_options_is_shared(self):
 
2462
        a_dict = dict()
 
2463
        section = config.Section(None, a_dict)
 
2464
        self.assertIs(a_dict, section.options)
 
2465
 
 
2466
 
 
2467
class TestMutableSection(tests.TestCase):
 
2468
 
 
2469
    # FIXME: Parametrize so that all sections (including os.environ and the
 
2470
    # ones produced by Stores) run these tests -- vila 2011-04-01
 
2471
 
 
2472
    def test_set(self):
 
2473
        a_dict = dict(foo='bar')
 
2474
        section = config.MutableSection('myID', a_dict)
 
2475
        section.set('foo', 'new_value')
 
2476
        self.assertEquals('new_value', section.get('foo'))
 
2477
        # The change appears in the shared section
 
2478
        self.assertEquals('new_value', a_dict.get('foo'))
 
2479
        # We keep track of the change
 
2480
        self.assertTrue('foo' in section.orig)
 
2481
        self.assertEquals('bar', section.orig.get('foo'))
 
2482
 
 
2483
    def test_set_preserve_original_once(self):
 
2484
        a_dict = dict(foo='bar')
 
2485
        section = config.MutableSection('myID', a_dict)
 
2486
        section.set('foo', 'first_value')
 
2487
        section.set('foo', 'second_value')
 
2488
        # We keep track of the original value
 
2489
        self.assertTrue('foo' in section.orig)
 
2490
        self.assertEquals('bar', section.orig.get('foo'))
 
2491
 
 
2492
    def test_remove(self):
 
2493
        a_dict = dict(foo='bar')
 
2494
        section = config.MutableSection('myID', a_dict)
 
2495
        section.remove('foo')
 
2496
        # We get None for unknown options via the default value
 
2497
        self.assertEquals(None, section.get('foo'))
 
2498
        # Or we just get the default value
 
2499
        self.assertEquals('unknown', section.get('foo', 'unknown'))
 
2500
        self.assertFalse('foo' in section.options)
 
2501
        # We keep track of the deletion
 
2502
        self.assertTrue('foo' in section.orig)
 
2503
        self.assertEquals('bar', section.orig.get('foo'))
 
2504
 
 
2505
    def test_remove_new_option(self):
 
2506
        a_dict = dict()
 
2507
        section = config.MutableSection('myID', a_dict)
 
2508
        section.set('foo', 'bar')
 
2509
        section.remove('foo')
 
2510
        self.assertFalse('foo' in section.options)
 
2511
        # The option didn't exist initially so it we need to keep track of it
 
2512
        # with a special value
 
2513
        self.assertTrue('foo' in section.orig)
 
2514
        self.assertEquals(config._NewlyCreatedOption, section.orig['foo'])
 
2515
 
 
2516
 
 
2517
class TestStore(tests.TestCaseWithTransport):
 
2518
 
 
2519
    def assertSectionContent(self, expected, section):
 
2520
        """Assert that some options have the proper values in a section."""
 
2521
        expected_name, expected_options = expected
 
2522
        self.assertEquals(expected_name, section.id)
 
2523
        self.assertEquals(
 
2524
            expected_options,
 
2525
            dict([(k, section.get(k)) for k in expected_options.keys()]))
 
2526
 
 
2527
 
 
2528
class TestReadonlyStore(TestStore):
 
2529
 
 
2530
    scenarios = [(key, {'get_store': builder}) for key, builder
 
2531
                 in config.test_store_builder_registry.iteritems()]
 
2532
 
 
2533
    def setUp(self):
 
2534
        super(TestReadonlyStore, self).setUp()
 
2535
 
 
2536
    def test_building_delays_load(self):
 
2537
        store = self.get_store(self)
 
2538
        self.assertEquals(False, store.is_loaded())
 
2539
        store._load_from_string('')
 
2540
        self.assertEquals(True, store.is_loaded())
 
2541
 
 
2542
    def test_get_no_sections_for_empty(self):
 
2543
        store = self.get_store(self)
 
2544
        store._load_from_string('')
 
2545
        self.assertEquals([], list(store.get_sections()))
 
2546
 
 
2547
    def test_get_default_section(self):
 
2548
        store = self.get_store(self)
 
2549
        store._load_from_string('foo=bar')
 
2550
        sections = list(store.get_sections())
 
2551
        self.assertLength(1, sections)
 
2552
        self.assertSectionContent((None, {'foo': 'bar'}), sections[0])
 
2553
 
 
2554
    def test_get_named_section(self):
 
2555
        store = self.get_store(self)
 
2556
        store._load_from_string('[baz]\nfoo=bar')
 
2557
        sections = list(store.get_sections())
 
2558
        self.assertLength(1, sections)
 
2559
        self.assertSectionContent(('baz', {'foo': 'bar'}), sections[0])
 
2560
 
 
2561
    def test_load_from_string_fails_for_non_empty_store(self):
 
2562
        store = self.get_store(self)
 
2563
        store._load_from_string('foo=bar')
 
2564
        self.assertRaises(AssertionError, store._load_from_string, 'bar=baz')
 
2565
 
 
2566
 
 
2567
class TestIniFileStoreContent(tests.TestCaseWithTransport):
 
2568
    """Simulate loading a config store without content of various encodings.
 
2569
 
 
2570
    All files produced by bzr are in utf8 content.
 
2571
 
 
2572
    Users may modify them manually and end up with a file that can't be
 
2573
    loaded. We need to issue proper error messages in this case.
 
2574
    """
 
2575
 
 
2576
    invalid_utf8_char = '\xff'
 
2577
 
 
2578
    def test_load_utf8(self):
 
2579
        """Ensure we can load an utf8-encoded file."""
 
2580
        t = self.get_transport()
 
2581
        # From http://pad.lv/799212
 
2582
        unicode_user = u'b\N{Euro Sign}ar'
 
2583
        unicode_content = u'user=%s' % (unicode_user,)
 
2584
        utf8_content = unicode_content.encode('utf8')
 
2585
        # Store the raw content in the config file
 
2586
        t.put_bytes('foo.conf', utf8_content)
 
2587
        store = config.IniFileStore(t, 'foo.conf')
 
2588
        store.load()
 
2589
        stack = config.Stack([store.get_sections], store)
 
2590
        self.assertEquals(unicode_user, stack.get('user'))
 
2591
 
 
2592
    def test_load_non_ascii(self):
 
2593
        """Ensure we display a proper error on non-ascii, non utf-8 content."""
 
2594
        t = self.get_transport()
 
2595
        t.put_bytes('foo.conf', 'user=foo\n#%s\n' % (self.invalid_utf8_char,))
 
2596
        store = config.IniFileStore(t, 'foo.conf')
 
2597
        self.assertRaises(errors.ConfigContentError, store.load)
 
2598
 
 
2599
    def test_load_erroneous_content(self):
 
2600
        """Ensure we display a proper error on content that can't be parsed."""
 
2601
        t = self.get_transport()
 
2602
        t.put_bytes('foo.conf', '[open_section\n')
 
2603
        store = config.IniFileStore(t, 'foo.conf')
 
2604
        self.assertRaises(errors.ParseConfigError, store.load)
 
2605
 
 
2606
    def test_load_permission_denied(self):
 
2607
        """Ensure we get warned when trying to load an inaccessible file."""
 
2608
        warnings = []
 
2609
        def warning(*args):
 
2610
            warnings.append(args[0] % args[1:])
 
2611
        self.overrideAttr(trace, 'warning', warning)
 
2612
 
 
2613
        t = self.get_transport()
 
2614
 
 
2615
        def get_bytes(relpath):
 
2616
            raise errors.PermissionDenied(relpath, "")
 
2617
        t.get_bytes = get_bytes
 
2618
        store = config.IniFileStore(t, 'foo.conf')
 
2619
        self.assertRaises(errors.PermissionDenied, store.load)
 
2620
        self.assertEquals(
 
2621
            warnings,
 
2622
            [u'Permission denied while trying to load configuration store %s.'
 
2623
             % store.external_url()])
 
2624
 
 
2625
 
 
2626
class TestIniConfigContent(tests.TestCaseWithTransport):
 
2627
    """Simulate loading a IniBasedConfig without content of various encodings.
 
2628
 
 
2629
    All files produced by bzr are in utf8 content.
 
2630
 
 
2631
    Users may modify them manually and end up with a file that can't be
 
2632
    loaded. We need to issue proper error messages in this case.
 
2633
    """
 
2634
 
 
2635
    invalid_utf8_char = '\xff'
 
2636
 
 
2637
    def test_load_utf8(self):
 
2638
        """Ensure we can load an utf8-encoded file."""
 
2639
        # From http://pad.lv/799212
 
2640
        unicode_user = u'b\N{Euro Sign}ar'
 
2641
        unicode_content = u'user=%s' % (unicode_user,)
 
2642
        utf8_content = unicode_content.encode('utf8')
 
2643
        # Store the raw content in the config file
 
2644
        with open('foo.conf', 'wb') as f:
 
2645
            f.write(utf8_content)
 
2646
        conf = config.IniBasedConfig(file_name='foo.conf')
 
2647
        self.assertEquals(unicode_user, conf.get_user_option('user'))
 
2648
 
 
2649
    def test_load_badly_encoded_content(self):
 
2650
        """Ensure we display a proper error on non-ascii, non utf-8 content."""
 
2651
        with open('foo.conf', 'wb') as f:
 
2652
            f.write('user=foo\n#%s\n' % (self.invalid_utf8_char,))
 
2653
        conf = config.IniBasedConfig(file_name='foo.conf')
 
2654
        self.assertRaises(errors.ConfigContentError, conf._get_parser)
 
2655
 
 
2656
    def test_load_erroneous_content(self):
 
2657
        """Ensure we display a proper error on content that can't be parsed."""
 
2658
        with open('foo.conf', 'wb') as f:
 
2659
            f.write('[open_section\n')
 
2660
        conf = config.IniBasedConfig(file_name='foo.conf')
 
2661
        self.assertRaises(errors.ParseConfigError, conf._get_parser)
 
2662
 
 
2663
 
 
2664
class TestMutableStore(TestStore):
 
2665
 
 
2666
    scenarios = [(key, {'store_id': key, 'get_store': builder}) for key, builder
 
2667
                 in config.test_store_builder_registry.iteritems()]
 
2668
 
 
2669
    def setUp(self):
 
2670
        super(TestMutableStore, self).setUp()
 
2671
        self.transport = self.get_transport()
 
2672
 
 
2673
    def has_store(self, store):
 
2674
        store_basename = urlutils.relative_url(self.transport.external_url(),
 
2675
                                               store.external_url())
 
2676
        return self.transport.has(store_basename)
 
2677
 
 
2678
    def test_save_empty_creates_no_file(self):
 
2679
        # FIXME: There should be a better way than relying on the test
 
2680
        # parametrization to identify branch.conf -- vila 2011-0526
 
2681
        if self.store_id in ('branch', 'remote_branch'):
 
2682
            raise tests.TestNotApplicable(
 
2683
                'branch.conf is *always* created when a branch is initialized')
 
2684
        store = self.get_store(self)
 
2685
        store.save()
 
2686
        self.assertEquals(False, self.has_store(store))
 
2687
 
 
2688
    def test_save_emptied_succeeds(self):
 
2689
        store = self.get_store(self)
 
2690
        store._load_from_string('foo=bar\n')
 
2691
        section = store.get_mutable_section(None)
 
2692
        section.remove('foo')
 
2693
        store.save()
 
2694
        self.assertEquals(True, self.has_store(store))
 
2695
        modified_store = self.get_store(self)
 
2696
        sections = list(modified_store.get_sections())
 
2697
        self.assertLength(0, sections)
 
2698
 
 
2699
    def test_save_with_content_succeeds(self):
 
2700
        # FIXME: There should be a better way than relying on the test
 
2701
        # parametrization to identify branch.conf -- vila 2011-0526
 
2702
        if self.store_id in ('branch', 'remote_branch'):
 
2703
            raise tests.TestNotApplicable(
 
2704
                'branch.conf is *always* created when a branch is initialized')
 
2705
        store = self.get_store(self)
 
2706
        store._load_from_string('foo=bar\n')
 
2707
        self.assertEquals(False, self.has_store(store))
 
2708
        store.save()
 
2709
        self.assertEquals(True, self.has_store(store))
 
2710
        modified_store = self.get_store(self)
 
2711
        sections = list(modified_store.get_sections())
 
2712
        self.assertLength(1, sections)
 
2713
        self.assertSectionContent((None, {'foo': 'bar'}), sections[0])
 
2714
 
 
2715
    def test_set_option_in_empty_store(self):
 
2716
        store = self.get_store(self)
 
2717
        section = store.get_mutable_section(None)
 
2718
        section.set('foo', 'bar')
 
2719
        store.save()
 
2720
        modified_store = self.get_store(self)
 
2721
        sections = list(modified_store.get_sections())
 
2722
        self.assertLength(1, sections)
 
2723
        self.assertSectionContent((None, {'foo': 'bar'}), sections[0])
 
2724
 
 
2725
    def test_set_option_in_default_section(self):
 
2726
        store = self.get_store(self)
 
2727
        store._load_from_string('')
 
2728
        section = store.get_mutable_section(None)
 
2729
        section.set('foo', 'bar')
 
2730
        store.save()
 
2731
        modified_store = self.get_store(self)
 
2732
        sections = list(modified_store.get_sections())
 
2733
        self.assertLength(1, sections)
 
2734
        self.assertSectionContent((None, {'foo': 'bar'}), sections[0])
 
2735
 
 
2736
    def test_set_option_in_named_section(self):
 
2737
        store = self.get_store(self)
 
2738
        store._load_from_string('')
 
2739
        section = store.get_mutable_section('baz')
 
2740
        section.set('foo', 'bar')
 
2741
        store.save()
 
2742
        modified_store = self.get_store(self)
 
2743
        sections = list(modified_store.get_sections())
 
2744
        self.assertLength(1, sections)
 
2745
        self.assertSectionContent(('baz', {'foo': 'bar'}), sections[0])
 
2746
 
 
2747
    def test_load_hook(self):
 
2748
        # We first needs to ensure that the store exists
 
2749
        store = self.get_store(self)
 
2750
        section = store.get_mutable_section('baz')
 
2751
        section.set('foo', 'bar')
 
2752
        store.save()
 
2753
        # Now we can try to load it
 
2754
        store = self.get_store(self)
 
2755
        calls = []
 
2756
        def hook(*args):
 
2757
            calls.append(args)
 
2758
        config.ConfigHooks.install_named_hook('load', hook, None)
 
2759
        self.assertLength(0, calls)
 
2760
        store.load()
 
2761
        self.assertLength(1, calls)
 
2762
        self.assertEquals((store,), calls[0])
 
2763
 
 
2764
    def test_save_hook(self):
 
2765
        calls = []
 
2766
        def hook(*args):
 
2767
            calls.append(args)
 
2768
        config.ConfigHooks.install_named_hook('save', hook, None)
 
2769
        self.assertLength(0, calls)
 
2770
        store = self.get_store(self)
 
2771
        section = store.get_mutable_section('baz')
 
2772
        section.set('foo', 'bar')
 
2773
        store.save()
 
2774
        self.assertLength(1, calls)
 
2775
        self.assertEquals((store,), calls[0])
 
2776
 
 
2777
 
 
2778
class TestIniFileStore(TestStore):
 
2779
 
 
2780
    def test_loading_unknown_file_fails(self):
 
2781
        store = config.IniFileStore(self.get_transport(), 'I-do-not-exist')
 
2782
        self.assertRaises(errors.NoSuchFile, store.load)
 
2783
 
 
2784
    def test_invalid_content(self):
 
2785
        store = config.IniFileStore(self.get_transport(), 'foo.conf', )
 
2786
        self.assertEquals(False, store.is_loaded())
 
2787
        exc = self.assertRaises(
 
2788
            errors.ParseConfigError, store._load_from_string,
 
2789
            'this is invalid !')
 
2790
        self.assertEndsWith(exc.filename, 'foo.conf')
 
2791
        # And the load failed
 
2792
        self.assertEquals(False, store.is_loaded())
 
2793
 
 
2794
    def test_get_embedded_sections(self):
 
2795
        # A more complicated example (which also shows that section names and
 
2796
        # option names share the same name space...)
 
2797
        # FIXME: This should be fixed by forbidding dicts as values ?
 
2798
        # -- vila 2011-04-05
 
2799
        store = config.IniFileStore(self.get_transport(), 'foo.conf', )
 
2800
        store._load_from_string('''
 
2801
foo=bar
 
2802
l=1,2
 
2803
[DEFAULT]
 
2804
foo_in_DEFAULT=foo_DEFAULT
 
2805
[bar]
 
2806
foo_in_bar=barbar
 
2807
[baz]
 
2808
foo_in_baz=barbaz
 
2809
[[qux]]
 
2810
foo_in_qux=quux
 
2811
''')
 
2812
        sections = list(store.get_sections())
 
2813
        self.assertLength(4, sections)
 
2814
        # The default section has no name.
 
2815
        # List values are provided as lists
 
2816
        self.assertSectionContent((None, {'foo': 'bar', 'l': ['1', '2']}),
 
2817
                                  sections[0])
 
2818
        self.assertSectionContent(
 
2819
            ('DEFAULT', {'foo_in_DEFAULT': 'foo_DEFAULT'}), sections[1])
 
2820
        self.assertSectionContent(
 
2821
            ('bar', {'foo_in_bar': 'barbar'}), sections[2])
 
2822
        # sub sections are provided as embedded dicts.
 
2823
        self.assertSectionContent(
 
2824
            ('baz', {'foo_in_baz': 'barbaz', 'qux': {'foo_in_qux': 'quux'}}),
 
2825
            sections[3])
 
2826
 
 
2827
 
 
2828
class TestLockableIniFileStore(TestStore):
 
2829
 
 
2830
    def test_create_store_in_created_dir(self):
 
2831
        self.assertPathDoesNotExist('dir')
 
2832
        t = self.get_transport('dir/subdir')
 
2833
        store = config.LockableIniFileStore(t, 'foo.conf')
 
2834
        store.get_mutable_section(None).set('foo', 'bar')
 
2835
        store.save()
 
2836
        self.assertPathExists('dir/subdir')
 
2837
 
 
2838
 
 
2839
class TestConcurrentStoreUpdates(TestStore):
 
2840
    """Test that Stores properly handle conccurent updates.
 
2841
 
 
2842
    New Store implementation may fail some of these tests but until such
 
2843
    implementations exist it's hard to properly filter them from the scenarios
 
2844
    applied here. If you encounter such a case, contact the bzr devs.
 
2845
    """
 
2846
 
 
2847
    scenarios = [(key, {'get_stack': builder}) for key, builder
 
2848
                 in config.test_stack_builder_registry.iteritems()]
 
2849
 
 
2850
    def setUp(self):
 
2851
        super(TestConcurrentStoreUpdates, self).setUp()
 
2852
        self._content = 'one=1\ntwo=2\n'
 
2853
        self.stack = self.get_stack(self)
 
2854
        if not isinstance(self.stack, config._CompatibleStack):
 
2855
            raise tests.TestNotApplicable(
 
2856
                '%s is not meant to be compatible with the old config design'
 
2857
                % (self.stack,))
 
2858
        self.stack.store._load_from_string(self._content)
 
2859
        # Flush the store
 
2860
        self.stack.store.save()
 
2861
 
 
2862
    def test_simple_read_access(self):
 
2863
        self.assertEquals('1', self.stack.get('one'))
 
2864
 
 
2865
    def test_simple_write_access(self):
 
2866
        self.stack.set('one', 'one')
 
2867
        self.assertEquals('one', self.stack.get('one'))
 
2868
 
 
2869
    def test_listen_to_the_last_speaker(self):
 
2870
        c1 = self.stack
 
2871
        c2 = self.get_stack(self)
 
2872
        c1.set('one', 'ONE')
 
2873
        c2.set('two', 'TWO')
 
2874
        self.assertEquals('ONE', c1.get('one'))
 
2875
        self.assertEquals('TWO', c2.get('two'))
 
2876
        # The second update respect the first one
 
2877
        self.assertEquals('ONE', c2.get('one'))
 
2878
 
 
2879
    def test_last_speaker_wins(self):
 
2880
        # If the same config is not shared, the same variable modified twice
 
2881
        # can only see a single result.
 
2882
        c1 = self.stack
 
2883
        c2 = self.get_stack(self)
 
2884
        c1.set('one', 'c1')
 
2885
        c2.set('one', 'c2')
 
2886
        self.assertEquals('c2', c2.get('one'))
 
2887
        # The first modification is still available until another refresh
 
2888
        # occur
 
2889
        self.assertEquals('c1', c1.get('one'))
 
2890
        c1.set('two', 'done')
 
2891
        self.assertEquals('c2', c1.get('one'))
 
2892
 
 
2893
    def test_writes_are_serialized(self):
 
2894
        c1 = self.stack
 
2895
        c2 = self.get_stack(self)
 
2896
 
 
2897
        # We spawn a thread that will pause *during* the config saving.
 
2898
        before_writing = threading.Event()
 
2899
        after_writing = threading.Event()
 
2900
        writing_done = threading.Event()
 
2901
        c1_save_without_locking_orig = c1.store.save_without_locking
 
2902
        def c1_save_without_locking():
 
2903
            before_writing.set()
 
2904
            c1_save_without_locking_orig()
 
2905
            # The lock is held. We wait for the main thread to decide when to
 
2906
            # continue
 
2907
            after_writing.wait()
 
2908
        c1.store.save_without_locking = c1_save_without_locking
 
2909
        def c1_set():
 
2910
            c1.set('one', 'c1')
 
2911
            writing_done.set()
 
2912
        t1 = threading.Thread(target=c1_set)
 
2913
        # Collect the thread after the test
 
2914
        self.addCleanup(t1.join)
 
2915
        # Be ready to unblock the thread if the test goes wrong
 
2916
        self.addCleanup(after_writing.set)
 
2917
        t1.start()
 
2918
        before_writing.wait()
 
2919
        self.assertRaises(errors.LockContention,
 
2920
                          c2.set, 'one', 'c2')
 
2921
        self.assertEquals('c1', c1.get('one'))
 
2922
        # Let the lock be released
 
2923
        after_writing.set()
 
2924
        writing_done.wait()
 
2925
        c2.set('one', 'c2')
 
2926
        self.assertEquals('c2', c2.get('one'))
 
2927
 
 
2928
    def test_read_while_writing(self):
 
2929
       c1 = self.stack
 
2930
       # We spawn a thread that will pause *during* the write
 
2931
       ready_to_write = threading.Event()
 
2932
       do_writing = threading.Event()
 
2933
       writing_done = threading.Event()
 
2934
       # We override the _save implementation so we know the store is locked
 
2935
       c1_save_without_locking_orig = c1.store.save_without_locking
 
2936
       def c1_save_without_locking():
 
2937
           ready_to_write.set()
 
2938
           # The lock is held. We wait for the main thread to decide when to
 
2939
           # continue
 
2940
           do_writing.wait()
 
2941
           c1_save_without_locking_orig()
 
2942
           writing_done.set()
 
2943
       c1.store.save_without_locking = c1_save_without_locking
 
2944
       def c1_set():
 
2945
           c1.set('one', 'c1')
 
2946
       t1 = threading.Thread(target=c1_set)
 
2947
       # Collect the thread after the test
 
2948
       self.addCleanup(t1.join)
 
2949
       # Be ready to unblock the thread if the test goes wrong
 
2950
       self.addCleanup(do_writing.set)
 
2951
       t1.start()
 
2952
       # Ensure the thread is ready to write
 
2953
       ready_to_write.wait()
 
2954
       self.assertEquals('c1', c1.get('one'))
 
2955
       # If we read during the write, we get the old value
 
2956
       c2 = self.get_stack(self)
 
2957
       self.assertEquals('1', c2.get('one'))
 
2958
       # Let the writing occur and ensure it occurred
 
2959
       do_writing.set()
 
2960
       writing_done.wait()
 
2961
       # Now we get the updated value
 
2962
       c3 = self.get_stack(self)
 
2963
       self.assertEquals('c1', c3.get('one'))
 
2964
 
 
2965
    # FIXME: It may be worth looking into removing the lock dir when it's not
 
2966
    # needed anymore and look at possible fallouts for concurrent lockers. This
 
2967
    # will matter if/when we use config files outside of bazaar directories
 
2968
    # (.bazaar or .bzr) -- vila 20110-04-11
 
2969
 
 
2970
 
 
2971
class TestSectionMatcher(TestStore):
 
2972
 
 
2973
    scenarios = [('location', {'matcher': config.LocationMatcher})]
 
2974
 
 
2975
    def get_store(self, file_name):
 
2976
        return config.IniFileStore(self.get_readonly_transport(), file_name)
 
2977
 
 
2978
    def test_no_matches_for_empty_stores(self):
 
2979
        store = self.get_store('foo.conf')
 
2980
        store._load_from_string('')
 
2981
        matcher = self.matcher(store, '/bar')
 
2982
        self.assertEquals([], list(matcher.get_sections()))
 
2983
 
 
2984
    def test_build_doesnt_load_store(self):
 
2985
        store = self.get_store('foo.conf')
 
2986
        matcher = self.matcher(store, '/bar')
 
2987
        self.assertFalse(store.is_loaded())
 
2988
 
 
2989
 
 
2990
class TestLocationSection(tests.TestCase):
 
2991
 
 
2992
    def get_section(self, options, extra_path):
 
2993
        section = config.Section('foo', options)
 
2994
        # We don't care about the length so we use '0'
 
2995
        return config.LocationSection(section, 0, extra_path)
 
2996
 
 
2997
    def test_simple_option(self):
 
2998
        section = self.get_section({'foo': 'bar'}, '')
 
2999
        self.assertEquals('bar', section.get('foo'))
 
3000
 
 
3001
    def test_option_with_extra_path(self):
 
3002
        section = self.get_section({'foo': 'bar', 'foo:policy': 'appendpath'},
 
3003
                                   'baz')
 
3004
        self.assertEquals('bar/baz', section.get('foo'))
 
3005
 
 
3006
    def test_invalid_policy(self):
 
3007
        section = self.get_section({'foo': 'bar', 'foo:policy': 'die'},
 
3008
                                   'baz')
 
3009
        # invalid policies are ignored
 
3010
        self.assertEquals('bar', section.get('foo'))
 
3011
 
 
3012
 
 
3013
class TestLocationMatcher(TestStore):
 
3014
 
 
3015
    def get_store(self, file_name):
 
3016
        return config.IniFileStore(self.get_readonly_transport(), file_name)
 
3017
 
 
3018
    def test_unrelated_section_excluded(self):
 
3019
        store = self.get_store('foo.conf')
 
3020
        store._load_from_string('''
 
3021
[/foo]
 
3022
section=/foo
 
3023
[/foo/baz]
 
3024
section=/foo/baz
 
3025
[/foo/bar]
 
3026
section=/foo/bar
 
3027
[/foo/bar/baz]
 
3028
section=/foo/bar/baz
 
3029
[/quux/quux]
 
3030
section=/quux/quux
 
3031
''')
 
3032
        self.assertEquals(['/foo', '/foo/baz', '/foo/bar', '/foo/bar/baz',
 
3033
                           '/quux/quux'],
 
3034
                          [section.id for section in store.get_sections()])
 
3035
        matcher = config.LocationMatcher(store, '/foo/bar/quux')
 
3036
        sections = list(matcher.get_sections())
 
3037
        self.assertEquals([3, 2],
 
3038
                          [section.length for section in sections])
 
3039
        self.assertEquals(['/foo/bar', '/foo'],
 
3040
                          [section.id for section in sections])
 
3041
        self.assertEquals(['quux', 'bar/quux'],
 
3042
                          [section.extra_path for section in sections])
 
3043
 
 
3044
    def test_more_specific_sections_first(self):
 
3045
        store = self.get_store('foo.conf')
 
3046
        store._load_from_string('''
 
3047
[/foo]
 
3048
section=/foo
 
3049
[/foo/bar]
 
3050
section=/foo/bar
 
3051
''')
 
3052
        self.assertEquals(['/foo', '/foo/bar'],
 
3053
                          [section.id for section in store.get_sections()])
 
3054
        matcher = config.LocationMatcher(store, '/foo/bar/baz')
 
3055
        sections = list(matcher.get_sections())
 
3056
        self.assertEquals([3, 2],
 
3057
                          [section.length for section in sections])
 
3058
        self.assertEquals(['/foo/bar', '/foo'],
 
3059
                          [section.id for section in sections])
 
3060
        self.assertEquals(['baz', 'bar/baz'],
 
3061
                          [section.extra_path for section in sections])
 
3062
 
 
3063
    def test_appendpath_in_no_name_section(self):
 
3064
        # It's a bit weird to allow appendpath in a no-name section, but
 
3065
        # someone may found a use for it
 
3066
        store = self.get_store('foo.conf')
 
3067
        store._load_from_string('''
 
3068
foo=bar
 
3069
foo:policy = appendpath
 
3070
''')
 
3071
        matcher = config.LocationMatcher(store, 'dir/subdir')
 
3072
        sections = list(matcher.get_sections())
 
3073
        self.assertLength(1, sections)
 
3074
        self.assertEquals('bar/dir/subdir', sections[0].get('foo'))
 
3075
 
 
3076
    def test_file_urls_are_normalized(self):
 
3077
        store = self.get_store('foo.conf')
 
3078
        if sys.platform == 'win32':
 
3079
            expected_url = 'file:///C:/dir/subdir'
 
3080
            expected_location = 'C:/dir/subdir'
 
3081
        else:
 
3082
            expected_url = 'file:///dir/subdir'
 
3083
            expected_location = '/dir/subdir'
 
3084
        matcher = config.LocationMatcher(store, expected_url)
 
3085
        self.assertEquals(expected_location, matcher.location)
 
3086
 
 
3087
 
 
3088
class TestStackGet(tests.TestCase):
 
3089
 
 
3090
    # FIXME: This should be parametrized for all known Stack or dedicated
 
3091
    # paramerized tests created to avoid bloating -- vila 2011-03-31
 
3092
 
 
3093
    def overrideOptionRegistry(self):
 
3094
        self.overrideAttr(config, 'option_registry', config.OptionRegistry())
 
3095
 
 
3096
    def test_single_config_get(self):
 
3097
        conf = dict(foo='bar')
 
3098
        conf_stack = config.Stack([conf])
 
3099
        self.assertEquals('bar', conf_stack.get('foo'))
 
3100
 
 
3101
    def test_get_with_registered_default_value(self):
 
3102
        conf_stack = config.Stack([dict()])
 
3103
        opt = config.Option('foo', default='bar')
 
3104
        self.overrideOptionRegistry()
 
3105
        config.option_registry.register('foo', opt)
 
3106
        self.assertEquals('bar', conf_stack.get('foo'))
 
3107
 
 
3108
    def test_get_without_registered_default_value(self):
 
3109
        conf_stack = config.Stack([dict()])
 
3110
        opt = config.Option('foo')
 
3111
        self.overrideOptionRegistry()
 
3112
        config.option_registry.register('foo', opt)
 
3113
        self.assertEquals(None, conf_stack.get('foo'))
 
3114
 
 
3115
    def test_get_without_default_value_for_not_registered(self):
 
3116
        conf_stack = config.Stack([dict()])
 
3117
        opt = config.Option('foo')
 
3118
        self.overrideOptionRegistry()
 
3119
        self.assertEquals(None, conf_stack.get('foo'))
 
3120
 
 
3121
    def test_get_first_definition(self):
 
3122
        conf1 = dict(foo='bar')
 
3123
        conf2 = dict(foo='baz')
 
3124
        conf_stack = config.Stack([conf1, conf2])
 
3125
        self.assertEquals('bar', conf_stack.get('foo'))
 
3126
 
 
3127
    def test_get_embedded_definition(self):
 
3128
        conf1 = dict(yy='12')
 
3129
        conf2 = config.Stack([dict(xx='42'), dict(foo='baz')])
 
3130
        conf_stack = config.Stack([conf1, conf2])
 
3131
        self.assertEquals('baz', conf_stack.get('foo'))
 
3132
 
 
3133
    def test_get_for_empty_section_callable(self):
 
3134
        conf_stack = config.Stack([lambda : []])
 
3135
        self.assertEquals(None, conf_stack.get('foo'))
 
3136
 
 
3137
    def test_get_for_broken_callable(self):
 
3138
        # Trying to use and invalid callable raises an exception on first use
 
3139
        conf_stack = config.Stack([lambda : object()])
 
3140
        self.assertRaises(TypeError, conf_stack.get, 'foo')
 
3141
 
 
3142
 
 
3143
class TestStackWithTransport(tests.TestCaseWithTransport):
 
3144
 
 
3145
    scenarios = [(key, {'get_stack': builder}) for key, builder
 
3146
                 in config.test_stack_builder_registry.iteritems()]
 
3147
 
 
3148
 
 
3149
class TestConcreteStacks(TestStackWithTransport):
 
3150
 
 
3151
    def test_build_stack(self):
 
3152
        # Just a smoke test to help debug builders
 
3153
        stack = self.get_stack(self)
 
3154
 
 
3155
 
 
3156
class TestStackGet(TestStackWithTransport):
 
3157
 
 
3158
    def setUp(self):
 
3159
        super(TestStackGet, self).setUp()
 
3160
        self.conf = self.get_stack(self)
 
3161
 
 
3162
    def test_get_for_empty_stack(self):
 
3163
        self.assertEquals(None, self.conf.get('foo'))
 
3164
 
 
3165
    def test_get_hook(self):
 
3166
        self.conf.store._load_from_string('foo=bar')
 
3167
        calls = []
 
3168
        def hook(*args):
 
3169
            calls.append(args)
 
3170
        config.ConfigHooks.install_named_hook('get', hook, None)
 
3171
        self.assertLength(0, calls)
 
3172
        value = self.conf.get('foo')
 
3173
        self.assertEquals('bar', value)
 
3174
        self.assertLength(1, calls)
 
3175
        self.assertEquals((self.conf, 'foo', 'bar'), calls[0])
 
3176
 
 
3177
 
 
3178
class TestStackGetWithConverter(TestStackGet):
 
3179
 
 
3180
    def setUp(self):
 
3181
        super(TestStackGetWithConverter, self).setUp()
 
3182
        self.overrideAttr(config, 'option_registry', config.OptionRegistry())
 
3183
        self.registry = config.option_registry
 
3184
 
 
3185
    def register_bool_option(self, name, default=None, default_from_env=None):
 
3186
        b = config.Option(name, help='A boolean.',
 
3187
                          default=default, default_from_env=default_from_env,
 
3188
                          from_unicode=config.bool_from_store)
 
3189
        self.registry.register(b)
 
3190
 
 
3191
    def test_get_default_bool_None(self):
 
3192
        self.register_bool_option('foo')
 
3193
        self.assertEquals(None, self.conf.get('foo'))
 
3194
 
 
3195
    def test_get_default_bool_True(self):
 
3196
        self.register_bool_option('foo', u'True')
 
3197
        self.assertEquals(True, self.conf.get('foo'))
 
3198
 
 
3199
    def test_get_default_bool_False(self):
 
3200
        self.register_bool_option('foo', False)
 
3201
        self.assertEquals(False, self.conf.get('foo'))
 
3202
 
 
3203
    def test_get_default_bool_False_as_string(self):
 
3204
        self.register_bool_option('foo', u'False')
 
3205
        self.assertEquals(False, self.conf.get('foo'))
 
3206
 
 
3207
    def test_get_default_bool_from_env_converted(self):
 
3208
        self.register_bool_option('foo', u'True', default_from_env=['FOO'])
 
3209
        self.overrideEnv('FOO', 'False')
 
3210
        self.assertEquals(False, self.conf.get('foo'))
 
3211
 
 
3212
    def test_get_default_bool_when_conversion_fails(self):
 
3213
        self.register_bool_option('foo', default='True')
 
3214
        self.conf.store._load_from_string('foo=invalid boolean')
 
3215
        self.assertEquals(True, self.conf.get('foo'))
 
3216
 
 
3217
    def register_integer_option(self, name,
 
3218
                                default=None, default_from_env=None):
 
3219
        i = config.Option(name, help='An integer.',
 
3220
                          default=default, default_from_env=default_from_env,
 
3221
                          from_unicode=config.int_from_store)
 
3222
        self.registry.register(i)
 
3223
 
 
3224
    def test_get_default_integer_None(self):
 
3225
        self.register_integer_option('foo')
 
3226
        self.assertEquals(None, self.conf.get('foo'))
 
3227
 
 
3228
    def test_get_default_integer(self):
 
3229
        self.register_integer_option('foo', 42)
 
3230
        self.assertEquals(42, self.conf.get('foo'))
 
3231
 
 
3232
    def test_get_default_integer_as_string(self):
 
3233
        self.register_integer_option('foo', u'42')
 
3234
        self.assertEquals(42, self.conf.get('foo'))
 
3235
 
 
3236
    def test_get_default_integer_from_env(self):
 
3237
        self.register_integer_option('foo', default_from_env=['FOO'])
 
3238
        self.overrideEnv('FOO', '18')
 
3239
        self.assertEquals(18, self.conf.get('foo'))
 
3240
 
 
3241
    def test_get_default_integer_when_conversion_fails(self):
 
3242
        self.register_integer_option('foo', default='12')
 
3243
        self.conf.store._load_from_string('foo=invalid integer')
 
3244
        self.assertEquals(12, self.conf.get('foo'))
 
3245
 
 
3246
    def register_list_option(self, name, default=None, default_from_env=None):
 
3247
        l = config.Option(name, help='A list.',
 
3248
                          default=default, default_from_env=default_from_env,
 
3249
                          from_unicode=config.list_from_store)
 
3250
        self.registry.register(l)
 
3251
 
 
3252
    def test_get_default_list_None(self):
 
3253
        self.register_list_option('foo')
 
3254
        self.assertEquals(None, self.conf.get('foo'))
 
3255
 
 
3256
    def test_get_default_list_empty(self):
 
3257
        self.register_list_option('foo', '')
 
3258
        self.assertEquals([], self.conf.get('foo'))
 
3259
 
 
3260
    def test_get_default_list_from_env(self):
 
3261
        self.register_list_option('foo', default_from_env=['FOO'])
 
3262
        self.overrideEnv('FOO', '')
 
3263
        self.assertEquals([], self.conf.get('foo'))
 
3264
 
 
3265
    def test_get_with_list_converter_no_item(self):
 
3266
        self.register_list_option('foo', None)
 
3267
        self.conf.store._load_from_string('foo=,')
 
3268
        self.assertEquals([], self.conf.get('foo'))
 
3269
 
 
3270
    def test_get_with_list_converter_many_items(self):
 
3271
        self.register_list_option('foo', None)
 
3272
        self.conf.store._load_from_string('foo=m,o,r,e')
 
3273
        self.assertEquals(['m', 'o', 'r', 'e'], self.conf.get('foo'))
 
3274
 
 
3275
 
 
3276
class TestStackSet(TestStackWithTransport):
 
3277
 
 
3278
    def test_simple_set(self):
 
3279
        conf = self.get_stack(self)
 
3280
        conf.store._load_from_string('foo=bar')
 
3281
        self.assertEquals('bar', conf.get('foo'))
 
3282
        conf.set('foo', 'baz')
 
3283
        # Did we get it back ?
 
3284
        self.assertEquals('baz', conf.get('foo'))
 
3285
 
 
3286
    def test_set_creates_a_new_section(self):
 
3287
        conf = self.get_stack(self)
 
3288
        conf.set('foo', 'baz')
 
3289
        self.assertEquals, 'baz', conf.get('foo')
 
3290
 
 
3291
    def test_set_hook(self):
 
3292
        calls = []
 
3293
        def hook(*args):
 
3294
            calls.append(args)
 
3295
        config.ConfigHooks.install_named_hook('set', hook, None)
 
3296
        self.assertLength(0, calls)
 
3297
        conf = self.get_stack(self)
 
3298
        conf.set('foo', 'bar')
 
3299
        self.assertLength(1, calls)
 
3300
        self.assertEquals((conf, 'foo', 'bar'), calls[0])
 
3301
 
 
3302
 
 
3303
class TestStackRemove(TestStackWithTransport):
 
3304
 
 
3305
    def test_remove_existing(self):
 
3306
        conf = self.get_stack(self)
 
3307
        conf.store._load_from_string('foo=bar')
 
3308
        self.assertEquals('bar', conf.get('foo'))
 
3309
        conf.remove('foo')
 
3310
        # Did we get it back ?
 
3311
        self.assertEquals(None, conf.get('foo'))
 
3312
 
 
3313
    def test_remove_unknown(self):
 
3314
        conf = self.get_stack(self)
 
3315
        self.assertRaises(KeyError, conf.remove, 'I_do_not_exist')
 
3316
 
 
3317
    def test_remove_hook(self):
 
3318
        calls = []
 
3319
        def hook(*args):
 
3320
            calls.append(args)
 
3321
        config.ConfigHooks.install_named_hook('remove', hook, None)
 
3322
        self.assertLength(0, calls)
 
3323
        conf = self.get_stack(self)
 
3324
        conf.store._load_from_string('foo=bar')
 
3325
        conf.remove('foo')
 
3326
        self.assertLength(1, calls)
 
3327
        self.assertEquals((conf, 'foo'), calls[0])
 
3328
 
 
3329
 
 
3330
class TestConfigGetOptions(tests.TestCaseWithTransport, TestOptionsMixin):
 
3331
 
 
3332
    def setUp(self):
 
3333
        super(TestConfigGetOptions, self).setUp()
 
3334
        create_configs(self)
 
3335
 
 
3336
    def test_no_variable(self):
 
3337
        # Using branch should query branch, locations and bazaar
 
3338
        self.assertOptions([], self.branch_config)
 
3339
 
 
3340
    def test_option_in_bazaar(self):
 
3341
        self.bazaar_config.set_user_option('file', 'bazaar')
 
3342
        self.assertOptions([('file', 'bazaar', 'DEFAULT', 'bazaar')],
 
3343
                           self.bazaar_config)
 
3344
 
 
3345
    def test_option_in_locations(self):
 
3346
        self.locations_config.set_user_option('file', 'locations')
 
3347
        self.assertOptions(
 
3348
            [('file', 'locations', self.tree.basedir, 'locations')],
 
3349
            self.locations_config)
 
3350
 
 
3351
    def test_option_in_branch(self):
 
3352
        self.branch_config.set_user_option('file', 'branch')
 
3353
        self.assertOptions([('file', 'branch', 'DEFAULT', 'branch')],
 
3354
                           self.branch_config)
 
3355
 
 
3356
    def test_option_in_bazaar_and_branch(self):
 
3357
        self.bazaar_config.set_user_option('file', 'bazaar')
 
3358
        self.branch_config.set_user_option('file', 'branch')
 
3359
        self.assertOptions([('file', 'branch', 'DEFAULT', 'branch'),
 
3360
                            ('file', 'bazaar', 'DEFAULT', 'bazaar'),],
 
3361
                           self.branch_config)
 
3362
 
 
3363
    def test_option_in_branch_and_locations(self):
 
3364
        # Hmm, locations override branch :-/
 
3365
        self.locations_config.set_user_option('file', 'locations')
 
3366
        self.branch_config.set_user_option('file', 'branch')
 
3367
        self.assertOptions(
 
3368
            [('file', 'locations', self.tree.basedir, 'locations'),
 
3369
             ('file', 'branch', 'DEFAULT', 'branch'),],
 
3370
            self.branch_config)
 
3371
 
 
3372
    def test_option_in_bazaar_locations_and_branch(self):
 
3373
        self.bazaar_config.set_user_option('file', 'bazaar')
 
3374
        self.locations_config.set_user_option('file', 'locations')
 
3375
        self.branch_config.set_user_option('file', 'branch')
 
3376
        self.assertOptions(
 
3377
            [('file', 'locations', self.tree.basedir, 'locations'),
 
3378
             ('file', 'branch', 'DEFAULT', 'branch'),
 
3379
             ('file', 'bazaar', 'DEFAULT', 'bazaar'),],
 
3380
            self.branch_config)
 
3381
 
 
3382
 
 
3383
class TestConfigRemoveOption(tests.TestCaseWithTransport, TestOptionsMixin):
 
3384
 
 
3385
    def setUp(self):
 
3386
        super(TestConfigRemoveOption, self).setUp()
 
3387
        create_configs_with_file_option(self)
 
3388
 
 
3389
    def test_remove_in_locations(self):
 
3390
        self.locations_config.remove_user_option('file', self.tree.basedir)
 
3391
        self.assertOptions(
 
3392
            [('file', 'branch', 'DEFAULT', 'branch'),
 
3393
             ('file', 'bazaar', 'DEFAULT', 'bazaar'),],
 
3394
            self.branch_config)
 
3395
 
 
3396
    def test_remove_in_branch(self):
 
3397
        self.branch_config.remove_user_option('file')
 
3398
        self.assertOptions(
 
3399
            [('file', 'locations', self.tree.basedir, 'locations'),
 
3400
             ('file', 'bazaar', 'DEFAULT', 'bazaar'),],
 
3401
            self.branch_config)
 
3402
 
 
3403
    def test_remove_in_bazaar(self):
 
3404
        self.bazaar_config.remove_user_option('file')
 
3405
        self.assertOptions(
 
3406
            [('file', 'locations', self.tree.basedir, 'locations'),
 
3407
             ('file', 'branch', 'DEFAULT', 'branch'),],
 
3408
            self.branch_config)
 
3409
 
 
3410
 
 
3411
class TestConfigGetSections(tests.TestCaseWithTransport):
 
3412
 
 
3413
    def setUp(self):
 
3414
        super(TestConfigGetSections, self).setUp()
 
3415
        create_configs(self)
 
3416
 
 
3417
    def assertSectionNames(self, expected, conf, name=None):
 
3418
        """Check which sections are returned for a given config.
 
3419
 
 
3420
        If fallback configurations exist their sections can be included.
 
3421
 
 
3422
        :param expected: A list of section names.
 
3423
 
 
3424
        :param conf: The configuration that will be queried.
 
3425
 
 
3426
        :param name: An optional section name that will be passed to
 
3427
            get_sections().
 
3428
        """
 
3429
        sections = list(conf._get_sections(name))
 
3430
        self.assertLength(len(expected), sections)
 
3431
        self.assertEqual(expected, [name for name, _, _ in sections])
 
3432
 
 
3433
    def test_bazaar_default_section(self):
 
3434
        self.assertSectionNames(['DEFAULT'], self.bazaar_config)
 
3435
 
 
3436
    def test_locations_default_section(self):
 
3437
        # No sections are defined in an empty file
 
3438
        self.assertSectionNames([], self.locations_config)
 
3439
 
 
3440
    def test_locations_named_section(self):
 
3441
        self.locations_config.set_user_option('file', 'locations')
 
3442
        self.assertSectionNames([self.tree.basedir], self.locations_config)
 
3443
 
 
3444
    def test_locations_matching_sections(self):
 
3445
        loc_config = self.locations_config
 
3446
        loc_config.set_user_option('file', 'locations')
 
3447
        # We need to cheat a bit here to create an option in sections above and
 
3448
        # below the 'location' one.
 
3449
        parser = loc_config._get_parser()
 
3450
        # locations.cong deals with '/' ignoring native os.sep
 
3451
        location_names = self.tree.basedir.split('/')
 
3452
        parent = '/'.join(location_names[:-1])
 
3453
        child = '/'.join(location_names + ['child'])
 
3454
        parser[parent] = {}
 
3455
        parser[parent]['file'] = 'parent'
 
3456
        parser[child] = {}
 
3457
        parser[child]['file'] = 'child'
 
3458
        self.assertSectionNames([self.tree.basedir, parent], loc_config)
 
3459
 
 
3460
    def test_branch_data_default_section(self):
 
3461
        self.assertSectionNames([None],
 
3462
                                self.branch_config._get_branch_data_config())
 
3463
 
 
3464
    def test_branch_default_sections(self):
 
3465
        # No sections are defined in an empty locations file
 
3466
        self.assertSectionNames([None, 'DEFAULT'],
 
3467
                                self.branch_config)
 
3468
        # Unless we define an option
 
3469
        self.branch_config._get_location_config().set_user_option(
 
3470
            'file', 'locations')
 
3471
        self.assertSectionNames([self.tree.basedir, None, 'DEFAULT'],
 
3472
                                self.branch_config)
 
3473
 
 
3474
    def test_bazaar_named_section(self):
 
3475
        # We need to cheat as the API doesn't give direct access to sections
 
3476
        # other than DEFAULT.
 
3477
        self.bazaar_config.set_alias('bazaar', 'bzr')
 
3478
        self.assertSectionNames(['ALIASES'], self.bazaar_config, 'ALIASES')
 
3479
 
 
3480
 
1315
3481
class TestAuthenticationConfigFile(tests.TestCase):
1316
3482
    """Test the authentication.conf file matching"""
1317
3483
 
1332
3498
        self.assertEquals({}, conf._get_config())
1333
3499
        self._got_user_passwd(None, None, conf, 'http', 'foo.net')
1334
3500
 
 
3501
    def test_non_utf8_config(self):
 
3502
        conf = config.AuthenticationConfig(_file=StringIO(
 
3503
                'foo = bar\xff'))
 
3504
        self.assertRaises(errors.ConfigContentError, conf._get_config)
 
3505
 
1335
3506
    def test_missing_auth_section_header(self):
1336
3507
        conf = config.AuthenticationConfig(_file=StringIO('foo = bar'))
1337
3508
        self.assertRaises(ValueError, conf.get_credentials, 'ftp', 'foo.net')
1595
3766
 
1596
3767
    def test_username_defaults_prompts(self):
1597
3768
        # HTTP prompts can't be tested here, see test_http.py
1598
 
        self._check_default_username_prompt('FTP %(host)s username: ', 'ftp')
1599
 
        self._check_default_username_prompt(
1600
 
            'FTP %(host)s:%(port)d username: ', 'ftp', port=10020)
1601
 
        self._check_default_username_prompt(
1602
 
            'SSH %(host)s:%(port)d username: ', 'ssh', port=12345)
 
3769
        self._check_default_username_prompt(u'FTP %(host)s username: ', 'ftp')
 
3770
        self._check_default_username_prompt(
 
3771
            u'FTP %(host)s:%(port)d username: ', 'ftp', port=10020)
 
3772
        self._check_default_username_prompt(
 
3773
            u'SSH %(host)s:%(port)d username: ', 'ssh', port=12345)
1603
3774
 
1604
3775
    def test_username_default_no_prompt(self):
1605
3776
        conf = config.AuthenticationConfig()
1611
3782
    def test_password_default_prompts(self):
1612
3783
        # HTTP prompts can't be tested here, see test_http.py
1613
3784
        self._check_default_password_prompt(
1614
 
            'FTP %(user)s@%(host)s password: ', 'ftp')
1615
 
        self._check_default_password_prompt(
1616
 
            'FTP %(user)s@%(host)s:%(port)d password: ', 'ftp', port=10020)
1617
 
        self._check_default_password_prompt(
1618
 
            'SSH %(user)s@%(host)s:%(port)d password: ', 'ssh', port=12345)
 
3785
            u'FTP %(user)s@%(host)s password: ', 'ftp')
 
3786
        self._check_default_password_prompt(
 
3787
            u'FTP %(user)s@%(host)s:%(port)d password: ', 'ftp', port=10020)
 
3788
        self._check_default_password_prompt(
 
3789
            u'SSH %(user)s@%(host)s:%(port)d password: ', 'ssh', port=12345)
1619
3790
        # SMTP port handling is a bit special (it's handled if embedded in the
1620
3791
        # host too)
1621
3792
        # FIXME: should we: forbid that, extend it to other schemes, leave
1622
3793
        # things as they are that's fine thank you ?
1623
 
        self._check_default_password_prompt('SMTP %(user)s@%(host)s password: ',
1624
 
                                            'smtp')
1625
 
        self._check_default_password_prompt('SMTP %(user)s@%(host)s password: ',
1626
 
                                            'smtp', host='bar.org:10025')
1627
 
        self._check_default_password_prompt(
1628
 
            'SMTP %(user)s@%(host)s:%(port)d password: ',
1629
 
            'smtp', port=10025)
 
3794
        self._check_default_password_prompt(
 
3795
            u'SMTP %(user)s@%(host)s password: ', 'smtp')
 
3796
        self._check_default_password_prompt(
 
3797
            u'SMTP %(user)s@%(host)s password: ', 'smtp', host='bar.org:10025')
 
3798
        self._check_default_password_prompt(
 
3799
            u'SMTP %(user)s@%(host)s:%(port)d password: ', 'smtp', port=10025)
1630
3800
 
1631
3801
    def test_ssh_password_emits_warning(self):
1632
3802
        conf = config.AuthenticationConfig(_file=StringIO(
1812
3982
# test_user_prompted ?
1813
3983
class TestAuthenticationRing(tests.TestCaseWithTransport):
1814
3984
    pass
 
3985
 
 
3986
 
 
3987
class TestAutoUserId(tests.TestCase):
 
3988
    """Test inferring an automatic user name."""
 
3989
 
 
3990
    def test_auto_user_id(self):
 
3991
        """Automatic inference of user name.
 
3992
        
 
3993
        This is a bit hard to test in an isolated way, because it depends on
 
3994
        system functions that go direct to /etc or perhaps somewhere else.
 
3995
        But it's reasonable to say that on Unix, with an /etc/mailname, we ought
 
3996
        to be able to choose a user name with no configuration.
 
3997
        """
 
3998
        if sys.platform == 'win32':
 
3999
            raise tests.TestSkipped(
 
4000
                "User name inference not implemented on win32")
 
4001
        realname, address = config._auto_user_id()
 
4002
        if os.path.exists('/etc/mailname'):
 
4003
            self.assertIsNot(None, realname)
 
4004
            self.assertIsNot(None, address)
 
4005
        else:
 
4006
            self.assertEquals((None, None), (realname, address))
 
4007