~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_config.py

  • Committer: Robert Collins
  • Date: 2005-10-07 23:35:42 UTC
  • mfrom: (1393.1.70)
  • mto: (1185.1.51)
  • mto: This revision was merged to the branch mainline in revision 1422.
  • Revision ID: robertc@robertcollins.net-20051007233541-eb80a1c86fa1405f
merge from martins newformat

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2011 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
"""Tests for finding and reading the bzr config file[s]."""
18
 
# import system imports here
19
 
from cStringIO import StringIO
20
 
import os
21
 
import sys
22
 
import threading
23
 
 
24
 
 
25
 
from testtools import matchers
26
 
 
27
 
#import bzrlib specific imports here
28
 
from bzrlib import (
29
 
    branch,
30
 
    bzrdir,
31
 
    config,
32
 
    diff,
33
 
    errors,
34
 
    osutils,
35
 
    mail_client,
36
 
    ui,
37
 
    urlutils,
38
 
    remote,
39
 
    tests,
40
 
    trace,
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,
50
 
    )
51
 
from bzrlib.util.configobj import configobj
52
 
 
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
 
 
172
 
sample_long_alias="log -r-15..-1 --line"
173
 
sample_config_text = u"""
174
 
[DEFAULT]
175
 
email=Erik B\u00e5gfors <erik@bagfors.nu>
176
 
editor=vim
177
 
change_editor=vimdiff -of @new_path @old_path
178
 
gpg_signing_command=gnome-gpg
179
 
gpg_signing_key=DD4D5088
180
 
log_format=short
181
 
validate_signatures_in_log=true
182
 
acceptable_keys=amy
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.mergetool.newtool='"newtool with spaces" {this_temp}'
187
 
bzr.default_mergetool=sometool
188
 
[ALIASES]
189
 
h=help
190
 
ll=""" + sample_long_alias + "\n"
191
 
 
192
 
 
193
 
sample_always_signatures = """
194
 
[DEFAULT]
195
 
check_signatures=ignore
196
 
create_signatures=always
197
 
"""
198
 
 
199
 
sample_ignore_signatures = """
200
 
[DEFAULT]
201
 
check_signatures=require
202
 
create_signatures=never
203
 
"""
204
 
 
205
 
sample_maybe_signatures = """
206
 
[DEFAULT]
207
 
check_signatures=ignore
208
 
create_signatures=when-required
209
 
"""
210
 
 
211
 
sample_branches_text = """
212
 
[http://www.example.com]
213
 
# Top level policy
214
 
email=Robert Collins <robertc@example.org>
215
 
normal_option = normal
216
 
appendpath_option = append
217
 
appendpath_option:policy = appendpath
218
 
norecurse_option = norecurse
219
 
norecurse_option:policy = norecurse
220
 
[http://www.example.com/ignoreparent]
221
 
# different project: ignore parent dir config
222
 
ignore_parents=true
223
 
[http://www.example.com/norecurse]
224
 
# configuration items that only apply to this dir
225
 
recurse=false
226
 
normal_option = norecurse
227
 
[http://www.example.com/dir]
228
 
appendpath_option = normal
229
 
[/b/]
230
 
check_signatures=require
231
 
# test trailing / matching with no children
232
 
[/a/]
233
 
check_signatures=check-available
234
 
gpg_signing_command=false
235
 
gpg_signing_key=default
236
 
user_local_option=local
237
 
# test trailing / matching
238
 
[/a/*]
239
 
#subdirs will match but not the parent
240
 
[/a/c]
241
 
check_signatures=ignore
242
 
post_commit=bzrlib.tests.test_config.post_commit
243
 
#testing explicit beats globs
244
 
"""
245
 
 
246
 
 
247
 
def create_configs(test):
248
 
    """Create configuration files for a given test.
249
 
 
250
 
    This requires creating a tree (and populate the ``test.tree`` attribute)
251
 
    and its associated branch and will populate the following attributes:
252
 
 
253
 
    - branch_config: A BranchConfig for the associated branch.
254
 
 
255
 
    - locations_config : A LocationConfig for the associated branch
256
 
 
257
 
    - bazaar_config: A GlobalConfig.
258
 
 
259
 
    The tree and branch are created in a 'tree' subdirectory so the tests can
260
 
    still use the test directory to stay outside of the branch.
261
 
    """
262
 
    tree = test.make_branch_and_tree('tree')
263
 
    test.tree = tree
264
 
    test.branch_config = config.BranchConfig(tree.branch)
265
 
    test.locations_config = config.LocationConfig(tree.basedir)
266
 
    test.bazaar_config = config.GlobalConfig()
267
 
 
268
 
 
269
 
def create_configs_with_file_option(test):
270
 
    """Create configuration files with a ``file`` option set in each.
271
 
 
272
 
    This builds on ``create_configs`` and add one ``file`` option in each
273
 
    configuration with a value which allows identifying the configuration file.
274
 
    """
275
 
    create_configs(test)
276
 
    test.bazaar_config.set_user_option('file', 'bazaar')
277
 
    test.locations_config.set_user_option('file', 'locations')
278
 
    test.branch_config.set_user_option('file', 'branch')
279
 
 
280
 
 
281
 
class TestOptionsMixin:
282
 
 
283
 
    def assertOptions(self, expected, conf):
284
 
        # We don't care about the parser (as it will make tests hard to write
285
 
        # and error-prone anyway)
286
 
        self.assertThat([opt[:4] for opt in conf._get_options()],
287
 
                        matchers.Equals(expected))
288
 
 
289
 
 
290
 
class InstrumentedConfigObj(object):
291
 
    """A config obj look-enough-alike to record calls made to it."""
292
 
 
293
 
    def __contains__(self, thing):
294
 
        self._calls.append(('__contains__', thing))
295
 
        return False
296
 
 
297
 
    def __getitem__(self, key):
298
 
        self._calls.append(('__getitem__', key))
299
 
        return self
300
 
 
301
 
    def __init__(self, input, encoding=None):
302
 
        self._calls = [('__init__', input, encoding)]
303
 
 
304
 
    def __setitem__(self, key, value):
305
 
        self._calls.append(('__setitem__', key, value))
306
 
 
307
 
    def __delitem__(self, key):
308
 
        self._calls.append(('__delitem__', key))
309
 
 
310
 
    def keys(self):
311
 
        self._calls.append(('keys',))
312
 
        return []
313
 
 
314
 
    def reload(self):
315
 
        self._calls.append(('reload',))
316
 
 
317
 
    def write(self, arg):
318
 
        self._calls.append(('write',))
319
 
 
320
 
    def as_bool(self, value):
321
 
        self._calls.append(('as_bool', value))
322
 
        return False
323
 
 
324
 
    def get_value(self, section, name):
325
 
        self._calls.append(('get_value', section, name))
326
 
        return None
327
 
 
328
 
 
329
 
class FakeBranch(object):
330
 
 
331
 
    def __init__(self, base=None, user_id=None):
332
 
        if base is None:
333
 
            self.base = "http://example.com/branches/demo"
334
 
        else:
335
 
            self.base = base
336
 
        self._transport = self.control_files = \
337
 
            FakeControlFilesAndTransport(user_id=user_id)
338
 
 
339
 
    def _get_config(self):
340
 
        return config.TransportConfig(self._transport, 'branch.conf')
341
 
 
342
 
    def lock_write(self):
343
 
        pass
344
 
 
345
 
    def unlock(self):
346
 
        pass
347
 
 
348
 
 
349
 
class FakeControlFilesAndTransport(object):
350
 
 
351
 
    def __init__(self, user_id=None):
352
 
        self.files = {}
353
 
        if user_id:
354
 
            self.files['email'] = user_id
355
 
        self._transport = self
356
 
 
357
 
    def get_utf8(self, filename):
358
 
        # from LockableFiles
359
 
        raise AssertionError("get_utf8 should no longer be used")
360
 
 
361
 
    def get(self, filename):
362
 
        # from Transport
363
 
        try:
364
 
            return StringIO(self.files[filename])
365
 
        except KeyError:
366
 
            raise errors.NoSuchFile(filename)
367
 
 
368
 
    def get_bytes(self, filename):
369
 
        # from Transport
370
 
        try:
371
 
            return self.files[filename]
372
 
        except KeyError:
373
 
            raise errors.NoSuchFile(filename)
374
 
 
375
 
    def put(self, filename, fileobj):
376
 
        self.files[filename] = fileobj.read()
377
 
 
378
 
    def put_file(self, filename, fileobj):
379
 
        return self.put(filename, fileobj)
380
 
 
381
 
 
382
 
class InstrumentedConfig(config.Config):
383
 
    """An instrumented config that supplies stubs for template methods."""
384
 
 
385
 
    def __init__(self):
386
 
        super(InstrumentedConfig, self).__init__()
387
 
        self._calls = []
388
 
        self._signatures = config.CHECK_NEVER
389
 
 
390
 
    def _get_user_id(self):
391
 
        self._calls.append('_get_user_id')
392
 
        return "Robert Collins <robert.collins@example.org>"
393
 
 
394
 
    def _get_signature_checking(self):
395
 
        self._calls.append('_get_signature_checking')
396
 
        return self._signatures
397
 
 
398
 
    def _get_change_editor(self):
399
 
        self._calls.append('_get_change_editor')
400
 
        return 'vimdiff -fo @new_path @old_path'
401
 
 
402
 
 
403
 
bool_config = """[DEFAULT]
404
 
active = true
405
 
inactive = false
406
 
[UPPERCASE]
407
 
active = True
408
 
nonactive = False
409
 
"""
410
 
 
411
 
 
412
 
class TestConfigObj(tests.TestCase):
413
 
 
414
 
    def test_get_bool(self):
415
 
        co = config.ConfigObj(StringIO(bool_config))
416
 
        self.assertIs(co.get_bool('DEFAULT', 'active'), True)
417
 
        self.assertIs(co.get_bool('DEFAULT', 'inactive'), False)
418
 
        self.assertIs(co.get_bool('UPPERCASE', 'active'), True)
419
 
        self.assertIs(co.get_bool('UPPERCASE', 'nonactive'), False)
420
 
 
421
 
    def test_hash_sign_in_value(self):
422
 
        """
423
 
        Before 4.5.0, ConfigObj did not quote # signs in values, so they'd be
424
 
        treated as comments when read in again. (#86838)
425
 
        """
426
 
        co = config.ConfigObj()
427
 
        co['test'] = 'foo#bar'
428
 
        outfile = StringIO()
429
 
        co.write(outfile=outfile)
430
 
        lines = outfile.getvalue().splitlines()
431
 
        self.assertEqual(lines, ['test = "foo#bar"'])
432
 
        co2 = config.ConfigObj(lines)
433
 
        self.assertEqual(co2['test'], 'foo#bar')
434
 
 
435
 
    def test_triple_quotes(self):
436
 
        # Bug #710410: if the value string has triple quotes
437
 
        # then ConfigObj versions up to 4.7.2 will quote them wrong
438
 
        # and won't able to read them back
439
 
        triple_quotes_value = '''spam
440
 
""" that's my spam """
441
 
eggs'''
442
 
        co = config.ConfigObj()
443
 
        co['test'] = triple_quotes_value
444
 
        # While writing this test another bug in ConfigObj has been found:
445
 
        # method co.write() without arguments produces list of lines
446
 
        # one option per line, and multiline values are not split
447
 
        # across multiple lines,
448
 
        # and that breaks the parsing these lines back by ConfigObj.
449
 
        # This issue only affects test, but it's better to avoid
450
 
        # `co.write()` construct at all.
451
 
        # [bialix 20110222] bug report sent to ConfigObj's author
452
 
        outfile = StringIO()
453
 
        co.write(outfile=outfile)
454
 
        output = outfile.getvalue()
455
 
        # now we're trying to read it back
456
 
        co2 = config.ConfigObj(StringIO(output))
457
 
        self.assertEquals(triple_quotes_value, co2['test'])
458
 
 
459
 
 
460
 
erroneous_config = """[section] # line 1
461
 
good=good # line 2
462
 
[section] # line 3
463
 
whocares=notme # line 4
464
 
"""
465
 
 
466
 
 
467
 
class TestConfigObjErrors(tests.TestCase):
468
 
 
469
 
    def test_duplicate_section_name_error_line(self):
470
 
        try:
471
 
            co = configobj.ConfigObj(StringIO(erroneous_config),
472
 
                                     raise_errors=True)
473
 
        except config.configobj.DuplicateError, e:
474
 
            self.assertEqual(3, e.line_number)
475
 
        else:
476
 
            self.fail('Error in config file not detected')
477
 
 
478
 
 
479
 
class TestConfig(tests.TestCase):
480
 
 
481
 
    def test_constructs(self):
482
 
        config.Config()
483
 
 
484
 
    def test_no_default_editor(self):
485
 
        self.assertRaises(
486
 
            NotImplementedError,
487
 
            self.applyDeprecated, deprecated_in((2, 4, 0)),
488
 
            config.Config().get_editor)
489
 
 
490
 
    def test_user_email(self):
491
 
        my_config = InstrumentedConfig()
492
 
        self.assertEqual('robert.collins@example.org', my_config.user_email())
493
 
        self.assertEqual(['_get_user_id'], my_config._calls)
494
 
 
495
 
    def test_username(self):
496
 
        my_config = InstrumentedConfig()
497
 
        self.assertEqual('Robert Collins <robert.collins@example.org>',
498
 
                         my_config.username())
499
 
        self.assertEqual(['_get_user_id'], my_config._calls)
500
 
 
501
 
    def test_signatures_default(self):
502
 
        my_config = config.Config()
503
 
        self.assertFalse(my_config.signature_needed())
504
 
        self.assertEqual(config.CHECK_IF_POSSIBLE,
505
 
                         my_config.signature_checking())
506
 
        self.assertEqual(config.SIGN_WHEN_REQUIRED,
507
 
                         my_config.signing_policy())
508
 
 
509
 
    def test_signatures_template_method(self):
510
 
        my_config = InstrumentedConfig()
511
 
        self.assertEqual(config.CHECK_NEVER, my_config.signature_checking())
512
 
        self.assertEqual(['_get_signature_checking'], my_config._calls)
513
 
 
514
 
    def test_signatures_template_method_none(self):
515
 
        my_config = InstrumentedConfig()
516
 
        my_config._signatures = None
517
 
        self.assertEqual(config.CHECK_IF_POSSIBLE,
518
 
                         my_config.signature_checking())
519
 
        self.assertEqual(['_get_signature_checking'], my_config._calls)
520
 
 
521
 
    def test_gpg_signing_command_default(self):
522
 
        my_config = config.Config()
523
 
        self.assertEqual('gpg', my_config.gpg_signing_command())
524
 
 
525
 
    def test_get_user_option_default(self):
526
 
        my_config = config.Config()
527
 
        self.assertEqual(None, my_config.get_user_option('no_option'))
528
 
 
529
 
    def test_post_commit_default(self):
530
 
        my_config = config.Config()
531
 
        self.assertEqual(None, my_config.post_commit())
532
 
 
533
 
    def test_log_format_default(self):
534
 
        my_config = config.Config()
535
 
        self.assertEqual('long', my_config.log_format())
536
 
 
537
 
    def test_acceptable_keys_default(self):
538
 
        my_config = config.Config()
539
 
        self.assertEqual(None, my_config.acceptable_keys())
540
 
 
541
 
    def test_validate_signatures_in_log_default(self):
542
 
        my_config = config.Config()
543
 
        self.assertEqual(False, my_config.validate_signatures_in_log())
544
 
 
545
 
    def test_get_change_editor(self):
546
 
        my_config = InstrumentedConfig()
547
 
        change_editor = my_config.get_change_editor('old_tree', 'new_tree')
548
 
        self.assertEqual(['_get_change_editor'], my_config._calls)
549
 
        self.assertIs(diff.DiffFromTool, change_editor.__class__)
550
 
        self.assertEqual(['vimdiff', '-fo', '@new_path', '@old_path'],
551
 
                         change_editor.command_template)
552
 
 
553
 
 
554
 
class TestConfigPath(tests.TestCase):
555
 
 
556
 
    def setUp(self):
557
 
        super(TestConfigPath, self).setUp()
558
 
        self.overrideEnv('HOME', '/home/bogus')
559
 
        self.overrideEnv('XDG_CACHE_DIR', '')
560
 
        if sys.platform == 'win32':
561
 
            self.overrideEnv(
562
 
                'BZR_HOME', r'C:\Documents and Settings\bogus\Application Data')
563
 
            self.bzr_home = \
564
 
                'C:/Documents and Settings/bogus/Application Data/bazaar/2.0'
565
 
        else:
566
 
            self.bzr_home = '/home/bogus/.bazaar'
567
 
 
568
 
    def test_config_dir(self):
569
 
        self.assertEqual(config.config_dir(), self.bzr_home)
570
 
 
571
 
    def test_config_filename(self):
572
 
        self.assertEqual(config.config_filename(),
573
 
                         self.bzr_home + '/bazaar.conf')
574
 
 
575
 
    def test_locations_config_filename(self):
576
 
        self.assertEqual(config.locations_config_filename(),
577
 
                         self.bzr_home + '/locations.conf')
578
 
 
579
 
    def test_authentication_config_filename(self):
580
 
        self.assertEqual(config.authentication_config_filename(),
581
 
                         self.bzr_home + '/authentication.conf')
582
 
 
583
 
    def test_xdg_cache_dir(self):
584
 
        self.assertEqual(config.xdg_cache_dir(),
585
 
            '/home/bogus/.cache')
586
 
 
587
 
 
588
 
class TestXDGConfigDir(tests.TestCaseInTempDir):
589
 
    # must be in temp dir because config tests for the existence of the bazaar
590
 
    # subdirectory of $XDG_CONFIG_HOME
591
 
 
592
 
    def setUp(self):
593
 
        if sys.platform in ('darwin', 'win32'):
594
 
            raise tests.TestNotApplicable(
595
 
                'XDG config dir not used on this platform')
596
 
        super(TestXDGConfigDir, self).setUp()
597
 
        self.overrideEnv('HOME', self.test_home_dir)
598
 
        # BZR_HOME overrides everything we want to test so unset it.
599
 
        self.overrideEnv('BZR_HOME', None)
600
 
 
601
 
    def test_xdg_config_dir_exists(self):
602
 
        """When ~/.config/bazaar exists, use it as the config dir."""
603
 
        newdir = osutils.pathjoin(self.test_home_dir, '.config', 'bazaar')
604
 
        os.makedirs(newdir)
605
 
        self.assertEqual(config.config_dir(), newdir)
606
 
 
607
 
    def test_xdg_config_home(self):
608
 
        """When XDG_CONFIG_HOME is set, use it."""
609
 
        xdgconfigdir = osutils.pathjoin(self.test_home_dir, 'xdgconfig')
610
 
        self.overrideEnv('XDG_CONFIG_HOME', xdgconfigdir)
611
 
        newdir = osutils.pathjoin(xdgconfigdir, 'bazaar')
612
 
        os.makedirs(newdir)
613
 
        self.assertEqual(config.config_dir(), newdir)
614
 
 
615
 
 
616
 
class TestIniConfig(tests.TestCaseInTempDir):
617
 
 
618
 
    def make_config_parser(self, s):
619
 
        conf = config.IniBasedConfig.from_string(s)
620
 
        return conf, conf._get_parser()
621
 
 
622
 
 
623
 
class TestIniConfigBuilding(TestIniConfig):
624
 
 
625
 
    def test_contructs(self):
626
 
        my_config = config.IniBasedConfig()
627
 
 
628
 
    def test_from_fp(self):
629
 
        my_config = config.IniBasedConfig.from_string(sample_config_text)
630
 
        self.assertIsInstance(my_config._get_parser(), configobj.ConfigObj)
631
 
 
632
 
    def test_cached(self):
633
 
        my_config = config.IniBasedConfig.from_string(sample_config_text)
634
 
        parser = my_config._get_parser()
635
 
        self.assertTrue(my_config._get_parser() is parser)
636
 
 
637
 
    def _dummy_chown(self, path, uid, gid):
638
 
        self.path, self.uid, self.gid = path, uid, gid
639
 
 
640
 
    def test_ini_config_ownership(self):
641
 
        """Ensure that chown is happening during _write_config_file"""
642
 
        self.requireFeature(features.chown_feature)
643
 
        self.overrideAttr(os, 'chown', self._dummy_chown)
644
 
        self.path = self.uid = self.gid = None
645
 
        conf = config.IniBasedConfig(file_name='./foo.conf')
646
 
        conf._write_config_file()
647
 
        self.assertEquals(self.path, './foo.conf')
648
 
        self.assertTrue(isinstance(self.uid, int))
649
 
        self.assertTrue(isinstance(self.gid, int))
650
 
 
651
 
    def test_get_filename_parameter_is_deprecated_(self):
652
 
        conf = self.callDeprecated([
653
 
            'IniBasedConfig.__init__(get_filename) was deprecated in 2.3.'
654
 
            ' Use file_name instead.'],
655
 
            config.IniBasedConfig, lambda: 'ini.conf')
656
 
        self.assertEqual('ini.conf', conf.file_name)
657
 
 
658
 
    def test_get_parser_file_parameter_is_deprecated_(self):
659
 
        config_file = StringIO(sample_config_text.encode('utf-8'))
660
 
        conf = config.IniBasedConfig.from_string(sample_config_text)
661
 
        conf = self.callDeprecated([
662
 
            'IniBasedConfig._get_parser(file=xxx) was deprecated in 2.3.'
663
 
            ' Use IniBasedConfig(_content=xxx) instead.'],
664
 
            conf._get_parser, file=config_file)
665
 
 
666
 
 
667
 
class TestIniConfigSaving(tests.TestCaseInTempDir):
668
 
 
669
 
    def test_cant_save_without_a_file_name(self):
670
 
        conf = config.IniBasedConfig()
671
 
        self.assertRaises(AssertionError, conf._write_config_file)
672
 
 
673
 
    def test_saved_with_content(self):
674
 
        content = 'foo = bar\n'
675
 
        conf = config.IniBasedConfig.from_string(
676
 
            content, file_name='./test.conf', save=True)
677
 
        self.assertFileEqual(content, 'test.conf')
678
 
 
679
 
 
680
 
class TestIniConfigOptionExpansionDefaultValue(tests.TestCaseInTempDir):
681
 
    """What is the default value of expand for config options.
682
 
 
683
 
    This is an opt-in beta feature used to evaluate whether or not option
684
 
    references can appear in dangerous place raising exceptions, disapearing
685
 
    (and as such corrupting data) or if it's safe to activate the option by
686
 
    default.
687
 
 
688
 
    Note that these tests relies on config._expand_default_value being already
689
 
    overwritten in the parent class setUp.
690
 
    """
691
 
 
692
 
    def setUp(self):
693
 
        super(TestIniConfigOptionExpansionDefaultValue, self).setUp()
694
 
        self.config = None
695
 
        self.warnings = []
696
 
        def warning(*args):
697
 
            self.warnings.append(args[0] % args[1:])
698
 
        self.overrideAttr(trace, 'warning', warning)
699
 
 
700
 
    def get_config(self, expand):
701
 
        c = config.GlobalConfig.from_string('bzr.config.expand=%s' % (expand,),
702
 
                                            save=True)
703
 
        return c
704
 
 
705
 
    def assertExpandIs(self, expected):
706
 
        actual = config._get_expand_default_value()
707
 
        #self.config.get_user_option_as_bool('bzr.config.expand')
708
 
        self.assertEquals(expected, actual)
709
 
 
710
 
    def test_default_is_None(self):
711
 
        self.assertEquals(None, config._expand_default_value)
712
 
 
713
 
    def test_default_is_False_even_if_None(self):
714
 
        self.config = self.get_config(None)
715
 
        self.assertExpandIs(False)
716
 
 
717
 
    def test_default_is_False_even_if_invalid(self):
718
 
        self.config = self.get_config('<your choice>')
719
 
        self.assertExpandIs(False)
720
 
        # ...
721
 
        # Huh ? My choice is False ? Thanks, always happy to hear that :D
722
 
        # Wait, you've been warned !
723
 
        self.assertLength(1, self.warnings)
724
 
        self.assertEquals(
725
 
            'Value "<your choice>" is not a boolean for "bzr.config.expand"',
726
 
            self.warnings[0])
727
 
 
728
 
    def test_default_is_True(self):
729
 
        self.config = self.get_config(True)
730
 
        self.assertExpandIs(True)
731
 
 
732
 
    def test_default_is_False(self):
733
 
        self.config = self.get_config(False)
734
 
        self.assertExpandIs(False)
735
 
 
736
 
 
737
 
class TestIniConfigOptionExpansion(tests.TestCase):
738
 
    """Test option expansion from the IniConfig level.
739
 
 
740
 
    What we really want here is to test the Config level, but the class being
741
 
    abstract as far as storing values is concerned, this can't be done
742
 
    properly (yet).
743
 
    """
744
 
    # FIXME: This should be rewritten when all configs share a storage
745
 
    # implementation -- vila 2011-02-18
746
 
 
747
 
    def get_config(self, string=None):
748
 
        if string is None:
749
 
            string = ''
750
 
        c = config.IniBasedConfig.from_string(string)
751
 
        return c
752
 
 
753
 
    def assertExpansion(self, expected, conf, string, env=None):
754
 
        self.assertEquals(expected, conf.expand_options(string, env))
755
 
 
756
 
    def test_no_expansion(self):
757
 
        c = self.get_config('')
758
 
        self.assertExpansion('foo', c, 'foo')
759
 
 
760
 
    def test_env_adding_options(self):
761
 
        c = self.get_config('')
762
 
        self.assertExpansion('bar', c, '{foo}', {'foo': 'bar'})
763
 
 
764
 
    def test_env_overriding_options(self):
765
 
        c = self.get_config('foo=baz')
766
 
        self.assertExpansion('bar', c, '{foo}', {'foo': 'bar'})
767
 
 
768
 
    def test_simple_ref(self):
769
 
        c = self.get_config('foo=xxx')
770
 
        self.assertExpansion('xxx', c, '{foo}')
771
 
 
772
 
    def test_unknown_ref(self):
773
 
        c = self.get_config('')
774
 
        self.assertRaises(errors.ExpandingUnknownOption,
775
 
                          c.expand_options, '{foo}')
776
 
 
777
 
    def test_indirect_ref(self):
778
 
        c = self.get_config('''
779
 
foo=xxx
780
 
bar={foo}
781
 
''')
782
 
        self.assertExpansion('xxx', c, '{bar}')
783
 
 
784
 
    def test_embedded_ref(self):
785
 
        c = self.get_config('''
786
 
foo=xxx
787
 
bar=foo
788
 
''')
789
 
        self.assertExpansion('xxx', c, '{{bar}}')
790
 
 
791
 
    def test_simple_loop(self):
792
 
        c = self.get_config('foo={foo}')
793
 
        self.assertRaises(errors.OptionExpansionLoop, c.expand_options, '{foo}')
794
 
 
795
 
    def test_indirect_loop(self):
796
 
        c = self.get_config('''
797
 
foo={bar}
798
 
bar={baz}
799
 
baz={foo}''')
800
 
        e = self.assertRaises(errors.OptionExpansionLoop,
801
 
                              c.expand_options, '{foo}')
802
 
        self.assertEquals('foo->bar->baz', e.refs)
803
 
        self.assertEquals('{foo}', e.string)
804
 
 
805
 
    def test_list(self):
806
 
        conf = self.get_config('''
807
 
foo=start
808
 
bar=middle
809
 
baz=end
810
 
list={foo},{bar},{baz}
811
 
''')
812
 
        self.assertEquals(['start', 'middle', 'end'],
813
 
                           conf.get_user_option('list', expand=True))
814
 
 
815
 
    def test_cascading_list(self):
816
 
        conf = self.get_config('''
817
 
foo=start,{bar}
818
 
bar=middle,{baz}
819
 
baz=end
820
 
list={foo}
821
 
''')
822
 
        self.assertEquals(['start', 'middle', 'end'],
823
 
                           conf.get_user_option('list', expand=True))
824
 
 
825
 
    def test_pathological_hidden_list(self):
826
 
        conf = self.get_config('''
827
 
foo=bin
828
 
bar=go
829
 
start={foo
830
 
middle=},{
831
 
end=bar}
832
 
hidden={start}{middle}{end}
833
 
''')
834
 
        # Nope, it's either a string or a list, and the list wins as soon as a
835
 
        # ',' appears, so the string concatenation never occur.
836
 
        self.assertEquals(['{foo', '}', '{', 'bar}'],
837
 
                          conf.get_user_option('hidden', expand=True))
838
 
 
839
 
 
840
 
class TestLocationConfigOptionExpansion(tests.TestCaseInTempDir):
841
 
 
842
 
    def get_config(self, location, string=None):
843
 
        if string is None:
844
 
            string = ''
845
 
        # Since we don't save the config we won't strictly require to inherit
846
 
        # from TestCaseInTempDir, but an error occurs so quickly...
847
 
        c = config.LocationConfig.from_string(string, location)
848
 
        return c
849
 
 
850
 
    def test_dont_cross_unrelated_section(self):
851
 
        c = self.get_config('/another/branch/path','''
852
 
[/one/branch/path]
853
 
foo = hello
854
 
bar = {foo}/2
855
 
 
856
 
[/another/branch/path]
857
 
bar = {foo}/2
858
 
''')
859
 
        self.assertRaises(errors.ExpandingUnknownOption,
860
 
                          c.get_user_option, 'bar', expand=True)
861
 
 
862
 
    def test_cross_related_sections(self):
863
 
        c = self.get_config('/project/branch/path','''
864
 
[/project]
865
 
foo = qu
866
 
 
867
 
[/project/branch/path]
868
 
bar = {foo}ux
869
 
''')
870
 
        self.assertEquals('quux', c.get_user_option('bar', expand=True))
871
 
 
872
 
 
873
 
class TestIniBaseConfigOnDisk(tests.TestCaseInTempDir):
874
 
 
875
 
    def test_cannot_reload_without_name(self):
876
 
        conf = config.IniBasedConfig.from_string(sample_config_text)
877
 
        self.assertRaises(AssertionError, conf.reload)
878
 
 
879
 
    def test_reload_see_new_value(self):
880
 
        c1 = config.IniBasedConfig.from_string('editor=vim\n',
881
 
                                               file_name='./test/conf')
882
 
        c1._write_config_file()
883
 
        c2 = config.IniBasedConfig.from_string('editor=emacs\n',
884
 
                                               file_name='./test/conf')
885
 
        c2._write_config_file()
886
 
        self.assertEqual('vim', c1.get_user_option('editor'))
887
 
        self.assertEqual('emacs', c2.get_user_option('editor'))
888
 
        # Make sure we get the Right value
889
 
        c1.reload()
890
 
        self.assertEqual('emacs', c1.get_user_option('editor'))
891
 
 
892
 
 
893
 
class TestLockableConfig(tests.TestCaseInTempDir):
894
 
 
895
 
    scenarios = lockable_config_scenarios()
896
 
 
897
 
    # Set by load_tests
898
 
    config_class = None
899
 
    config_args = None
900
 
    config_section = None
901
 
 
902
 
    def setUp(self):
903
 
        super(TestLockableConfig, self).setUp()
904
 
        self._content = '[%s]\none=1\ntwo=2\n' % (self.config_section,)
905
 
        self.config = self.create_config(self._content)
906
 
 
907
 
    def get_existing_config(self):
908
 
        return self.config_class(*self.config_args)
909
 
 
910
 
    def create_config(self, content):
911
 
        kwargs = dict(save=True)
912
 
        c = self.config_class.from_string(content, *self.config_args, **kwargs)
913
 
        return c
914
 
 
915
 
    def test_simple_read_access(self):
916
 
        self.assertEquals('1', self.config.get_user_option('one'))
917
 
 
918
 
    def test_simple_write_access(self):
919
 
        self.config.set_user_option('one', 'one')
920
 
        self.assertEquals('one', self.config.get_user_option('one'))
921
 
 
922
 
    def test_listen_to_the_last_speaker(self):
923
 
        c1 = self.config
924
 
        c2 = self.get_existing_config()
925
 
        c1.set_user_option('one', 'ONE')
926
 
        c2.set_user_option('two', 'TWO')
927
 
        self.assertEquals('ONE', c1.get_user_option('one'))
928
 
        self.assertEquals('TWO', c2.get_user_option('two'))
929
 
        # The second update respect the first one
930
 
        self.assertEquals('ONE', c2.get_user_option('one'))
931
 
 
932
 
    def test_last_speaker_wins(self):
933
 
        # If the same config is not shared, the same variable modified twice
934
 
        # can only see a single result.
935
 
        c1 = self.config
936
 
        c2 = self.get_existing_config()
937
 
        c1.set_user_option('one', 'c1')
938
 
        c2.set_user_option('one', 'c2')
939
 
        self.assertEquals('c2', c2._get_user_option('one'))
940
 
        # The first modification is still available until another refresh
941
 
        # occur
942
 
        self.assertEquals('c1', c1._get_user_option('one'))
943
 
        c1.set_user_option('two', 'done')
944
 
        self.assertEquals('c2', c1._get_user_option('one'))
945
 
 
946
 
    def test_writes_are_serialized(self):
947
 
        c1 = self.config
948
 
        c2 = self.get_existing_config()
949
 
 
950
 
        # We spawn a thread that will pause *during* the write
951
 
        before_writing = threading.Event()
952
 
        after_writing = threading.Event()
953
 
        writing_done = threading.Event()
954
 
        c1_orig = c1._write_config_file
955
 
        def c1_write_config_file():
956
 
            before_writing.set()
957
 
            c1_orig()
958
 
            # The lock is held. We wait for the main thread to decide when to
959
 
            # continue
960
 
            after_writing.wait()
961
 
        c1._write_config_file = c1_write_config_file
962
 
        def c1_set_option():
963
 
            c1.set_user_option('one', 'c1')
964
 
            writing_done.set()
965
 
        t1 = threading.Thread(target=c1_set_option)
966
 
        # Collect the thread after the test
967
 
        self.addCleanup(t1.join)
968
 
        # Be ready to unblock the thread if the test goes wrong
969
 
        self.addCleanup(after_writing.set)
970
 
        t1.start()
971
 
        before_writing.wait()
972
 
        self.assertTrue(c1._lock.is_held)
973
 
        self.assertRaises(errors.LockContention,
974
 
                          c2.set_user_option, 'one', 'c2')
975
 
        self.assertEquals('c1', c1.get_user_option('one'))
976
 
        # Let the lock be released
977
 
        after_writing.set()
978
 
        writing_done.wait()
979
 
        c2.set_user_option('one', 'c2')
980
 
        self.assertEquals('c2', c2.get_user_option('one'))
981
 
 
982
 
    def test_read_while_writing(self):
983
 
       c1 = self.config
984
 
       # We spawn a thread that will pause *during* the write
985
 
       ready_to_write = threading.Event()
986
 
       do_writing = threading.Event()
987
 
       writing_done = threading.Event()
988
 
       c1_orig = c1._write_config_file
989
 
       def c1_write_config_file():
990
 
           ready_to_write.set()
991
 
           # The lock is held. We wait for the main thread to decide when to
992
 
           # continue
993
 
           do_writing.wait()
994
 
           c1_orig()
995
 
           writing_done.set()
996
 
       c1._write_config_file = c1_write_config_file
997
 
       def c1_set_option():
998
 
           c1.set_user_option('one', 'c1')
999
 
       t1 = threading.Thread(target=c1_set_option)
1000
 
       # Collect the thread after the test
1001
 
       self.addCleanup(t1.join)
1002
 
       # Be ready to unblock the thread if the test goes wrong
1003
 
       self.addCleanup(do_writing.set)
1004
 
       t1.start()
1005
 
       # Ensure the thread is ready to write
1006
 
       ready_to_write.wait()
1007
 
       self.assertTrue(c1._lock.is_held)
1008
 
       self.assertEquals('c1', c1.get_user_option('one'))
1009
 
       # If we read during the write, we get the old value
1010
 
       c2 = self.get_existing_config()
1011
 
       self.assertEquals('1', c2.get_user_option('one'))
1012
 
       # Let the writing occur and ensure it occurred
1013
 
       do_writing.set()
1014
 
       writing_done.wait()
1015
 
       # Now we get the updated value
1016
 
       c3 = self.get_existing_config()
1017
 
       self.assertEquals('c1', c3.get_user_option('one'))
1018
 
 
1019
 
 
1020
 
class TestGetUserOptionAs(TestIniConfig):
1021
 
 
1022
 
    def test_get_user_option_as_bool(self):
1023
 
        conf, parser = self.make_config_parser("""
1024
 
a_true_bool = true
1025
 
a_false_bool = 0
1026
 
an_invalid_bool = maybe
1027
 
a_list = hmm, who knows ? # This is interpreted as a list !
1028
 
""")
1029
 
        get_bool = conf.get_user_option_as_bool
1030
 
        self.assertEqual(True, get_bool('a_true_bool'))
1031
 
        self.assertEqual(False, get_bool('a_false_bool'))
1032
 
        warnings = []
1033
 
        def warning(*args):
1034
 
            warnings.append(args[0] % args[1:])
1035
 
        self.overrideAttr(trace, 'warning', warning)
1036
 
        msg = 'Value "%s" is not a boolean for "%s"'
1037
 
        self.assertIs(None, get_bool('an_invalid_bool'))
1038
 
        self.assertEquals(msg % ('maybe', 'an_invalid_bool'), warnings[0])
1039
 
        warnings = []
1040
 
        self.assertIs(None, get_bool('not_defined_in_this_config'))
1041
 
        self.assertEquals([], warnings)
1042
 
 
1043
 
    def test_get_user_option_as_list(self):
1044
 
        conf, parser = self.make_config_parser("""
1045
 
a_list = a,b,c
1046
 
length_1 = 1,
1047
 
one_item = x
1048
 
""")
1049
 
        get_list = conf.get_user_option_as_list
1050
 
        self.assertEqual(['a', 'b', 'c'], get_list('a_list'))
1051
 
        self.assertEqual(['1'], get_list('length_1'))
1052
 
        self.assertEqual('x', conf.get_user_option('one_item'))
1053
 
        # automatically cast to list
1054
 
        self.assertEqual(['x'], get_list('one_item'))
1055
 
 
1056
 
    def test_get_user_option_as_int_from_SI(self):
1057
 
        conf, parser = self.make_config_parser("""
1058
 
plain = 100
1059
 
si_k = 5k,
1060
 
si_kb = 5kb,
1061
 
si_m = 5M,
1062
 
si_mb = 5MB,
1063
 
si_g = 5g,
1064
 
si_gb = 5gB,
1065
 
""")
1066
 
        get_si = conf.get_user_option_as_int_from_SI
1067
 
        self.assertEqual(100, get_si('plain'))
1068
 
        self.assertEqual(5000, get_si('si_k'))
1069
 
        self.assertEqual(5000, get_si('si_kb'))
1070
 
        self.assertEqual(5000000, get_si('si_m'))
1071
 
        self.assertEqual(5000000, get_si('si_mb'))
1072
 
        self.assertEqual(5000000000, get_si('si_g'))
1073
 
        self.assertEqual(5000000000, get_si('si_gb'))
1074
 
        self.assertEqual(None, get_si('non-exist'))
1075
 
        self.assertEqual(42, get_si('non-exist-with-default',  42))
1076
 
 
1077
 
class TestSupressWarning(TestIniConfig):
1078
 
 
1079
 
    def make_warnings_config(self, s):
1080
 
        conf, parser = self.make_config_parser(s)
1081
 
        return conf.suppress_warning
1082
 
 
1083
 
    def test_suppress_warning_unknown(self):
1084
 
        suppress_warning = self.make_warnings_config('')
1085
 
        self.assertEqual(False, suppress_warning('unknown_warning'))
1086
 
 
1087
 
    def test_suppress_warning_known(self):
1088
 
        suppress_warning = self.make_warnings_config('suppress_warnings=a,b')
1089
 
        self.assertEqual(False, suppress_warning('c'))
1090
 
        self.assertEqual(True, suppress_warning('a'))
1091
 
        self.assertEqual(True, suppress_warning('b'))
1092
 
 
1093
 
 
1094
 
class TestGetConfig(tests.TestCase):
1095
 
 
1096
 
    def test_constructs(self):
1097
 
        my_config = config.GlobalConfig()
1098
 
 
1099
 
    def test_calls_read_filenames(self):
1100
 
        # replace the class that is constructed, to check its parameters
1101
 
        oldparserclass = config.ConfigObj
1102
 
        config.ConfigObj = InstrumentedConfigObj
1103
 
        my_config = config.GlobalConfig()
1104
 
        try:
1105
 
            parser = my_config._get_parser()
1106
 
        finally:
1107
 
            config.ConfigObj = oldparserclass
1108
 
        self.assertIsInstance(parser, InstrumentedConfigObj)
1109
 
        self.assertEqual(parser._calls, [('__init__', config.config_filename(),
1110
 
                                          'utf-8')])
1111
 
 
1112
 
 
1113
 
class TestBranchConfig(tests.TestCaseWithTransport):
1114
 
 
1115
 
    def test_constructs(self):
1116
 
        branch = FakeBranch()
1117
 
        my_config = config.BranchConfig(branch)
1118
 
        self.assertRaises(TypeError, config.BranchConfig)
1119
 
 
1120
 
    def test_get_location_config(self):
1121
 
        branch = FakeBranch()
1122
 
        my_config = config.BranchConfig(branch)
1123
 
        location_config = my_config._get_location_config()
1124
 
        self.assertEqual(branch.base, location_config.location)
1125
 
        self.assertIs(location_config, my_config._get_location_config())
1126
 
 
1127
 
    def test_get_config(self):
1128
 
        """The Branch.get_config method works properly"""
1129
 
        b = bzrdir.BzrDir.create_standalone_workingtree('.').branch
1130
 
        my_config = b.get_config()
1131
 
        self.assertIs(my_config.get_user_option('wacky'), None)
1132
 
        my_config.set_user_option('wacky', 'unlikely')
1133
 
        self.assertEqual(my_config.get_user_option('wacky'), 'unlikely')
1134
 
 
1135
 
        # Ensure we get the same thing if we start again
1136
 
        b2 = branch.Branch.open('.')
1137
 
        my_config2 = b2.get_config()
1138
 
        self.assertEqual(my_config2.get_user_option('wacky'), 'unlikely')
1139
 
 
1140
 
    def test_has_explicit_nickname(self):
1141
 
        b = self.make_branch('.')
1142
 
        self.assertFalse(b.get_config().has_explicit_nickname())
1143
 
        b.nick = 'foo'
1144
 
        self.assertTrue(b.get_config().has_explicit_nickname())
1145
 
 
1146
 
    def test_config_url(self):
1147
 
        """The Branch.get_config will use section that uses a local url"""
1148
 
        branch = self.make_branch('branch')
1149
 
        self.assertEqual('branch', branch.nick)
1150
 
 
1151
 
        local_url = urlutils.local_path_to_url('branch')
1152
 
        conf = config.LocationConfig.from_string(
1153
 
            '[%s]\nnickname = foobar' % (local_url,),
1154
 
            local_url, save=True)
1155
 
        self.assertEqual('foobar', branch.nick)
1156
 
 
1157
 
    def test_config_local_path(self):
1158
 
        """The Branch.get_config will use a local system path"""
1159
 
        branch = self.make_branch('branch')
1160
 
        self.assertEqual('branch', branch.nick)
1161
 
 
1162
 
        local_path = osutils.getcwd().encode('utf8')
1163
 
        conf = config.LocationConfig.from_string(
1164
 
            '[%s/branch]\nnickname = barry' % (local_path,),
1165
 
            'branch',  save=True)
1166
 
        self.assertEqual('barry', branch.nick)
1167
 
 
1168
 
    def test_config_creates_local(self):
1169
 
        """Creating a new entry in config uses a local path."""
1170
 
        branch = self.make_branch('branch', format='knit')
1171
 
        branch.set_push_location('http://foobar')
1172
 
        local_path = osutils.getcwd().encode('utf8')
1173
 
        # Surprisingly ConfigObj doesn't create a trailing newline
1174
 
        self.check_file_contents(config.locations_config_filename(),
1175
 
                                 '[%s/branch]\n'
1176
 
                                 'push_location = http://foobar\n'
1177
 
                                 'push_location:policy = norecurse\n'
1178
 
                                 % (local_path,))
1179
 
 
1180
 
    def test_autonick_urlencoded(self):
1181
 
        b = self.make_branch('!repo')
1182
 
        self.assertEqual('!repo', b.get_config().get_nickname())
1183
 
 
1184
 
    def test_warn_if_masked(self):
1185
 
        warnings = []
1186
 
        def warning(*args):
1187
 
            warnings.append(args[0] % args[1:])
1188
 
        self.overrideAttr(trace, 'warning', warning)
1189
 
 
1190
 
        def set_option(store, warn_masked=True):
1191
 
            warnings[:] = []
1192
 
            conf.set_user_option('example_option', repr(store), store=store,
1193
 
                                 warn_masked=warn_masked)
1194
 
        def assertWarning(warning):
1195
 
            if warning is None:
1196
 
                self.assertEqual(0, len(warnings))
1197
 
            else:
1198
 
                self.assertEqual(1, len(warnings))
1199
 
                self.assertEqual(warning, warnings[0])
1200
 
        branch = self.make_branch('.')
1201
 
        conf = branch.get_config()
1202
 
        set_option(config.STORE_GLOBAL)
1203
 
        assertWarning(None)
1204
 
        set_option(config.STORE_BRANCH)
1205
 
        assertWarning(None)
1206
 
        set_option(config.STORE_GLOBAL)
1207
 
        assertWarning('Value "4" is masked by "3" from branch.conf')
1208
 
        set_option(config.STORE_GLOBAL, warn_masked=False)
1209
 
        assertWarning(None)
1210
 
        set_option(config.STORE_LOCATION)
1211
 
        assertWarning(None)
1212
 
        set_option(config.STORE_BRANCH)
1213
 
        assertWarning('Value "3" is masked by "0" from locations.conf')
1214
 
        set_option(config.STORE_BRANCH, warn_masked=False)
1215
 
        assertWarning(None)
1216
 
 
1217
 
 
1218
 
class TestGlobalConfigItems(tests.TestCaseInTempDir):
1219
 
 
1220
 
    def test_user_id(self):
1221
 
        my_config = config.GlobalConfig.from_string(sample_config_text)
1222
 
        self.assertEqual(u"Erik B\u00e5gfors <erik@bagfors.nu>",
1223
 
                         my_config._get_user_id())
1224
 
 
1225
 
    def test_absent_user_id(self):
1226
 
        my_config = config.GlobalConfig()
1227
 
        self.assertEqual(None, my_config._get_user_id())
1228
 
 
1229
 
    def test_configured_editor(self):
1230
 
        my_config = config.GlobalConfig.from_string(sample_config_text)
1231
 
        editor = self.applyDeprecated(
1232
 
            deprecated_in((2, 4, 0)), my_config.get_editor)
1233
 
        self.assertEqual('vim', editor)
1234
 
 
1235
 
    def test_signatures_always(self):
1236
 
        my_config = config.GlobalConfig.from_string(sample_always_signatures)
1237
 
        self.assertEqual(config.CHECK_NEVER,
1238
 
                         my_config.signature_checking())
1239
 
        self.assertEqual(config.SIGN_ALWAYS,
1240
 
                         my_config.signing_policy())
1241
 
        self.assertEqual(True, my_config.signature_needed())
1242
 
 
1243
 
    def test_signatures_if_possible(self):
1244
 
        my_config = config.GlobalConfig.from_string(sample_maybe_signatures)
1245
 
        self.assertEqual(config.CHECK_NEVER,
1246
 
                         my_config.signature_checking())
1247
 
        self.assertEqual(config.SIGN_WHEN_REQUIRED,
1248
 
                         my_config.signing_policy())
1249
 
        self.assertEqual(False, my_config.signature_needed())
1250
 
 
1251
 
    def test_signatures_ignore(self):
1252
 
        my_config = config.GlobalConfig.from_string(sample_ignore_signatures)
1253
 
        self.assertEqual(config.CHECK_ALWAYS,
1254
 
                         my_config.signature_checking())
1255
 
        self.assertEqual(config.SIGN_NEVER,
1256
 
                         my_config.signing_policy())
1257
 
        self.assertEqual(False, my_config.signature_needed())
1258
 
 
1259
 
    def _get_sample_config(self):
1260
 
        my_config = config.GlobalConfig.from_string(sample_config_text)
1261
 
        return my_config
1262
 
 
1263
 
    def test_gpg_signing_command(self):
1264
 
        my_config = self._get_sample_config()
1265
 
        self.assertEqual("gnome-gpg", my_config.gpg_signing_command())
1266
 
        self.assertEqual(False, my_config.signature_needed())
1267
 
 
1268
 
    def test_gpg_signing_key(self):
1269
 
        my_config = self._get_sample_config()
1270
 
        self.assertEqual("DD4D5088", my_config.gpg_signing_key())
1271
 
 
1272
 
    def _get_empty_config(self):
1273
 
        my_config = config.GlobalConfig()
1274
 
        return my_config
1275
 
 
1276
 
    def test_gpg_signing_command_unset(self):
1277
 
        my_config = self._get_empty_config()
1278
 
        self.assertEqual("gpg", my_config.gpg_signing_command())
1279
 
 
1280
 
    def test_get_user_option_default(self):
1281
 
        my_config = self._get_empty_config()
1282
 
        self.assertEqual(None, my_config.get_user_option('no_option'))
1283
 
 
1284
 
    def test_get_user_option_global(self):
1285
 
        my_config = self._get_sample_config()
1286
 
        self.assertEqual("something",
1287
 
                         my_config.get_user_option('user_global_option'))
1288
 
 
1289
 
    def test_post_commit_default(self):
1290
 
        my_config = self._get_sample_config()
1291
 
        self.assertEqual(None, my_config.post_commit())
1292
 
 
1293
 
    def test_configured_logformat(self):
1294
 
        my_config = self._get_sample_config()
1295
 
        self.assertEqual("short", my_config.log_format())
1296
 
 
1297
 
    def test_configured_acceptable_keys(self):
1298
 
        my_config = self._get_sample_config()
1299
 
        self.assertEqual("amy", my_config.acceptable_keys())
1300
 
 
1301
 
    def test_configured_validate_signatures_in_log(self):
1302
 
        my_config = self._get_sample_config()
1303
 
        self.assertEqual(True, my_config.validate_signatures_in_log())
1304
 
 
1305
 
    def test_get_alias(self):
1306
 
        my_config = self._get_sample_config()
1307
 
        self.assertEqual('help', my_config.get_alias('h'))
1308
 
 
1309
 
    def test_get_aliases(self):
1310
 
        my_config = self._get_sample_config()
1311
 
        aliases = my_config.get_aliases()
1312
 
        self.assertEqual(2, len(aliases))
1313
 
        sorted_keys = sorted(aliases)
1314
 
        self.assertEqual('help', aliases[sorted_keys[0]])
1315
 
        self.assertEqual(sample_long_alias, aliases[sorted_keys[1]])
1316
 
 
1317
 
    def test_get_no_alias(self):
1318
 
        my_config = self._get_sample_config()
1319
 
        self.assertEqual(None, my_config.get_alias('foo'))
1320
 
 
1321
 
    def test_get_long_alias(self):
1322
 
        my_config = self._get_sample_config()
1323
 
        self.assertEqual(sample_long_alias, my_config.get_alias('ll'))
1324
 
 
1325
 
    def test_get_change_editor(self):
1326
 
        my_config = self._get_sample_config()
1327
 
        change_editor = my_config.get_change_editor('old', 'new')
1328
 
        self.assertIs(diff.DiffFromTool, change_editor.__class__)
1329
 
        self.assertEqual('vimdiff -of @new_path @old_path',
1330
 
                         ' '.join(change_editor.command_template))
1331
 
 
1332
 
    def test_get_no_change_editor(self):
1333
 
        my_config = self._get_empty_config()
1334
 
        change_editor = my_config.get_change_editor('old', 'new')
1335
 
        self.assertIs(None, change_editor)
1336
 
 
1337
 
    def test_get_merge_tools(self):
1338
 
        conf = self._get_sample_config()
1339
 
        tools = conf.get_merge_tools()
1340
 
        self.log(repr(tools))
1341
 
        self.assertEqual(
1342
 
            {u'funkytool' : u'funkytool "arg with spaces" {this_temp}',
1343
 
            u'sometool' : u'sometool {base} {this} {other} -o {result}',
1344
 
            u'newtool' : u'"newtool with spaces" {this_temp}'},
1345
 
            tools)
1346
 
 
1347
 
    def test_get_merge_tools_empty(self):
1348
 
        conf = self._get_empty_config()
1349
 
        tools = conf.get_merge_tools()
1350
 
        self.assertEqual({}, tools)
1351
 
 
1352
 
    def test_find_merge_tool(self):
1353
 
        conf = self._get_sample_config()
1354
 
        cmdline = conf.find_merge_tool('sometool')
1355
 
        self.assertEqual('sometool {base} {this} {other} -o {result}', cmdline)
1356
 
 
1357
 
    def test_find_merge_tool_not_found(self):
1358
 
        conf = self._get_sample_config()
1359
 
        cmdline = conf.find_merge_tool('DOES NOT EXIST')
1360
 
        self.assertIs(cmdline, None)
1361
 
 
1362
 
    def test_find_merge_tool_known(self):
1363
 
        conf = self._get_empty_config()
1364
 
        cmdline = conf.find_merge_tool('kdiff3')
1365
 
        self.assertEquals('kdiff3 {base} {this} {other} -o {result}', cmdline)
1366
 
 
1367
 
    def test_find_merge_tool_override_known(self):
1368
 
        conf = self._get_empty_config()
1369
 
        conf.set_user_option('bzr.mergetool.kdiff3', 'kdiff3 blah')
1370
 
        cmdline = conf.find_merge_tool('kdiff3')
1371
 
        self.assertEqual('kdiff3 blah', cmdline)
1372
 
 
1373
 
 
1374
 
class TestGlobalConfigSavingOptions(tests.TestCaseInTempDir):
1375
 
 
1376
 
    def test_empty(self):
1377
 
        my_config = config.GlobalConfig()
1378
 
        self.assertEqual(0, len(my_config.get_aliases()))
1379
 
 
1380
 
    def test_set_alias(self):
1381
 
        my_config = config.GlobalConfig()
1382
 
        alias_value = 'commit --strict'
1383
 
        my_config.set_alias('commit', alias_value)
1384
 
        new_config = config.GlobalConfig()
1385
 
        self.assertEqual(alias_value, new_config.get_alias('commit'))
1386
 
 
1387
 
    def test_remove_alias(self):
1388
 
        my_config = config.GlobalConfig()
1389
 
        my_config.set_alias('commit', 'commit --strict')
1390
 
        # Now remove the alias again.
1391
 
        my_config.unset_alias('commit')
1392
 
        new_config = config.GlobalConfig()
1393
 
        self.assertIs(None, new_config.get_alias('commit'))
1394
 
 
1395
 
 
1396
 
class TestLocationConfig(tests.TestCaseInTempDir, TestOptionsMixin):
1397
 
 
1398
 
    def test_constructs(self):
1399
 
        my_config = config.LocationConfig('http://example.com')
1400
 
        self.assertRaises(TypeError, config.LocationConfig)
1401
 
 
1402
 
    def test_branch_calls_read_filenames(self):
1403
 
        # This is testing the correct file names are provided.
1404
 
        # TODO: consolidate with the test for GlobalConfigs filename checks.
1405
 
        #
1406
 
        # replace the class that is constructed, to check its parameters
1407
 
        oldparserclass = config.ConfigObj
1408
 
        config.ConfigObj = InstrumentedConfigObj
1409
 
        try:
1410
 
            my_config = config.LocationConfig('http://www.example.com')
1411
 
            parser = my_config._get_parser()
1412
 
        finally:
1413
 
            config.ConfigObj = oldparserclass
1414
 
        self.assertIsInstance(parser, InstrumentedConfigObj)
1415
 
        self.assertEqual(parser._calls,
1416
 
                         [('__init__', config.locations_config_filename(),
1417
 
                           'utf-8')])
1418
 
 
1419
 
    def test_get_global_config(self):
1420
 
        my_config = config.BranchConfig(FakeBranch('http://example.com'))
1421
 
        global_config = my_config._get_global_config()
1422
 
        self.assertIsInstance(global_config, config.GlobalConfig)
1423
 
        self.assertIs(global_config, my_config._get_global_config())
1424
 
 
1425
 
    def assertLocationMatching(self, expected):
1426
 
        self.assertEqual(expected,
1427
 
                         list(self.my_location_config._get_matching_sections()))
1428
 
 
1429
 
    def test__get_matching_sections_no_match(self):
1430
 
        self.get_branch_config('/')
1431
 
        self.assertLocationMatching([])
1432
 
 
1433
 
    def test__get_matching_sections_exact(self):
1434
 
        self.get_branch_config('http://www.example.com')
1435
 
        self.assertLocationMatching([('http://www.example.com', '')])
1436
 
 
1437
 
    def test__get_matching_sections_suffix_does_not(self):
1438
 
        self.get_branch_config('http://www.example.com-com')
1439
 
        self.assertLocationMatching([])
1440
 
 
1441
 
    def test__get_matching_sections_subdir_recursive(self):
1442
 
        self.get_branch_config('http://www.example.com/com')
1443
 
        self.assertLocationMatching([('http://www.example.com', 'com')])
1444
 
 
1445
 
    def test__get_matching_sections_ignoreparent(self):
1446
 
        self.get_branch_config('http://www.example.com/ignoreparent')
1447
 
        self.assertLocationMatching([('http://www.example.com/ignoreparent',
1448
 
                                      '')])
1449
 
 
1450
 
    def test__get_matching_sections_ignoreparent_subdir(self):
1451
 
        self.get_branch_config(
1452
 
            'http://www.example.com/ignoreparent/childbranch')
1453
 
        self.assertLocationMatching([('http://www.example.com/ignoreparent',
1454
 
                                      'childbranch')])
1455
 
 
1456
 
    def test__get_matching_sections_subdir_trailing_slash(self):
1457
 
        self.get_branch_config('/b')
1458
 
        self.assertLocationMatching([('/b/', '')])
1459
 
 
1460
 
    def test__get_matching_sections_subdir_child(self):
1461
 
        self.get_branch_config('/a/foo')
1462
 
        self.assertLocationMatching([('/a/*', ''), ('/a/', 'foo')])
1463
 
 
1464
 
    def test__get_matching_sections_subdir_child_child(self):
1465
 
        self.get_branch_config('/a/foo/bar')
1466
 
        self.assertLocationMatching([('/a/*', 'bar'), ('/a/', 'foo/bar')])
1467
 
 
1468
 
    def test__get_matching_sections_trailing_slash_with_children(self):
1469
 
        self.get_branch_config('/a/')
1470
 
        self.assertLocationMatching([('/a/', '')])
1471
 
 
1472
 
    def test__get_matching_sections_explicit_over_glob(self):
1473
 
        # XXX: 2006-09-08 jamesh
1474
 
        # This test only passes because ord('c') > ord('*').  If there
1475
 
        # was a config section for '/a/?', it would get precedence
1476
 
        # over '/a/c'.
1477
 
        self.get_branch_config('/a/c')
1478
 
        self.assertLocationMatching([('/a/c', ''), ('/a/*', ''), ('/a/', 'c')])
1479
 
 
1480
 
    def test__get_option_policy_normal(self):
1481
 
        self.get_branch_config('http://www.example.com')
1482
 
        self.assertEqual(
1483
 
            self.my_location_config._get_config_policy(
1484
 
            'http://www.example.com', 'normal_option'),
1485
 
            config.POLICY_NONE)
1486
 
 
1487
 
    def test__get_option_policy_norecurse(self):
1488
 
        self.get_branch_config('http://www.example.com')
1489
 
        self.assertEqual(
1490
 
            self.my_location_config._get_option_policy(
1491
 
            'http://www.example.com', 'norecurse_option'),
1492
 
            config.POLICY_NORECURSE)
1493
 
        # Test old recurse=False setting:
1494
 
        self.assertEqual(
1495
 
            self.my_location_config._get_option_policy(
1496
 
            'http://www.example.com/norecurse', 'normal_option'),
1497
 
            config.POLICY_NORECURSE)
1498
 
 
1499
 
    def test__get_option_policy_normal(self):
1500
 
        self.get_branch_config('http://www.example.com')
1501
 
        self.assertEqual(
1502
 
            self.my_location_config._get_option_policy(
1503
 
            'http://www.example.com', 'appendpath_option'),
1504
 
            config.POLICY_APPENDPATH)
1505
 
 
1506
 
    def test__get_options_with_policy(self):
1507
 
        self.get_branch_config('/dir/subdir',
1508
 
                               location_config="""\
1509
 
[/dir]
1510
 
other_url = /other-dir
1511
 
other_url:policy = appendpath
1512
 
[/dir/subdir]
1513
 
other_url = /other-subdir
1514
 
""")
1515
 
        self.assertOptions(
1516
 
            [(u'other_url', u'/other-subdir', u'/dir/subdir', 'locations'),
1517
 
             (u'other_url', u'/other-dir', u'/dir', 'locations'),
1518
 
             (u'other_url:policy', u'appendpath', u'/dir', 'locations')],
1519
 
            self.my_location_config)
1520
 
 
1521
 
    def test_location_without_username(self):
1522
 
        self.get_branch_config('http://www.example.com/ignoreparent')
1523
 
        self.assertEqual(u'Erik B\u00e5gfors <erik@bagfors.nu>',
1524
 
                         self.my_config.username())
1525
 
 
1526
 
    def test_location_not_listed(self):
1527
 
        """Test that the global username is used when no location matches"""
1528
 
        self.get_branch_config('/home/robertc/sources')
1529
 
        self.assertEqual(u'Erik B\u00e5gfors <erik@bagfors.nu>',
1530
 
                         self.my_config.username())
1531
 
 
1532
 
    def test_overriding_location(self):
1533
 
        self.get_branch_config('http://www.example.com/foo')
1534
 
        self.assertEqual('Robert Collins <robertc@example.org>',
1535
 
                         self.my_config.username())
1536
 
 
1537
 
    def test_signatures_not_set(self):
1538
 
        self.get_branch_config('http://www.example.com',
1539
 
                                 global_config=sample_ignore_signatures)
1540
 
        self.assertEqual(config.CHECK_ALWAYS,
1541
 
                         self.my_config.signature_checking())
1542
 
        self.assertEqual(config.SIGN_NEVER,
1543
 
                         self.my_config.signing_policy())
1544
 
 
1545
 
    def test_signatures_never(self):
1546
 
        self.get_branch_config('/a/c')
1547
 
        self.assertEqual(config.CHECK_NEVER,
1548
 
                         self.my_config.signature_checking())
1549
 
 
1550
 
    def test_signatures_when_available(self):
1551
 
        self.get_branch_config('/a/', global_config=sample_ignore_signatures)
1552
 
        self.assertEqual(config.CHECK_IF_POSSIBLE,
1553
 
                         self.my_config.signature_checking())
1554
 
 
1555
 
    def test_signatures_always(self):
1556
 
        self.get_branch_config('/b')
1557
 
        self.assertEqual(config.CHECK_ALWAYS,
1558
 
                         self.my_config.signature_checking())
1559
 
 
1560
 
    def test_gpg_signing_command(self):
1561
 
        self.get_branch_config('/b')
1562
 
        self.assertEqual("gnome-gpg", self.my_config.gpg_signing_command())
1563
 
 
1564
 
    def test_gpg_signing_command_missing(self):
1565
 
        self.get_branch_config('/a')
1566
 
        self.assertEqual("false", self.my_config.gpg_signing_command())
1567
 
 
1568
 
    def test_gpg_signing_key(self):
1569
 
        self.get_branch_config('/b')
1570
 
        self.assertEqual("DD4D5088", self.my_config.gpg_signing_key())
1571
 
 
1572
 
    def test_gpg_signing_key_default(self):
1573
 
        self.get_branch_config('/a')
1574
 
        self.assertEqual("erik@bagfors.nu", self.my_config.gpg_signing_key())
1575
 
 
1576
 
    def test_get_user_option_global(self):
1577
 
        self.get_branch_config('/a')
1578
 
        self.assertEqual('something',
1579
 
                         self.my_config.get_user_option('user_global_option'))
1580
 
 
1581
 
    def test_get_user_option_local(self):
1582
 
        self.get_branch_config('/a')
1583
 
        self.assertEqual('local',
1584
 
                         self.my_config.get_user_option('user_local_option'))
1585
 
 
1586
 
    def test_get_user_option_appendpath(self):
1587
 
        # returned as is for the base path:
1588
 
        self.get_branch_config('http://www.example.com')
1589
 
        self.assertEqual('append',
1590
 
                         self.my_config.get_user_option('appendpath_option'))
1591
 
        # Extra path components get appended:
1592
 
        self.get_branch_config('http://www.example.com/a/b/c')
1593
 
        self.assertEqual('append/a/b/c',
1594
 
                         self.my_config.get_user_option('appendpath_option'))
1595
 
        # Overriden for http://www.example.com/dir, where it is a
1596
 
        # normal option:
1597
 
        self.get_branch_config('http://www.example.com/dir/a/b/c')
1598
 
        self.assertEqual('normal',
1599
 
                         self.my_config.get_user_option('appendpath_option'))
1600
 
 
1601
 
    def test_get_user_option_norecurse(self):
1602
 
        self.get_branch_config('http://www.example.com')
1603
 
        self.assertEqual('norecurse',
1604
 
                         self.my_config.get_user_option('norecurse_option'))
1605
 
        self.get_branch_config('http://www.example.com/dir')
1606
 
        self.assertEqual(None,
1607
 
                         self.my_config.get_user_option('norecurse_option'))
1608
 
        # http://www.example.com/norecurse is a recurse=False section
1609
 
        # that redefines normal_option.  Subdirectories do not pick up
1610
 
        # this redefinition.
1611
 
        self.get_branch_config('http://www.example.com/norecurse')
1612
 
        self.assertEqual('norecurse',
1613
 
                         self.my_config.get_user_option('normal_option'))
1614
 
        self.get_branch_config('http://www.example.com/norecurse/subdir')
1615
 
        self.assertEqual('normal',
1616
 
                         self.my_config.get_user_option('normal_option'))
1617
 
 
1618
 
    def test_set_user_option_norecurse(self):
1619
 
        self.get_branch_config('http://www.example.com')
1620
 
        self.my_config.set_user_option('foo', 'bar',
1621
 
                                       store=config.STORE_LOCATION_NORECURSE)
1622
 
        self.assertEqual(
1623
 
            self.my_location_config._get_option_policy(
1624
 
            'http://www.example.com', 'foo'),
1625
 
            config.POLICY_NORECURSE)
1626
 
 
1627
 
    def test_set_user_option_appendpath(self):
1628
 
        self.get_branch_config('http://www.example.com')
1629
 
        self.my_config.set_user_option('foo', 'bar',
1630
 
                                       store=config.STORE_LOCATION_APPENDPATH)
1631
 
        self.assertEqual(
1632
 
            self.my_location_config._get_option_policy(
1633
 
            'http://www.example.com', 'foo'),
1634
 
            config.POLICY_APPENDPATH)
1635
 
 
1636
 
    def test_set_user_option_change_policy(self):
1637
 
        self.get_branch_config('http://www.example.com')
1638
 
        self.my_config.set_user_option('norecurse_option', 'normal',
1639
 
                                       store=config.STORE_LOCATION)
1640
 
        self.assertEqual(
1641
 
            self.my_location_config._get_option_policy(
1642
 
            'http://www.example.com', 'norecurse_option'),
1643
 
            config.POLICY_NONE)
1644
 
 
1645
 
    def test_set_user_option_recurse_false_section(self):
1646
 
        # The following section has recurse=False set.  The test is to
1647
 
        # make sure that a normal option can be added to the section,
1648
 
        # converting recurse=False to the norecurse policy.
1649
 
        self.get_branch_config('http://www.example.com/norecurse')
1650
 
        self.callDeprecated(['The recurse option is deprecated as of 0.14.  '
1651
 
                             'The section "http://www.example.com/norecurse" '
1652
 
                             'has been converted to use policies.'],
1653
 
                            self.my_config.set_user_option,
1654
 
                            'foo', 'bar', store=config.STORE_LOCATION)
1655
 
        self.assertEqual(
1656
 
            self.my_location_config._get_option_policy(
1657
 
            'http://www.example.com/norecurse', 'foo'),
1658
 
            config.POLICY_NONE)
1659
 
        # The previously existing option is still norecurse:
1660
 
        self.assertEqual(
1661
 
            self.my_location_config._get_option_policy(
1662
 
            'http://www.example.com/norecurse', 'normal_option'),
1663
 
            config.POLICY_NORECURSE)
1664
 
 
1665
 
    def test_post_commit_default(self):
1666
 
        self.get_branch_config('/a/c')
1667
 
        self.assertEqual('bzrlib.tests.test_config.post_commit',
1668
 
                         self.my_config.post_commit())
1669
 
 
1670
 
    def get_branch_config(self, location, global_config=None,
1671
 
                          location_config=None):
1672
 
        my_branch = FakeBranch(location)
1673
 
        if global_config is None:
1674
 
            global_config = sample_config_text
1675
 
        if location_config is None:
1676
 
            location_config = sample_branches_text
1677
 
 
1678
 
        my_global_config = config.GlobalConfig.from_string(global_config,
1679
 
                                                           save=True)
1680
 
        my_location_config = config.LocationConfig.from_string(
1681
 
            location_config, my_branch.base, save=True)
1682
 
        my_config = config.BranchConfig(my_branch)
1683
 
        self.my_config = my_config
1684
 
        self.my_location_config = my_config._get_location_config()
1685
 
 
1686
 
    def test_set_user_setting_sets_and_saves(self):
1687
 
        self.get_branch_config('/a/c')
1688
 
        record = InstrumentedConfigObj("foo")
1689
 
        self.my_location_config._parser = record
1690
 
 
1691
 
        self.callDeprecated(['The recurse option is deprecated as of '
1692
 
                             '0.14.  The section "/a/c" has been '
1693
 
                             'converted to use policies.'],
1694
 
                            self.my_config.set_user_option,
1695
 
                            'foo', 'bar', store=config.STORE_LOCATION)
1696
 
        self.assertEqual([('reload',),
1697
 
                          ('__contains__', '/a/c'),
1698
 
                          ('__contains__', '/a/c/'),
1699
 
                          ('__setitem__', '/a/c', {}),
1700
 
                          ('__getitem__', '/a/c'),
1701
 
                          ('__setitem__', 'foo', 'bar'),
1702
 
                          ('__getitem__', '/a/c'),
1703
 
                          ('as_bool', 'recurse'),
1704
 
                          ('__getitem__', '/a/c'),
1705
 
                          ('__delitem__', 'recurse'),
1706
 
                          ('__getitem__', '/a/c'),
1707
 
                          ('keys',),
1708
 
                          ('__getitem__', '/a/c'),
1709
 
                          ('__contains__', 'foo:policy'),
1710
 
                          ('write',)],
1711
 
                         record._calls[1:])
1712
 
 
1713
 
    def test_set_user_setting_sets_and_saves2(self):
1714
 
        self.get_branch_config('/a/c')
1715
 
        self.assertIs(self.my_config.get_user_option('foo'), None)
1716
 
        self.my_config.set_user_option('foo', 'bar')
1717
 
        self.assertEqual(
1718
 
            self.my_config.branch.control_files.files['branch.conf'].strip(),
1719
 
            'foo = bar')
1720
 
        self.assertEqual(self.my_config.get_user_option('foo'), 'bar')
1721
 
        self.my_config.set_user_option('foo', 'baz',
1722
 
                                       store=config.STORE_LOCATION)
1723
 
        self.assertEqual(self.my_config.get_user_option('foo'), 'baz')
1724
 
        self.my_config.set_user_option('foo', 'qux')
1725
 
        self.assertEqual(self.my_config.get_user_option('foo'), 'baz')
1726
 
 
1727
 
    def test_get_bzr_remote_path(self):
1728
 
        my_config = config.LocationConfig('/a/c')
1729
 
        self.assertEqual('bzr', my_config.get_bzr_remote_path())
1730
 
        my_config.set_user_option('bzr_remote_path', '/path-bzr')
1731
 
        self.assertEqual('/path-bzr', my_config.get_bzr_remote_path())
1732
 
        self.overrideEnv('BZR_REMOTE_PATH', '/environ-bzr')
1733
 
        self.assertEqual('/environ-bzr', my_config.get_bzr_remote_path())
1734
 
 
1735
 
 
1736
 
precedence_global = 'option = global'
1737
 
precedence_branch = 'option = branch'
1738
 
precedence_location = """
1739
 
[http://]
1740
 
recurse = true
1741
 
option = recurse
1742
 
[http://example.com/specific]
1743
 
option = exact
1744
 
"""
1745
 
 
1746
 
class TestBranchConfigItems(tests.TestCaseInTempDir):
1747
 
 
1748
 
    def get_branch_config(self, global_config=None, location=None,
1749
 
                          location_config=None, branch_data_config=None):
1750
 
        my_branch = FakeBranch(location)
1751
 
        if global_config is not None:
1752
 
            my_global_config = config.GlobalConfig.from_string(global_config,
1753
 
                                                               save=True)
1754
 
        if location_config is not None:
1755
 
            my_location_config = config.LocationConfig.from_string(
1756
 
                location_config, my_branch.base, save=True)
1757
 
        my_config = config.BranchConfig(my_branch)
1758
 
        if branch_data_config is not None:
1759
 
            my_config.branch.control_files.files['branch.conf'] = \
1760
 
                branch_data_config
1761
 
        return my_config
1762
 
 
1763
 
    def test_user_id(self):
1764
 
        branch = FakeBranch(user_id='Robert Collins <robertc@example.net>')
1765
 
        my_config = config.BranchConfig(branch)
1766
 
        self.assertEqual("Robert Collins <robertc@example.net>",
1767
 
                         my_config.username())
1768
 
        my_config.branch.control_files.files['email'] = "John"
1769
 
        my_config.set_user_option('email',
1770
 
                                  "Robert Collins <robertc@example.org>")
1771
 
        self.assertEqual("John", my_config.username())
1772
 
        del my_config.branch.control_files.files['email']
1773
 
        self.assertEqual("Robert Collins <robertc@example.org>",
1774
 
                         my_config.username())
1775
 
 
1776
 
    def test_not_set_in_branch(self):
1777
 
        my_config = self.get_branch_config(global_config=sample_config_text)
1778
 
        self.assertEqual(u"Erik B\u00e5gfors <erik@bagfors.nu>",
1779
 
                         my_config._get_user_id())
1780
 
        my_config.branch.control_files.files['email'] = "John"
1781
 
        self.assertEqual("John", my_config._get_user_id())
1782
 
 
1783
 
    def test_BZR_EMAIL_OVERRIDES(self):
1784
 
        self.overrideEnv('BZR_EMAIL', "Robert Collins <robertc@example.org>")
1785
 
        branch = FakeBranch()
1786
 
        my_config = config.BranchConfig(branch)
1787
 
        self.assertEqual("Robert Collins <robertc@example.org>",
1788
 
                         my_config.username())
1789
 
 
1790
 
    def test_signatures_forced(self):
1791
 
        my_config = self.get_branch_config(
1792
 
            global_config=sample_always_signatures)
1793
 
        self.assertEqual(config.CHECK_NEVER, my_config.signature_checking())
1794
 
        self.assertEqual(config.SIGN_ALWAYS, my_config.signing_policy())
1795
 
        self.assertTrue(my_config.signature_needed())
1796
 
 
1797
 
    def test_signatures_forced_branch(self):
1798
 
        my_config = self.get_branch_config(
1799
 
            global_config=sample_ignore_signatures,
1800
 
            branch_data_config=sample_always_signatures)
1801
 
        self.assertEqual(config.CHECK_NEVER, my_config.signature_checking())
1802
 
        self.assertEqual(config.SIGN_ALWAYS, my_config.signing_policy())
1803
 
        self.assertTrue(my_config.signature_needed())
1804
 
 
1805
 
    def test_gpg_signing_command(self):
1806
 
        my_config = self.get_branch_config(
1807
 
            global_config=sample_config_text,
1808
 
            # branch data cannot set gpg_signing_command
1809
 
            branch_data_config="gpg_signing_command=pgp")
1810
 
        self.assertEqual('gnome-gpg', my_config.gpg_signing_command())
1811
 
 
1812
 
    def test_get_user_option_global(self):
1813
 
        my_config = self.get_branch_config(global_config=sample_config_text)
1814
 
        self.assertEqual('something',
1815
 
                         my_config.get_user_option('user_global_option'))
1816
 
 
1817
 
    def test_post_commit_default(self):
1818
 
        my_config = self.get_branch_config(global_config=sample_config_text,
1819
 
                                      location='/a/c',
1820
 
                                      location_config=sample_branches_text)
1821
 
        self.assertEqual(my_config.branch.base, '/a/c')
1822
 
        self.assertEqual('bzrlib.tests.test_config.post_commit',
1823
 
                         my_config.post_commit())
1824
 
        my_config.set_user_option('post_commit', 'rmtree_root')
1825
 
        # post-commit is ignored when present in branch data
1826
 
        self.assertEqual('bzrlib.tests.test_config.post_commit',
1827
 
                         my_config.post_commit())
1828
 
        my_config.set_user_option('post_commit', 'rmtree_root',
1829
 
                                  store=config.STORE_LOCATION)
1830
 
        self.assertEqual('rmtree_root', my_config.post_commit())
1831
 
 
1832
 
    def test_config_precedence(self):
1833
 
        # FIXME: eager test, luckily no persitent config file makes it fail
1834
 
        # -- vila 20100716
1835
 
        my_config = self.get_branch_config(global_config=precedence_global)
1836
 
        self.assertEqual(my_config.get_user_option('option'), 'global')
1837
 
        my_config = self.get_branch_config(global_config=precedence_global,
1838
 
                                           branch_data_config=precedence_branch)
1839
 
        self.assertEqual(my_config.get_user_option('option'), 'branch')
1840
 
        my_config = self.get_branch_config(
1841
 
            global_config=precedence_global,
1842
 
            branch_data_config=precedence_branch,
1843
 
            location_config=precedence_location)
1844
 
        self.assertEqual(my_config.get_user_option('option'), 'recurse')
1845
 
        my_config = self.get_branch_config(
1846
 
            global_config=precedence_global,
1847
 
            branch_data_config=precedence_branch,
1848
 
            location_config=precedence_location,
1849
 
            location='http://example.com/specific')
1850
 
        self.assertEqual(my_config.get_user_option('option'), 'exact')
1851
 
 
1852
 
    def test_get_mail_client(self):
1853
 
        config = self.get_branch_config()
1854
 
        client = config.get_mail_client()
1855
 
        self.assertIsInstance(client, mail_client.DefaultMail)
1856
 
 
1857
 
        # Specific clients
1858
 
        config.set_user_option('mail_client', 'evolution')
1859
 
        client = config.get_mail_client()
1860
 
        self.assertIsInstance(client, mail_client.Evolution)
1861
 
 
1862
 
        config.set_user_option('mail_client', 'kmail')
1863
 
        client = config.get_mail_client()
1864
 
        self.assertIsInstance(client, mail_client.KMail)
1865
 
 
1866
 
        config.set_user_option('mail_client', 'mutt')
1867
 
        client = config.get_mail_client()
1868
 
        self.assertIsInstance(client, mail_client.Mutt)
1869
 
 
1870
 
        config.set_user_option('mail_client', 'thunderbird')
1871
 
        client = config.get_mail_client()
1872
 
        self.assertIsInstance(client, mail_client.Thunderbird)
1873
 
 
1874
 
        # Generic options
1875
 
        config.set_user_option('mail_client', 'default')
1876
 
        client = config.get_mail_client()
1877
 
        self.assertIsInstance(client, mail_client.DefaultMail)
1878
 
 
1879
 
        config.set_user_option('mail_client', 'editor')
1880
 
        client = config.get_mail_client()
1881
 
        self.assertIsInstance(client, mail_client.Editor)
1882
 
 
1883
 
        config.set_user_option('mail_client', 'mapi')
1884
 
        client = config.get_mail_client()
1885
 
        self.assertIsInstance(client, mail_client.MAPIClient)
1886
 
 
1887
 
        config.set_user_option('mail_client', 'xdg-email')
1888
 
        client = config.get_mail_client()
1889
 
        self.assertIsInstance(client, mail_client.XDGEmail)
1890
 
 
1891
 
        config.set_user_option('mail_client', 'firebird')
1892
 
        self.assertRaises(errors.UnknownMailClient, config.get_mail_client)
1893
 
 
1894
 
 
1895
 
class TestMailAddressExtraction(tests.TestCase):
1896
 
 
1897
 
    def test_extract_email_address(self):
1898
 
        self.assertEqual('jane@test.com',
1899
 
                         config.extract_email_address('Jane <jane@test.com>'))
1900
 
        self.assertRaises(errors.NoEmailInUsername,
1901
 
                          config.extract_email_address, 'Jane Tester')
1902
 
 
1903
 
    def test_parse_username(self):
1904
 
        self.assertEqual(('', 'jdoe@example.com'),
1905
 
                         config.parse_username('jdoe@example.com'))
1906
 
        self.assertEqual(('', 'jdoe@example.com'),
1907
 
                         config.parse_username('<jdoe@example.com>'))
1908
 
        self.assertEqual(('John Doe', 'jdoe@example.com'),
1909
 
                         config.parse_username('John Doe <jdoe@example.com>'))
1910
 
        self.assertEqual(('John Doe', ''),
1911
 
                         config.parse_username('John Doe'))
1912
 
        self.assertEqual(('John Doe', 'jdoe@example.com'),
1913
 
                         config.parse_username('John Doe jdoe@example.com'))
1914
 
 
1915
 
class TestTreeConfig(tests.TestCaseWithTransport):
1916
 
 
1917
 
    def test_get_value(self):
1918
 
        """Test that retreiving a value from a section is possible"""
1919
 
        branch = self.make_branch('.')
1920
 
        tree_config = config.TreeConfig(branch)
1921
 
        tree_config.set_option('value', 'key', 'SECTION')
1922
 
        tree_config.set_option('value2', 'key2')
1923
 
        tree_config.set_option('value3-top', 'key3')
1924
 
        tree_config.set_option('value3-section', 'key3', 'SECTION')
1925
 
        value = tree_config.get_option('key', 'SECTION')
1926
 
        self.assertEqual(value, 'value')
1927
 
        value = tree_config.get_option('key2')
1928
 
        self.assertEqual(value, 'value2')
1929
 
        self.assertEqual(tree_config.get_option('non-existant'), None)
1930
 
        value = tree_config.get_option('non-existant', 'SECTION')
1931
 
        self.assertEqual(value, None)
1932
 
        value = tree_config.get_option('non-existant', default='default')
1933
 
        self.assertEqual(value, 'default')
1934
 
        self.assertEqual(tree_config.get_option('key2', 'NOSECTION'), None)
1935
 
        value = tree_config.get_option('key2', 'NOSECTION', default='default')
1936
 
        self.assertEqual(value, 'default')
1937
 
        value = tree_config.get_option('key3')
1938
 
        self.assertEqual(value, 'value3-top')
1939
 
        value = tree_config.get_option('key3', 'SECTION')
1940
 
        self.assertEqual(value, 'value3-section')
1941
 
 
1942
 
 
1943
 
class TestTransportConfig(tests.TestCaseWithTransport):
1944
 
 
1945
 
    def test_load_utf8(self):
1946
 
        """Ensure we can load an utf8-encoded file."""
1947
 
        t = self.get_transport()
1948
 
        unicode_user = u'b\N{Euro Sign}ar'
1949
 
        unicode_content = u'user=%s' % (unicode_user,)
1950
 
        utf8_content = unicode_content.encode('utf8')
1951
 
        # Store the raw content in the config file
1952
 
        t.put_bytes('foo.conf', utf8_content)
1953
 
        conf = config.TransportConfig(t, 'foo.conf')
1954
 
        self.assertEquals(unicode_user, conf.get_option('user'))
1955
 
 
1956
 
    def test_load_non_ascii(self):
1957
 
        """Ensure we display a proper error on non-ascii, non utf-8 content."""
1958
 
        t = self.get_transport()
1959
 
        t.put_bytes('foo.conf', 'user=foo\n#\xff\n')
1960
 
        conf = config.TransportConfig(t, 'foo.conf')
1961
 
        self.assertRaises(errors.ConfigContentError, conf._get_configobj)
1962
 
 
1963
 
    def test_load_erroneous_content(self):
1964
 
        """Ensure we display a proper error on content that can't be parsed."""
1965
 
        t = self.get_transport()
1966
 
        t.put_bytes('foo.conf', '[open_section\n')
1967
 
        conf = config.TransportConfig(t, 'foo.conf')
1968
 
        self.assertRaises(errors.ParseConfigError, conf._get_configobj)
1969
 
 
1970
 
    def test_load_permission_denied(self):
1971
 
        """Ensure we get an empty config file if the file is inaccessible."""
1972
 
        warnings = []
1973
 
        def warning(*args):
1974
 
            warnings.append(args[0] % args[1:])
1975
 
        self.overrideAttr(trace, 'warning', warning)
1976
 
 
1977
 
        class DenyingTransport(object):
1978
 
 
1979
 
            def __init__(self, base):
1980
 
                self.base = base
1981
 
 
1982
 
            def get_bytes(self, relpath):
1983
 
                raise errors.PermissionDenied(relpath, "")
1984
 
 
1985
 
        cfg = config.TransportConfig(
1986
 
            DenyingTransport("nonexisting://"), 'control.conf')
1987
 
        self.assertIs(None, cfg.get_option('non-existant', 'SECTION'))
1988
 
        self.assertEquals(
1989
 
            warnings,
1990
 
            [u'Permission denied while trying to open configuration file '
1991
 
             u'nonexisting:///control.conf.'])
1992
 
 
1993
 
    def test_get_value(self):
1994
 
        """Test that retreiving a value from a section is possible"""
1995
 
        bzrdir_config = config.TransportConfig(self.get_transport('.'),
1996
 
                                               'control.conf')
1997
 
        bzrdir_config.set_option('value', 'key', 'SECTION')
1998
 
        bzrdir_config.set_option('value2', 'key2')
1999
 
        bzrdir_config.set_option('value3-top', 'key3')
2000
 
        bzrdir_config.set_option('value3-section', 'key3', 'SECTION')
2001
 
        value = bzrdir_config.get_option('key', 'SECTION')
2002
 
        self.assertEqual(value, 'value')
2003
 
        value = bzrdir_config.get_option('key2')
2004
 
        self.assertEqual(value, 'value2')
2005
 
        self.assertEqual(bzrdir_config.get_option('non-existant'), None)
2006
 
        value = bzrdir_config.get_option('non-existant', 'SECTION')
2007
 
        self.assertEqual(value, None)
2008
 
        value = bzrdir_config.get_option('non-existant', default='default')
2009
 
        self.assertEqual(value, 'default')
2010
 
        self.assertEqual(bzrdir_config.get_option('key2', 'NOSECTION'), None)
2011
 
        value = bzrdir_config.get_option('key2', 'NOSECTION',
2012
 
                                         default='default')
2013
 
        self.assertEqual(value, 'default')
2014
 
        value = bzrdir_config.get_option('key3')
2015
 
        self.assertEqual(value, 'value3-top')
2016
 
        value = bzrdir_config.get_option('key3', 'SECTION')
2017
 
        self.assertEqual(value, 'value3-section')
2018
 
 
2019
 
    def test_set_unset_default_stack_on(self):
2020
 
        my_dir = self.make_bzrdir('.')
2021
 
        bzrdir_config = config.BzrDirConfig(my_dir)
2022
 
        self.assertIs(None, bzrdir_config.get_default_stack_on())
2023
 
        bzrdir_config.set_default_stack_on('Foo')
2024
 
        self.assertEqual('Foo', bzrdir_config._config.get_option(
2025
 
                         'default_stack_on'))
2026
 
        self.assertEqual('Foo', bzrdir_config.get_default_stack_on())
2027
 
        bzrdir_config.set_default_stack_on(None)
2028
 
        self.assertIs(None, bzrdir_config.get_default_stack_on())
2029
 
 
2030
 
 
2031
 
class TestOldConfigHooks(tests.TestCaseWithTransport):
2032
 
 
2033
 
    def setUp(self):
2034
 
        super(TestOldConfigHooks, self).setUp()
2035
 
        create_configs_with_file_option(self)
2036
 
 
2037
 
    def assertGetHook(self, conf, name, value):
2038
 
        calls = []
2039
 
        def hook(*args):
2040
 
            calls.append(args)
2041
 
        config.OldConfigHooks.install_named_hook('get', hook, None)
2042
 
        self.addCleanup(
2043
 
            config.OldConfigHooks.uninstall_named_hook, 'get', None)
2044
 
        self.assertLength(0, calls)
2045
 
        actual_value = conf.get_user_option(name)
2046
 
        self.assertEquals(value, actual_value)
2047
 
        self.assertLength(1, calls)
2048
 
        self.assertEquals((conf, name, value), calls[0])
2049
 
 
2050
 
    def test_get_hook_bazaar(self):
2051
 
        self.assertGetHook(self.bazaar_config, 'file', 'bazaar')
2052
 
 
2053
 
    def test_get_hook_locations(self):
2054
 
        self.assertGetHook(self.locations_config, 'file', 'locations')
2055
 
 
2056
 
    def test_get_hook_branch(self):
2057
 
        # Since locations masks branch, we define a different option
2058
 
        self.branch_config.set_user_option('file2', 'branch')
2059
 
        self.assertGetHook(self.branch_config, 'file2', 'branch')
2060
 
 
2061
 
    def assertSetHook(self, conf, name, value):
2062
 
        calls = []
2063
 
        def hook(*args):
2064
 
            calls.append(args)
2065
 
        config.OldConfigHooks.install_named_hook('set', hook, None)
2066
 
        self.addCleanup(
2067
 
            config.OldConfigHooks.uninstall_named_hook, 'set', None)
2068
 
        self.assertLength(0, calls)
2069
 
        conf.set_user_option(name, value)
2070
 
        self.assertLength(1, calls)
2071
 
        # We can't assert the conf object below as different configs use
2072
 
        # different means to implement set_user_option and we care only about
2073
 
        # coverage here.
2074
 
        self.assertEquals((name, value), calls[0][1:])
2075
 
 
2076
 
    def test_set_hook_bazaar(self):
2077
 
        self.assertSetHook(self.bazaar_config, 'foo', 'bazaar')
2078
 
 
2079
 
    def test_set_hook_locations(self):
2080
 
        self.assertSetHook(self.locations_config, 'foo', 'locations')
2081
 
 
2082
 
    def test_set_hook_branch(self):
2083
 
        self.assertSetHook(self.branch_config, 'foo', 'branch')
2084
 
 
2085
 
    def assertRemoveHook(self, conf, name, section_name=None):
2086
 
        calls = []
2087
 
        def hook(*args):
2088
 
            calls.append(args)
2089
 
        config.OldConfigHooks.install_named_hook('remove', hook, None)
2090
 
        self.addCleanup(
2091
 
            config.OldConfigHooks.uninstall_named_hook, 'remove', None)
2092
 
        self.assertLength(0, calls)
2093
 
        conf.remove_user_option(name, section_name)
2094
 
        self.assertLength(1, calls)
2095
 
        # We can't assert the conf object below as different configs use
2096
 
        # different means to implement remove_user_option and we care only about
2097
 
        # coverage here.
2098
 
        self.assertEquals((name,), calls[0][1:])
2099
 
 
2100
 
    def test_remove_hook_bazaar(self):
2101
 
        self.assertRemoveHook(self.bazaar_config, 'file')
2102
 
 
2103
 
    def test_remove_hook_locations(self):
2104
 
        self.assertRemoveHook(self.locations_config, 'file',
2105
 
                              self.locations_config.location)
2106
 
 
2107
 
    def test_remove_hook_branch(self):
2108
 
        self.assertRemoveHook(self.branch_config, 'file')
2109
 
 
2110
 
    def assertLoadHook(self, name, conf_class, *conf_args):
2111
 
        calls = []
2112
 
        def hook(*args):
2113
 
            calls.append(args)
2114
 
        config.OldConfigHooks.install_named_hook('load', hook, None)
2115
 
        self.addCleanup(
2116
 
            config.OldConfigHooks.uninstall_named_hook, 'load', None)
2117
 
        self.assertLength(0, calls)
2118
 
        # Build a config
2119
 
        conf = conf_class(*conf_args)
2120
 
        # Access an option to trigger a load
2121
 
        conf.get_user_option(name)
2122
 
        self.assertLength(1, calls)
2123
 
        # Since we can't assert about conf, we just use the number of calls ;-/
2124
 
 
2125
 
    def test_load_hook_bazaar(self):
2126
 
        self.assertLoadHook('file', config.GlobalConfig)
2127
 
 
2128
 
    def test_load_hook_locations(self):
2129
 
        self.assertLoadHook('file', config.LocationConfig, self.tree.basedir)
2130
 
 
2131
 
    def test_load_hook_branch(self):
2132
 
        self.assertLoadHook('file', config.BranchConfig, self.tree.branch)
2133
 
 
2134
 
    def assertSaveHook(self, conf):
2135
 
        calls = []
2136
 
        def hook(*args):
2137
 
            calls.append(args)
2138
 
        config.OldConfigHooks.install_named_hook('save', hook, None)
2139
 
        self.addCleanup(
2140
 
            config.OldConfigHooks.uninstall_named_hook, 'save', None)
2141
 
        self.assertLength(0, calls)
2142
 
        # Setting an option triggers a save
2143
 
        conf.set_user_option('foo', 'bar')
2144
 
        self.assertLength(1, calls)
2145
 
        # Since we can't assert about conf, we just use the number of calls ;-/
2146
 
 
2147
 
    def test_save_hook_bazaar(self):
2148
 
        self.assertSaveHook(self.bazaar_config)
2149
 
 
2150
 
    def test_save_hook_locations(self):
2151
 
        self.assertSaveHook(self.locations_config)
2152
 
 
2153
 
    def test_save_hook_branch(self):
2154
 
        self.assertSaveHook(self.branch_config)
2155
 
 
2156
 
 
2157
 
class TestOldConfigHooksForRemote(tests.TestCaseWithTransport):
2158
 
    """Tests config hooks for remote configs.
2159
 
 
2160
 
    No tests for the remove hook as this is not implemented there.
2161
 
    """
2162
 
 
2163
 
    def setUp(self):
2164
 
        super(TestOldConfigHooksForRemote, self).setUp()
2165
 
        self.transport_server = test_server.SmartTCPServer_for_testing
2166
 
        create_configs_with_file_option(self)
2167
 
 
2168
 
    def assertGetHook(self, conf, name, value):
2169
 
        calls = []
2170
 
        def hook(*args):
2171
 
            calls.append(args)
2172
 
        config.OldConfigHooks.install_named_hook('get', hook, None)
2173
 
        self.addCleanup(
2174
 
            config.OldConfigHooks.uninstall_named_hook, 'get', None)
2175
 
        self.assertLength(0, calls)
2176
 
        actual_value = conf.get_option(name)
2177
 
        self.assertEquals(value, actual_value)
2178
 
        self.assertLength(1, calls)
2179
 
        self.assertEquals((conf, name, value), calls[0])
2180
 
 
2181
 
    def test_get_hook_remote_branch(self):
2182
 
        remote_branch = branch.Branch.open(self.get_url('tree'))
2183
 
        self.assertGetHook(remote_branch._get_config(), 'file', 'branch')
2184
 
 
2185
 
    def test_get_hook_remote_bzrdir(self):
2186
 
        remote_bzrdir = bzrdir.BzrDir.open(self.get_url('tree'))
2187
 
        conf = remote_bzrdir._get_config()
2188
 
        conf.set_option('remotedir', 'file')
2189
 
        self.assertGetHook(conf, 'file', 'remotedir')
2190
 
 
2191
 
    def assertSetHook(self, conf, name, value):
2192
 
        calls = []
2193
 
        def hook(*args):
2194
 
            calls.append(args)
2195
 
        config.OldConfigHooks.install_named_hook('set', hook, None)
2196
 
        self.addCleanup(
2197
 
            config.OldConfigHooks.uninstall_named_hook, 'set', None)
2198
 
        self.assertLength(0, calls)
2199
 
        conf.set_option(value, name)
2200
 
        self.assertLength(1, calls)
2201
 
        # We can't assert the conf object below as different configs use
2202
 
        # different means to implement set_user_option and we care only about
2203
 
        # coverage here.
2204
 
        self.assertEquals((name, value), calls[0][1:])
2205
 
 
2206
 
    def test_set_hook_remote_branch(self):
2207
 
        remote_branch = branch.Branch.open(self.get_url('tree'))
2208
 
        self.addCleanup(remote_branch.lock_write().unlock)
2209
 
        self.assertSetHook(remote_branch._get_config(), 'file', 'remote')
2210
 
 
2211
 
    def test_set_hook_remote_bzrdir(self):
2212
 
        remote_branch = branch.Branch.open(self.get_url('tree'))
2213
 
        self.addCleanup(remote_branch.lock_write().unlock)
2214
 
        remote_bzrdir = bzrdir.BzrDir.open(self.get_url('tree'))
2215
 
        self.assertSetHook(remote_bzrdir._get_config(), 'file', 'remotedir')
2216
 
 
2217
 
    def assertLoadHook(self, expected_nb_calls, name, conf_class, *conf_args):
2218
 
        calls = []
2219
 
        def hook(*args):
2220
 
            calls.append(args)
2221
 
        config.OldConfigHooks.install_named_hook('load', hook, None)
2222
 
        self.addCleanup(
2223
 
            config.OldConfigHooks.uninstall_named_hook, 'load', None)
2224
 
        self.assertLength(0, calls)
2225
 
        # Build a config
2226
 
        conf = conf_class(*conf_args)
2227
 
        # Access an option to trigger a load
2228
 
        conf.get_option(name)
2229
 
        self.assertLength(expected_nb_calls, calls)
2230
 
        # Since we can't assert about conf, we just use the number of calls ;-/
2231
 
 
2232
 
    def test_load_hook_remote_branch(self):
2233
 
        remote_branch = branch.Branch.open(self.get_url('tree'))
2234
 
        self.assertLoadHook(1, 'file', remote.RemoteBranchConfig, remote_branch)
2235
 
 
2236
 
    def test_load_hook_remote_bzrdir(self):
2237
 
        remote_bzrdir = bzrdir.BzrDir.open(self.get_url('tree'))
2238
 
        # The config file doesn't exist, set an option to force its creation
2239
 
        conf = remote_bzrdir._get_config()
2240
 
        conf.set_option('remotedir', 'file')
2241
 
        # We get one call for the server and one call for the client, this is
2242
 
        # caused by the differences in implementations betwen
2243
 
        # SmartServerBzrDirRequestConfigFile (in smart/bzrdir.py) and
2244
 
        # SmartServerBranchGetConfigFile (in smart/branch.py)
2245
 
        self.assertLoadHook(2 ,'file', remote.RemoteBzrDirConfig, remote_bzrdir)
2246
 
 
2247
 
    def assertSaveHook(self, conf):
2248
 
        calls = []
2249
 
        def hook(*args):
2250
 
            calls.append(args)
2251
 
        config.OldConfigHooks.install_named_hook('save', hook, None)
2252
 
        self.addCleanup(
2253
 
            config.OldConfigHooks.uninstall_named_hook, 'save', None)
2254
 
        self.assertLength(0, calls)
2255
 
        # Setting an option triggers a save
2256
 
        conf.set_option('foo', 'bar')
2257
 
        self.assertLength(1, calls)
2258
 
        # Since we can't assert about conf, we just use the number of calls ;-/
2259
 
 
2260
 
    def test_save_hook_remote_branch(self):
2261
 
        remote_branch = branch.Branch.open(self.get_url('tree'))
2262
 
        self.addCleanup(remote_branch.lock_write().unlock)
2263
 
        self.assertSaveHook(remote_branch._get_config())
2264
 
 
2265
 
    def test_save_hook_remote_bzrdir(self):
2266
 
        remote_branch = branch.Branch.open(self.get_url('tree'))
2267
 
        self.addCleanup(remote_branch.lock_write().unlock)
2268
 
        remote_bzrdir = bzrdir.BzrDir.open(self.get_url('tree'))
2269
 
        self.assertSaveHook(remote_bzrdir._get_config())
2270
 
 
2271
 
 
2272
 
class TestOption(tests.TestCase):
2273
 
 
2274
 
    def test_default_value(self):
2275
 
        opt = config.Option('foo', default='bar')
2276
 
        self.assertEquals('bar', opt.get_default())
2277
 
 
2278
 
    def test_default_value_from_env(self):
2279
 
        opt = config.Option('foo', default='bar', default_from_env=['FOO'])
2280
 
        self.overrideEnv('FOO', 'quux')
2281
 
        # Env variable provides a default taking over the option one
2282
 
        self.assertEquals('quux', opt.get_default())
2283
 
 
2284
 
    def test_first_default_value_from_env_wins(self):
2285
 
        opt = config.Option('foo', default='bar',
2286
 
                            default_from_env=['NO_VALUE', 'FOO', 'BAZ'])
2287
 
        self.overrideEnv('FOO', 'foo')
2288
 
        self.overrideEnv('BAZ', 'baz')
2289
 
        # The first env var set wins
2290
 
        self.assertEquals('foo', opt.get_default())
2291
 
 
2292
 
    def test_not_supported_list_default_value(self):
2293
 
        self.assertRaises(AssertionError, config.Option, 'foo', default=[1])
2294
 
 
2295
 
    def test_not_supported_object_default_value(self):
2296
 
        self.assertRaises(AssertionError, config.Option, 'foo',
2297
 
                          default=object())
2298
 
 
2299
 
 
2300
 
class TestOptionConverterMixin(object):
2301
 
 
2302
 
    def assertConverted(self, expected, opt, value):
2303
 
        self.assertEquals(expected, opt.convert_from_unicode(value))
2304
 
 
2305
 
    def assertWarns(self, opt, value):
2306
 
        warnings = []
2307
 
        def warning(*args):
2308
 
            warnings.append(args[0] % args[1:])
2309
 
        self.overrideAttr(trace, 'warning', warning)
2310
 
        self.assertEquals(None, opt.convert_from_unicode(value))
2311
 
        self.assertLength(1, warnings)
2312
 
        self.assertEquals(
2313
 
            'Value "%s" is not valid for "%s"' % (value, opt.name),
2314
 
            warnings[0])
2315
 
 
2316
 
    def assertErrors(self, opt, value):
2317
 
        self.assertRaises(errors.ConfigOptionValueError,
2318
 
                          opt.convert_from_unicode, value)
2319
 
 
2320
 
    def assertConvertInvalid(self, opt, invalid_value):
2321
 
        opt.invalid = None
2322
 
        self.assertEquals(None, opt.convert_from_unicode(invalid_value))
2323
 
        opt.invalid = 'warning'
2324
 
        self.assertWarns(opt, invalid_value)
2325
 
        opt.invalid = 'error'
2326
 
        self.assertErrors(opt, invalid_value)
2327
 
 
2328
 
 
2329
 
class TestOptionWithBooleanConverter(tests.TestCase, TestOptionConverterMixin):
2330
 
 
2331
 
    def get_option(self):
2332
 
        return config.Option('foo', help='A boolean.',
2333
 
                             from_unicode=config.bool_from_store)
2334
 
 
2335
 
    def test_convert_invalid(self):
2336
 
        opt = self.get_option()
2337
 
        # A string that is not recognized as a boolean
2338
 
        self.assertConvertInvalid(opt, u'invalid-boolean')
2339
 
        # A list of strings is never recognized as a boolean
2340
 
        self.assertConvertInvalid(opt, [u'not', u'a', u'boolean'])
2341
 
 
2342
 
    def test_convert_valid(self):
2343
 
        opt = self.get_option()
2344
 
        self.assertConverted(True, opt, u'True')
2345
 
        self.assertConverted(True, opt, u'1')
2346
 
        self.assertConverted(False, opt, u'False')
2347
 
 
2348
 
 
2349
 
class TestOptionWithIntegerConverter(tests.TestCase, TestOptionConverterMixin):
2350
 
 
2351
 
    def get_option(self):
2352
 
        return config.Option('foo', help='An integer.',
2353
 
                             from_unicode=config.int_from_store)
2354
 
 
2355
 
    def test_convert_invalid(self):
2356
 
        opt = self.get_option()
2357
 
        # A string that is not recognized as an integer
2358
 
        self.assertConvertInvalid(opt, u'forty-two')
2359
 
        # A list of strings is never recognized as an integer
2360
 
        self.assertConvertInvalid(opt, [u'a', u'list'])
2361
 
 
2362
 
    def test_convert_valid(self):
2363
 
        opt = self.get_option()
2364
 
        self.assertConverted(16, opt, u'16')
2365
 
 
2366
 
class TestOptionWithListConverter(tests.TestCase, TestOptionConverterMixin):
2367
 
 
2368
 
    def get_option(self):
2369
 
        return config.Option('foo', help='A list.',
2370
 
                             from_unicode=config.list_from_store)
2371
 
 
2372
 
    def test_convert_invalid(self):
2373
 
        # No string is invalid as all forms can be converted to a list
2374
 
        pass
2375
 
 
2376
 
    def test_convert_valid(self):
2377
 
        opt = self.get_option()
2378
 
        # An empty string is an empty list
2379
 
        self.assertConverted([], opt, '') # Using a bare str() just in case
2380
 
        self.assertConverted([], opt, u'')
2381
 
        # A boolean
2382
 
        self.assertConverted([u'True'], opt, u'True')
2383
 
        # An integer
2384
 
        self.assertConverted([u'42'], opt, u'42')
2385
 
        # A single string
2386
 
        self.assertConverted([u'bar'], opt, u'bar')
2387
 
        # A list remains a list (configObj will turn a string containing commas
2388
 
        # into a list, but that's not what we're testing here)
2389
 
        self.assertConverted([u'foo', u'1', u'True'],
2390
 
                             opt, [u'foo', u'1', u'True'])
2391
 
 
2392
 
 
2393
 
class TestOptionConverterMixin(object):
2394
 
 
2395
 
    def assertConverted(self, expected, opt, value):
2396
 
        self.assertEquals(expected, opt.convert_from_unicode(value))
2397
 
 
2398
 
    def assertWarns(self, opt, value):
2399
 
        warnings = []
2400
 
        def warning(*args):
2401
 
            warnings.append(args[0] % args[1:])
2402
 
        self.overrideAttr(trace, 'warning', warning)
2403
 
        self.assertEquals(None, opt.convert_from_unicode(value))
2404
 
        self.assertLength(1, warnings)
2405
 
        self.assertEquals(
2406
 
            'Value "%s" is not valid for "%s"' % (value, opt.name),
2407
 
            warnings[0])
2408
 
 
2409
 
    def assertErrors(self, opt, value):
2410
 
        self.assertRaises(errors.ConfigOptionValueError,
2411
 
                          opt.convert_from_unicode, value)
2412
 
 
2413
 
    def assertConvertInvalid(self, opt, invalid_value):
2414
 
        opt.invalid = None
2415
 
        self.assertEquals(None, opt.convert_from_unicode(invalid_value))
2416
 
        opt.invalid = 'warning'
2417
 
        self.assertWarns(opt, invalid_value)
2418
 
        opt.invalid = 'error'
2419
 
        self.assertErrors(opt, invalid_value)
2420
 
 
2421
 
 
2422
 
class TestOptionWithBooleanConverter(tests.TestCase, TestOptionConverterMixin):
2423
 
 
2424
 
    def get_option(self):
2425
 
        return config.Option('foo', help='A boolean.',
2426
 
                             from_unicode=config.bool_from_store)
2427
 
 
2428
 
    def test_convert_invalid(self):
2429
 
        opt = self.get_option()
2430
 
        # A string that is not recognized as a boolean
2431
 
        self.assertConvertInvalid(opt, u'invalid-boolean')
2432
 
        # A list of strings is never recognized as a boolean
2433
 
        self.assertConvertInvalid(opt, [u'not', u'a', u'boolean'])
2434
 
 
2435
 
    def test_convert_valid(self):
2436
 
        opt = self.get_option()
2437
 
        self.assertConverted(True, opt, u'True')
2438
 
        self.assertConverted(True, opt, u'1')
2439
 
        self.assertConverted(False, opt, u'False')
2440
 
 
2441
 
 
2442
 
class TestOptionWithIntegerConverter(tests.TestCase, TestOptionConverterMixin):
2443
 
 
2444
 
    def get_option(self):
2445
 
        return config.Option('foo', help='An integer.',
2446
 
                             from_unicode=config.int_from_store)
2447
 
 
2448
 
    def test_convert_invalid(self):
2449
 
        opt = self.get_option()
2450
 
        # A string that is not recognized as an integer
2451
 
        self.assertConvertInvalid(opt, u'forty-two')
2452
 
        # A list of strings is never recognized as an integer
2453
 
        self.assertConvertInvalid(opt, [u'a', u'list'])
2454
 
 
2455
 
    def test_convert_valid(self):
2456
 
        opt = self.get_option()
2457
 
        self.assertConverted(16, opt, u'16')
2458
 
 
2459
 
 
2460
 
class TestOptionWithListConverter(tests.TestCase, TestOptionConverterMixin):
2461
 
 
2462
 
    def get_option(self):
2463
 
        return config.Option('foo', help='A list.',
2464
 
                             from_unicode=config.list_from_store)
2465
 
 
2466
 
    def test_convert_invalid(self):
2467
 
        opt = self.get_option()
2468
 
        # We don't even try to convert a list into a list, we only expect
2469
 
        # strings
2470
 
        self.assertConvertInvalid(opt, [1])
2471
 
        # No string is invalid as all forms can be converted to a list
2472
 
 
2473
 
    def test_convert_valid(self):
2474
 
        opt = self.get_option()
2475
 
        # An empty string is an empty list
2476
 
        self.assertConverted([], opt, '') # Using a bare str() just in case
2477
 
        self.assertConverted([], opt, u'')
2478
 
        # A boolean
2479
 
        self.assertConverted([u'True'], opt, u'True')
2480
 
        # An integer
2481
 
        self.assertConverted([u'42'], opt, u'42')
2482
 
        # A single string
2483
 
        self.assertConverted([u'bar'], opt, u'bar')
2484
 
 
2485
 
 
2486
 
class TestOptionRegistry(tests.TestCase):
2487
 
 
2488
 
    def setUp(self):
2489
 
        super(TestOptionRegistry, self).setUp()
2490
 
        # Always start with an empty registry
2491
 
        self.overrideAttr(config, 'option_registry', config.OptionRegistry())
2492
 
        self.registry = config.option_registry
2493
 
 
2494
 
    def test_register(self):
2495
 
        opt = config.Option('foo')
2496
 
        self.registry.register(opt)
2497
 
        self.assertIs(opt, self.registry.get('foo'))
2498
 
 
2499
 
    def test_registered_help(self):
2500
 
        opt = config.Option('foo', help='A simple option')
2501
 
        self.registry.register(opt)
2502
 
        self.assertEquals('A simple option', self.registry.get_help('foo'))
2503
 
 
2504
 
    lazy_option = config.Option('lazy_foo', help='Lazy help')
2505
 
 
2506
 
    def test_register_lazy(self):
2507
 
        self.registry.register_lazy('lazy_foo', self.__module__,
2508
 
                                    'TestOptionRegistry.lazy_option')
2509
 
        self.assertIs(self.lazy_option, self.registry.get('lazy_foo'))
2510
 
 
2511
 
    def test_registered_lazy_help(self):
2512
 
        self.registry.register_lazy('lazy_foo', self.__module__,
2513
 
                                    'TestOptionRegistry.lazy_option')
2514
 
        self.assertEquals('Lazy help', self.registry.get_help('lazy_foo'))
2515
 
 
2516
 
 
2517
 
class TestRegisteredOptions(tests.TestCase):
2518
 
    """All registered options should verify some constraints."""
2519
 
 
2520
 
    scenarios = [(key, {'option_name': key, 'option': option}) for key, option
2521
 
                 in config.option_registry.iteritems()]
2522
 
 
2523
 
    def setUp(self):
2524
 
        super(TestRegisteredOptions, self).setUp()
2525
 
        self.registry = config.option_registry
2526
 
 
2527
 
    def test_proper_name(self):
2528
 
        # An option should be registered under its own name, this can't be
2529
 
        # checked at registration time for the lazy ones.
2530
 
        self.assertEquals(self.option_name, self.option.name)
2531
 
 
2532
 
    def test_help_is_set(self):
2533
 
        option_help = self.registry.get_help(self.option_name)
2534
 
        self.assertNotEquals(None, option_help)
2535
 
        # Come on, think about the user, he really wants to know what the
2536
 
        # option is about
2537
 
        self.assertIsNot(None, option_help)
2538
 
        self.assertNotEquals('', option_help)
2539
 
 
2540
 
 
2541
 
class TestSection(tests.TestCase):
2542
 
 
2543
 
    # FIXME: Parametrize so that all sections produced by Stores run these
2544
 
    # tests -- vila 2011-04-01
2545
 
 
2546
 
    def test_get_a_value(self):
2547
 
        a_dict = dict(foo='bar')
2548
 
        section = config.Section('myID', a_dict)
2549
 
        self.assertEquals('bar', section.get('foo'))
2550
 
 
2551
 
    def test_get_unknown_option(self):
2552
 
        a_dict = dict()
2553
 
        section = config.Section(None, a_dict)
2554
 
        self.assertEquals('out of thin air',
2555
 
                          section.get('foo', 'out of thin air'))
2556
 
 
2557
 
    def test_options_is_shared(self):
2558
 
        a_dict = dict()
2559
 
        section = config.Section(None, a_dict)
2560
 
        self.assertIs(a_dict, section.options)
2561
 
 
2562
 
 
2563
 
class TestMutableSection(tests.TestCase):
2564
 
 
2565
 
    # FIXME: Parametrize so that all sections (including os.environ and the
2566
 
    # ones produced by Stores) run these tests -- vila 2011-04-01
2567
 
 
2568
 
    def test_set(self):
2569
 
        a_dict = dict(foo='bar')
2570
 
        section = config.MutableSection('myID', a_dict)
2571
 
        section.set('foo', 'new_value')
2572
 
        self.assertEquals('new_value', section.get('foo'))
2573
 
        # The change appears in the shared section
2574
 
        self.assertEquals('new_value', a_dict.get('foo'))
2575
 
        # We keep track of the change
2576
 
        self.assertTrue('foo' in section.orig)
2577
 
        self.assertEquals('bar', section.orig.get('foo'))
2578
 
 
2579
 
    def test_set_preserve_original_once(self):
2580
 
        a_dict = dict(foo='bar')
2581
 
        section = config.MutableSection('myID', a_dict)
2582
 
        section.set('foo', 'first_value')
2583
 
        section.set('foo', 'second_value')
2584
 
        # We keep track of the original value
2585
 
        self.assertTrue('foo' in section.orig)
2586
 
        self.assertEquals('bar', section.orig.get('foo'))
2587
 
 
2588
 
    def test_remove(self):
2589
 
        a_dict = dict(foo='bar')
2590
 
        section = config.MutableSection('myID', a_dict)
2591
 
        section.remove('foo')
2592
 
        # We get None for unknown options via the default value
2593
 
        self.assertEquals(None, section.get('foo'))
2594
 
        # Or we just get the default value
2595
 
        self.assertEquals('unknown', section.get('foo', 'unknown'))
2596
 
        self.assertFalse('foo' in section.options)
2597
 
        # We keep track of the deletion
2598
 
        self.assertTrue('foo' in section.orig)
2599
 
        self.assertEquals('bar', section.orig.get('foo'))
2600
 
 
2601
 
    def test_remove_new_option(self):
2602
 
        a_dict = dict()
2603
 
        section = config.MutableSection('myID', a_dict)
2604
 
        section.set('foo', 'bar')
2605
 
        section.remove('foo')
2606
 
        self.assertFalse('foo' in section.options)
2607
 
        # The option didn't exist initially so it we need to keep track of it
2608
 
        # with a special value
2609
 
        self.assertTrue('foo' in section.orig)
2610
 
        self.assertEquals(config._NewlyCreatedOption, section.orig['foo'])
2611
 
 
2612
 
 
2613
 
class TestStore(tests.TestCaseWithTransport):
2614
 
 
2615
 
    def assertSectionContent(self, expected, section):
2616
 
        """Assert that some options have the proper values in a section."""
2617
 
        expected_name, expected_options = expected
2618
 
        self.assertEquals(expected_name, section.id)
2619
 
        self.assertEquals(
2620
 
            expected_options,
2621
 
            dict([(k, section.get(k)) for k in expected_options.keys()]))
2622
 
 
2623
 
 
2624
 
class TestReadonlyStore(TestStore):
2625
 
 
2626
 
    scenarios = [(key, {'get_store': builder}) for key, builder
2627
 
                 in config.test_store_builder_registry.iteritems()]
2628
 
 
2629
 
    def test_building_delays_load(self):
2630
 
        store = self.get_store(self)
2631
 
        self.assertEquals(False, store.is_loaded())
2632
 
        store._load_from_string('')
2633
 
        self.assertEquals(True, store.is_loaded())
2634
 
 
2635
 
    def test_get_no_sections_for_empty(self):
2636
 
        store = self.get_store(self)
2637
 
        store._load_from_string('')
2638
 
        self.assertEquals([], list(store.get_sections()))
2639
 
 
2640
 
    def test_get_default_section(self):
2641
 
        store = self.get_store(self)
2642
 
        store._load_from_string('foo=bar')
2643
 
        sections = list(store.get_sections())
2644
 
        self.assertLength(1, sections)
2645
 
        self.assertSectionContent((None, {'foo': 'bar'}), sections[0])
2646
 
 
2647
 
    def test_get_named_section(self):
2648
 
        store = self.get_store(self)
2649
 
        store._load_from_string('[baz]\nfoo=bar')
2650
 
        sections = list(store.get_sections())
2651
 
        self.assertLength(1, sections)
2652
 
        self.assertSectionContent(('baz', {'foo': 'bar'}), sections[0])
2653
 
 
2654
 
    def test_load_from_string_fails_for_non_empty_store(self):
2655
 
        store = self.get_store(self)
2656
 
        store._load_from_string('foo=bar')
2657
 
        self.assertRaises(AssertionError, store._load_from_string, 'bar=baz')
2658
 
 
2659
 
 
2660
 
class TestIniFileStoreContent(tests.TestCaseWithTransport):
2661
 
    """Simulate loading a config store with content of various encodings.
2662
 
 
2663
 
    All files produced by bzr are in utf8 content.
2664
 
 
2665
 
    Users may modify them manually and end up with a file that can't be
2666
 
    loaded. We need to issue proper error messages in this case.
2667
 
    """
2668
 
 
2669
 
    invalid_utf8_char = '\xff'
2670
 
 
2671
 
    def test_load_utf8(self):
2672
 
        """Ensure we can load an utf8-encoded file."""
2673
 
        t = self.get_transport()
2674
 
        # From http://pad.lv/799212
2675
 
        unicode_user = u'b\N{Euro Sign}ar'
2676
 
        unicode_content = u'user=%s' % (unicode_user,)
2677
 
        utf8_content = unicode_content.encode('utf8')
2678
 
        # Store the raw content in the config file
2679
 
        t.put_bytes('foo.conf', utf8_content)
2680
 
        store = config.IniFileStore(t, 'foo.conf')
2681
 
        store.load()
2682
 
        stack = config.Stack([store.get_sections], store)
2683
 
        self.assertEquals(unicode_user, stack.get('user'))
2684
 
 
2685
 
    def test_load_non_ascii(self):
2686
 
        """Ensure we display a proper error on non-ascii, non utf-8 content."""
2687
 
        t = self.get_transport()
2688
 
        t.put_bytes('foo.conf', 'user=foo\n#%s\n' % (self.invalid_utf8_char,))
2689
 
        store = config.IniFileStore(t, 'foo.conf')
2690
 
        self.assertRaises(errors.ConfigContentError, store.load)
2691
 
 
2692
 
    def test_load_erroneous_content(self):
2693
 
        """Ensure we display a proper error on content that can't be parsed."""
2694
 
        t = self.get_transport()
2695
 
        t.put_bytes('foo.conf', '[open_section\n')
2696
 
        store = config.IniFileStore(t, 'foo.conf')
2697
 
        self.assertRaises(errors.ParseConfigError, store.load)
2698
 
 
2699
 
    def test_load_permission_denied(self):
2700
 
        """Ensure we get warned when trying to load an inaccessible file."""
2701
 
        warnings = []
2702
 
        def warning(*args):
2703
 
            warnings.append(args[0] % args[1:])
2704
 
        self.overrideAttr(trace, 'warning', warning)
2705
 
 
2706
 
        t = self.get_transport()
2707
 
 
2708
 
        def get_bytes(relpath):
2709
 
            raise errors.PermissionDenied(relpath, "")
2710
 
        t.get_bytes = get_bytes
2711
 
        store = config.IniFileStore(t, 'foo.conf')
2712
 
        self.assertRaises(errors.PermissionDenied, store.load)
2713
 
        self.assertEquals(
2714
 
            warnings,
2715
 
            [u'Permission denied while trying to load configuration store %s.'
2716
 
             % store.external_url()])
2717
 
 
2718
 
 
2719
 
class TestIniConfigContent(tests.TestCaseWithTransport):
2720
 
    """Simulate loading a IniBasedConfig with content of various encodings.
2721
 
 
2722
 
    All files produced by bzr are in utf8 content.
2723
 
 
2724
 
    Users may modify them manually and end up with a file that can't be
2725
 
    loaded. We need to issue proper error messages in this case.
2726
 
    """
2727
 
 
2728
 
    invalid_utf8_char = '\xff'
2729
 
 
2730
 
    def test_load_utf8(self):
2731
 
        """Ensure we can load an utf8-encoded file."""
2732
 
        # From http://pad.lv/799212
2733
 
        unicode_user = u'b\N{Euro Sign}ar'
2734
 
        unicode_content = u'user=%s' % (unicode_user,)
2735
 
        utf8_content = unicode_content.encode('utf8')
2736
 
        # Store the raw content in the config file
2737
 
        with open('foo.conf', 'wb') as f:
2738
 
            f.write(utf8_content)
2739
 
        conf = config.IniBasedConfig(file_name='foo.conf')
2740
 
        self.assertEquals(unicode_user, conf.get_user_option('user'))
2741
 
 
2742
 
    def test_load_badly_encoded_content(self):
2743
 
        """Ensure we display a proper error on non-ascii, non utf-8 content."""
2744
 
        with open('foo.conf', 'wb') as f:
2745
 
            f.write('user=foo\n#%s\n' % (self.invalid_utf8_char,))
2746
 
        conf = config.IniBasedConfig(file_name='foo.conf')
2747
 
        self.assertRaises(errors.ConfigContentError, conf._get_parser)
2748
 
 
2749
 
    def test_load_erroneous_content(self):
2750
 
        """Ensure we display a proper error on content that can't be parsed."""
2751
 
        with open('foo.conf', 'wb') as f:
2752
 
            f.write('[open_section\n')
2753
 
        conf = config.IniBasedConfig(file_name='foo.conf')
2754
 
        self.assertRaises(errors.ParseConfigError, conf._get_parser)
2755
 
 
2756
 
 
2757
 
class TestMutableStore(TestStore):
2758
 
 
2759
 
    scenarios = [(key, {'store_id': key, 'get_store': builder}) for key, builder
2760
 
                 in config.test_store_builder_registry.iteritems()]
2761
 
 
2762
 
    def setUp(self):
2763
 
        super(TestMutableStore, self).setUp()
2764
 
        self.transport = self.get_transport()
2765
 
 
2766
 
    def has_store(self, store):
2767
 
        store_basename = urlutils.relative_url(self.transport.external_url(),
2768
 
                                               store.external_url())
2769
 
        return self.transport.has(store_basename)
2770
 
 
2771
 
    def test_save_empty_creates_no_file(self):
2772
 
        # FIXME: There should be a better way than relying on the test
2773
 
        # parametrization to identify branch.conf -- vila 2011-0526
2774
 
        if self.store_id in ('branch', 'remote_branch'):
2775
 
            raise tests.TestNotApplicable(
2776
 
                'branch.conf is *always* created when a branch is initialized')
2777
 
        store = self.get_store(self)
2778
 
        store.save()
2779
 
        self.assertEquals(False, self.has_store(store))
2780
 
 
2781
 
    def test_save_emptied_succeeds(self):
2782
 
        store = self.get_store(self)
2783
 
        store._load_from_string('foo=bar\n')
2784
 
        section = store.get_mutable_section(None)
2785
 
        section.remove('foo')
2786
 
        store.save()
2787
 
        self.assertEquals(True, self.has_store(store))
2788
 
        modified_store = self.get_store(self)
2789
 
        sections = list(modified_store.get_sections())
2790
 
        self.assertLength(0, sections)
2791
 
 
2792
 
    def test_save_with_content_succeeds(self):
2793
 
        # FIXME: There should be a better way than relying on the test
2794
 
        # parametrization to identify branch.conf -- vila 2011-0526
2795
 
        if self.store_id in ('branch', 'remote_branch'):
2796
 
            raise tests.TestNotApplicable(
2797
 
                'branch.conf is *always* created when a branch is initialized')
2798
 
        store = self.get_store(self)
2799
 
        store._load_from_string('foo=bar\n')
2800
 
        self.assertEquals(False, self.has_store(store))
2801
 
        store.save()
2802
 
        self.assertEquals(True, self.has_store(store))
2803
 
        modified_store = self.get_store(self)
2804
 
        sections = list(modified_store.get_sections())
2805
 
        self.assertLength(1, sections)
2806
 
        self.assertSectionContent((None, {'foo': 'bar'}), sections[0])
2807
 
 
2808
 
    def test_set_option_in_empty_store(self):
2809
 
        store = self.get_store(self)
2810
 
        section = store.get_mutable_section(None)
2811
 
        section.set('foo', 'bar')
2812
 
        store.save()
2813
 
        modified_store = self.get_store(self)
2814
 
        sections = list(modified_store.get_sections())
2815
 
        self.assertLength(1, sections)
2816
 
        self.assertSectionContent((None, {'foo': 'bar'}), sections[0])
2817
 
 
2818
 
    def test_set_option_in_default_section(self):
2819
 
        store = self.get_store(self)
2820
 
        store._load_from_string('')
2821
 
        section = store.get_mutable_section(None)
2822
 
        section.set('foo', 'bar')
2823
 
        store.save()
2824
 
        modified_store = self.get_store(self)
2825
 
        sections = list(modified_store.get_sections())
2826
 
        self.assertLength(1, sections)
2827
 
        self.assertSectionContent((None, {'foo': 'bar'}), sections[0])
2828
 
 
2829
 
    def test_set_option_in_named_section(self):
2830
 
        store = self.get_store(self)
2831
 
        store._load_from_string('')
2832
 
        section = store.get_mutable_section('baz')
2833
 
        section.set('foo', 'bar')
2834
 
        store.save()
2835
 
        modified_store = self.get_store(self)
2836
 
        sections = list(modified_store.get_sections())
2837
 
        self.assertLength(1, sections)
2838
 
        self.assertSectionContent(('baz', {'foo': 'bar'}), sections[0])
2839
 
 
2840
 
    def test_load_hook(self):
2841
 
        # We first needs to ensure that the store exists
2842
 
        store = self.get_store(self)
2843
 
        section = store.get_mutable_section('baz')
2844
 
        section.set('foo', 'bar')
2845
 
        store.save()
2846
 
        # Now we can try to load it
2847
 
        store = self.get_store(self)
2848
 
        calls = []
2849
 
        def hook(*args):
2850
 
            calls.append(args)
2851
 
        config.ConfigHooks.install_named_hook('load', hook, None)
2852
 
        self.assertLength(0, calls)
2853
 
        store.load()
2854
 
        self.assertLength(1, calls)
2855
 
        self.assertEquals((store,), calls[0])
2856
 
 
2857
 
    def test_save_hook(self):
2858
 
        calls = []
2859
 
        def hook(*args):
2860
 
            calls.append(args)
2861
 
        config.ConfigHooks.install_named_hook('save', hook, None)
2862
 
        self.assertLength(0, calls)
2863
 
        store = self.get_store(self)
2864
 
        section = store.get_mutable_section('baz')
2865
 
        section.set('foo', 'bar')
2866
 
        store.save()
2867
 
        self.assertLength(1, calls)
2868
 
        self.assertEquals((store,), calls[0])
2869
 
 
2870
 
 
2871
 
class TestIniFileStore(TestStore):
2872
 
 
2873
 
    def test_loading_unknown_file_fails(self):
2874
 
        store = config.IniFileStore(self.get_transport(), 'I-do-not-exist')
2875
 
        self.assertRaises(errors.NoSuchFile, store.load)
2876
 
 
2877
 
    def test_invalid_content(self):
2878
 
        store = config.IniFileStore(self.get_transport(), 'foo.conf', )
2879
 
        self.assertEquals(False, store.is_loaded())
2880
 
        exc = self.assertRaises(
2881
 
            errors.ParseConfigError, store._load_from_string,
2882
 
            'this is invalid !')
2883
 
        self.assertEndsWith(exc.filename, 'foo.conf')
2884
 
        # And the load failed
2885
 
        self.assertEquals(False, store.is_loaded())
2886
 
 
2887
 
    def test_get_embedded_sections(self):
2888
 
        # A more complicated example (which also shows that section names and
2889
 
        # option names share the same name space...)
2890
 
        # FIXME: This should be fixed by forbidding dicts as values ?
2891
 
        # -- vila 2011-04-05
2892
 
        store = config.IniFileStore(self.get_transport(), 'foo.conf', )
2893
 
        store._load_from_string('''
2894
 
foo=bar
2895
 
l=1,2
2896
 
[DEFAULT]
2897
 
foo_in_DEFAULT=foo_DEFAULT
2898
 
[bar]
2899
 
foo_in_bar=barbar
2900
 
[baz]
2901
 
foo_in_baz=barbaz
2902
 
[[qux]]
2903
 
foo_in_qux=quux
2904
 
''')
2905
 
        sections = list(store.get_sections())
2906
 
        self.assertLength(4, sections)
2907
 
        # The default section has no name.
2908
 
        # List values are provided as strings and need to be explicitly
2909
 
        # converted by specifying from_unicode=list_from_store at option
2910
 
        # registration
2911
 
        self.assertSectionContent((None, {'foo': 'bar', 'l': u'1,2'}),
2912
 
                                  sections[0])
2913
 
        self.assertSectionContent(
2914
 
            ('DEFAULT', {'foo_in_DEFAULT': 'foo_DEFAULT'}), sections[1])
2915
 
        self.assertSectionContent(
2916
 
            ('bar', {'foo_in_bar': 'barbar'}), sections[2])
2917
 
        # sub sections are provided as embedded dicts.
2918
 
        self.assertSectionContent(
2919
 
            ('baz', {'foo_in_baz': 'barbaz', 'qux': {'foo_in_qux': 'quux'}}),
2920
 
            sections[3])
2921
 
 
2922
 
 
2923
 
class TestLockableIniFileStore(TestStore):
2924
 
 
2925
 
    def test_create_store_in_created_dir(self):
2926
 
        self.assertPathDoesNotExist('dir')
2927
 
        t = self.get_transport('dir/subdir')
2928
 
        store = config.LockableIniFileStore(t, 'foo.conf')
2929
 
        store.get_mutable_section(None).set('foo', 'bar')
2930
 
        store.save()
2931
 
        self.assertPathExists('dir/subdir')
2932
 
 
2933
 
 
2934
 
class TestConcurrentStoreUpdates(TestStore):
2935
 
    """Test that Stores properly handle conccurent updates.
2936
 
 
2937
 
    New Store implementation may fail some of these tests but until such
2938
 
    implementations exist it's hard to properly filter them from the scenarios
2939
 
    applied here. If you encounter such a case, contact the bzr devs.
2940
 
    """
2941
 
 
2942
 
    scenarios = [(key, {'get_stack': builder}) for key, builder
2943
 
                 in config.test_stack_builder_registry.iteritems()]
2944
 
 
2945
 
    def setUp(self):
2946
 
        super(TestConcurrentStoreUpdates, self).setUp()
2947
 
        self._content = 'one=1\ntwo=2\n'
2948
 
        self.stack = self.get_stack(self)
2949
 
        if not isinstance(self.stack, config._CompatibleStack):
2950
 
            raise tests.TestNotApplicable(
2951
 
                '%s is not meant to be compatible with the old config design'
2952
 
                % (self.stack,))
2953
 
        self.stack.store._load_from_string(self._content)
2954
 
        # Flush the store
2955
 
        self.stack.store.save()
2956
 
 
2957
 
    def test_simple_read_access(self):
2958
 
        self.assertEquals('1', self.stack.get('one'))
2959
 
 
2960
 
    def test_simple_write_access(self):
2961
 
        self.stack.set('one', 'one')
2962
 
        self.assertEquals('one', self.stack.get('one'))
2963
 
 
2964
 
    def test_listen_to_the_last_speaker(self):
2965
 
        c1 = self.stack
2966
 
        c2 = self.get_stack(self)
2967
 
        c1.set('one', 'ONE')
2968
 
        c2.set('two', 'TWO')
2969
 
        self.assertEquals('ONE', c1.get('one'))
2970
 
        self.assertEquals('TWO', c2.get('two'))
2971
 
        # The second update respect the first one
2972
 
        self.assertEquals('ONE', c2.get('one'))
2973
 
 
2974
 
    def test_last_speaker_wins(self):
2975
 
        # If the same config is not shared, the same variable modified twice
2976
 
        # can only see a single result.
2977
 
        c1 = self.stack
2978
 
        c2 = self.get_stack(self)
2979
 
        c1.set('one', 'c1')
2980
 
        c2.set('one', 'c2')
2981
 
        self.assertEquals('c2', c2.get('one'))
2982
 
        # The first modification is still available until another refresh
2983
 
        # occur
2984
 
        self.assertEquals('c1', c1.get('one'))
2985
 
        c1.set('two', 'done')
2986
 
        self.assertEquals('c2', c1.get('one'))
2987
 
 
2988
 
    def test_writes_are_serialized(self):
2989
 
        c1 = self.stack
2990
 
        c2 = self.get_stack(self)
2991
 
 
2992
 
        # We spawn a thread that will pause *during* the config saving.
2993
 
        before_writing = threading.Event()
2994
 
        after_writing = threading.Event()
2995
 
        writing_done = threading.Event()
2996
 
        c1_save_without_locking_orig = c1.store.save_without_locking
2997
 
        def c1_save_without_locking():
2998
 
            before_writing.set()
2999
 
            c1_save_without_locking_orig()
3000
 
            # The lock is held. We wait for the main thread to decide when to
3001
 
            # continue
3002
 
            after_writing.wait()
3003
 
        c1.store.save_without_locking = c1_save_without_locking
3004
 
        def c1_set():
3005
 
            c1.set('one', 'c1')
3006
 
            writing_done.set()
3007
 
        t1 = threading.Thread(target=c1_set)
3008
 
        # Collect the thread after the test
3009
 
        self.addCleanup(t1.join)
3010
 
        # Be ready to unblock the thread if the test goes wrong
3011
 
        self.addCleanup(after_writing.set)
3012
 
        t1.start()
3013
 
        before_writing.wait()
3014
 
        self.assertRaises(errors.LockContention,
3015
 
                          c2.set, 'one', 'c2')
3016
 
        self.assertEquals('c1', c1.get('one'))
3017
 
        # Let the lock be released
3018
 
        after_writing.set()
3019
 
        writing_done.wait()
3020
 
        c2.set('one', 'c2')
3021
 
        self.assertEquals('c2', c2.get('one'))
3022
 
 
3023
 
    def test_read_while_writing(self):
3024
 
       c1 = self.stack
3025
 
       # We spawn a thread that will pause *during* the write
3026
 
       ready_to_write = threading.Event()
3027
 
       do_writing = threading.Event()
3028
 
       writing_done = threading.Event()
3029
 
       # We override the _save implementation so we know the store is locked
3030
 
       c1_save_without_locking_orig = c1.store.save_without_locking
3031
 
       def c1_save_without_locking():
3032
 
           ready_to_write.set()
3033
 
           # The lock is held. We wait for the main thread to decide when to
3034
 
           # continue
3035
 
           do_writing.wait()
3036
 
           c1_save_without_locking_orig()
3037
 
           writing_done.set()
3038
 
       c1.store.save_without_locking = c1_save_without_locking
3039
 
       def c1_set():
3040
 
           c1.set('one', 'c1')
3041
 
       t1 = threading.Thread(target=c1_set)
3042
 
       # Collect the thread after the test
3043
 
       self.addCleanup(t1.join)
3044
 
       # Be ready to unblock the thread if the test goes wrong
3045
 
       self.addCleanup(do_writing.set)
3046
 
       t1.start()
3047
 
       # Ensure the thread is ready to write
3048
 
       ready_to_write.wait()
3049
 
       self.assertEquals('c1', c1.get('one'))
3050
 
       # If we read during the write, we get the old value
3051
 
       c2 = self.get_stack(self)
3052
 
       self.assertEquals('1', c2.get('one'))
3053
 
       # Let the writing occur and ensure it occurred
3054
 
       do_writing.set()
3055
 
       writing_done.wait()
3056
 
       # Now we get the updated value
3057
 
       c3 = self.get_stack(self)
3058
 
       self.assertEquals('c1', c3.get('one'))
3059
 
 
3060
 
    # FIXME: It may be worth looking into removing the lock dir when it's not
3061
 
    # needed anymore and look at possible fallouts for concurrent lockers. This
3062
 
    # will matter if/when we use config files outside of bazaar directories
3063
 
    # (.bazaar or .bzr) -- vila 20110-04-111
3064
 
 
3065
 
 
3066
 
class TestSectionMatcher(TestStore):
3067
 
 
3068
 
    scenarios = [('location', {'matcher': config.LocationMatcher}),
3069
 
                 ('id', {'matcher': config.NameMatcher}),]
3070
 
 
3071
 
    def get_store(self, file_name):
3072
 
        return config.IniFileStore(self.get_readonly_transport(), file_name)
3073
 
 
3074
 
    def test_no_matches_for_empty_stores(self):
3075
 
        store = self.get_store('foo.conf')
3076
 
        store._load_from_string('')
3077
 
        matcher = self.matcher(store, '/bar')
3078
 
        self.assertEquals([], list(matcher.get_sections()))
3079
 
 
3080
 
    def test_build_doesnt_load_store(self):
3081
 
        store = self.get_store('foo.conf')
3082
 
        matcher = self.matcher(store, '/bar')
3083
 
        self.assertFalse(store.is_loaded())
3084
 
 
3085
 
 
3086
 
class TestLocationSection(tests.TestCase):
3087
 
 
3088
 
    def get_section(self, options, extra_path):
3089
 
        section = config.Section('foo', options)
3090
 
        # We don't care about the length so we use '0'
3091
 
        return config.LocationSection(section, 0, extra_path)
3092
 
 
3093
 
    def test_simple_option(self):
3094
 
        section = self.get_section({'foo': 'bar'}, '')
3095
 
        self.assertEquals('bar', section.get('foo'))
3096
 
 
3097
 
    def test_option_with_extra_path(self):
3098
 
        section = self.get_section({'foo': 'bar', 'foo:policy': 'appendpath'},
3099
 
                                   'baz')
3100
 
        self.assertEquals('bar/baz', section.get('foo'))
3101
 
 
3102
 
    def test_invalid_policy(self):
3103
 
        section = self.get_section({'foo': 'bar', 'foo:policy': 'die'},
3104
 
                                   'baz')
3105
 
        # invalid policies are ignored
3106
 
        self.assertEquals('bar', section.get('foo'))
3107
 
 
3108
 
 
3109
 
class TestLocationMatcher(TestStore):
3110
 
 
3111
 
    def get_store(self, file_name):
3112
 
        return config.IniFileStore(self.get_readonly_transport(), file_name)
3113
 
 
3114
 
    def test_unrelated_section_excluded(self):
3115
 
        store = self.get_store('foo.conf')
3116
 
        store._load_from_string('''
3117
 
[/foo]
3118
 
section=/foo
3119
 
[/foo/baz]
3120
 
section=/foo/baz
3121
 
[/foo/bar]
3122
 
section=/foo/bar
3123
 
[/foo/bar/baz]
3124
 
section=/foo/bar/baz
3125
 
[/quux/quux]
3126
 
section=/quux/quux
3127
 
''')
3128
 
        self.assertEquals(['/foo', '/foo/baz', '/foo/bar', '/foo/bar/baz',
3129
 
                           '/quux/quux'],
3130
 
                          [section.id for section in store.get_sections()])
3131
 
        matcher = config.LocationMatcher(store, '/foo/bar/quux')
3132
 
        sections = list(matcher.get_sections())
3133
 
        self.assertEquals([3, 2],
3134
 
                          [section.length for section in sections])
3135
 
        self.assertEquals(['/foo/bar', '/foo'],
3136
 
                          [section.id for section in sections])
3137
 
        self.assertEquals(['quux', 'bar/quux'],
3138
 
                          [section.extra_path for section in sections])
3139
 
 
3140
 
    def test_more_specific_sections_first(self):
3141
 
        store = self.get_store('foo.conf')
3142
 
        store._load_from_string('''
3143
 
[/foo]
3144
 
section=/foo
3145
 
[/foo/bar]
3146
 
section=/foo/bar
3147
 
''')
3148
 
        self.assertEquals(['/foo', '/foo/bar'],
3149
 
                          [section.id for section in store.get_sections()])
3150
 
        matcher = config.LocationMatcher(store, '/foo/bar/baz')
3151
 
        sections = list(matcher.get_sections())
3152
 
        self.assertEquals([3, 2],
3153
 
                          [section.length for section in sections])
3154
 
        self.assertEquals(['/foo/bar', '/foo'],
3155
 
                          [section.id for section in sections])
3156
 
        self.assertEquals(['baz', 'bar/baz'],
3157
 
                          [section.extra_path for section in sections])
3158
 
 
3159
 
    def test_appendpath_in_no_name_section(self):
3160
 
        # It's a bit weird to allow appendpath in a no-name section, but
3161
 
        # someone may found a use for it
3162
 
        store = self.get_store('foo.conf')
3163
 
        store._load_from_string('''
3164
 
foo=bar
3165
 
foo:policy = appendpath
3166
 
''')
3167
 
        matcher = config.LocationMatcher(store, 'dir/subdir')
3168
 
        sections = list(matcher.get_sections())
3169
 
        self.assertLength(1, sections)
3170
 
        self.assertEquals('bar/dir/subdir', sections[0].get('foo'))
3171
 
 
3172
 
    def test_file_urls_are_normalized(self):
3173
 
        store = self.get_store('foo.conf')
3174
 
        if sys.platform == 'win32':
3175
 
            expected_url = 'file:///C:/dir/subdir'
3176
 
            expected_location = 'C:/dir/subdir'
3177
 
        else:
3178
 
            expected_url = 'file:///dir/subdir'
3179
 
            expected_location = '/dir/subdir'
3180
 
        matcher = config.LocationMatcher(store, expected_url)
3181
 
        self.assertEquals(expected_location, matcher.location)
3182
 
 
3183
 
 
3184
 
class TestNameMatcher(TestStore):
3185
 
 
3186
 
    def setUp(self):
3187
 
        super(TestNameMatcher, self).setUp()
3188
 
        self.store = config.IniFileStore(self.get_readonly_transport(),
3189
 
                                         'foo.conf')
3190
 
        self.store._load_from_string('''
3191
 
[foo]
3192
 
option=foo
3193
 
[foo/baz]
3194
 
option=foo/baz
3195
 
[bar]
3196
 
option=bar
3197
 
''')
3198
 
 
3199
 
    def get_matching_sections(self, name):
3200
 
        matcher = config.NameMatcher(self.store, name)
3201
 
        return list(matcher.get_sections())
3202
 
 
3203
 
    def test_matching(self):
3204
 
        sections = self.get_matching_sections('foo')
3205
 
        self.assertLength(1, sections)
3206
 
        self.assertSectionContent(('foo', {'option': 'foo'}), sections[0])
3207
 
 
3208
 
    def test_not_matching(self):
3209
 
        sections = self.get_matching_sections('baz')
3210
 
        self.assertLength(0, sections)
3211
 
 
3212
 
 
3213
 
class TestStackGet(tests.TestCase):
3214
 
 
3215
 
    # FIXME: This should be parametrized for all known Stack or dedicated
3216
 
    # paramerized tests created to avoid bloating -- vila 2011-03-31
3217
 
 
3218
 
    def overrideOptionRegistry(self):
3219
 
        self.overrideAttr(config, 'option_registry', config.OptionRegistry())
3220
 
 
3221
 
    def test_single_config_get(self):
3222
 
        conf = dict(foo='bar')
3223
 
        conf_stack = config.Stack([conf])
3224
 
        self.assertEquals('bar', conf_stack.get('foo'))
3225
 
 
3226
 
    def test_get_with_registered_default_value(self):
3227
 
        conf_stack = config.Stack([dict()])
3228
 
        opt = config.Option('foo', default='bar')
3229
 
        self.overrideOptionRegistry()
3230
 
        config.option_registry.register('foo', opt)
3231
 
        self.assertEquals('bar', conf_stack.get('foo'))
3232
 
 
3233
 
    def test_get_without_registered_default_value(self):
3234
 
        conf_stack = config.Stack([dict()])
3235
 
        opt = config.Option('foo')
3236
 
        self.overrideOptionRegistry()
3237
 
        config.option_registry.register('foo', opt)
3238
 
        self.assertEquals(None, conf_stack.get('foo'))
3239
 
 
3240
 
    def test_get_without_default_value_for_not_registered(self):
3241
 
        conf_stack = config.Stack([dict()])
3242
 
        opt = config.Option('foo')
3243
 
        self.overrideOptionRegistry()
3244
 
        self.assertEquals(None, conf_stack.get('foo'))
3245
 
 
3246
 
    def test_get_first_definition(self):
3247
 
        conf1 = dict(foo='bar')
3248
 
        conf2 = dict(foo='baz')
3249
 
        conf_stack = config.Stack([conf1, conf2])
3250
 
        self.assertEquals('bar', conf_stack.get('foo'))
3251
 
 
3252
 
    def test_get_embedded_definition(self):
3253
 
        conf1 = dict(yy='12')
3254
 
        conf2 = config.Stack([dict(xx='42'), dict(foo='baz')])
3255
 
        conf_stack = config.Stack([conf1, conf2])
3256
 
        self.assertEquals('baz', conf_stack.get('foo'))
3257
 
 
3258
 
    def test_get_for_empty_section_callable(self):
3259
 
        conf_stack = config.Stack([lambda : []])
3260
 
        self.assertEquals(None, conf_stack.get('foo'))
3261
 
 
3262
 
    def test_get_for_broken_callable(self):
3263
 
        # Trying to use and invalid callable raises an exception on first use
3264
 
        conf_stack = config.Stack([lambda : object()])
3265
 
        self.assertRaises(TypeError, conf_stack.get, 'foo')
3266
 
 
3267
 
 
3268
 
class TestStackWithTransport(tests.TestCaseWithTransport):
3269
 
 
3270
 
    scenarios = [(key, {'get_stack': builder}) for key, builder
3271
 
                 in config.test_stack_builder_registry.iteritems()]
3272
 
 
3273
 
 
3274
 
class TestConcreteStacks(TestStackWithTransport):
3275
 
 
3276
 
    def test_build_stack(self):
3277
 
        # Just a smoke test to help debug builders
3278
 
        stack = self.get_stack(self)
3279
 
 
3280
 
 
3281
 
class TestStackGet(TestStackWithTransport):
3282
 
 
3283
 
    def setUp(self):
3284
 
        super(TestStackGet, self).setUp()
3285
 
        self.conf = self.get_stack(self)
3286
 
 
3287
 
    def test_get_for_empty_stack(self):
3288
 
        self.assertEquals(None, self.conf.get('foo'))
3289
 
 
3290
 
    def test_get_hook(self):
3291
 
        self.conf.store._load_from_string('foo=bar')
3292
 
        calls = []
3293
 
        def hook(*args):
3294
 
            calls.append(args)
3295
 
        config.ConfigHooks.install_named_hook('get', hook, None)
3296
 
        self.assertLength(0, calls)
3297
 
        value = self.conf.get('foo')
3298
 
        self.assertEquals('bar', value)
3299
 
        self.assertLength(1, calls)
3300
 
        self.assertEquals((self.conf, 'foo', 'bar'), calls[0])
3301
 
 
3302
 
 
3303
 
class TestStackGetWithConverter(TestStackGet):
3304
 
 
3305
 
    def setUp(self):
3306
 
        super(TestStackGetWithConverter, self).setUp()
3307
 
        self.overrideAttr(config, 'option_registry', config.OptionRegistry())
3308
 
        self.registry = config.option_registry
3309
 
 
3310
 
    def register_bool_option(self, name, default=None, default_from_env=None):
3311
 
        b = config.Option(name, help='A boolean.',
3312
 
                          default=default, default_from_env=default_from_env,
3313
 
                          from_unicode=config.bool_from_store)
3314
 
        self.registry.register(b)
3315
 
 
3316
 
    def test_get_default_bool_None(self):
3317
 
        self.register_bool_option('foo')
3318
 
        self.assertEquals(None, self.conf.get('foo'))
3319
 
 
3320
 
    def test_get_default_bool_True(self):
3321
 
        self.register_bool_option('foo', u'True')
3322
 
        self.assertEquals(True, self.conf.get('foo'))
3323
 
 
3324
 
    def test_get_default_bool_False(self):
3325
 
        self.register_bool_option('foo', False)
3326
 
        self.assertEquals(False, self.conf.get('foo'))
3327
 
 
3328
 
    def test_get_default_bool_False_as_string(self):
3329
 
        self.register_bool_option('foo', u'False')
3330
 
        self.assertEquals(False, self.conf.get('foo'))
3331
 
 
3332
 
    def test_get_default_bool_from_env_converted(self):
3333
 
        self.register_bool_option('foo', u'True', default_from_env=['FOO'])
3334
 
        self.overrideEnv('FOO', 'False')
3335
 
        self.assertEquals(False, self.conf.get('foo'))
3336
 
 
3337
 
    def test_get_default_bool_when_conversion_fails(self):
3338
 
        self.register_bool_option('foo', default='True')
3339
 
        self.conf.store._load_from_string('foo=invalid boolean')
3340
 
        self.assertEquals(True, self.conf.get('foo'))
3341
 
 
3342
 
    def register_integer_option(self, name,
3343
 
                                default=None, default_from_env=None):
3344
 
        i = config.Option(name, help='An integer.',
3345
 
                          default=default, default_from_env=default_from_env,
3346
 
                          from_unicode=config.int_from_store)
3347
 
        self.registry.register(i)
3348
 
 
3349
 
    def test_get_default_integer_None(self):
3350
 
        self.register_integer_option('foo')
3351
 
        self.assertEquals(None, self.conf.get('foo'))
3352
 
 
3353
 
    def test_get_default_integer(self):
3354
 
        self.register_integer_option('foo', 42)
3355
 
        self.assertEquals(42, self.conf.get('foo'))
3356
 
 
3357
 
    def test_get_default_integer_as_string(self):
3358
 
        self.register_integer_option('foo', u'42')
3359
 
        self.assertEquals(42, self.conf.get('foo'))
3360
 
 
3361
 
    def test_get_default_integer_from_env(self):
3362
 
        self.register_integer_option('foo', default_from_env=['FOO'])
3363
 
        self.overrideEnv('FOO', '18')
3364
 
        self.assertEquals(18, self.conf.get('foo'))
3365
 
 
3366
 
    def test_get_default_integer_when_conversion_fails(self):
3367
 
        self.register_integer_option('foo', default='12')
3368
 
        self.conf.store._load_from_string('foo=invalid integer')
3369
 
        self.assertEquals(12, self.conf.get('foo'))
3370
 
 
3371
 
    def register_list_option(self, name, default=None, default_from_env=None):
3372
 
        l = config.Option(name, help='A list.',
3373
 
                          default=default, default_from_env=default_from_env,
3374
 
                          from_unicode=config.list_from_store)
3375
 
        self.registry.register(l)
3376
 
 
3377
 
    def test_get_default_list_None(self):
3378
 
        self.register_list_option('foo')
3379
 
        self.assertEquals(None, self.conf.get('foo'))
3380
 
 
3381
 
    def test_get_default_list_empty(self):
3382
 
        self.register_list_option('foo', '')
3383
 
        self.assertEquals([], self.conf.get('foo'))
3384
 
 
3385
 
    def test_get_default_list_from_env(self):
3386
 
        self.register_list_option('foo', default_from_env=['FOO'])
3387
 
        self.overrideEnv('FOO', '')
3388
 
        self.assertEquals([], self.conf.get('foo'))
3389
 
 
3390
 
    def test_get_with_list_converter_no_item(self):
3391
 
        self.register_list_option('foo', None)
3392
 
        self.conf.store._load_from_string('foo=,')
3393
 
        self.assertEquals([], self.conf.get('foo'))
3394
 
 
3395
 
    def test_get_with_list_converter_many_items(self):
3396
 
        self.register_list_option('foo', None)
3397
 
        self.conf.store._load_from_string('foo=m,o,r,e')
3398
 
        self.assertEquals(['m', 'o', 'r', 'e'], self.conf.get('foo'))
3399
 
 
3400
 
    def test_get_with_list_converter_embedded_spaces_many_items(self):
3401
 
        self.register_list_option('foo', None)
3402
 
        self.conf.store._load_from_string('foo=" bar", "baz "')
3403
 
        self.assertEquals([' bar', 'baz '], self.conf.get('foo'))
3404
 
 
3405
 
    def test_get_with_list_converter_stripped_spaces_many_items(self):
3406
 
        self.register_list_option('foo', None)
3407
 
        self.conf.store._load_from_string('foo= bar ,  baz ')
3408
 
        self.assertEquals(['bar', 'baz'], self.conf.get('foo'))
3409
 
 
3410
 
 
3411
 
class TestStackExpandOptions(tests.TestCaseWithTransport):
3412
 
 
3413
 
    def setUp(self):
3414
 
        super(TestStackExpandOptions, self).setUp()
3415
 
        self.overrideAttr(config, 'option_registry', config.OptionRegistry())
3416
 
        self.registry = config.option_registry
3417
 
        self.conf = build_branch_stack(self)
3418
 
 
3419
 
    def assertExpansion(self, expected, string, env=None):
3420
 
        self.assertEquals(expected, self.conf.expand_options(string, env))
3421
 
 
3422
 
    def test_no_expansion(self):
3423
 
        self.assertExpansion('foo', 'foo')
3424
 
 
3425
 
    def test_expand_default_value(self):
3426
 
        self.conf.store._load_from_string('bar=baz')
3427
 
        self.registry.register(config.Option('foo', default=u'{bar}'))
3428
 
        self.assertEquals('baz', self.conf.get('foo', expand=True))
3429
 
 
3430
 
    def test_expand_default_from_env(self):
3431
 
        self.conf.store._load_from_string('bar=baz')
3432
 
        self.registry.register(config.Option('foo', default_from_env=['FOO']))
3433
 
        self.overrideEnv('FOO', '{bar}')
3434
 
        self.assertEquals('baz', self.conf.get('foo', expand=True))
3435
 
 
3436
 
    def test_expand_default_on_failed_conversion(self):
3437
 
        self.conf.store._load_from_string('baz=bogus\nbar=42\nfoo={baz}')
3438
 
        self.registry.register(
3439
 
            config.Option('foo', default=u'{bar}',
3440
 
                          from_unicode=config.int_from_store))
3441
 
        self.assertEquals(42, self.conf.get('foo', expand=True))
3442
 
 
3443
 
    def test_env_adding_options(self):
3444
 
        self.assertExpansion('bar', '{foo}', {'foo': 'bar'})
3445
 
 
3446
 
    def test_env_overriding_options(self):
3447
 
        self.conf.store._load_from_string('foo=baz')
3448
 
        self.assertExpansion('bar', '{foo}', {'foo': 'bar'})
3449
 
 
3450
 
    def test_simple_ref(self):
3451
 
        self.conf.store._load_from_string('foo=xxx')
3452
 
        self.assertExpansion('xxx', '{foo}')
3453
 
 
3454
 
    def test_unknown_ref(self):
3455
 
        self.assertRaises(errors.ExpandingUnknownOption,
3456
 
                          self.conf.expand_options, '{foo}')
3457
 
 
3458
 
    def test_indirect_ref(self):
3459
 
        self.conf.store._load_from_string('''
3460
 
foo=xxx
3461
 
bar={foo}
3462
 
''')
3463
 
        self.assertExpansion('xxx', '{bar}')
3464
 
 
3465
 
    def test_embedded_ref(self):
3466
 
        self.conf.store._load_from_string('''
3467
 
foo=xxx
3468
 
bar=foo
3469
 
''')
3470
 
        self.assertExpansion('xxx', '{{bar}}')
3471
 
 
3472
 
    def test_simple_loop(self):
3473
 
        self.conf.store._load_from_string('foo={foo}')
3474
 
        self.assertRaises(errors.OptionExpansionLoop,
3475
 
                          self.conf.expand_options, '{foo}')
3476
 
 
3477
 
    def test_indirect_loop(self):
3478
 
        self.conf.store._load_from_string('''
3479
 
foo={bar}
3480
 
bar={baz}
3481
 
baz={foo}''')
3482
 
        e = self.assertRaises(errors.OptionExpansionLoop,
3483
 
                              self.conf.expand_options, '{foo}')
3484
 
        self.assertEquals('foo->bar->baz', e.refs)
3485
 
        self.assertEquals('{foo}', e.string)
3486
 
 
3487
 
    def test_list(self):
3488
 
        self.conf.store._load_from_string('''
3489
 
foo=start
3490
 
bar=middle
3491
 
baz=end
3492
 
list={foo},{bar},{baz}
3493
 
''')
3494
 
        self.registry.register(
3495
 
            config.Option('list', from_unicode=config.list_from_store))
3496
 
        self.assertEquals(['start', 'middle', 'end'],
3497
 
                           self.conf.get('list', expand=True))
3498
 
 
3499
 
    def test_cascading_list(self):
3500
 
        self.conf.store._load_from_string('''
3501
 
foo=start,{bar}
3502
 
bar=middle,{baz}
3503
 
baz=end
3504
 
list={foo}
3505
 
''')
3506
 
        self.registry.register(
3507
 
            config.Option('list', from_unicode=config.list_from_store))
3508
 
        self.assertEquals(['start', 'middle', 'end'],
3509
 
                           self.conf.get('list', expand=True))
3510
 
 
3511
 
    def test_pathologically_hidden_list(self):
3512
 
        self.conf.store._load_from_string('''
3513
 
foo=bin
3514
 
bar=go
3515
 
start={foo
3516
 
middle=},{
3517
 
end=bar}
3518
 
hidden={start}{middle}{end}
3519
 
''')
3520
 
        # What matters is what the registration says, the conversion happens
3521
 
        # only after all expansions have been performed
3522
 
        self.registry.register(
3523
 
            config.Option('hidden', from_unicode=config.list_from_store))
3524
 
        self.assertEquals(['bin', 'go'],
3525
 
                          self.conf.get('hidden', expand=True))
3526
 
 
3527
 
 
3528
 
class TestStackCrossSectionsExpand(tests.TestCaseWithTransport):
3529
 
 
3530
 
    def setUp(self):
3531
 
        super(TestStackCrossSectionsExpand, self).setUp()
3532
 
 
3533
 
    def get_config(self, location, string):
3534
 
        if string is None:
3535
 
            string = ''
3536
 
        # Since we don't save the config we won't strictly require to inherit
3537
 
        # from TestCaseInTempDir, but an error occurs so quickly...
3538
 
        c = config.LocationStack(location)
3539
 
        c.store._load_from_string(string)
3540
 
        return c
3541
 
 
3542
 
    def test_dont_cross_unrelated_section(self):
3543
 
        c = self.get_config('/another/branch/path','''
3544
 
[/one/branch/path]
3545
 
foo = hello
3546
 
bar = {foo}/2
3547
 
 
3548
 
[/another/branch/path]
3549
 
bar = {foo}/2
3550
 
''')
3551
 
        self.assertRaises(errors.ExpandingUnknownOption,
3552
 
                          c.get, 'bar', expand=True)
3553
 
 
3554
 
    def test_cross_related_sections(self):
3555
 
        c = self.get_config('/project/branch/path','''
3556
 
[/project]
3557
 
foo = qu
3558
 
 
3559
 
[/project/branch/path]
3560
 
bar = {foo}ux
3561
 
''')
3562
 
        self.assertEquals('quux', c.get('bar', expand=True))
3563
 
 
3564
 
 
3565
 
class TestStackSet(TestStackWithTransport):
3566
 
 
3567
 
    def test_simple_set(self):
3568
 
        conf = self.get_stack(self)
3569
 
        conf.store._load_from_string('foo=bar')
3570
 
        self.assertEquals('bar', conf.get('foo'))
3571
 
        conf.set('foo', 'baz')
3572
 
        # Did we get it back ?
3573
 
        self.assertEquals('baz', conf.get('foo'))
3574
 
 
3575
 
    def test_set_creates_a_new_section(self):
3576
 
        conf = self.get_stack(self)
3577
 
        conf.set('foo', 'baz')
3578
 
        self.assertEquals, 'baz', conf.get('foo')
3579
 
 
3580
 
    def test_set_hook(self):
3581
 
        calls = []
3582
 
        def hook(*args):
3583
 
            calls.append(args)
3584
 
        config.ConfigHooks.install_named_hook('set', hook, None)
3585
 
        self.assertLength(0, calls)
3586
 
        conf = self.get_stack(self)
3587
 
        conf.set('foo', 'bar')
3588
 
        self.assertLength(1, calls)
3589
 
        self.assertEquals((conf, 'foo', 'bar'), calls[0])
3590
 
 
3591
 
 
3592
 
class TestStackRemove(TestStackWithTransport):
3593
 
 
3594
 
    def test_remove_existing(self):
3595
 
        conf = self.get_stack(self)
3596
 
        conf.store._load_from_string('foo=bar')
3597
 
        self.assertEquals('bar', conf.get('foo'))
3598
 
        conf.remove('foo')
3599
 
        # Did we get it back ?
3600
 
        self.assertEquals(None, conf.get('foo'))
3601
 
 
3602
 
    def test_remove_unknown(self):
3603
 
        conf = self.get_stack(self)
3604
 
        self.assertRaises(KeyError, conf.remove, 'I_do_not_exist')
3605
 
 
3606
 
    def test_remove_hook(self):
3607
 
        calls = []
3608
 
        def hook(*args):
3609
 
            calls.append(args)
3610
 
        config.ConfigHooks.install_named_hook('remove', hook, None)
3611
 
        self.assertLength(0, calls)
3612
 
        conf = self.get_stack(self)
3613
 
        conf.store._load_from_string('foo=bar')
3614
 
        conf.remove('foo')
3615
 
        self.assertLength(1, calls)
3616
 
        self.assertEquals((conf, 'foo'), calls[0])
3617
 
 
3618
 
 
3619
 
class TestConfigGetOptions(tests.TestCaseWithTransport, TestOptionsMixin):
3620
 
 
3621
 
    def setUp(self):
3622
 
        super(TestConfigGetOptions, self).setUp()
3623
 
        create_configs(self)
3624
 
 
3625
 
    def test_no_variable(self):
3626
 
        # Using branch should query branch, locations and bazaar
3627
 
        self.assertOptions([], self.branch_config)
3628
 
 
3629
 
    def test_option_in_bazaar(self):
3630
 
        self.bazaar_config.set_user_option('file', 'bazaar')
3631
 
        self.assertOptions([('file', 'bazaar', 'DEFAULT', 'bazaar')],
3632
 
                           self.bazaar_config)
3633
 
 
3634
 
    def test_option_in_locations(self):
3635
 
        self.locations_config.set_user_option('file', 'locations')
3636
 
        self.assertOptions(
3637
 
            [('file', 'locations', self.tree.basedir, 'locations')],
3638
 
            self.locations_config)
3639
 
 
3640
 
    def test_option_in_branch(self):
3641
 
        self.branch_config.set_user_option('file', 'branch')
3642
 
        self.assertOptions([('file', 'branch', 'DEFAULT', 'branch')],
3643
 
                           self.branch_config)
3644
 
 
3645
 
    def test_option_in_bazaar_and_branch(self):
3646
 
        self.bazaar_config.set_user_option('file', 'bazaar')
3647
 
        self.branch_config.set_user_option('file', 'branch')
3648
 
        self.assertOptions([('file', 'branch', 'DEFAULT', 'branch'),
3649
 
                            ('file', 'bazaar', 'DEFAULT', 'bazaar'),],
3650
 
                           self.branch_config)
3651
 
 
3652
 
    def test_option_in_branch_and_locations(self):
3653
 
        # Hmm, locations override branch :-/
3654
 
        self.locations_config.set_user_option('file', 'locations')
3655
 
        self.branch_config.set_user_option('file', 'branch')
3656
 
        self.assertOptions(
3657
 
            [('file', 'locations', self.tree.basedir, 'locations'),
3658
 
             ('file', 'branch', 'DEFAULT', 'branch'),],
3659
 
            self.branch_config)
3660
 
 
3661
 
    def test_option_in_bazaar_locations_and_branch(self):
3662
 
        self.bazaar_config.set_user_option('file', 'bazaar')
3663
 
        self.locations_config.set_user_option('file', 'locations')
3664
 
        self.branch_config.set_user_option('file', 'branch')
3665
 
        self.assertOptions(
3666
 
            [('file', 'locations', self.tree.basedir, 'locations'),
3667
 
             ('file', 'branch', 'DEFAULT', 'branch'),
3668
 
             ('file', 'bazaar', 'DEFAULT', 'bazaar'),],
3669
 
            self.branch_config)
3670
 
 
3671
 
 
3672
 
class TestConfigRemoveOption(tests.TestCaseWithTransport, TestOptionsMixin):
3673
 
 
3674
 
    def setUp(self):
3675
 
        super(TestConfigRemoveOption, self).setUp()
3676
 
        create_configs_with_file_option(self)
3677
 
 
3678
 
    def test_remove_in_locations(self):
3679
 
        self.locations_config.remove_user_option('file', self.tree.basedir)
3680
 
        self.assertOptions(
3681
 
            [('file', 'branch', 'DEFAULT', 'branch'),
3682
 
             ('file', 'bazaar', 'DEFAULT', 'bazaar'),],
3683
 
            self.branch_config)
3684
 
 
3685
 
    def test_remove_in_branch(self):
3686
 
        self.branch_config.remove_user_option('file')
3687
 
        self.assertOptions(
3688
 
            [('file', 'locations', self.tree.basedir, 'locations'),
3689
 
             ('file', 'bazaar', 'DEFAULT', 'bazaar'),],
3690
 
            self.branch_config)
3691
 
 
3692
 
    def test_remove_in_bazaar(self):
3693
 
        self.bazaar_config.remove_user_option('file')
3694
 
        self.assertOptions(
3695
 
            [('file', 'locations', self.tree.basedir, 'locations'),
3696
 
             ('file', 'branch', 'DEFAULT', 'branch'),],
3697
 
            self.branch_config)
3698
 
 
3699
 
 
3700
 
class TestConfigGetSections(tests.TestCaseWithTransport):
3701
 
 
3702
 
    def setUp(self):
3703
 
        super(TestConfigGetSections, self).setUp()
3704
 
        create_configs(self)
3705
 
 
3706
 
    def assertSectionNames(self, expected, conf, name=None):
3707
 
        """Check which sections are returned for a given config.
3708
 
 
3709
 
        If fallback configurations exist their sections can be included.
3710
 
 
3711
 
        :param expected: A list of section names.
3712
 
 
3713
 
        :param conf: The configuration that will be queried.
3714
 
 
3715
 
        :param name: An optional section name that will be passed to
3716
 
            get_sections().
3717
 
        """
3718
 
        sections = list(conf._get_sections(name))
3719
 
        self.assertLength(len(expected), sections)
3720
 
        self.assertEqual(expected, [name for name, _, _ in sections])
3721
 
 
3722
 
    def test_bazaar_default_section(self):
3723
 
        self.assertSectionNames(['DEFAULT'], self.bazaar_config)
3724
 
 
3725
 
    def test_locations_default_section(self):
3726
 
        # No sections are defined in an empty file
3727
 
        self.assertSectionNames([], self.locations_config)
3728
 
 
3729
 
    def test_locations_named_section(self):
3730
 
        self.locations_config.set_user_option('file', 'locations')
3731
 
        self.assertSectionNames([self.tree.basedir], self.locations_config)
3732
 
 
3733
 
    def test_locations_matching_sections(self):
3734
 
        loc_config = self.locations_config
3735
 
        loc_config.set_user_option('file', 'locations')
3736
 
        # We need to cheat a bit here to create an option in sections above and
3737
 
        # below the 'location' one.
3738
 
        parser = loc_config._get_parser()
3739
 
        # locations.cong deals with '/' ignoring native os.sep
3740
 
        location_names = self.tree.basedir.split('/')
3741
 
        parent = '/'.join(location_names[:-1])
3742
 
        child = '/'.join(location_names + ['child'])
3743
 
        parser[parent] = {}
3744
 
        parser[parent]['file'] = 'parent'
3745
 
        parser[child] = {}
3746
 
        parser[child]['file'] = 'child'
3747
 
        self.assertSectionNames([self.tree.basedir, parent], loc_config)
3748
 
 
3749
 
    def test_branch_data_default_section(self):
3750
 
        self.assertSectionNames([None],
3751
 
                                self.branch_config._get_branch_data_config())
3752
 
 
3753
 
    def test_branch_default_sections(self):
3754
 
        # No sections are defined in an empty locations file
3755
 
        self.assertSectionNames([None, 'DEFAULT'],
3756
 
                                self.branch_config)
3757
 
        # Unless we define an option
3758
 
        self.branch_config._get_location_config().set_user_option(
3759
 
            'file', 'locations')
3760
 
        self.assertSectionNames([self.tree.basedir, None, 'DEFAULT'],
3761
 
                                self.branch_config)
3762
 
 
3763
 
    def test_bazaar_named_section(self):
3764
 
        # We need to cheat as the API doesn't give direct access to sections
3765
 
        # other than DEFAULT.
3766
 
        self.bazaar_config.set_alias('bazaar', 'bzr')
3767
 
        self.assertSectionNames(['ALIASES'], self.bazaar_config, 'ALIASES')
3768
 
 
3769
 
 
3770
 
class TestAuthenticationConfigFile(tests.TestCase):
3771
 
    """Test the authentication.conf file matching"""
3772
 
 
3773
 
    def _got_user_passwd(self, expected_user, expected_password,
3774
 
                         config, *args, **kwargs):
3775
 
        credentials = config.get_credentials(*args, **kwargs)
3776
 
        if credentials is None:
3777
 
            user = None
3778
 
            password = None
3779
 
        else:
3780
 
            user = credentials['user']
3781
 
            password = credentials['password']
3782
 
        self.assertEquals(expected_user, user)
3783
 
        self.assertEquals(expected_password, password)
3784
 
 
3785
 
    def test_empty_config(self):
3786
 
        conf = config.AuthenticationConfig(_file=StringIO())
3787
 
        self.assertEquals({}, conf._get_config())
3788
 
        self._got_user_passwd(None, None, conf, 'http', 'foo.net')
3789
 
 
3790
 
    def test_non_utf8_config(self):
3791
 
        conf = config.AuthenticationConfig(_file=StringIO(
3792
 
                'foo = bar\xff'))
3793
 
        self.assertRaises(errors.ConfigContentError, conf._get_config)
3794
 
 
3795
 
    def test_missing_auth_section_header(self):
3796
 
        conf = config.AuthenticationConfig(_file=StringIO('foo = bar'))
3797
 
        self.assertRaises(ValueError, conf.get_credentials, 'ftp', 'foo.net')
3798
 
 
3799
 
    def test_auth_section_header_not_closed(self):
3800
 
        conf = config.AuthenticationConfig(_file=StringIO('[DEF'))
3801
 
        self.assertRaises(errors.ParseConfigError, conf._get_config)
3802
 
 
3803
 
    def test_auth_value_not_boolean(self):
3804
 
        conf = config.AuthenticationConfig(_file=StringIO(
3805
 
                """[broken]
3806
 
scheme=ftp
3807
 
user=joe
3808
 
verify_certificates=askme # Error: Not a boolean
3809
 
"""))
3810
 
        self.assertRaises(ValueError, conf.get_credentials, 'ftp', 'foo.net')
3811
 
 
3812
 
    def test_auth_value_not_int(self):
3813
 
        conf = config.AuthenticationConfig(_file=StringIO(
3814
 
                """[broken]
3815
 
scheme=ftp
3816
 
user=joe
3817
 
port=port # Error: Not an int
3818
 
"""))
3819
 
        self.assertRaises(ValueError, conf.get_credentials, 'ftp', 'foo.net')
3820
 
 
3821
 
    def test_unknown_password_encoding(self):
3822
 
        conf = config.AuthenticationConfig(_file=StringIO(
3823
 
                """[broken]
3824
 
scheme=ftp
3825
 
user=joe
3826
 
password_encoding=unknown
3827
 
"""))
3828
 
        self.assertRaises(ValueError, conf.get_password,
3829
 
                          'ftp', 'foo.net', 'joe')
3830
 
 
3831
 
    def test_credentials_for_scheme_host(self):
3832
 
        conf = config.AuthenticationConfig(_file=StringIO(
3833
 
                """# Identity on foo.net
3834
 
[ftp definition]
3835
 
scheme=ftp
3836
 
host=foo.net
3837
 
user=joe
3838
 
password=secret-pass
3839
 
"""))
3840
 
        # Basic matching
3841
 
        self._got_user_passwd('joe', 'secret-pass', conf, 'ftp', 'foo.net')
3842
 
        # different scheme
3843
 
        self._got_user_passwd(None, None, conf, 'http', 'foo.net')
3844
 
        # different host
3845
 
        self._got_user_passwd(None, None, conf, 'ftp', 'bar.net')
3846
 
 
3847
 
    def test_credentials_for_host_port(self):
3848
 
        conf = config.AuthenticationConfig(_file=StringIO(
3849
 
                """# Identity on foo.net
3850
 
[ftp definition]
3851
 
scheme=ftp
3852
 
port=10021
3853
 
host=foo.net
3854
 
user=joe
3855
 
password=secret-pass
3856
 
"""))
3857
 
        # No port
3858
 
        self._got_user_passwd('joe', 'secret-pass',
3859
 
                              conf, 'ftp', 'foo.net', port=10021)
3860
 
        # different port
3861
 
        self._got_user_passwd(None, None, conf, 'ftp', 'foo.net')
3862
 
 
3863
 
    def test_for_matching_host(self):
3864
 
        conf = config.AuthenticationConfig(_file=StringIO(
3865
 
                """# Identity on foo.net
3866
 
[sourceforge]
3867
 
scheme=bzr
3868
 
host=bzr.sf.net
3869
 
user=joe
3870
 
password=joepass
3871
 
[sourceforge domain]
3872
 
scheme=bzr
3873
 
host=.bzr.sf.net
3874
 
user=georges
3875
 
password=bendover
3876
 
"""))
3877
 
        # matching domain
3878
 
        self._got_user_passwd('georges', 'bendover',
3879
 
                              conf, 'bzr', 'foo.bzr.sf.net')
3880
 
        # phishing attempt
3881
 
        self._got_user_passwd(None, None,
3882
 
                              conf, 'bzr', 'bbzr.sf.net')
3883
 
 
3884
 
    def test_for_matching_host_None(self):
3885
 
        conf = config.AuthenticationConfig(_file=StringIO(
3886
 
                """# Identity on foo.net
3887
 
[catchup bzr]
3888
 
scheme=bzr
3889
 
user=joe
3890
 
password=joepass
3891
 
[DEFAULT]
3892
 
user=georges
3893
 
password=bendover
3894
 
"""))
3895
 
        # match no host
3896
 
        self._got_user_passwd('joe', 'joepass',
3897
 
                              conf, 'bzr', 'quux.net')
3898
 
        # no host but different scheme
3899
 
        self._got_user_passwd('georges', 'bendover',
3900
 
                              conf, 'ftp', 'quux.net')
3901
 
 
3902
 
    def test_credentials_for_path(self):
3903
 
        conf = config.AuthenticationConfig(_file=StringIO(
3904
 
                """
3905
 
[http dir1]
3906
 
scheme=http
3907
 
host=bar.org
3908
 
path=/dir1
3909
 
user=jim
3910
 
password=jimpass
3911
 
[http dir2]
3912
 
scheme=http
3913
 
host=bar.org
3914
 
path=/dir2
3915
 
user=georges
3916
 
password=bendover
3917
 
"""))
3918
 
        # no path no dice
3919
 
        self._got_user_passwd(None, None,
3920
 
                              conf, 'http', host='bar.org', path='/dir3')
3921
 
        # matching path
3922
 
        self._got_user_passwd('georges', 'bendover',
3923
 
                              conf, 'http', host='bar.org', path='/dir2')
3924
 
        # matching subdir
3925
 
        self._got_user_passwd('jim', 'jimpass',
3926
 
                              conf, 'http', host='bar.org',path='/dir1/subdir')
3927
 
 
3928
 
    def test_credentials_for_user(self):
3929
 
        conf = config.AuthenticationConfig(_file=StringIO(
3930
 
                """
3931
 
[with user]
3932
 
scheme=http
3933
 
host=bar.org
3934
 
user=jim
3935
 
password=jimpass
3936
 
"""))
3937
 
        # Get user
3938
 
        self._got_user_passwd('jim', 'jimpass',
3939
 
                              conf, 'http', 'bar.org')
3940
 
        # Get same user
3941
 
        self._got_user_passwd('jim', 'jimpass',
3942
 
                              conf, 'http', 'bar.org', user='jim')
3943
 
        # Don't get a different user if one is specified
3944
 
        self._got_user_passwd(None, None,
3945
 
                              conf, 'http', 'bar.org', user='georges')
3946
 
 
3947
 
    def test_credentials_for_user_without_password(self):
3948
 
        conf = config.AuthenticationConfig(_file=StringIO(
3949
 
                """
3950
 
[without password]
3951
 
scheme=http
3952
 
host=bar.org
3953
 
user=jim
3954
 
"""))
3955
 
        # Get user but no password
3956
 
        self._got_user_passwd('jim', None,
3957
 
                              conf, 'http', 'bar.org')
3958
 
 
3959
 
    def test_verify_certificates(self):
3960
 
        conf = config.AuthenticationConfig(_file=StringIO(
3961
 
                """
3962
 
[self-signed]
3963
 
scheme=https
3964
 
host=bar.org
3965
 
user=jim
3966
 
password=jimpass
3967
 
verify_certificates=False
3968
 
[normal]
3969
 
scheme=https
3970
 
host=foo.net
3971
 
user=georges
3972
 
password=bendover
3973
 
"""))
3974
 
        credentials = conf.get_credentials('https', 'bar.org')
3975
 
        self.assertEquals(False, credentials.get('verify_certificates'))
3976
 
        credentials = conf.get_credentials('https', 'foo.net')
3977
 
        self.assertEquals(True, credentials.get('verify_certificates'))
3978
 
 
3979
 
 
3980
 
class TestAuthenticationStorage(tests.TestCaseInTempDir):
3981
 
 
3982
 
    def test_set_credentials(self):
3983
 
        conf = config.AuthenticationConfig()
3984
 
        conf.set_credentials('name', 'host', 'user', 'scheme', 'password',
3985
 
        99, path='/foo', verify_certificates=False, realm='realm')
3986
 
        credentials = conf.get_credentials(host='host', scheme='scheme',
3987
 
                                           port=99, path='/foo',
3988
 
                                           realm='realm')
3989
 
        CREDENTIALS = {'name': 'name', 'user': 'user', 'password': 'password',
3990
 
                       'verify_certificates': False, 'scheme': 'scheme', 
3991
 
                       'host': 'host', 'port': 99, 'path': '/foo', 
3992
 
                       'realm': 'realm'}
3993
 
        self.assertEqual(CREDENTIALS, credentials)
3994
 
        credentials_from_disk = config.AuthenticationConfig().get_credentials(
3995
 
            host='host', scheme='scheme', port=99, path='/foo', realm='realm')
3996
 
        self.assertEqual(CREDENTIALS, credentials_from_disk)
3997
 
 
3998
 
    def test_reset_credentials_different_name(self):
3999
 
        conf = config.AuthenticationConfig()
4000
 
        conf.set_credentials('name', 'host', 'user', 'scheme', 'password'),
4001
 
        conf.set_credentials('name2', 'host', 'user2', 'scheme', 'password'),
4002
 
        self.assertIs(None, conf._get_config().get('name'))
4003
 
        credentials = conf.get_credentials(host='host', scheme='scheme')
4004
 
        CREDENTIALS = {'name': 'name2', 'user': 'user2', 'password':
4005
 
                       'password', 'verify_certificates': True, 
4006
 
                       'scheme': 'scheme', 'host': 'host', 'port': None, 
4007
 
                       'path': None, 'realm': None}
4008
 
        self.assertEqual(CREDENTIALS, credentials)
4009
 
 
4010
 
 
4011
 
class TestAuthenticationConfig(tests.TestCase):
4012
 
    """Test AuthenticationConfig behaviour"""
4013
 
 
4014
 
    def _check_default_password_prompt(self, expected_prompt_format, scheme,
4015
 
                                       host=None, port=None, realm=None,
4016
 
                                       path=None):
4017
 
        if host is None:
4018
 
            host = 'bar.org'
4019
 
        user, password = 'jim', 'precious'
4020
 
        expected_prompt = expected_prompt_format % {
4021
 
            'scheme': scheme, 'host': host, 'port': port,
4022
 
            'user': user, 'realm': realm}
4023
 
 
4024
 
        stdout = tests.StringIOWrapper()
4025
 
        stderr = tests.StringIOWrapper()
4026
 
        ui.ui_factory = tests.TestUIFactory(stdin=password + '\n',
4027
 
                                            stdout=stdout, stderr=stderr)
4028
 
        # We use an empty conf so that the user is always prompted
4029
 
        conf = config.AuthenticationConfig()
4030
 
        self.assertEquals(password,
4031
 
                          conf.get_password(scheme, host, user, port=port,
4032
 
                                            realm=realm, path=path))
4033
 
        self.assertEquals(expected_prompt, stderr.getvalue())
4034
 
        self.assertEquals('', stdout.getvalue())
4035
 
 
4036
 
    def _check_default_username_prompt(self, expected_prompt_format, scheme,
4037
 
                                       host=None, port=None, realm=None,
4038
 
                                       path=None):
4039
 
        if host is None:
4040
 
            host = 'bar.org'
4041
 
        username = 'jim'
4042
 
        expected_prompt = expected_prompt_format % {
4043
 
            'scheme': scheme, 'host': host, 'port': port,
4044
 
            'realm': realm}
4045
 
        stdout = tests.StringIOWrapper()
4046
 
        stderr = tests.StringIOWrapper()
4047
 
        ui.ui_factory = tests.TestUIFactory(stdin=username+ '\n',
4048
 
                                            stdout=stdout, stderr=stderr)
4049
 
        # We use an empty conf so that the user is always prompted
4050
 
        conf = config.AuthenticationConfig()
4051
 
        self.assertEquals(username, conf.get_user(scheme, host, port=port,
4052
 
                          realm=realm, path=path, ask=True))
4053
 
        self.assertEquals(expected_prompt, stderr.getvalue())
4054
 
        self.assertEquals('', stdout.getvalue())
4055
 
 
4056
 
    def test_username_defaults_prompts(self):
4057
 
        # HTTP prompts can't be tested here, see test_http.py
4058
 
        self._check_default_username_prompt(u'FTP %(host)s username: ', 'ftp')
4059
 
        self._check_default_username_prompt(
4060
 
            u'FTP %(host)s:%(port)d username: ', 'ftp', port=10020)
4061
 
        self._check_default_username_prompt(
4062
 
            u'SSH %(host)s:%(port)d username: ', 'ssh', port=12345)
4063
 
 
4064
 
    def test_username_default_no_prompt(self):
4065
 
        conf = config.AuthenticationConfig()
4066
 
        self.assertEquals(None,
4067
 
            conf.get_user('ftp', 'example.com'))
4068
 
        self.assertEquals("explicitdefault",
4069
 
            conf.get_user('ftp', 'example.com', default="explicitdefault"))
4070
 
 
4071
 
    def test_password_default_prompts(self):
4072
 
        # HTTP prompts can't be tested here, see test_http.py
4073
 
        self._check_default_password_prompt(
4074
 
            u'FTP %(user)s@%(host)s password: ', 'ftp')
4075
 
        self._check_default_password_prompt(
4076
 
            u'FTP %(user)s@%(host)s:%(port)d password: ', 'ftp', port=10020)
4077
 
        self._check_default_password_prompt(
4078
 
            u'SSH %(user)s@%(host)s:%(port)d password: ', 'ssh', port=12345)
4079
 
        # SMTP port handling is a bit special (it's handled if embedded in the
4080
 
        # host too)
4081
 
        # FIXME: should we: forbid that, extend it to other schemes, leave
4082
 
        # things as they are that's fine thank you ?
4083
 
        self._check_default_password_prompt(
4084
 
            u'SMTP %(user)s@%(host)s password: ', 'smtp')
4085
 
        self._check_default_password_prompt(
4086
 
            u'SMTP %(user)s@%(host)s password: ', 'smtp', host='bar.org:10025')
4087
 
        self._check_default_password_prompt(
4088
 
            u'SMTP %(user)s@%(host)s:%(port)d password: ', 'smtp', port=10025)
4089
 
 
4090
 
    def test_ssh_password_emits_warning(self):
4091
 
        conf = config.AuthenticationConfig(_file=StringIO(
4092
 
                """
4093
 
[ssh with password]
4094
 
scheme=ssh
4095
 
host=bar.org
4096
 
user=jim
4097
 
password=jimpass
4098
 
"""))
4099
 
        entered_password = 'typed-by-hand'
4100
 
        stdout = tests.StringIOWrapper()
4101
 
        stderr = tests.StringIOWrapper()
4102
 
        ui.ui_factory = tests.TestUIFactory(stdin=entered_password + '\n',
4103
 
                                            stdout=stdout, stderr=stderr)
4104
 
 
4105
 
        # Since the password defined in the authentication config is ignored,
4106
 
        # the user is prompted
4107
 
        self.assertEquals(entered_password,
4108
 
                          conf.get_password('ssh', 'bar.org', user='jim'))
4109
 
        self.assertContainsRe(
4110
 
            self.get_log(),
4111
 
            'password ignored in section \[ssh with password\]')
4112
 
 
4113
 
    def test_ssh_without_password_doesnt_emit_warning(self):
4114
 
        conf = config.AuthenticationConfig(_file=StringIO(
4115
 
                """
4116
 
[ssh with password]
4117
 
scheme=ssh
4118
 
host=bar.org
4119
 
user=jim
4120
 
"""))
4121
 
        entered_password = 'typed-by-hand'
4122
 
        stdout = tests.StringIOWrapper()
4123
 
        stderr = tests.StringIOWrapper()
4124
 
        ui.ui_factory = tests.TestUIFactory(stdin=entered_password + '\n',
4125
 
                                            stdout=stdout,
4126
 
                                            stderr=stderr)
4127
 
 
4128
 
        # Since the password defined in the authentication config is ignored,
4129
 
        # the user is prompted
4130
 
        self.assertEquals(entered_password,
4131
 
                          conf.get_password('ssh', 'bar.org', user='jim'))
4132
 
        # No warning shoud be emitted since there is no password. We are only
4133
 
        # providing "user".
4134
 
        self.assertNotContainsRe(
4135
 
            self.get_log(),
4136
 
            'password ignored in section \[ssh with password\]')
4137
 
 
4138
 
    def test_uses_fallback_stores(self):
4139
 
        self.overrideAttr(config, 'credential_store_registry',
4140
 
                          config.CredentialStoreRegistry())
4141
 
        store = StubCredentialStore()
4142
 
        store.add_credentials("http", "example.com", "joe", "secret")
4143
 
        config.credential_store_registry.register("stub", store, fallback=True)
4144
 
        conf = config.AuthenticationConfig(_file=StringIO())
4145
 
        creds = conf.get_credentials("http", "example.com")
4146
 
        self.assertEquals("joe", creds["user"])
4147
 
        self.assertEquals("secret", creds["password"])
4148
 
 
4149
 
 
4150
 
class StubCredentialStore(config.CredentialStore):
4151
 
 
4152
 
    def __init__(self):
4153
 
        self._username = {}
4154
 
        self._password = {}
4155
 
 
4156
 
    def add_credentials(self, scheme, host, user, password=None):
4157
 
        self._username[(scheme, host)] = user
4158
 
        self._password[(scheme, host)] = password
4159
 
 
4160
 
    def get_credentials(self, scheme, host, port=None, user=None,
4161
 
        path=None, realm=None):
4162
 
        key = (scheme, host)
4163
 
        if not key in self._username:
4164
 
            return None
4165
 
        return { "scheme": scheme, "host": host, "port": port,
4166
 
                "user": self._username[key], "password": self._password[key]}
4167
 
 
4168
 
 
4169
 
class CountingCredentialStore(config.CredentialStore):
4170
 
 
4171
 
    def __init__(self):
4172
 
        self._calls = 0
4173
 
 
4174
 
    def get_credentials(self, scheme, host, port=None, user=None,
4175
 
        path=None, realm=None):
4176
 
        self._calls += 1
4177
 
        return None
4178
 
 
4179
 
 
4180
 
class TestCredentialStoreRegistry(tests.TestCase):
4181
 
 
4182
 
    def _get_cs_registry(self):
4183
 
        return config.credential_store_registry
4184
 
 
4185
 
    def test_default_credential_store(self):
4186
 
        r = self._get_cs_registry()
4187
 
        default = r.get_credential_store(None)
4188
 
        self.assertIsInstance(default, config.PlainTextCredentialStore)
4189
 
 
4190
 
    def test_unknown_credential_store(self):
4191
 
        r = self._get_cs_registry()
4192
 
        # It's hard to imagine someone creating a credential store named
4193
 
        # 'unknown' so we use that as an never registered key.
4194
 
        self.assertRaises(KeyError, r.get_credential_store, 'unknown')
4195
 
 
4196
 
    def test_fallback_none_registered(self):
4197
 
        r = config.CredentialStoreRegistry()
4198
 
        self.assertEquals(None,
4199
 
                          r.get_fallback_credentials("http", "example.com"))
4200
 
 
4201
 
    def test_register(self):
4202
 
        r = config.CredentialStoreRegistry()
4203
 
        r.register("stub", StubCredentialStore(), fallback=False)
4204
 
        r.register("another", StubCredentialStore(), fallback=True)
4205
 
        self.assertEquals(["another", "stub"], r.keys())
4206
 
 
4207
 
    def test_register_lazy(self):
4208
 
        r = config.CredentialStoreRegistry()
4209
 
        r.register_lazy("stub", "bzrlib.tests.test_config",
4210
 
                        "StubCredentialStore", fallback=False)
4211
 
        self.assertEquals(["stub"], r.keys())
4212
 
        self.assertIsInstance(r.get_credential_store("stub"),
4213
 
                              StubCredentialStore)
4214
 
 
4215
 
    def test_is_fallback(self):
4216
 
        r = config.CredentialStoreRegistry()
4217
 
        r.register("stub1", None, fallback=False)
4218
 
        r.register("stub2", None, fallback=True)
4219
 
        self.assertEquals(False, r.is_fallback("stub1"))
4220
 
        self.assertEquals(True, r.is_fallback("stub2"))
4221
 
 
4222
 
    def test_no_fallback(self):
4223
 
        r = config.CredentialStoreRegistry()
4224
 
        store = CountingCredentialStore()
4225
 
        r.register("count", store, fallback=False)
4226
 
        self.assertEquals(None,
4227
 
                          r.get_fallback_credentials("http", "example.com"))
4228
 
        self.assertEquals(0, store._calls)
4229
 
 
4230
 
    def test_fallback_credentials(self):
4231
 
        r = config.CredentialStoreRegistry()
4232
 
        store = StubCredentialStore()
4233
 
        store.add_credentials("http", "example.com",
4234
 
                              "somebody", "geheim")
4235
 
        r.register("stub", store, fallback=True)
4236
 
        creds = r.get_fallback_credentials("http", "example.com")
4237
 
        self.assertEquals("somebody", creds["user"])
4238
 
        self.assertEquals("geheim", creds["password"])
4239
 
 
4240
 
    def test_fallback_first_wins(self):
4241
 
        r = config.CredentialStoreRegistry()
4242
 
        stub1 = StubCredentialStore()
4243
 
        stub1.add_credentials("http", "example.com",
4244
 
                              "somebody", "stub1")
4245
 
        r.register("stub1", stub1, fallback=True)
4246
 
        stub2 = StubCredentialStore()
4247
 
        stub2.add_credentials("http", "example.com",
4248
 
                              "somebody", "stub2")
4249
 
        r.register("stub2", stub1, fallback=True)
4250
 
        creds = r.get_fallback_credentials("http", "example.com")
4251
 
        self.assertEquals("somebody", creds["user"])
4252
 
        self.assertEquals("stub1", creds["password"])
4253
 
 
4254
 
 
4255
 
class TestPlainTextCredentialStore(tests.TestCase):
4256
 
 
4257
 
    def test_decode_password(self):
4258
 
        r = config.credential_store_registry
4259
 
        plain_text = r.get_credential_store()
4260
 
        decoded = plain_text.decode_password(dict(password='secret'))
4261
 
        self.assertEquals('secret', decoded)
4262
 
 
4263
 
 
4264
 
# FIXME: Once we have a way to declare authentication to all test servers, we
4265
 
# can implement generic tests.
4266
 
# test_user_password_in_url
4267
 
# test_user_in_url_password_from_config
4268
 
# test_user_in_url_password_prompted
4269
 
# test_user_in_config
4270
 
# test_user_getpass.getuser
4271
 
# test_user_prompted ?
4272
 
class TestAuthenticationRing(tests.TestCaseWithTransport):
4273
 
    pass
4274
 
 
4275
 
 
4276
 
class TestAutoUserId(tests.TestCase):
4277
 
    """Test inferring an automatic user name."""
4278
 
 
4279
 
    def test_auto_user_id(self):
4280
 
        """Automatic inference of user name.
4281
 
        
4282
 
        This is a bit hard to test in an isolated way, because it depends on
4283
 
        system functions that go direct to /etc or perhaps somewhere else.
4284
 
        But it's reasonable to say that on Unix, with an /etc/mailname, we ought
4285
 
        to be able to choose a user name with no configuration.
4286
 
        """
4287
 
        if sys.platform == 'win32':
4288
 
            raise tests.TestSkipped(
4289
 
                "User name inference not implemented on win32")
4290
 
        realname, address = config._auto_user_id()
4291
 
        if os.path.exists('/etc/mailname'):
4292
 
            self.assertIsNot(None, realname)
4293
 
            self.assertIsNot(None, address)
4294
 
        else:
4295
 
            self.assertEquals((None, None), (realname, address))
4296