def test_encrypt_email_address(self): """ Test function `gdpr.encryption.encrypt_email_address` by using email address from Faker lib. """ cleartext = self.faker.email() ciphertext = encrypt_email_address(self.encryption_key, cleartext) assert_not_equal(cleartext, ciphertext, "The encrypted email address is equal to the original email address.") decrypted = decrypt_email_address(self.encryption_key, ciphertext) assert_equal(cleartext, decrypted, "The decrypted email address is not equal to the original email address.")
def test_encrypt_email_address_with_single_level_domain(self): """ Test function `gdpr.encryption.encrypt_email_address` by using email address from Faker lib. """ cleartext = f'{self.faker.email().split("@")[0]}@localhost' ciphertext = encrypt_email_address(self.encryption_key, cleartext) assert_not_equal(cleartext, ciphertext, "The encrypted email address is equal to the original email address.") assert_equal(ciphertext.split('@')[1], 'localhost') decrypted = decrypt_email_address(self.encryption_key, ciphertext) assert_equal(cleartext, decrypted, "The decrypted email address is not equal to the original email address.")
def get_encrypted_value(self, value, encryption_key: str): split = value.split(':', 1) if len(split) == 2: return f'{split[0]}:{encrypt_email_address(encryption_key, split[1])}' return encrypt_email_address(encryption_key, value)
def get_encrypted_value(self, value, encryption_key: str): return encrypt_email_address(encryption_key, value)