diff options
author | David Reid <dreid@dreid.org> | 2014-01-20 16:05:53 -0800 |
---|---|---|
committer | David Reid <dreid@dreid.org> | 2014-02-03 10:05:26 -0800 |
commit | 66c9cd928601725e27aa64255e56b3a7e481a08d (patch) | |
tree | 2d746623c5bdb603d62ec28a3a765a5b9fd4d20d /tests/hazmat/primitives/test_hkdf_vectors.py | |
parent | ab33266b16d9a1cd3cf6abcf0a7b80e86f915d95 (diff) | |
download | cryptography-66c9cd928601725e27aa64255e56b3a7e481a08d.tar.gz cryptography-66c9cd928601725e27aa64255e56b3a7e481a08d.tar.bz2 cryptography-66c9cd928601725e27aa64255e56b3a7e481a08d.zip |
Refactor HKDF support and provide vectors for tests.
Diffstat (limited to 'tests/hazmat/primitives/test_hkdf_vectors.py')
-rw-r--r-- | tests/hazmat/primitives/test_hkdf_vectors.py | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/tests/hazmat/primitives/test_hkdf_vectors.py b/tests/hazmat/primitives/test_hkdf_vectors.py new file mode 100644 index 00000000..2595c956 --- /dev/null +++ b/tests/hazmat/primitives/test_hkdf_vectors.py @@ -0,0 +1,50 @@ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or +# implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +from __future__ import absolute_import, division, print_function + +import os + +import pytest + +from cryptography.hazmat.primitives import hashes + +from .utils import generate_hkdf_test +from ...utils import load_hkdf_vectors + +@pytest.mark.supported( + only_if=lambda backend: backend.hash_supported(hashes.SHA1()), + skip_message="Does not support SHA1." +) +@pytest.mark.hash +class TestHKDFSHA1(object): + test_HKDFSHA1 = generate_hkdf_test( + load_hkdf_vectors, + os.path.join("kdf"), + ["rfc-5869-HKDF-SHA1.txt"], + hashes.SHA1() + ) + + +@pytest.mark.supported( + only_if=lambda backend: backend.hash_supported(hashes.SHA256()), + skip_message="Does not support SHA256." +) +@pytest.mark.hash +class TestHKDFSHA256(object): + test_HKDFSHA1 = generate_hkdf_test( + load_hkdf_vectors, + os.path.join("kdf"), + ["rfc-5869-HKDF-SHA256.txt"], + hashes.SHA256() + ) |