Spaces:
Sleeping
Sleeping
File size: 758 Bytes
868b252 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import pytest
from cryptography.fernet import Fernet
from fastapi import HTTPException
from reworkd_platform.services.security import EncryptionService
def test_encrypt_decrypt():
key = Fernet.generate_key()
service = EncryptionService(key)
original_text = "Hello, world!"
encrypted = service.encrypt(original_text)
decrypted = service.decrypt(encrypted)
assert original_text == decrypted
def test_invalid_key():
key = Fernet.generate_key()
different_key = Fernet.generate_key()
different_service = EncryptionService(different_key)
original_text = "Hello, world!"
encrypted = Fernet(key).encrypt(original_text.encode())
with pytest.raises(HTTPException):
different_service.decrypt(encrypted)
|