mirror of
https://github.com/ceph/ceph
synced 2024-12-28 22:43:29 +00:00
105 lines
3.7 KiB
Python
105 lines
3.7 KiB
Python
from teuthology import config
|
|
|
|
import fudge
|
|
|
|
from .util import assert_raises
|
|
|
|
from .. import connection
|
|
|
|
|
|
class TestConnection(object):
|
|
def clear_config(self):
|
|
config.config.teuthology_yaml = ''
|
|
config.config.load_files()
|
|
|
|
def test_split_user_just_host(self):
|
|
got = connection.split_user('somehost.example.com')
|
|
assert got == (None, 'somehost.example.com')
|
|
|
|
def test_split_user_both(self):
|
|
got = connection.split_user('jdoe@somehost.example.com')
|
|
assert got == ('jdoe', 'somehost.example.com')
|
|
|
|
def test_split_user_empty_user(self):
|
|
s = '@somehost.example.com'
|
|
e = assert_raises(AssertionError, connection.split_user, s)
|
|
assert str(e) == 'Bad input to split_user: {s!r}'.format(s=s)
|
|
|
|
@fudge.with_fakes
|
|
def test_connect(self):
|
|
self.clear_config()
|
|
config.config.verify_host_keys = True
|
|
fudge.clear_expectations()
|
|
ssh = fudge.Fake('SSHClient')
|
|
ssh.expects_call().with_args().returns(ssh)
|
|
ssh.expects('set_missing_host_key_policy')
|
|
ssh.expects('load_system_host_keys').with_args()
|
|
ssh.expects('connect').with_args(
|
|
hostname='orchestra.test.newdream.net.invalid',
|
|
username='jdoe',
|
|
timeout=60,
|
|
)
|
|
transport = ssh.expects('get_transport').with_args().returns_fake()
|
|
transport.remember_order()
|
|
transport.expects('set_keepalive').with_args(False)
|
|
got = connection.connect(
|
|
'jdoe@orchestra.test.newdream.net.invalid',
|
|
_SSHClient=ssh,
|
|
)
|
|
assert got is ssh
|
|
|
|
@fudge.with_fakes
|
|
def test_connect_no_verify_host_keys(self):
|
|
self.clear_config()
|
|
config.config.verify_host_keys = False
|
|
fudge.clear_expectations()
|
|
ssh = fudge.Fake('SSHClient')
|
|
ssh.expects_call().with_args().returns(ssh)
|
|
ssh.expects('set_missing_host_key_policy')
|
|
ssh.expects('connect').with_args(
|
|
hostname='orchestra.test.newdream.net.invalid',
|
|
username='jdoe',
|
|
timeout=60,
|
|
)
|
|
transport = ssh.expects('get_transport').with_args().returns_fake()
|
|
transport.remember_order()
|
|
transport.expects('set_keepalive').with_args(False)
|
|
got = connection.connect(
|
|
'jdoe@orchestra.test.newdream.net.invalid',
|
|
_SSHClient=ssh,
|
|
)
|
|
assert got is ssh
|
|
|
|
@fudge.with_fakes
|
|
def test_connect_override_hostkeys(self):
|
|
self.clear_config()
|
|
fudge.clear_expectations()
|
|
sshclient = fudge.Fake('SSHClient')
|
|
ssh = sshclient.expects_call().with_args().returns_fake()
|
|
ssh.remember_order()
|
|
host_keys = fudge.Fake('HostKeys')
|
|
host_keys.expects('add').with_args(
|
|
hostname='orchestra.test.newdream.net.invalid',
|
|
keytype='ssh-rsa',
|
|
key='frobnitz',
|
|
)
|
|
ssh.expects('get_host_keys').with_args().returns(host_keys)
|
|
ssh.expects('connect').with_args(
|
|
hostname='orchestra.test.newdream.net.invalid',
|
|
username='jdoe',
|
|
timeout=60,
|
|
)
|
|
transport = ssh.expects('get_transport').with_args().returns_fake()
|
|
transport.remember_order()
|
|
transport.expects('set_keepalive').with_args(False)
|
|
create_key = fudge.Fake('create_key')
|
|
create_key.expects_call().with_args('ssh-rsa',
|
|
'testkey').returns('frobnitz')
|
|
got = connection.connect(
|
|
'jdoe@orchestra.test.newdream.net.invalid',
|
|
host_key='ssh-rsa testkey',
|
|
_SSHClient=sshclient,
|
|
_create_key=create_key,
|
|
)
|
|
assert got is ssh
|