def it_raises_keychainlocked_exception_when_trying_to_get_item_from_locked_keychain(
self):
key_repository = getMock(EncryptionKeyRepository)
item_repository = getMock(KeychainItemRepository)
keychain = Keychain(key_repository, item_repository)
keychain.get_item_by_unique_id('some_random_item')
def it_returns_all_the_items_in_an_item_repository(self):
encryption_key_repository = getMock(EncryptionKeyRepository)
keychain_item = getMock(KeychainItem)
keychain_item_repository = getMock(KeychainItemRepository)
all_items = (keychain_item, )
keychain_item_repository.provides('all_items').returns(all_items)
keychain = Keychain(encryption_key_repository,
keychain_item_repository)
assert keychain.all_items() is all_items
def it_returns_a_decrypted_item_by_unique_id(self):
encryption_key = self._encryption_key_that_provides_decrypt()
encryption_key_repository = self._encryption_key_repository_that_returns_key(
encryption_key)
keychain_item = getMock(KeychainItem)
keychain_item.expects('decrypt').with_args("master_key")
keychain_item_repository = getMock(KeychainItemRepository)
keychain_item_repository.provides('item_by_unique_id').with_args(
'random_unique_id').returns(keychain_item)
keychain = Keychain(encryption_key_repository,
keychain_item_repository)
keychain.unlock("password")
eq_(keychain_item, keychain.get_item_by_unique_id('random_unique_id'))
fudge.verify()
def it_raises_invalidpasswordexception_with_wrong_password(self):
encryption_key = self._encryption_key_that_raises_invalid_password_exception(
)
encryption_key_repository = self._encryption_key_repository_that_returns_key(
encryption_key)
keychain_item_repository = getMock(KeychainItemRepository)
keychain = Keychain(encryption_key_repository,
keychain_item_repository)
keychain.unlock('wrongpassword')
def it_unlocks_the_keychain_with_the_right_password(self):
encryption_key = self._encryption_key_that_provides_decrypt()
encryption_key_repository = self._encryption_key_repository_that_returns_key(
encryption_key)
keychain_item_repository = getMock(KeychainItemRepository)
keychain = Keychain(encryption_key_repository,
keychain_item_repository)
keychain.unlock('rightpassword')
eq_(keychain.is_locked(), False)
def it_locks_when_lock_is_called(self):
encryption_key = self._encryption_key_that_provides_decrypt()
encryption_key_repository = self._encryption_key_repository_that_returns_key(
encryption_key)
keychain_item_repository = getMock(KeychainItemRepository)
keychain = Keychain(encryption_key_repository,
keychain_item_repository)
keychain.unlock('rightpassword')
eq_(keychain.is_locked(), False)
keychain.lock()
eq_(keychain.is_locked(), True)
def it_fails_to_unlock_with_wrong_password(self):
encryption_key = self._encryption_key_that_raises_invalid_password_exception(
)
encryption_key_repository = self._encryption_key_repository_that_returns_key(
encryption_key)
keychain_item_repository = getMock(KeychainItemRepository)
keychain = Keychain(encryption_key_repository,
keychain_item_repository)
try:
keychain.unlock('wrongpassword')
except InvalidPasswordException:
pass
eq_(keychain.is_locked(), True)
def _encryption_key_repository_that_returns_key(self, key):
key_repository = getMock(EncryptionKeyRepository)
key_repository.provides("key_for_security_level").with_args(
"SL5").returns(key)
return key_repository
def _encryption_key_that_provides_decrypt(self):
encryption_key = getMock(EncryptionKey)
encryption_key.provides("decrypt").returns("master_key")
return encryption_key
def _encryption_key_that_raises_invalid_password_exception(self):
encryption_key = getMock(EncryptionKey)
encryption_key.provides("decrypt").raises(InvalidPasswordException)
return encryption_key
