def test_key_filename_makes_new_key(self):
config = _Keys(client='test-client--test_key_filename')
cnx = MockConnection('key', 'secret')
cnx._key_pairs.append(_Keys(name='another-host'))
with mock.patch('socket.gethostname', lambda: 'test-host'):
dir_made = []
with mock.patch('os.mkdir', lambda p: dir_made.append(p)):
filename = get_private_key_filename(config, cnx)
self.assertTrue(dir_made)
self.assertIn('test-host', cnx._key_pairs_created)
cnx._key_pairs_created = []
filename = get_private_key_filename(config, cnx)
self.assertNotIn('test-host', cnx._key_pairs_created)
def wrapper(server, *args, **kwargs):
"""Wrapped method
"""
keyfile = get_private_key_filename(server.config, server.cnx)
with settings(host_string=server.eip or server.instance.dns_name,
user='******', key_filename=keyfile, connection_attempts=10):
function(server, *args, **kwargs)
def wrapper(server, *args, **kwargs):
"""Wrapped method
"""
keyfile = get_private_key_filename(server.config, server.cnx)
with settings(host_string=server.eip or server.instance.dns_name,
user='******', key_filename=keyfile, connection_attempts=10):
function(server, *args, **kwargs)
def test_key_filename(self):
config = _Keys(client='test-client--test_key_filename')
cnx = MockConnection('key', 'secret')
with mock.patch('socket.gethostname', lambda: 'test-host'):
dir_made = []
with mock.patch('os.mkdir', lambda p: dir_made.append(p)):
filename = get_private_key_filename(config, cnx)
self.assertTrue(dir_made)
