def test_build_ssh_config(self):
contents = """
[aaa]
hostname = 1.2.3.4
user = toto
[bbb]
inherits = aaa
port = 23
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='test',
configfiles=[DEFAULT_CONFIG])
config = advssh.prepare_sshconfig()
arr = advssh.build_sshconfig()
string = '\n'.join(arr)
self.assertEquals(len(arr), 9)
dest = """
Host aaa
user toto
# hostname 1.2.3.4
Host bbb
port 23
user toto
# inherits aaa
"""
self.assertEquals(string.strip(), dest.strip())
def test_build_ssh_config_with_multiline_localforward(self):
contents = """
[localhost]
user = toto
localforward = 1 2.3.4.5 6
7 8.9.10.11 12
port = 22
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='localhost',
configfiles=[DEFAULT_CONFIG])
config = advssh.prepare_sshconfig()
arr = advssh.build_sshconfig()
string = '\n'.join(arr)
self.assertEquals(len(arr), 11)
dest = """
# assh version: {}
Host localhost
localforward 1 2.3.4.5 6
localforward 7 8.9.10.11 12
port 22
user toto
Host *
proxycommand assh connect %h --port=%p
""".format(__version__)
self.assertEquals(string.strip(), dest.strip())
def test_build_ssh_config(self):
contents = """
[aaa]
hostname = 1.2.3.4
user = toto
[bbb]
inherits = aaa
port = 23
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='test',
configfiles=[DEFAULT_CONFIG])
config = advssh.prepare_sshconfig()
arr = advssh.build_sshconfig()
string = '\n'.join(arr)
self.assertEquals(len(arr), 14)
dest = """
# assh version: {}
Host aaa
user toto
# hostname 1.2.3.4
Host bbb
port 23
user toto
# inherits aaa
Host *
proxycommand assh connect %h --port=%p
""".format(__version__)
self.assertEquals(string.strip(), dest.strip())
def test_routing_via_two_other_hosts(self):
advssh = AdvancedSshConfig(hostname='aaa.com/bbb.com/ccc.com')
routing = advssh.get_routing()
self.assertEqual(routing['hostname'], 'aaa.com')
self.assertEqual(routing['proxy_type'], 'nc')
self.assertEqual(routing['gateways'], ['direct'])
self.assertEqual(routing['proxy_commands'][0],
['nc', '-w', 180, '-G', 5, 'aaa.com', 22])
self.assertEqual(routing['gateway_route'], ['bbb.com', 'ccc.com'])
def test_build_ssh_config_sorted(self):
contents = """
[ddd]
inherits = aaa
port = 23
user = titi
[bbb]
user = titi
inherits = aaa
port = 23
hostname = 1.1.1.1
[ccc]
hostname = 5.4.3.2
inherits = aaa
port = 23
[aaa]
hostname = 1.2.3.4
user = toto
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='test',
configfiles=[DEFAULT_CONFIG])
config = advssh.prepare_sshconfig()
arr = advssh.build_sshconfig()
string = '\n'.join(arr)
dest = """
# assh version: {}
Host aaa
user toto
# hostname 1.2.3.4
Host bbb
port 23
user titi
# hostname 1.1.1.1
# inherits aaa
Host ccc
port 23
user toto
# hostname 5.4.3.2
# inherits aaa
Host ddd
port 23
user titi
# inherits aaa
Host *
proxycommand assh connect %h --port=%p
""".format(__version__)
self.assertEquals(string.strip(), dest.strip())
def test_comment_simple(self):
contents = """
[test]
comment = Hello
"""
config = set_config(contents)
advssh = AdvancedSshConfig(hostname='test',
configfiles=[DEFAULT_CONFIG])
routing = advssh.get_routing()
self.assertEquals(routing['comment'], ['Hello'])
def test_prepare_sshconfig_multiline(self):
contents = """
[test]
localforward = 1 2.3.4.5 6 \n 7 8.9.10.11 12
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='test',
configfiles=[DEFAULT_CONFIG])
config = advssh.prepare_sshconfig()
self.assertEqual(config['test'].host, 'test')
self.assertEqual(config['test'].config,
[('localforward', '1 2.3.4.5 6'),
('localforward', '7 8.9.10.11 12')])
def test_comment_multiline(self):
contents = """
[test]
comment = Hello
World
!
port = 22
"""
config = set_config(contents)
advssh = AdvancedSshConfig(hostname='test',
configfiles=[DEFAULT_CONFIG])
routing = advssh.get_routing()
self.assertEquals(routing['comment'], ['Hello', 'World', '!'])
def test_routing_simple(self):
advssh = AdvancedSshConfig(hostname='test',
port=23,
verbose=True,
dry_run=True)
routing = advssh.get_routing()
self.assertEqual(routing['port'], 23)
self.assertEqual(routing['hostname'], 'test')
self.assertEqual(routing['reallocalcommand'], None)
self.assertEqual(routing['gateways'], ['direct'])
self.assertEqual(routing['verbose'], True)
self.assertEqual(routing['proxy_type'], 'nc')
self.assertEqual(routing['proxy_commands'][0],
['nc', '-v', '-w', 180, '-G', 5, 'test', 23])
def test_routing_via_two_other_hosts_with_config_one(self):
contents = """
[ddd.com]
hostname = 1.2.3.4
port = 25
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='ddd.com/eee.com',
configfiles=[DEFAULT_CONFIG])
routing = advssh.get_routing()
self.assertEqual(routing['hostname'], '1.2.3.4')
self.assertEqual(routing['proxy_type'], 'nc')
self.assertEqual(routing['gateways'], ['direct'])
self.assertEqual(routing['proxy_commands'][0],
['nc', '-w', 180, '-G', 5, '1.2.3.4', 25])
self.assertEqual(routing['gateway_route'], ['eee.com'])
def test_prepare_sshconfig_simple(self):
contents = """
[test]
port = 25
[default]
port = 24
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='test',
configfiles=[DEFAULT_CONFIG])
config = advssh.prepare_sshconfig()
self.assertEqual(len(config.keys()), 2)
self.assertEqual(config['test'].host, 'test')
self.assertEqual(config['test'].config, [('port', '25')])
self.assertEqual(config['default'].host, 'default')
self.assertEqual(config['default'].config, [('port', '24')])
def test_routing_config_override(self):
contents = """
[test.com]
port = 25
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='test.com',
port=23,
verbose=True,
dry_run=True,
configfiles=[DEFAULT_CONFIG])
routing = advssh.get_routing()
self.assertEqual(routing['port'], 23)
self.assertEqual(routing['hostname'], 'test.com')
self.assertEqual(routing['proxy_type'], 'nc')
self.assertEqual(routing['proxy_commands'][0],
['nc', '-v', '-w', 180, '-G', 5, 'test.com', 23])
def test_build_ssh_config_with_multiline_comment(self):
contents = """
[localhost]
port = 22
comment = .
. O
. _______O_
. / O \\
. / _ _ O _ _ \\
. | _ |
. | o | | o |
. |___|_|___|
.
user = toto
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='localhost',
configfiles=[DEFAULT_CONFIG])
config = advssh.prepare_sshconfig()
arr = advssh.build_sshconfig()
string = '\n'.join(arr)
dest = """
# assh version: {}
Host localhost
port 22
user toto
# comment .
# comment . O
# comment . _______O_
# comment . / O \\
# comment . / _ _ O _ _ \\
# comment . | _ |
# comment . | o | | o |
# comment . |___|_|___|
# comment .
Host *
proxycommand assh connect %h --port=%p
""".format(__version__)
self.assertEquals(string.strip(), dest.strip())
def test_build_ssh_config_with_multiline_localforward_onliner(self):
contents = """
[localhost]
user = toto
localforward = 1 2.3.4.5 6 \n 7 8.9.10.11 12
port = 22
"""
set_config(contents)
advssh = AdvancedSshConfig(hostname='localhost',
configfiles=[DEFAULT_CONFIG])
config = advssh.prepare_sshconfig()
arr = advssh.build_sshconfig()
string = '\n'.join(arr)
self.assertEquals(len(arr), 6)
dest = """
Host localhost
localforward 1 2.3.4.5 6
localforward 7 8.9.10.11 12
port 22
user toto
"""
self.assertEquals(string.strip(), dest.strip())
