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_add_host_key(self): """Testing FileSSHStorage.add_host_key""" storage = FileSSHStorage() storage.add_host_key('host1', self.key1) 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-rsa %s\n' % self.key1_b64)
def test_add_host_key(self): """Testing FileSSHStorage.add_host_key""" storage = FileSSHStorage() storage.add_host_key('host1', self.key1) 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-rsa %s\n' % self.key1_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)
def test_read_host_keys(self): """Testing FileSSHStorage.read_host_keys""" storage = FileSSHStorage() storage.ensure_ssh_dir() line1 = 'host1 ssh-rsa %s' % self.key1_b64 line2 = 'host2 ssh-dss %s' % self.key2_b64 filename = storage.get_host_keys_filename() with open(filename, 'w') as fp: fp.write('%s\n' % line1) fp.write('\n') fp.write('# foo\n') fp.write('%s \n' % line2) lines = storage.read_host_keys() self.assertEqual(len(lines), 2) self.assertEqual(lines[0], line1) self.assertEqual(lines[1], line2)
def test_read_host_keys(self): """Testing FileSSHStorage.read_host_keys""" storage = FileSSHStorage() storage.ensure_ssh_dir() line1 = 'host1 ssh-rsa %s' % self.key1_b64 line2 = 'host2 ssh-dss %s' % self.key2_b64 filename = storage.get_host_keys_filename() with open(filename, 'w') as fp: fp.write('%s\n' % line1) fp.write('\n') fp.write('# foo\n') fp.write('%s \n' % line2) lines = storage.read_host_keys() self.assertEqual(len(lines), 2) self.assertEqual(lines[0], line1) self.assertEqual(lines[1], line2)