##// END OF EJS Templates
Aligned expanded sumarry box content better and moved comments as last column
Aligned expanded sumarry box content better and moved comments as last column

File last commit:

r3522:3910c057 default
r3639:7ee2e326 new-ui
Show More
test_encrypt.py
133 lines | 4.3 KiB | text/x-python | PythonLexer
project: added all source files and assets
r1 # -*- coding: utf-8 -*-
docs: updated copyrights to 2019
r3363 # Copyright (C) 2010-2019 RhodeCode GmbH
project: added all source files and assets
r1 #
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/
import pytest
encryption: Implement a slightly improved AesCipher encryption....
r281 from rhodecode.lib.encrypt import (
AESCipher, SignatureVerificationError, InvalidDecryptedValue)
encryption: added new backend using cryptography + Fernet encryption....
r3522 from rhodecode.lib.encrypt2 import (Encryptor, InvalidToken)
project: added all source files and assets
r1
class TestEncryptModule(object):
@pytest.mark.parametrize(
"key, text",
[
('a', 'short'),
('a'*64, 'too long(trimmed to 32)'),
('a'*32, 'just enough'),
('ąćęćę', 'non asci'),
('$asa$asa', 'special $ used'),
]
)
def test_encryption(self, key, text):
enc = AESCipher(key).encrypt(text)
assert AESCipher(key).decrypt(enc) == text
encryption: Implement a slightly improved AesCipher encryption....
r281
def test_encryption_with_hmac(self):
key = 'secret'
text = 'ihatemysql'
enc = AESCipher(key, hmac=True).encrypt(text)
assert AESCipher(key, hmac=True).decrypt(enc) == text
def test_encryption_with_hmac_with_bad_key(self):
key = 'secretstring'
text = 'ihatemysql'
enc = AESCipher(key, hmac=True).encrypt(text)
with pytest.raises(SignatureVerificationError) as e:
assert AESCipher('differentsecret', hmac=True).decrypt(enc) == ''
assert 'Encryption signature verification failed' in str(e)
def test_encryption_with_hmac_with_bad_data(self):
key = 'secret'
text = 'ihatemysql'
enc = AESCipher(key, hmac=True).encrypt(text)
enc = 'xyz' + enc[3:]
with pytest.raises(SignatureVerificationError) as e:
assert AESCipher(key, hmac=True).decrypt(enc) == text
assert 'Encryption signature verification failed' in str(e)
def test_encryption_with_hmac_with_bad_key_not_strict(self):
key = 'secretstring'
text = 'ihatemysql'
enc = AESCipher(key, hmac=True).encrypt(text)
assert isinstance(AESCipher(
'differentsecret', hmac=True, strict_verification=False
).decrypt(enc), InvalidDecryptedValue)
encryption: added new backend using cryptography + Fernet encryption....
r3522
class TestEncryptModule2(object):
@pytest.mark.parametrize(
"key, text",
[
('a', 'short'),
('a'*64, 'too long(trimmed to 32)'),
('a'*32, 'just enough'),
('ąćęćę', 'non asci'),
('$asa$asa', 'special $ used'),
]
)
def test_encryption(self, key, text):
enc = Encryptor(key).encrypt(text)
assert Encryptor(key).decrypt(enc) == text
def test_encryption_with_bad_key(self):
key = 'secretstring'
text = 'ihatemysql'
enc = Encryptor(key).encrypt(text)
assert Encryptor('differentsecret').decrypt(enc) == ''
def test_encryption_with_bad_key_raises(self):
key = 'secretstring'
text = 'ihatemysql'
enc = Encryptor(key).encrypt(text)
with pytest.raises(InvalidToken) as e:
Encryptor('differentsecret').decrypt(enc, safe=False)
assert 'InvalidToken' in str(e)
def test_encryption_with_bad_format_data(self):
key = 'secret'
text = 'ihatemysql'
enc = Encryptor(key).encrypt(text)
enc = '$xyz' + enc[3:]
with pytest.raises(ValueError) as e:
Encryptor(key).decrypt(enc, safe=False)
assert 'Encrypted Data has invalid format' in str(e)
def test_encryption_with_bad_data(self):
key = 'secret'
text = 'ihatemysql'
enc = Encryptor(key).encrypt(text)
enc = enc[:-5]
with pytest.raises(InvalidToken) as e:
Encryptor(key).decrypt(enc, safe=False)
assert 'InvalidToken' in str(e)