diff options
Diffstat (limited to 'tests/test_utils.py')
-rw-r--r-- | tests/test_utils.py | 57 |
1 files changed, 56 insertions, 1 deletions
diff --git a/tests/test_utils.py b/tests/test_utils.py index 28e7407b..b56e106e 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -11,11 +11,14 @@ # See the License for the specific language governing permissions and # limitations under the License. +import pytest + import textwrap from .utils import (load_nist_vectors, load_nist_vectors_from_file, load_cryptrec_vectors, load_cryptrec_vectors_from_file, - load_openssl_vectors, load_openssl_vectors_from_file) + load_openssl_vectors, load_openssl_vectors_from_file, load_hash_vectors, + load_hash_vectors_from_file) def test_load_nist_vectors_encrypt(): @@ -360,3 +363,55 @@ def test_load_openssl_vectors_from_file(): "ciphertext": b"D776379BE0E50825E681DA1A4C980E8E", }, ] + + +def test_load_hash_vectors(): + vector_data = textwrap.dedent(""" + + # http://tools.ietf.org/html/rfc1321 + + Len = 0 + Msg = 00 + MD = d41d8cd98f00b204e9800998ecf8427e + + Len = 8 + Msg = 61 + MD = 0cc175b9c0f1b6a831c399e269772661 + + Len = 24 + Msg = 616263 + MD = 900150983cd24fb0d6963f7d28e17f72 + + Len = 112 + Msg = 6d65737361676520646967657374 + MD = f96b697d7cb7938d525a2f31aaf161d0 + """).splitlines() + assert load_hash_vectors(vector_data) == [ + ('', 'd41d8cd98f00b204e9800998ecf8427e'), + ('61', '0cc175b9c0f1b6a831c399e269772661'), + ('616263', '900150983cd24fb0d6963f7d28e17f72'), + ('6d65737361676520646967657374', 'f96b697d7cb7938d525a2f31aaf161d0'), + ] + + +def test_load_hash_vectors_bad_data(): + vector_data = textwrap.dedent(""" + # http://tools.ietf.org/html/rfc1321 + + Len = 0 + Msg = 00 + UNKNOWN=Hello World + """).splitlines() + with pytest.raises(ValueError): + load_hash_vectors(vector_data) + + +def test_load_hash_vectors_from_file(): + test_list = load_hash_vectors_from_file("RFC/MD5/rfc-1321.txt") + assert len(test_list) == 7 + assert test_list[:4] == [ + ('', 'd41d8cd98f00b204e9800998ecf8427e'), + ('61', '0cc175b9c0f1b6a831c399e269772661'), + ('616263', '900150983cd24fb0d6963f7d28e17f72'), + ('6d65737361676520646967657374', 'f96b697d7cb7938d525a2f31aaf161d0'), + ] |