1
# Copyright (C) 2005, 2006, 2008, 2009 Canonical Ltd
1
# Copyright (C) 2005, 2006, 2008 Canonical Ltd
2
# Authors: Robert Collins <robert.collins@canonical.com>
3
4
# This program is free software; you can redistribute it and/or modify
4
5
# it under the terms of the GNU General Public License as published by
210
210
self._calls.append('_get_signature_checking')
211
211
return self._signatures
213
def _get_change_editor(self):
214
self._calls.append('_get_change_editor')
215
return 'vimdiff -fo @new_path @old_path'
218
214
bool_config = """[DEFAULT]
320
316
my_config = config.Config()
321
317
self.assertEqual('long', my_config.log_format())
323
def test_get_change_editor(self):
324
my_config = InstrumentedConfig()
325
change_editor = my_config.get_change_editor('old_tree', 'new_tree')
326
self.assertEqual(['_get_change_editor'], my_config._calls)
327
self.assertIs(diff.DiffFromTool, change_editor.__class__)
328
self.assertEqual(['vimdiff', '-fo', '@new_path', '@old_path'],
329
change_editor.command_template)
332
320
class TestConfigPath(tests.TestCase):
335
323
super(TestConfigPath, self).setUp()
336
324
os.environ['HOME'] = '/home/bogus'
337
os.environ['XDG_CACHE_DIR'] = ''
338
325
if sys.platform == 'win32':
339
326
os.environ['BZR_HOME'] = \
340
327
r'C:\Documents and Settings\bogus\Application Data'
362
349
self.assertEqual(config.authentication_config_filename(),
363
350
self.bzr_home + '/authentication.conf')
365
def test_xdg_cache_dir(self):
366
self.assertEqual(config.xdg_cache_dir(),
367
'/home/bogus/.cache')
370
353
class TestIniConfig(tests.TestCase):
372
def make_config_parser(self, s):
373
conf = config.IniBasedConfig(None)
374
parser = conf._get_parser(file=StringIO(s.encode('utf-8')))
378
class TestIniConfigBuilding(TestIniConfig):
380
355
def test_contructs(self):
381
356
my_config = config.IniBasedConfig("nothing")
394
369
self.failUnless(my_config._get_parser() is parser)
397
class TestGetUserOptionAs(TestIniConfig):
399
def test_get_user_option_as_bool(self):
400
conf, parser = self.make_config_parser("""
403
an_invalid_bool = maybe
404
a_list = hmm, who knows ? # This is interpreted as a list !
406
get_bool = conf.get_user_option_as_bool
407
self.assertEqual(True, get_bool('a_true_bool'))
408
self.assertEqual(False, get_bool('a_false_bool'))
409
self.assertIs(None, get_bool('an_invalid_bool'))
410
self.assertIs(None, get_bool('not_defined_in_this_config'))
413
def test_get_user_option_as_list(self):
414
conf, parser = self.make_config_parser("""
419
get_list = conf.get_user_option_as_list
420
self.assertEqual(['a', 'b', 'c'], get_list('a_list'))
421
self.assertEqual(['1'], get_list('length_1'))
422
self.assertEqual('x', conf.get_user_option('one_item'))
423
# automatically cast to list
424
self.assertEqual(['x'], get_list('one_item'))
427
class TestSupressWarning(TestIniConfig):
429
def make_warnings_config(self, s):
430
conf, parser = self.make_config_parser(s)
431
return conf.suppress_warning
433
def test_suppress_warning_unknown(self):
434
suppress_warning = self.make_warnings_config('')
435
self.assertEqual(False, suppress_warning('unknown_warning'))
437
def test_suppress_warning_known(self):
438
suppress_warning = self.make_warnings_config('suppress_warnings=a,b')
439
self.assertEqual(False, suppress_warning('c'))
440
self.assertEqual(True, suppress_warning('a'))
441
self.assertEqual(True, suppress_warning('b'))
444
372
class TestGetConfig(tests.TestCase):
446
374
def test_constructs(self):
680
608
my_config = self._get_sample_config()
681
609
self.assertEqual(sample_long_alias, my_config.get_alias('ll'))
683
def test_get_change_editor(self):
684
my_config = self._get_sample_config()
685
change_editor = my_config.get_change_editor('old', 'new')
686
self.assertIs(diff.DiffFromTool, change_editor.__class__)
687
self.assertEqual('vimdiff -of @new_path @old_path',
688
' '.join(change_editor.command_template))
690
def test_get_no_change_editor(self):
691
my_config = self._get_empty_config()
692
change_editor = my_config.get_change_editor('old', 'new')
693
self.assertIs(None, change_editor)
696
612
class TestGlobalConfigSavingOptions(tests.TestCaseInTempDir):
1545
1461
"""Test AuthenticationConfig behaviour"""
1547
1463
def _check_default_password_prompt(self, expected_prompt_format, scheme,
1548
host=None, port=None, realm=None,
1464
host=None, port=None, realm=None, path=None):
1550
1465
if host is None:
1551
1466
host = 'bar.org'
1552
1467
user, password = 'jim', 'precious'
1555
1470
'user': user, 'realm': realm}
1557
1472
stdout = tests.StringIOWrapper()
1558
stderr = tests.StringIOWrapper()
1559
1473
ui.ui_factory = tests.TestUIFactory(stdin=password + '\n',
1560
stdout=stdout, stderr=stderr)
1561
1475
# We use an empty conf so that the user is always prompted
1562
1476
conf = config.AuthenticationConfig()
1563
1477
self.assertEquals(password,
1564
1478
conf.get_password(scheme, host, user, port=port,
1565
1479
realm=realm, path=path))
1566
self.assertEquals(expected_prompt, stderr.getvalue())
1567
self.assertEquals('', stdout.getvalue())
1480
self.assertEquals(stdout.getvalue(), expected_prompt)
1569
1482
def _check_default_username_prompt(self, expected_prompt_format, scheme,
1570
host=None, port=None, realm=None,
1483
host=None, port=None, realm=None, path=None):
1572
1484
if host is None:
1573
1485
host = 'bar.org'
1574
1486
username = 'jim'
1576
1488
'scheme': scheme, 'host': host, 'port': port,
1577
1489
'realm': realm}
1578
1490
stdout = tests.StringIOWrapper()
1579
stderr = tests.StringIOWrapper()
1580
1491
ui.ui_factory = tests.TestUIFactory(stdin=username+ '\n',
1581
stdout=stdout, stderr=stderr)
1582
1493
# We use an empty conf so that the user is always prompted
1583
1494
conf = config.AuthenticationConfig()
1584
1495
self.assertEquals(username, conf.get_user(scheme, host, port=port,
1585
1496
realm=realm, path=path, ask=True))
1586
self.assertEquals(expected_prompt, stderr.getvalue())
1587
self.assertEquals('', stdout.getvalue())
1497
self.assertEquals(stdout.getvalue(), expected_prompt)
1589
1499
def test_username_defaults_prompts(self):
1590
1500
# HTTP prompts can't be tested here, see test_http.py
1633
1543
entered_password = 'typed-by-hand'
1634
1544
stdout = tests.StringIOWrapper()
1635
stderr = tests.StringIOWrapper()
1636
1545
ui.ui_factory = tests.TestUIFactory(stdin=entered_password + '\n',
1637
stdout=stdout, stderr=stderr)
1639
1548
# Since the password defined in the authentication config is ignored,
1640
1549
# the user is prompted
1641
1550
self.assertEquals(entered_password,
1642
1551
conf.get_password('ssh', 'bar.org', user='jim'))
1643
1552
self.assertContainsRe(
1553
self._get_log(keep_log_file=True),
1645
1554
'password ignored in section \[ssh with password\]')
1647
1556
def test_ssh_without_password_doesnt_emit_warning(self):
1655
1564
entered_password = 'typed-by-hand'
1656
1565
stdout = tests.StringIOWrapper()
1657
stderr = tests.StringIOWrapper()
1658
1566
ui.ui_factory = tests.TestUIFactory(stdin=entered_password + '\n',
1662
1569
# Since the password defined in the authentication config is ignored,
1663
1570
# the user is prompted
1666
1573
# No warning shoud be emitted since there is no password. We are only
1667
1574
# providing "user".
1668
1575
self.assertNotContainsRe(
1576
self._get_log(keep_log_file=True),
1670
1577
'password ignored in section \[ssh with password\]')
1672
1579
def test_uses_fallback_stores(self):
1673
self.overrideAttr(config, 'credential_store_registry',
1674
config.CredentialStoreRegistry())
1580
self._old_cs_registry = config.credential_store_registry
1582
config.credential_store_registry = self._old_cs_registry
1583
self.addCleanup(restore)
1584
config.credential_store_registry = config.CredentialStoreRegistry()
1675
1585
store = StubCredentialStore()
1676
1586
store.add_credentials("http", "example.com", "joe", "secret")
1677
1587
config.credential_store_registry.register("stub", store, fallback=True)