def test_write_key(self): tmp = self.tmp ssh.writeKey(tmp, 'harry', key1) ssh.writeKey(tmp, 'harry', key2) ssh.writeKey(tmp, 'harry', key3) f = open(tmp, 'r') self.assertEqual(k1, f.readline()) self.assertEqual(k2, f.readline()) self.assertEqual(k3, f.readline()) f.close()
def test_remove_key(self): tmp = self.tmp ssh.writeKey(tmp, 'harry', key1) ssh.writeKey(tmp, 'harry', key2) ssh.writeKey(tmp, 'harry', key3) f = open(tmp, 'r') self.assertEqual(f.readlines(), [k1, k2, k3]) f.close() ssh.removeKey(tmp, 'harry', key2) f = open(tmp, 'r') self.assertEqual(f.readlines(), [k1, k3]) f.close()