commit 0d492db1be3e287b5f49a5ce408196401bdd0a2b
author David Reid <dreid@dreid.org> Mon Jan 27 17:05:49 2014 -0800
committer David Reid <dreid@dreid.org> Mon Feb 03 10:05:27 2014 -0800
tree 8c9cbc30464819cfd173d751600d92e21fb7a836
parent 14367303f16bc271f4a8f11f09b02342f44c3a7e

Closer to proposed interface in #513.

tests/hazmat/primitives/utils.py

diff --git a/tests/hazmat/primitives/utils.py b/tests/hazmat/primitives/utils.py
index 9e9088a..2584272 100644
--- a/tests/hazmat/primitives/utils.py
+++ b/tests/hazmat/primitives/utils.py
@@ -8,9 +8,7 @@
 from cryptography.hazmat.primitives import hashes, hmac
 from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
 from cryptography.hazmat.primitives.ciphers import Cipher
-from cryptography.hazmat.primitives.kdf.hkdf import (
-    hkdf_derive, hkdf_extract, hkdf_expand
-)
+from cryptography.hazmat.primitives.kdf.hkdf import HKDF
 
 from cryptography.exceptions import (
     AlreadyFinalized, NotYetFinalized, AlreadyUpdated, InvalidTag,
@@ -306,38 +304,44 @@
 
 
 def hkdf_derive_test(backend, algorithm, params):
-    okm = hkdf_derive(
-        binascii.unhexlify(params["ikm"]),
-        int(params["l"]),
-        binascii.unhexlify(params["salt"]),
-        binascii.unhexlify(params["info"]),
+    hkdf = HKDF(
         algorithm,
+        int(params["l"]),
+        salt=binascii.unhexlify(params["salt"]) or None,
+        info=binascii.unhexlify(params["info"]) or None,
         backend=backend
     )
 
+    okm = hkdf.derive(binascii.unhexlify(params["ikm"]))
+
     assert okm == binascii.unhexlify(params["okm"])
 
 
 def hkdf_extract_test(backend, algorithm, params):
-    prk = hkdf_extract(
+    hkdf = HKDF(
         algorithm,
-        binascii.unhexlify(params["ikm"]),
-        binascii.unhexlify(params["salt"]),
+        int(params["l"]),
+        salt=binascii.unhexlify(params["salt"]) or None,
+        info=binascii.unhexlify(params["info"]) or None,
         backend=backend
     )
 
+    prk = hkdf.extract(binascii.unhexlify(params["ikm"]))
+
     assert prk == binascii.unhexlify(params["prk"])
 
 
 def hkdf_expand_test(backend, algorithm, params):
-    okm = hkdf_expand(
+    hkdf = HKDF(
         algorithm,
-        binascii.unhexlify(params["prk"]),
-        binascii.unhexlify(params["info"]),
         int(params["l"]),
+        salt=binascii.unhexlify(params["salt"]) or None,
+        info=binascii.unhexlify(params["info"]) or None,
         backend=backend
     )
 
+    okm = hkdf.expand(binascii.unhexlify(params["prk"]))
+
     assert okm == binascii.unhexlify(params["okm"])