def test_list_remove():
key = "begin_rsa 1"
mock_file = MagicMock()
mock_file.__str__ = lambda x: key
mock_file.exists.return_value = True
mock_path = MagicMock(return_value=mock_file)
mock_hashlib = MagicMock()
mock_hashlib.md5.hexdigest.return_value = "HASH"
mock_user = MagicMock()
mock_user.path = "path"
mock_user.name = "test"
with patch.multiple('pyolite.models.lists.keys', Path=mock_path,
hashlib=mock_hashlib):
keys = ListKeys(mock_user)
keys.remove(key)
mock_path.has_calls([
call("path", 'keydir', 'HASH'),
call(mock_file, "test.pub"),
])
commit_message = "Removed key for user test"
mock_user.git.commit.has_calls([call(["my_awesome_key"],
commit_message,
action='remove')])
def test_list_remove():
key = "begin_rsa 1"
mock_file = MagicMock()
mock_file.__str__ = lambda x: key
mock_file.exists.return_value = True
mock_path = MagicMock(return_value=mock_file)
mock_hashlib = MagicMock()
mock_hashlib.md5.hexdigest.return_value = "HASH"
mock_user = MagicMock()
mock_user.path = "path"
mock_user.name = "test"
with patch.multiple('pyolite.models.lists.keys',
Path=mock_path,
hashlib=mock_hashlib):
keys = ListKeys(mock_user)
keys.remove(key)
mock_path.has_calls([
call("path", 'keydir', 'HASH'),
call(mock_file, "test.pub"),
])
commit_message = "Removed key for user test"
mock_user.git.commit.has_calls(
[call(["my_awesome_key"], commit_message, action='remove')])
def test_if_we_commit_after_a_key_append():
key_path = "tests/fixtures/simple_key.pub"
mock_file = MagicMock()
mock_file.__str__ = lambda x: key_path
mock_path = MagicMock(return_value=mock_file)
mock_hashlib = MagicMock()
mock_hashlib.md5.hexdigest.return_value = "HASH"
mock_user = MagicMock()
mock_user.path = "path"
mock_user.name = "test"
with patch.multiple('pyolite.models.lists.keys', Path=mock_path,
hashlib=mock_hashlib):
keys = ListKeys(mock_user)
keys.append(key_path)
mock_path.has_calls([
call("path", key_path),
call("path", "keydir", "HASH"),
call(mock_file, "test"),
])
assert mock_file.isfile.call_count == 1
assert mock_file.mkdir.call_count == 1
mock_file.write_file.assert_called_once_with(b'nothing to see here\n', mode='wb')
def test_if_we_commit_after_a_key_append():
key_path = "tests/fixtures/simple_key.pub"
mock_file = MagicMock()
mock_file.__str__ = lambda x: key_path
mock_path = MagicMock(return_value=mock_file)
mock_hashlib = MagicMock()
mock_hashlib.md5.hexdigest.return_value = "HASH"
mock_user = MagicMock()
mock_user.path = "path"
mock_user.name = "test"
with patch.multiple('pyolite.models.lists.keys',
Path=mock_path,
hashlib=mock_hashlib):
keys = ListKeys(mock_user)
keys.append(key_path)
mock_path.has_calls([
call("path", key_path),
call("path", "keydir", "HASH"),
call(mock_file, "test"),
])
assert mock_file.isfile.call_count == 1
assert mock_file.mkdir.call_count == 1
mock_file.write_file.assert_called_once_with(b'nothing to see here\n',
mode='wb')
def test_list_addition(self):
mock_user = MagicMock()
mock_append = MagicMock()
keys = ListKeys(mock_user)
keys.append = mock_append
keys = keys + ["first_key", "second_key"]
mock_append.has_calls([call("first_key"), call("second_key")])
def test_list_addition():
mock_user = MagicMock()
mock_append = MagicMock()
keys = ListKeys(mock_user)
keys.append = mock_append
keys += ['first_key', 'second_key']
mock_append.has_calls([
call('first_key'),
call('second_key'),
])
def test_list_addition():
mock_user = MagicMock()
mock_append = MagicMock()
keys = ListKeys(mock_user)
keys.append = mock_append
keys += ['first_key', 'second_key']
mock_append.has_calls([
call('first_key'),
call('second_key'),
])
def __init__(self, path, git, name, **kwargs):
self.name = name
self.repos = kwargs.get('repos') or []
self.path = path
self.git = git
self.keys = ListKeys(self, kwargs.get('keys') or [])
def __init__(self, path, git, name, repos=None, keys=None):
self.name = name
self.repos = repos or []
self.path = path
self.git = git
keys = keys or []
self.keys = ListKeys(self, keys)