def test_authorized_keys_list_extend(fx_authorized_sftp):
sftp_client, path, keys = fx_authorized_sftp
key_list = AuthorizedKeyList(sftp_client)
new_keys = [RSAKey.generate(1024) for _ in range(3)]
key_list.extend(new_keys)
with path.join('.ssh', 'authorized_keys').open() as f:
for i in range(6):
assert parse_openssh_pubkey(f.readline().strip()) == keys[i]
for i in range(3):
assert parse_openssh_pubkey(f.readline().strip()) == new_keys[i]
assert not f.readline().strip()
def test_authorized_keys_list_extend(fx_authorized_sftp):
sftp_client, path, keys = fx_authorized_sftp
key_list = AuthorizedKeyList(sftp_client)
new_keys = [RSAKey.generate(1024) for _ in range(3)]
key_list.extend(new_keys)
with path.join('.ssh', 'authorized_keys').open() as f:
for i in range(6):
assert parse_openssh_pubkey(f.readline().strip()) == keys[i]
for i in range(3):
assert parse_openssh_pubkey(f.readline().strip()) == new_keys[i]
assert not f.readline().strip()
def test_authorized_keys_list_insert(fx_authorized_sftp):
sftp_client, path, keys = fx_authorized_sftp
key_list = AuthorizedKeyList(sftp_client)
new_key = RSAKey.generate(1024)
key_list.insert(2, new_key)
with path.join('.ssh', 'authorized_keys').open() as f:
assert parse_openssh_pubkey(f.readline().strip()) == keys[0]
assert parse_openssh_pubkey(f.readline().strip()) == keys[1]
assert parse_openssh_pubkey(f.readline().strip()) == new_key
for i in range(2, 6):
assert parse_openssh_pubkey(f.readline().strip()) == keys[i]
assert not f.readline().strip()
def test_authorized_keys_list_insert(fx_authorized_sftp):
sftp_client, path, keys = fx_authorized_sftp
key_list = AuthorizedKeyList(sftp_client)
new_key = RSAKey.generate(1024)
key_list.insert(2, new_key)
with path.join('.ssh', 'authorized_keys').open() as f:
assert parse_openssh_pubkey(f.readline().strip()) == keys[0]
assert parse_openssh_pubkey(f.readline().strip()) == keys[1]
assert parse_openssh_pubkey(f.readline().strip()) == new_key
for i in range(2, 6):
assert parse_openssh_pubkey(f.readline().strip()) == keys[i]
assert not f.readline().strip()
def test_authorized_keys_list_len(fx_authorized_sftp):
sftp_client, path, keys = fx_authorized_sftp
key_list = AuthorizedKeyList(sftp_client)
assert len(key_list) == 6
# It's lazily evaluated; changes should reflect
with path.join('.ssh', 'authorized_keys').open('w') as f:
f.write(format_openssh_pubkey(keys[0]))
assert len(key_list) == 1
def test_authorized_keys_list_iter(fx_authorized_sftp):
sftp_client, path, keys = fx_authorized_sftp
key_list = AuthorizedKeyList(sftp_client)
it = iter(key_list)
assert next(it) == keys[0]
assert next(it) == keys[1]
assert next(it) == keys[2]
assert next(it) == keys[3]
assert next(it) == keys[4]
assert next(it) == keys[5]
with raises(StopIteration):
next(it)
# It's lazily evaluated; changes should reflect
with path.join('.ssh', 'authorized_keys').open('w') as f:
f.write(format_openssh_pubkey(keys[0]))
it = iter(key_list)
assert next(it) == keys[0]
with raises(StopIteration):
next(it)
def test_authorized_keys_list_delitem(fx_authorized_sftp):
sftp_client, path, keys = fx_authorized_sftp
key_list = AuthorizedKeyList(sftp_client)
# Slice deletion
del key_list[3:]
with path.join('.ssh', 'authorized_keys').open() as f:
for i in range(3):
assert parse_openssh_pubkey(f.readline().strip()) == keys[i]
assert not f.readline().strip()
# Positive index
del key_list[2]
with path.join('.ssh', 'authorized_keys').open() as f:
assert parse_openssh_pubkey(f.readline().strip()) == keys[0]
assert parse_openssh_pubkey(f.readline().strip()) == keys[1]
assert not f.readline().strip()
# Negative index
del key_list[-1]
with path.join('.ssh', 'authorized_keys').open() as f:
assert parse_openssh_pubkey(f.readline().strip()) == keys[0]
assert not f.readline().strip()
def test_authorized_keys_list_getitem(fx_authorized_sftp):
sftp_client, path, keys = fx_authorized_sftp
key_list = AuthorizedKeyList(sftp_client)
for i in range(-6, 6):
assert key_list[i] == keys[i]
assert key_list[i:] == keys[i:]
assert key_list[:i] == keys[:i]
assert key_list[i:i + 3] == keys[i:i + 3]
with raises(IndexError):
assert key_list[-7]
with raises(IndexError):
assert key_list[6]
with raises(TypeError):
key_list['key']
# It's lazily evaluated; changes should reflect
with path.join('.ssh', 'authorized_keys').open('w') as f:
f.write(format_openssh_pubkey(keys[0]))
assert key_list[0] == key_list[-1] == keys[0]
with raises(IndexError):
key_list[1]
with raises(IndexError):
key_list[-2]
def test_authorized_keys_list_setitem(fx_authorized_sftp):
sftp_client, path, keys = fx_authorized_sftp
key_list = AuthorizedKeyList(sftp_client)
# Slice assignment
key_list[3:] = []
with path.join('.ssh', 'authorized_keys').open() as f:
for i in range(3):
assert parse_openssh_pubkey(get_next_line(f)) == keys[i]
assert not get_next_line(f)
# Positive index
key_list[2] = keys[3]
with path.join('.ssh', 'authorized_keys').open() as f:
assert parse_openssh_pubkey(get_next_line(f)) == keys[0]
assert parse_openssh_pubkey(get_next_line(f)) == keys[1]
assert parse_openssh_pubkey(get_next_line(f)) == keys[3]
assert not get_next_line(f)
# Negative index
key_list[-1] = keys[4]
with path.join('.ssh', 'authorized_keys').open() as f:
assert parse_openssh_pubkey(get_next_line(f)) == keys[0]
assert parse_openssh_pubkey(get_next_line(f)) == keys[1]
assert parse_openssh_pubkey(get_next_line(f)) == keys[4]
assert not get_next_line(f)
