From f3f8df349f78b38a9a713030e25b7a238e32cb21 Mon Sep 17 00:00:00 2001 From: David Reid Date: Tue, 28 Jan 2014 16:51:49 -0800 Subject: Complete test coverage --- tests/hazmat/primitives/test_hkdf.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) (limited to 'tests') diff --git a/tests/hazmat/primitives/test_hkdf.py b/tests/hazmat/primitives/test_hkdf.py index 171cf552..f3345b05 100644 --- a/tests/hazmat/primitives/test_hkdf.py +++ b/tests/hazmat/primitives/test_hkdf.py @@ -63,6 +63,32 @@ class TestHKDF(object): with pytest.raises(exceptions.AlreadyFinalized): hkdf.verify(b"\x02" * 16, b"gJ\xfb{\xb1Oi\xc5sMC\xb7\xe4@\xf7u") + hkdf = HKDF( + hashes.SHA256(), + 16, + salt=None, + info=None, + backend=backend + ) + + hkdf.extract(b"\x01" * 16) + + with pytest.raises(exceptions.AlreadyFinalized): + hkdf.extract(b"\x02" * 16) + + hkdf = HKDF( + hashes.SHA256(), + 16, + salt=None, + info=None, + backend=backend + ) + + hkdf.expand(b"\x01" * 16) + + with pytest.raises(exceptions.AlreadyFinalized): + hkdf.expand(b"\x02" * 16) + def test_verify(self, backend): hkdf = HKDF( hashes.SHA256(), -- cgit v1.2.3