def test_replace_host_key_no_known_hosts(self):
"""Testing FileSSHStorage.replace_host_key with no known hosts file"""
storage = FileSSHStorage()
storage.replace_host_key('host1', self.key1, self.key2)
filename = storage.get_host_keys_filename()
with open(filename, 'r') as fp:
lines = fp.readlines()
self.assertEqual(len(lines), 1)
self.assertEqual(lines[0], 'host1 ssh-dss %s\n' % self.key2_b64)
def test_replace_host_key_no_known_hosts(self):
"""Testing FileSSHStorage.replace_host_key with no known hosts file"""
storage = FileSSHStorage()
storage.replace_host_key('host1', self.key1, self.key2)
filename = storage.get_host_keys_filename()
with open(filename, 'r') as fp:
lines = fp.readlines()
self.assertEqual(len(lines), 1)
self.assertEqual(lines[0], 'host1 ssh-dss %s\n' % self.key2_b64)
def test_replace_host_key(self):
"""Testing FileSSHStorage.replace_host_key"""
storage = FileSSHStorage()
storage.add_host_key('host1', self.key1)
storage.replace_host_key('host1', self.key1, self.key2)
filename = storage.get_host_keys_filename()
fp = open(filename, 'r')
lines = fp.readlines()
fp.close()
self.assertEqual(len(lines), 1)
self.assertEqual(lines[0], 'host1 ssh-dss %s\n' % self.key2_b64)
def test_replace_host_key(self):
"""Testing FileSSHStorage.replace_host_key"""
storage = FileSSHStorage()
storage.add_host_key('host1', self.key1)
storage.replace_host_key('host1', self.key1, self.key2)
filename = storage.get_host_keys_filename()
fp = open(filename, 'r')
lines = fp.readlines()
fp.close()
self.assertEqual(len(lines), 1)
self.assertEqual(lines[0], 'host1 ssh-dss %s\n' % self.key2_b64)
def test_replace_host_key(self):
"""Testing FileSSHStorage.replace_host_key"""
storage = FileSSHStorage()
storage.add_host_key('host1', test_keys.rsa_key)
storage.replace_host_key('host1', test_keys.rsa_key, test_keys.dsa_key)
filename = storage.get_host_keys_filename()
with open(filename, 'r') as fp:
lines = fp.readlines()
self.assertEqual(len(lines), 1)
self.assertEqual(lines[0],
'host1 ssh-dss %s\n' % test_keys.dsa_key_b64)
