diff options
Diffstat (limited to 'tests/hazmat/primitives/test_pbkdf2.py')
| -rw-r--r-- | tests/hazmat/primitives/test_pbkdf2.py | 10 | 
1 files changed, 10 insertions, 0 deletions
| diff --git a/tests/hazmat/primitives/test_pbkdf2.py b/tests/hazmat/primitives/test_pbkdf2.py index 41123557..6ad225a8 100644 --- a/tests/hazmat/primitives/test_pbkdf2.py +++ b/tests/hazmat/primitives/test_pbkdf2.py @@ -14,6 +14,7 @@  from __future__ import absolute_import, division, print_function  import pytest +import six  from cryptography import utils  from cryptography.exceptions import ( @@ -57,3 +58,12 @@ class TestPBKDF2HMAC(object):          kdf = PBKDF2HMAC(hashes.SHA1(), 20, b"salt", 10, default_backend())          with pytest.raises(InvalidKey):              kdf.verify(b"password2", key) + +    def test_unicode_error_with_salt(self): +        with pytest.raises(TypeError): +            PBKDF2HMAC(hashes.SHA1(), 20, six.u("salt"), 10, default_backend()) + +    def test_unicode_error_with_key_material(self): +        kdf = PBKDF2HMAC(hashes.SHA1(), 20, b"salt", 10, default_backend()) +        with pytest.raises(TypeError): +            kdf.derive(six.u("unicode here")) | 
