ensure hkdf can return entirety of expand

wfatherley · Aug 4, 2024 · ac0d99f · ac0d99f
1 parent a68d70c
commit ac0d99f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/hkdfref/_hkdfref.py b/src/hkdfref/_hkdfref.py
@@ -7,7 +7,7 @@
 from typing import Any, Callable
 
 
-logger = getLogger(__name__)
+logger = getLogger(__name__[1:])
 
 
 default_hash = hashlib.sha512
@@ -39,7 +39,9 @@ def hkdf(*ikm: bytes, **kwargs: Any) -> bytes:
         prk = extract(salt, *ikm, hash=extract_hash)
     else:
         prk = b"".join(ikm)
-    return expand(prk, info, l, expand_hash)
+    return expand(
+        prk, info, l, expand_hash, return_all=kwargs.get("return_all", False)
+    )
 
 
 def expand(