mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 22:24:23 +01:00
90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
from os import urandom
|
|
|
|
from pisa import logging
|
|
from apps.cli.blob import Blob
|
|
from pisa.conf import SUPPORTED_CIPHERS, SUPPORTED_HASH_FUNCTIONS
|
|
|
|
logging.getLogger().disabled = True
|
|
|
|
|
|
def test_init_blob():
|
|
data = urandom(64).hex()
|
|
|
|
# Fixed (valid) hash function, try different valid ciphers
|
|
hash_function = SUPPORTED_HASH_FUNCTIONS[0]
|
|
for cipher in SUPPORTED_CIPHERS:
|
|
cipher_cases = [cipher, cipher.lower(), cipher.capitalize()]
|
|
|
|
for case in cipher_cases:
|
|
blob = Blob(data, case, hash_function)
|
|
assert(blob.data == data and blob.cipher == case and blob.hash_function == hash_function)
|
|
|
|
# Fixed (valid) cipher, try different valid hash functions
|
|
cipher = SUPPORTED_CIPHERS[0]
|
|
for hash_function in SUPPORTED_HASH_FUNCTIONS:
|
|
hash_function_cases = [hash_function, hash_function.lower(), hash_function.capitalize()]
|
|
|
|
for case in hash_function_cases:
|
|
blob = Blob(data, cipher, case)
|
|
assert(blob.data == data and blob.cipher == cipher and blob.hash_function == case)
|
|
|
|
# Invalid data
|
|
data = urandom(64)
|
|
cipher = SUPPORTED_CIPHERS[0]
|
|
hash_function = SUPPORTED_HASH_FUNCTIONS[0]
|
|
|
|
try:
|
|
Blob(data, cipher, hash_function)
|
|
assert False, "Able to create blob with wrong data"
|
|
|
|
except ValueError:
|
|
assert True
|
|
|
|
# Invalid cipher
|
|
data = urandom(64).hex()
|
|
cipher = "A" * 10
|
|
hash_function = SUPPORTED_HASH_FUNCTIONS[0]
|
|
|
|
try:
|
|
Blob(data, cipher, hash_function)
|
|
assert False, "Able to create blob with wrong data"
|
|
|
|
except ValueError:
|
|
assert True
|
|
|
|
# Invalid hash function
|
|
data = urandom(64).hex()
|
|
cipher = SUPPORTED_CIPHERS[0]
|
|
hash_function = "A" * 10
|
|
|
|
try:
|
|
Blob(data, cipher, hash_function)
|
|
assert(False, "Able to create blob with wrong data")
|
|
|
|
except ValueError:
|
|
assert True
|
|
|
|
|
|
def test_encrypt():
|
|
# Valid data, valid key
|
|
data = urandom(64).hex()
|
|
blob = Blob(data, SUPPORTED_CIPHERS[0], SUPPORTED_HASH_FUNCTIONS[0])
|
|
key = urandom(32).hex()
|
|
|
|
encrypted_blob = blob.encrypt(key)
|
|
|
|
# Invalid key (note that encrypt cannot be called with invalid data since that's checked when the Blob is created)
|
|
invalid_key = urandom(32)
|
|
|
|
try:
|
|
blob.encrypt(invalid_key)
|
|
assert (False, "Able to create encrypt with invalid key")
|
|
|
|
except ValueError:
|
|
assert True
|
|
|
|
# Check that two encryptions of the same data have the same result
|
|
encrypted_blob2 = blob.encrypt(key)
|
|
|
|
assert(encrypted_blob == encrypted_blob2 and id(encrypted_blob) != id(encrypted_blob2))
|