From 69e0652b1a48815c7d61fed8b625264419ac2796 Mon Sep 17 00:00:00 2001
From: Paul Kehrer <paul.l.kehrer@gmail.com>
Date: Fri, 18 Oct 2013 17:28:39 -0500
Subject: hash vector loader and tests

---
 tests/test_utils.py | 57 ++++++++++++++++++++++++++++++++++++++++++++++++++++-
 tests/utils.py      | 35 ++++++++++++++++++++++++++++++++
 2 files changed, 91 insertions(+), 1 deletion(-)

(limited to 'tests')

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'),
+    ]
diff --git a/tests/utils.py b/tests/utils.py
index 6b1cfd79..cf255334 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -119,3 +119,38 @@ def load_openssl_vectors(vector_data):
             "ciphertext": vector[4].encode("ascii"),
         })
     return vectors
+
+
+def load_hash_vectors(vector_data):
+    vectors = []
+
+    for line in vector_data:
+        line = line.strip()
+
+        if not line or line.startswith("#"):
+            continue
+
+        if line.startswith("Len"):
+            length = int(line.split(" = ")[1])
+        elif line.startswith("Msg"):
+            """
+            In the NIST vectors they have chosen to represent an empty
+            string as hex 00, which is of course not actually an empty
+            string. So we parse the provided length and catch this edge case.
+            """
+            msg = line.split(" = ")[1].encode("ascii") if length > 0 else b""
+        elif line.startswith("MD"):
+            md = line.split(" = ")[1]
+            # after MD is found the Msg+MD tuple is complete
+            vectors.append((msg, md))
+        else:
+            raise ValueError("Unknown line in hash vector")
+    return vectors
+
+
+def load_hash_vectors_from_file(filename):
+    base = os.path.join(
+        os.path.dirname(__file__), "primitives", "vectors"
+    )
+    with open(os.path.join(base, filename), "r") as vector_file:
+        return load_hash_vectors(vector_file)
-- 
cgit v1.2.3


From 79c16e9ef8d0454c64310ccd5faa7fd7dca26313 Mon Sep 17 00:00:00 2001
From: Paul Kehrer <paul.l.kehrer@gmail.com>
Date: Fri, 18 Oct 2013 17:44:36 -0500
Subject: alphabetical import order and quote consistency...

---
 tests/test_utils.py | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

(limited to 'tests')

diff --git a/tests/test_utils.py b/tests/test_utils.py
index b56e106e..e68f6a7f 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -11,10 +11,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import pytest
-
 import textwrap
 
+import pytest
+
 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_hash_vectors,
@@ -387,10 +387,10 @@ def test_load_hash_vectors():
         MD = f96b697d7cb7938d525a2f31aaf161d0
     """).splitlines()
     assert load_hash_vectors(vector_data) == [
-        ('', 'd41d8cd98f00b204e9800998ecf8427e'),
-        ('61', '0cc175b9c0f1b6a831c399e269772661'),
-        ('616263', '900150983cd24fb0d6963f7d28e17f72'),
-        ('6d65737361676520646967657374', 'f96b697d7cb7938d525a2f31aaf161d0'),
+        (b"", "d41d8cd98f00b204e9800998ecf8427e"),
+        (b"61", "0cc175b9c0f1b6a831c399e269772661"),
+        (b"616263", "900150983cd24fb0d6963f7d28e17f72"),
+        (b"6d65737361676520646967657374", "f96b697d7cb7938d525a2f31aaf161d0"),
     ]
 
 
@@ -410,8 +410,8 @@ 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'),
+        (b"", "d41d8cd98f00b204e9800998ecf8427e"),
+        (b"61", "0cc175b9c0f1b6a831c399e269772661"),
+        (b"616263", "900150983cd24fb0d6963f7d28e17f72"),
+        (b"6d65737361676520646967657374", "f96b697d7cb7938d525a2f31aaf161d0"),
     ]
-- 
cgit v1.2.3


From 87cd0db396e157b3fb160b8b6fd770e2c661ace2 Mon Sep 17 00:00:00 2001
From: Paul Kehrer <paul.l.kehrer@gmail.com>
Date: Fri, 18 Oct 2013 18:01:26 -0500
Subject: update loader and test to handle lines with brackets

---
 tests/test_utils.py | 1 +
 tests/utils.py      | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

(limited to 'tests')

diff --git a/tests/test_utils.py b/tests/test_utils.py
index e68f6a7f..a9bb6a87 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -369,6 +369,7 @@ def test_load_hash_vectors():
     vector_data = textwrap.dedent("""
 
         # http://tools.ietf.org/html/rfc1321
+        [irrelevant]
 
         Len = 0
         Msg = 00
diff --git a/tests/utils.py b/tests/utils.py
index cf255334..03b780f8 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -127,7 +127,7 @@ def load_hash_vectors(vector_data):
     for line in vector_data:
         line = line.strip()
 
-        if not line or line.startswith("#"):
+        if not line or line.startswith("#") or line.startswith("["):
             continue
 
         if line.startswith("Len"):
-- 
cgit v1.2.3


From bde6fb52129909cf319157dba95d65fb557d5013 Mon Sep 17 00:00:00 2001
From: Paul Kehrer <paul.l.kehrer@gmail.com>
Date: Fri, 18 Oct 2013 18:08:49 -0500
Subject: Hash Saga Part 3 - API changes + SHA1 support + tests

---
 tests/primitives/test_hash_vectors.py | 35 ++++++++++++++++++++++
 tests/primitives/test_hashes.py       | 28 ++++++++++++++++++
 tests/primitives/test_utils.py        | 24 ++++++++++++++-
 tests/primitives/utils.py             | 55 +++++++++++++++++++++++++++++++++++
 4 files changed, 141 insertions(+), 1 deletion(-)
 create mode 100644 tests/primitives/test_hash_vectors.py
 create mode 100644 tests/primitives/test_hashes.py

(limited to 'tests')

diff --git a/tests/primitives/test_hash_vectors.py b/tests/primitives/test_hash_vectors.py
new file mode 100644
index 00000000..4b71ad70
--- /dev/null
+++ b/tests/primitives/test_hash_vectors.py
@@ -0,0 +1,35 @@
+# 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
+
+from cryptography.primitives import hashes
+
+from .utils import generate_hash_test
+from ..utils import load_hash_vectors_from_file
+
+
+class TestSHA1(object):
+    test_SHA1 = generate_hash_test(
+        load_hash_vectors_from_file,
+        os.path.join("NIST", "SHABYTE"),
+        [
+            "SHA1LongMsg.rsp",
+            "SHA1ShortMsg.rsp",
+        ],
+        lambda api: hashes.SHA1(api=api),
+        only_if=lambda api: api.supports_hash("sha1"),
+        skip_message="Does not support SHA1",
+    )
diff --git a/tests/primitives/test_hashes.py b/tests/primitives/test_hashes.py
new file mode 100644
index 00000000..3419b14d
--- /dev/null
+++ b/tests/primitives/test_hashes.py
@@ -0,0 +1,28 @@
+# 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
+
+from cryptography.primitives import hashes
+
+from .utils import generate_base_hash_test
+
+
+class TestSHA1(object):
+    test_SHA1 = generate_base_hash_test(
+        lambda api: hashes.SHA1(api=api),
+        digest_size=20,
+        block_size=64,
+        only_if=lambda api: api.supports_hash("sha1"),
+        skip_message="Does not support SHA1",
+    )
diff --git a/tests/primitives/test_utils.py b/tests/primitives/test_utils.py
index 4666ece7..43ec8a71 100644
--- a/tests/primitives/test_utils.py
+++ b/tests/primitives/test_utils.py
@@ -1,6 +1,6 @@
 import pytest
 
-from .utils import encrypt_test
+from .utils import encrypt_test, hash_test, base_hash_test
 
 
 class TestEncryptTest(object):
@@ -12,3 +12,25 @@ class TestEncryptTest(object):
                 skip_message="message!"
             )
         assert exc_info.value.args[0] == "message!"
+
+
+class TestHashTest(object):
+    def test_skips_if_only_if_returns_false(self):
+        with pytest.raises(pytest.skip.Exception) as exc_info:
+            hash_test(
+                None, None, None,
+                only_if=lambda api: False,
+                skip_message="message!"
+            )
+        assert exc_info.value.args[0] == "message!"
+
+
+class TestBaseHashTest(object):
+    def test_skips_if_only_if_returns_false(self):
+        with pytest.raises(pytest.skip.Exception) as exc_info:
+            base_hash_test(
+                None, None, None, None,
+                only_if=lambda api: False,
+                skip_message="message!"
+            )
+        assert exc_info.value.args[0] == "message!"
diff --git a/tests/primitives/utils.py b/tests/primitives/utils.py
index 3cf08c28..f301199c 100644
--- a/tests/primitives/utils.py
+++ b/tests/primitives/utils.py
@@ -40,3 +40,58 @@ def encrypt_test(api, cipher_factory, mode_factory, params, only_if,
     actual_ciphertext = cipher.encrypt(binascii.unhexlify(plaintext))
     actual_ciphertext += cipher.finalize()
     assert actual_ciphertext == binascii.unhexlify(ciphertext)
+
+
+def generate_hash_test(param_loader, path, file_names, hash_factory,
+                       only_if=lambda api: True, skip_message=None):
+    def test_hash(self):
+        for api in _ALL_APIS:
+            for file_name in file_names:
+                for params in param_loader(os.path.join(path, file_name)):
+                    yield (
+                        hash_test,
+                        api,
+                        hash_factory,
+                        params,
+                        only_if,
+                        skip_message
+                    )
+    return test_hash
+
+
+def hash_test(api, hash_factory, params, only_if, skip_message):
+    if not only_if(api):
+        pytest.skip(skip_message)
+    msg = params[0]
+    md = params[1]
+    m = hash_factory(api)
+    m.update(binascii.unhexlify(msg))
+    assert m.hexdigest() == md.replace(" ", "").lower()
+
+
+def generate_base_hash_test(hash_factory, digest_size, block_size,
+                            only_if=lambda api: True, skip_message=None):
+    def test_base_hash(self):
+        for api in _ALL_APIS:
+            yield (
+                base_hash_test,
+                api,
+                hash_factory,
+                digest_size,
+                block_size,
+                only_if,
+                skip_message,
+            )
+    return test_base_hash
+
+
+def base_hash_test(api, hash_factory, digest_size, block_size, only_if,
+                   skip_message):
+    if not only_if(api):
+        pytest.skip(skip_message)
+    m = hash_factory(api=api)
+    assert m.digest_size == digest_size
+    assert m.block_size == block_size
+    m_copy = m.copy()
+    assert m != m_copy
+    assert m._ctx != m_copy._ctx
-- 
cgit v1.2.3


From bb069c2fee6460185ee435ea848d80bab2ccec6c Mon Sep 17 00:00:00 2001
From: Paul Kehrer <paul.l.kehrer@gmail.com>
Date: Fri, 18 Oct 2013 19:51:01 -0500
Subject: remove unneeded lambdas from tests

---
 tests/primitives/test_hash_vectors.py |  2 +-
 tests/primitives/test_hashes.py       |  2 +-
 tests/primitives/utils.py             | 16 ++++++++--------
 3 files changed, 10 insertions(+), 10 deletions(-)

(limited to 'tests')

diff --git a/tests/primitives/test_hash_vectors.py b/tests/primitives/test_hash_vectors.py
index 4b71ad70..8198b086 100644
--- a/tests/primitives/test_hash_vectors.py
+++ b/tests/primitives/test_hash_vectors.py
@@ -29,7 +29,7 @@ class TestSHA1(object):
             "SHA1LongMsg.rsp",
             "SHA1ShortMsg.rsp",
         ],
-        lambda api: hashes.SHA1(api=api),
+        hashes.SHA1,
         only_if=lambda api: api.supports_hash("sha1"),
         skip_message="Does not support SHA1",
     )
diff --git a/tests/primitives/test_hashes.py b/tests/primitives/test_hashes.py
index 3419b14d..1bc2e9e9 100644
--- a/tests/primitives/test_hashes.py
+++ b/tests/primitives/test_hashes.py
@@ -20,7 +20,7 @@ from .utils import generate_base_hash_test
 
 class TestSHA1(object):
     test_SHA1 = generate_base_hash_test(
-        lambda api: hashes.SHA1(api=api),
+        hashes.SHA1,
         digest_size=20,
         block_size=64,
         only_if=lambda api: api.supports_hash("sha1"),
diff --git a/tests/primitives/utils.py b/tests/primitives/utils.py
index f301199c..0d4c0eb3 100644
--- a/tests/primitives/utils.py
+++ b/tests/primitives/utils.py
@@ -42,7 +42,7 @@ def encrypt_test(api, cipher_factory, mode_factory, params, only_if,
     assert actual_ciphertext == binascii.unhexlify(ciphertext)
 
 
-def generate_hash_test(param_loader, path, file_names, hash_factory,
+def generate_hash_test(param_loader, path, file_names, hash_cls,
                        only_if=lambda api: True, skip_message=None):
     def test_hash(self):
         for api in _ALL_APIS:
@@ -51,7 +51,7 @@ def generate_hash_test(param_loader, path, file_names, hash_factory,
                     yield (
                         hash_test,
                         api,
-                        hash_factory,
+                        hash_cls,
                         params,
                         only_if,
                         skip_message
@@ -59,24 +59,24 @@ def generate_hash_test(param_loader, path, file_names, hash_factory,
     return test_hash
 
 
-def hash_test(api, hash_factory, params, only_if, skip_message):
+def hash_test(api, hash_cls, params, only_if, skip_message):
     if not only_if(api):
         pytest.skip(skip_message)
     msg = params[0]
     md = params[1]
-    m = hash_factory(api)
+    m = hash_cls(api=api)
     m.update(binascii.unhexlify(msg))
     assert m.hexdigest() == md.replace(" ", "").lower()
 
 
-def generate_base_hash_test(hash_factory, digest_size, block_size,
+def generate_base_hash_test(hash_cls, digest_size, block_size,
                             only_if=lambda api: True, skip_message=None):
     def test_base_hash(self):
         for api in _ALL_APIS:
             yield (
                 base_hash_test,
                 api,
-                hash_factory,
+                hash_cls,
                 digest_size,
                 block_size,
                 only_if,
@@ -85,11 +85,11 @@ def generate_base_hash_test(hash_factory, digest_size, block_size,
     return test_base_hash
 
 
-def base_hash_test(api, hash_factory, digest_size, block_size, only_if,
+def base_hash_test(api, hash_cls, digest_size, block_size, only_if,
                    skip_message):
     if not only_if(api):
         pytest.skip(skip_message)
-    m = hash_factory(api=api)
+    m = hash_cls(api=api)
     assert m.digest_size == digest_size
     assert m.block_size == block_size
     m_copy = m.copy()
-- 
cgit v1.2.3


From ba3b471e4b0f56e65acdb9d5daf64eb726d9c371 Mon Sep 17 00:00:00 2001
From: Paul Kehrer <paul.l.kehrer@gmail.com>
Date: Fri, 18 Oct 2013 20:53:04 -0500
Subject: change api.supports_hash to take a hash class rather than a str

* This change means hash class names will be byte strings and we no
  longer need to encode to ascii on hashobject.name in
  create_hash_context
---
 tests/primitives/test_hash_vectors.py | 2 +-
 tests/primitives/test_hashes.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

(limited to 'tests')

diff --git a/tests/primitives/test_hash_vectors.py b/tests/primitives/test_hash_vectors.py
index 8198b086..9a925e27 100644
--- a/tests/primitives/test_hash_vectors.py
+++ b/tests/primitives/test_hash_vectors.py
@@ -30,6 +30,6 @@ class TestSHA1(object):
             "SHA1ShortMsg.rsp",
         ],
         hashes.SHA1,
-        only_if=lambda api: api.supports_hash("sha1"),
+        only_if=lambda api: api.supports_hash(hashes.SHA1),
         skip_message="Does not support SHA1",
     )
diff --git a/tests/primitives/test_hashes.py b/tests/primitives/test_hashes.py
index 1bc2e9e9..4ad5c89a 100644
--- a/tests/primitives/test_hashes.py
+++ b/tests/primitives/test_hashes.py
@@ -23,6 +23,6 @@ class TestSHA1(object):
         hashes.SHA1,
         digest_size=20,
         block_size=64,
-        only_if=lambda api: api.supports_hash("sha1"),
+        only_if=lambda api: api.supports_hash(hashes.SHA1),
         skip_message="Does not support SHA1",
     )
-- 
cgit v1.2.3


From 7e5697c6f597ac6b0550ae3dbcc19482d10f3efe Mon Sep 17 00:00:00 2001
From: Paul Kehrer <paul.l.kehrer@gmail.com>
Date: Fri, 18 Oct 2013 21:07:36 -0500
Subject: SHA-2 family support

---
 tests/primitives/test_hash_vectors.py | 56 +++++++++++++++++++++++++++++++++++
 tests/primitives/test_hashes.py       | 40 +++++++++++++++++++++++++
 2 files changed, 96 insertions(+)

(limited to 'tests')

diff --git a/tests/primitives/test_hash_vectors.py b/tests/primitives/test_hash_vectors.py
index 9a925e27..d0fe46d1 100644
--- a/tests/primitives/test_hash_vectors.py
+++ b/tests/primitives/test_hash_vectors.py
@@ -33,3 +33,59 @@ class TestSHA1(object):
         only_if=lambda api: api.supports_hash(hashes.SHA1),
         skip_message="Does not support SHA1",
     )
+
+
+class TestSHA224(object):
+    test_SHA224 = generate_hash_test(
+        load_hash_vectors_from_file,
+        os.path.join("NIST", "SHABYTE"),
+        [
+            "SHA224LongMsg.rsp",
+            "SHA224ShortMsg.rsp",
+        ],
+        hashes.SHA224,
+        only_if=lambda api: api.supports_hash(hashes.SHA224),
+        skip_message="Does not support SHA224",
+    )
+
+
+class TestSHA256(object):
+    test_SHA256 = generate_hash_test(
+        load_hash_vectors_from_file,
+        os.path.join("NIST", "SHABYTE"),
+        [
+            "SHA256LongMsg.rsp",
+            "SHA256ShortMsg.rsp",
+        ],
+        hashes.SHA256,
+        only_if=lambda api: api.supports_hash(hashes.SHA256),
+        skip_message="Does not support SHA256",
+    )
+
+
+class TestSHA384(object):
+    test_SHA384 = generate_hash_test(
+        load_hash_vectors_from_file,
+        os.path.join("NIST", "SHABYTE"),
+        [
+            "SHA384LongMsg.rsp",
+            "SHA384ShortMsg.rsp",
+        ],
+        hashes.SHA384,
+        only_if=lambda api: api.supports_hash(hashes.SHA384),
+        skip_message="Does not support SHA384",
+    )
+
+
+class TestSHA512(object):
+    test_SHA512 = generate_hash_test(
+        load_hash_vectors_from_file,
+        os.path.join("NIST", "SHABYTE"),
+        [
+            "SHA512LongMsg.rsp",
+            "SHA512ShortMsg.rsp",
+        ],
+        hashes.SHA512,
+        only_if=lambda api: api.supports_hash(hashes.SHA512),
+        skip_message="Does not support SHA512",
+    )
diff --git a/tests/primitives/test_hashes.py b/tests/primitives/test_hashes.py
index 4ad5c89a..2f2dd1c7 100644
--- a/tests/primitives/test_hashes.py
+++ b/tests/primitives/test_hashes.py
@@ -26,3 +26,43 @@ class TestSHA1(object):
         only_if=lambda api: api.supports_hash(hashes.SHA1),
         skip_message="Does not support SHA1",
     )
+
+
+class TestSHA224(object):
+    test_SHA224 = generate_base_hash_test(
+        hashes.SHA224,
+        digest_size=28,
+        block_size=64,
+        only_if=lambda api: api.supports_hash(hashes.SHA224),
+        skip_message="Does not support SHA224",
+    )
+
+
+class TestSHA256(object):
+    test_SHA256 = generate_base_hash_test(
+        hashes.SHA256,
+        digest_size=32,
+        block_size=64,
+        only_if=lambda api: api.supports_hash(hashes.SHA256),
+        skip_message="Does not support SHA256",
+    )
+
+
+class TestSHA384(object):
+    test_SHA384 = generate_base_hash_test(
+        hashes.SHA384,
+        digest_size=48,
+        block_size=128,
+        only_if=lambda api: api.supports_hash(hashes.SHA384),
+        skip_message="Does not support SHA384",
+    )
+
+
+class TestSHA512(object):
+    test_SHA512 = generate_base_hash_test(
+        hashes.SHA512,
+        digest_size=64,
+        block_size=128,
+        only_if=lambda api: api.supports_hash(hashes.SHA512),
+        skip_message="Does not support SHA512",
+    )
-- 
cgit v1.2.3


From c179407406f0ef5c2b7b5b6316521408ba3803b3 Mon Sep 17 00:00:00 2001
From: Paul Kehrer <paul.l.kehrer@gmail.com>
Date: Fri, 18 Oct 2013 21:42:57 -0500
Subject: ripemd160 support + long string hash test

* Note that the long string hash test for RIPEMD160 adds a vector in the
  test. You can verify this vector (for b"a" * 1000000) on the RIPE
  homepage: http://homes.esat.kuleuven.be/~bosselae/ripemd160.html
---
 tests/primitives/test_hash_vectors.py | 22 +++++++++++++++++++++-
 tests/primitives/test_hashes.py       | 10 ++++++++++
 tests/primitives/test_utils.py        | 14 +++++++++++++-
 tests/primitives/utils.py             | 23 +++++++++++++++++++++++
 4 files changed, 67 insertions(+), 2 deletions(-)

(limited to 'tests')

diff --git a/tests/primitives/test_hash_vectors.py b/tests/primitives/test_hash_vectors.py
index d0fe46d1..51c4b85d 100644
--- a/tests/primitives/test_hash_vectors.py
+++ b/tests/primitives/test_hash_vectors.py
@@ -17,7 +17,7 @@ import os
 
 from cryptography.primitives import hashes
 
-from .utils import generate_hash_test
+from .utils import generate_hash_test, generate_long_string_hash_test
 from ..utils import load_hash_vectors_from_file
 
 
@@ -89,3 +89,23 @@ class TestSHA512(object):
         only_if=lambda api: api.supports_hash(hashes.SHA512),
         skip_message="Does not support SHA512",
     )
+
+
+class TestRIPEMD160(object):
+    test_RIPEMD160 = generate_hash_test(
+        load_hash_vectors_from_file,
+        os.path.join("ISO", "ripemd160"),
+        [
+            "ripevectors.txt",
+        ],
+        hashes.RIPEMD160,
+        only_if=lambda api: api.supports_hash(hashes.RIPEMD160),
+        skip_message="Does not support RIPEMD160",
+    )
+
+    test_RIPEMD160_long_string = generate_long_string_hash_test(
+        hashes.RIPEMD160,
+        "52783243c1697bdbe16d37f97f68f08325dc1528",
+        only_if=lambda api: api.supports_hash(hashes.RIPEMD160),
+        skip_message="Does not support RIPEMD160",
+    )
diff --git a/tests/primitives/test_hashes.py b/tests/primitives/test_hashes.py
index 2f2dd1c7..bfb45037 100644
--- a/tests/primitives/test_hashes.py
+++ b/tests/primitives/test_hashes.py
@@ -66,3 +66,13 @@ class TestSHA512(object):
         only_if=lambda api: api.supports_hash(hashes.SHA512),
         skip_message="Does not support SHA512",
     )
+
+
+class TestRIPEMD160(object):
+    test_RIPEMD160 = generate_base_hash_test(
+        hashes.RIPEMD160,
+        digest_size=20,
+        block_size=64,
+        only_if=lambda api: api.supports_hash(hashes.RIPEMD160),
+        skip_message="Does not support RIPEMD160",
+    )
diff --git a/tests/primitives/test_utils.py b/tests/primitives/test_utils.py
index 43ec8a71..9888309e 100644
--- a/tests/primitives/test_utils.py
+++ b/tests/primitives/test_utils.py
@@ -1,6 +1,7 @@
 import pytest
 
-from .utils import encrypt_test, hash_test, base_hash_test
+from .utils import (base_hash_test, encrypt_test, hash_test,
+    long_string_hash_test)
 
 
 class TestEncryptTest(object):
@@ -34,3 +35,14 @@ class TestBaseHashTest(object):
                 skip_message="message!"
             )
         assert exc_info.value.args[0] == "message!"
+
+
+class TestLongHashTest(object):
+    def test_skips_if_only_if_returns_false(self):
+        with pytest.raises(pytest.skip.Exception) as exc_info:
+            long_string_hash_test(
+                None, None, None,
+                only_if=lambda api: False,
+                skip_message="message!"
+            )
+        assert exc_info.value.args[0] == "message!"
diff --git a/tests/primitives/utils.py b/tests/primitives/utils.py
index 0d4c0eb3..8b32700b 100644
--- a/tests/primitives/utils.py
+++ b/tests/primitives/utils.py
@@ -95,3 +95,26 @@ def base_hash_test(api, hash_cls, digest_size, block_size, only_if,
     m_copy = m.copy()
     assert m != m_copy
     assert m._ctx != m_copy._ctx
+
+
+def generate_long_string_hash_test(hash_factory, md, only_if=lambda api: True,
+                                   skip_message=None):
+    def test_long_string_hash(self):
+        for api in _ALL_APIS:
+            yield(
+                long_string_hash_test,
+                api,
+                hash_factory,
+                md,
+                only_if,
+                skip_message
+            )
+    return test_long_string_hash
+
+
+def long_string_hash_test(api, hash_factory, md, only_if, skip_message):
+    if not only_if(api):
+        pytest.skip(skip_message)
+    m = hash_factory(api)
+    m.update(b"a" * 1000000)
+    assert m.hexdigest() == md.lower()
-- 
cgit v1.2.3